/*
* This file is part of the LIRE project: http://lire-project.net
* LIRE is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* LIRE is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with LIRE; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* We kindly ask you to refer the any or one of the following publications in
* any publication mentioning or employing Lire:
*
* Lux Mathias, Savvas A. Chatzichristofis. Lire: Lucene Image Retrieval –
* An Extensible Java CBIR Library. In proceedings of the 16th ACM International
* Conference on Multimedia, pp. 1085-1088, Vancouver, Canada, 2008
* URL: http://doi.acm.org/10.1145/1459359.1459577
*
* Lux Mathias. Content Based Image Retrieval with LIRE. In proceedings of the
* 19th ACM International Conference on Multimedia, pp. 735-738, Scottsdale,
* Arizona, USA, 2011
* URL: http://dl.acm.org/citation.cfm?id=2072432
*
* Mathias Lux, Oge Marques. Visual Information Retrieval using Java and LIRE
* Morgan & Claypool, 2013
* URL: http://www.morganclaypool.com/doi/abs/10.2200/S00468ED1V01Y201301ICR025
*
* Copyright statement:
* --------------------
* (c) 2002-2013 by Mathias Lux (mathias@juggle.at)
* http://www.semanticmetadata.net/lire, http://www.lire-project.net
*/
package net.semanticmetadata.lire.searchers;
import org.apache.lucene.analysis.core.WhitespaceAnalyzer;
import org.apache.lucene.document.Document;
import org.apache.lucene.index.FieldInvertState;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.queryparser.classic.ParseException;
import org.apache.lucene.queryparser.classic.QueryParser;
import org.apache.lucene.search.BooleanQuery;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.TopDocs;
import org.apache.lucene.search.similarities.ClassicSimilarity;
import org.apache.lucene.search.similarities.Similarity;
import org.apache.lucene.util.BytesRef;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.LinkedList;
/**
* Provides a general searcher for visual words implementation. Can be used for SIFT, SURF and MSER.
* Date: 28.09.2010
* Time: 13:58:33
* Mathias Lux, mathias@juggle.at
*/
public class VisualWordsImageSearcher extends AbstractImageSearcher {
private int numMaxHits;
private String fieldName;
private Similarity similarity = new ClassicSimilarity();
// private Similarity similarity = new MySimilarity();
// private Similarity similarity = new BM25Similarity();
QueryParser qp;
public VisualWordsImageSearcher(int numMaxHits, Similarity similarity, String fieldName) {
this.similarity = similarity;
this.numMaxHits = numMaxHits;
this.fieldName = fieldName;
qp = new QueryParser(fieldName, new WhitespaceAnalyzer());
BooleanQuery.setMaxClauseCount(10000);
}
public VisualWordsImageSearcher(int numMaxHits, String fieldName) {
this.numMaxHits = numMaxHits;
this.fieldName = fieldName;
qp = new QueryParser(fieldName, new WhitespaceAnalyzer());
BooleanQuery.setMaxClauseCount(10000);
}
public ImageSearchHits search(BufferedImage image, IndexReader reader) throws IOException {
throw new UnsupportedOperationException("Not implemented!");
}
public ImageSearchHits search(Document doc, IndexReader reader) throws IOException {
SimpleImageSearchHits sh = null;
IndexSearcher isearcher = new IndexSearcher(reader);
isearcher.setSimilarity(similarity);
String queryString = doc.getValues(fieldName)[0];
Query tq = null;
try {
tq = qp.parse(queryString);
TopDocs docs = isearcher.search(tq, numMaxHits);
LinkedList<SimpleResult> res = new LinkedList<SimpleResult>();
double maxDistance = 0d;
for (int i = 0; i < docs.scoreDocs.length; i++) {
double d = 1d / docs.scoreDocs[i].score;
maxDistance = Math.max(d, maxDistance);
SimpleResult sr = new SimpleResult(d, docs.scoreDocs[i].doc);
res.add(sr);
}
sh = new SimpleImageSearchHits(res, maxDistance);
} catch (ParseException e) {
System.err.println(queryString);
e.printStackTrace();
}
return sh;
}
public ImageDuplicates findDuplicates(IndexReader reader) throws IOException {
throw new UnsupportedOperationException("Not implemented!");
}
/**
* This implementation has shown formidable results with the Nister UKBench data set.
*/
@SuppressWarnings("unused")
private static class MySimilarity extends ClassicSimilarity {
@Override
public float coord(int overlap, int maxOverlap) {
return super.coord(overlap, maxOverlap);
}
@Override
public float queryNorm(float sumOfSquaredWeights) {
return super.queryNorm(sumOfSquaredWeights);
}
@Override
public float lengthNorm(FieldInvertState state) {
return super.lengthNorm(state);
}
@Override
public float tf(float freq) {
return super.tf(freq);
}
@Override
public float sloppyFreq(int distance) {
return super.sloppyFreq(distance);
}
@Override
public float scorePayload(int doc, int start, int end, BytesRef payload) {
return super.scorePayload(doc, start, end, payload);
}
@Override
public float idf(long docFreq, long numDocs) {
return 1f; //super.idf(docFreq, numDocs);
}
@Override
public void setDiscountOverlaps(boolean v) {
super.setDiscountOverlaps(v);
}
@Override
public boolean getDiscountOverlaps() {
return super.getDiscountOverlaps();
}
@Override
public String toString() {
return super.toString();
}
}
}