https://www.mediawiki.org/wiki/Special:Code/MediaWiki/109159
Revision: 109159 Author: oren Date: 2012-01-17 16:34:10 +0000 (Tue, 17 Jan 2012) Log Message: ----------- reverting deletions Added Paths: ----------- trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java Removed Paths: ------------- trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java Deleted: trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java =================================================================== --- trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java 2012-01-17 16:32:49 UTC (rev 109158) +++ trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java 2012-01-17 16:34:10 UTC (rev 109159) @@ -1,135 +0,0 @@ -package org.wikimedia.lsearch.analyzers; - -import java.io.IOException; -import java.io.StringReader; -import java.util.ArrayList; -import java.util.HashSet; - -import org.apache.lucene.analysis.Analyzer; -import org.apache.lucene.analysis.Token; -import org.apache.lucene.analysis.TokenStream; -import org.wikimedia.lsearch.config.IndexId; - -/** - * Aggregate bean that captures information about one - * item going into the some index aggregate field. - * - * @author rainman - * - */ -public class Aggregate { - protected ArrayList<Token> tokens; - protected float boost; - protected int noStopWordsLength; - protected Flags flags; - - public enum Flags { NONE, ALTTITLE, ANCHOR, RELATED, SECTION }; - - /** Construct from arbitrary text that will be tokenized - * @throws IOException */ - public Aggregate(String text, float boost, IndexId iid, Analyzer analyzer, - String field, HashSet<String> stopWords, Flags flags) throws IOException{ - setTokens(toTokenArray(analyzer.tokenStream(field,new StringReader(text))),stopWords); - this.boost = boost; - this.flags = flags; - - } - /** Set new token array, calc length, etc.. */ - public void setTokens(ArrayList<Token> tokens, HashSet<String> stopWords){ - this.tokens = tokens; - if(stopWords != null){ - noStopWordsLength = 0; - for(Token t : tokens){ - if(!stopWords.contains(t.termText()) && t.getPositionIncrement()!=0) - noStopWordsLength++; - } - } else{ - noStopWordsLength = noAliasLength(); - } - } - /** Number of tokens without aliases */ - public int noAliasLength(){ - int len = 0; - for(Token t : tokens){ - if(t.getPositionIncrement() != 0) - len++; - } - return len; - } - - /** Construct with specific analyzer - * @throws IOException */ - public Aggregate(String text, float boost, IndexId iid, Analyzer analyzer, - String field, Flags flags) throws IOException{ - this.tokens = toTokenArray(analyzer.tokenStream(field,new StringReader(text))); - this.boost = boost; - this.noStopWordsLength = noAliasLength(); - this.flags = flags; - } - - private ArrayList<Token> toTokenArray(TokenStream stream) throws IOException { - ArrayList<Token> tt = new ArrayList<Token>(); - Token t = null; - while( (t = stream.next()) != null && tt.size() < 0xff-1){ - tt.add(t); - } - return tt; - } - - /** Number of tokens */ - public int length(){ - if(tokens != null) - return tokens.size(); - else - return 0; - } - - /** Number of tokens when stop words are excluded */ - public int getNoStopWordsLength(){ - return noStopWordsLength; - } - - /** boost factor */ - public float boost(){ - return boost; - } - - public Token getToken(int index){ - return tokens.get(index); - } - - public ArrayList<Token> getTokens() { - return tokens; - } - - public Flags getFlags() { - return flags; - } - /** - * Generate the meta field stored contents - * format: [length] [length without stop words] [boost] [complete length] [flags] (1+1+4+1+1 bytes) - */ - public static byte[] serializeAggregate(ArrayList<Aggregate> items){ - byte[] buf = new byte[items.size() * 8]; - - for(int i=0;i<items.size();i++){ - Aggregate ag = items.get(i); - assert ag.length() < 0xff; - assert ag.noAliasLength() < 0xff; - assert ag.getNoStopWordsLength() < 0xff; - buf[i*8] = (byte)(ag.noAliasLength() & 0xff); - buf[i*8+1] = (byte)(ag.getNoStopWordsLength() & 0xff); - int boost = Float.floatToIntBits(ag.boost()); - buf[i*8+2] = (byte)((boost >>> 24) & 0xff); - buf[i*8+3] = (byte)((boost >>> 16) & 0xff); - buf[i*8+4] = (byte)((boost >>> 8) & 0xff); - buf[i*8+5] = (byte)((boost >>> 0) & 0xff); - buf[i*8+6] = (byte)(ag.length() & 0xff); - buf[i*8+7] = (byte)(ag.getFlags().ordinal() & 0xff); - } - - return buf; - } - - -} Copied: trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java (from rev 109149, trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java) =================================================================== --- trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java (rev 0) +++ trunk/lucene-search-2/src/org/wikimedia/lsearch/analyzers/Aggregate.java 2012-01-17 16:34:10 UTC (rev 109159) @@ -0,0 +1,134 @@ +package org.wikimedia.lsearch.analyzers; + +import java.io.IOException; +import java.util.ArrayList; +import java.util.HashSet; + +import org.apache.lucene.analysis.Analyzer; +import org.apache.lucene.analysis.Token; +import org.apache.lucene.analysis.TokenStream; +import org.wikimedia.lsearch.config.IndexId; + +/** + * Aggregate bean that captures information about one + * item going into the some index aggregate field. + * + * @author rainman + * + */ +public class Aggregate { + protected ArrayList<Token> tokens; + protected float boost; + protected int noStopWordsLength; + protected Flags flags; + + public enum Flags { NONE, ALTTITLE, ANCHOR, RELATED, SECTION }; + + /** Construct from arbitrary text that will be tokenized + * @throws IOException */ + public Aggregate(String text, float boost, IndexId iid, Analyzer analyzer, + String field, HashSet<String> stopWords, Flags flags) throws IOException{ + setTokens(toTokenArray(analyzer.tokenStream(field,text)),stopWords); + this.boost = boost; + this.flags = flags; + + } + /** Set new token array, calc length, etc.. */ + public void setTokens(ArrayList<Token> tokens, HashSet<String> stopWords){ + this.tokens = tokens; + if(stopWords != null){ + noStopWordsLength = 0; + for(Token t : tokens){ + if(!stopWords.contains(t.termText()) && t.getPositionIncrement()!=0) + noStopWordsLength++; + } + } else{ + noStopWordsLength = noAliasLength(); + } + } + /** Number of tokens without aliases */ + public int noAliasLength(){ + int len = 0; + for(Token t : tokens){ + if(t.getPositionIncrement() != 0) + len++; + } + return len; + } + + /** Construct with specific analyzer + * @throws IOException */ + public Aggregate(String text, float boost, IndexId iid, Analyzer analyzer, + String field, Flags flags) throws IOException{ + this.tokens = toTokenArray(analyzer.tokenStream(field,text)); + this.boost = boost; + this.noStopWordsLength = noAliasLength(); + this.flags = flags; + } + + private ArrayList<Token> toTokenArray(TokenStream stream) throws IOException { + ArrayList<Token> tt = new ArrayList<Token>(); + Token t = null; + while( (t = stream.next()) != null && tt.size() < 0xff-1){ + tt.add(t); + } + return tt; + } + + /** Number of tokens */ + public int length(){ + if(tokens != null) + return tokens.size(); + else + return 0; + } + + /** Number of tokens when stop words are excluded */ + public int getNoStopWordsLength(){ + return noStopWordsLength; + } + + /** boost factor */ + public float boost(){ + return boost; + } + + public Token getToken(int index){ + return tokens.get(index); + } + + public ArrayList<Token> getTokens() { + return tokens; + } + + public Flags getFlags() { + return flags; + } + /** + * Generate the meta field stored contents + * format: [length] [length without stop words] [boost] [complete length] [flags] (1+1+4+1+1 bytes) + */ + public static byte[] serializeAggregate(ArrayList<Aggregate> items){ + byte[] buf = new byte[items.size() * 8]; + + for(int i=0;i<items.size();i++){ + Aggregate ag = items.get(i); + assert ag.length() < 0xff; + assert ag.noAliasLength() < 0xff; + assert ag.getNoStopWordsLength() < 0xff; + buf[i*8] = (byte)(ag.noAliasLength() & 0xff); + buf[i*8+1] = (byte)(ag.getNoStopWordsLength() & 0xff); + int boost = Float.floatToIntBits(ag.boost()); + buf[i*8+2] = (byte)((boost >>> 24) & 0xff); + buf[i*8+3] = (byte)((boost >>> 16) & 0xff); + buf[i*8+4] = (byte)((boost >>> 8) & 0xff); + buf[i*8+5] = (byte)((boost >>> 0) & 0xff); + buf[i*8+6] = (byte)(ag.length() & 0xff); + buf[i*8+7] = (byte)(ag.getFlags().ordinal() & 0xff); + } + + return buf; + } + + +} _______________________________________________ MediaWiki-CVS mailing list MediaWiki-CVS@lists.wikimedia.org https://lists.wikimedia.org/mailman/listinfo/mediawiki-cvs