Author: pwang
Date: 2010-06-01 15:53:49 -0700 (Tue, 01 Jun 2010)
New Revision: 20391
Added:
cytoscape/trunk/src/cytoscape/dialogs/plugins/PluginIndex.java
Log:
Add Lucene-based searching
Added: cytoscape/trunk/src/cytoscape/dialogs/plugins/PluginIndex.java
===================================================================
--- cytoscape/trunk/src/cytoscape/dialogs/plugins/PluginIndex.java
(rev 0)
+++ cytoscape/trunk/src/cytoscape/dialogs/plugins/PluginIndex.java
2010-06-01 22:53:49 UTC (rev 20391)
@@ -0,0 +1,96 @@
+package cytoscape.dialogs.plugins;
+
+import java.io.IOException;
+import java.util.Vector;
+import java.util.HashMap;
+
+import org.apache.lucene.analysis.standard.StandardAnalyzer;
+import org.apache.lucene.document.Document;
+import org.apache.lucene.document.Field;
+import org.apache.lucene.index.IndexWriter;
+import org.apache.lucene.queryParser.ParseException;
+import org.apache.lucene.queryParser.QueryParser;
+import org.apache.lucene.search.*;
+import org.apache.lucene.store.Directory;
+import org.apache.lucene.store.RAMDirectory;
+import org.apache.lucene.util.Version;
+import cytoscape.plugin.DownloadableInfo;
+
+public class PluginIndex {
+
+ private static StandardAnalyzer analyzer = new
StandardAnalyzer(Version.LUCENE_30);
+
+ private static Vector indices = new Vector();
+ private static HashMap<String,Directory> indexTracker = new
HashMap<String,Directory>();
+ private static HashMap<String,Vector> pluginVectTracker = new
HashMap<String,Vector>();
+
+ private static Vector allPluginVect;
+
+ public static void setAllPluginVector( Vector allPluginVect, boolean
versionCheck, String downloadLocText ) throws Exception {
+
+ String index_id = downloadLocText.trim() +
Boolean.toString(versionCheck);
+
+ pluginVectTracker.put(index_id, allPluginVect);
+
+ // build an index for this id
+ Directory index = new RAMDirectory();
+
+ // the boolean arg in the IndexWriter ctor means to
+ // create a new index, overwriting any existing index
+ IndexWriter w = new IndexWriter(index, analyzer, true,
IndexWriter.MaxFieldLength.UNLIMITED);
+
+ for (int i=0; i<allPluginVect.size(); i++ ){
+ Vector aPlugin = (Vector) allPluginVect.elementAt(i);
+ DownloadableInfo info = (DownloadableInfo) aPlugin.elementAt(2);
+ addDoc(w, info.getDescription(), "description");
+ }
+
+ w.close();
+
+ indexTracker.put(index_id, index);
+
+ }
+
+ private static void addDoc(IndexWriter w, String value, String title)
throws IOException {
+ Document doc = new Document();
+ doc.add(new Field(title, value, Field.Store.YES,
Field.Index.ANALYZED));
+ w.addDocument(doc);
+ }
+
+
+ public static Vector getSearchResult(String querystr, boolean
versionCheck, String downloadLocText ) throws Exception {
+
+ String index_id = downloadLocText.trim() +
Boolean.toString(versionCheck);
+
+ // Check if the index for this case already existed
+ if (!indexTracker.containsKey(index_id)){
+ // The index does not exist, so we can not do search,
just return null
+ return null;
+ }
+
+ // The index does exist, do the search now
+ Directory index = (Directory) indexTracker.get(index_id);
+
+ Query q = new QueryParser(Version.LUCENE_30, "description",
analyzer).parse(querystr);
+
+ // search
+ int hitsPerPage = 1000;
+ IndexSearcher searcher = new IndexSearcher(index, true);
+ TopScoreDocCollector collector =
TopScoreDocCollector.create(hitsPerPage, true);
+ searcher.search(q, collector);
+ ScoreDoc[] hits = collector.topDocs().scoreDocs;
+
+ // Retrive all the plugins for this situation
+ Vector pluginVect = pluginVectTracker.get(index_id);
+
+ // This will hold the filered set of plugins
+ Vector filteredPluginVector = new Vector();
+
+ for(int i=0;i<hits.length;++i) {
+ int docId = hits[i].doc;
+ filteredPluginVector.add(pluginVect.elementAt(docId));
+ }
+
+ return filteredPluginVector;
+ }
+}
--
You received this message because you are subscribed to the Google Groups
"cytoscape-cvs" group.
To post to this group, send email to [email protected].
To unsubscribe from this group, send email to
[email protected].
For more options, visit this group at
http://groups.google.com/group/cytoscape-cvs?hl=en.
