/*
* This file is part of Caliph & Emir.
*
* Caliph & Emir 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.
*
* Caliph & Emir 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 Caliph & Emir; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* Copyright statement:
* --------------------
* (c) 2002-2005 by Mathias Lux (mathias@juggle.at)
* http://www.juggle.at, http://caliph-emir.sourceforge.net
*/
package at.lux.fotoretrieval;
import at.lux.imageanalysis.ColorLayout;
import at.lux.imageanalysis.EdgeHistogram;
import at.lux.imageanalysis.ScalableColor;
import at.lux.retrieval.fastmap.ArrayFastmapDistanceMatrix;
import at.lux.retrieval.fastmap.FastMap;
import at.lux.retrieval.fastmap.FastmapDistanceMatrix;
import at.lux.retrieval.fastmap.VisualDescriptorDistanceCalculator;
import junit.framework.TestCase;
import org.jdom.Element;
import org.jdom.JDOMException;
import org.jdom.Namespace;
import org.jdom.input.SAXBuilder;
import javax.swing.*;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.File;
import java.io.IOException;
import java.util.*;
/**
* Date: 13.01.2005
* Time: 22:50:04
*
* @author Mathias Lux, mathias@juggle.at
*/
public class FastMapTest extends TestCase {
Namespace xsi = Namespace.getNamespace("xsi", "http://www.w3.org/2001/XMLSchema-instance");
public void testMatrix() throws IOException, JDOMException {
String[] files;
files = FileOperations.getAllDescriptions(new File("C:\\Dokumente und Einstellungen\\Mathias\\Eigene Dateien\\JavaProjects\\Caliph\\test-cbir"), true);
// files = FileOperations.getAllImages(new File("C:\\Dokumente und Einstellungen\\Mathias\\Eigene Dateien\\JavaProjects\\Caliph"), true);
LinkedList<ColorLayout> colorLayoutList = new LinkedList<ColorLayout>();
LinkedList<ScalableColor> scalableColorList = new LinkedList<ScalableColor>();
LinkedList<EdgeHistogram> edgeHistogramList = new LinkedList<EdgeHistogram>();
LinkedList<Date> dateList = new LinkedList<Date>();
for (int i = 0; i < files.length; i++) {
String file = files[i];
SAXBuilder builder = new SAXBuilder();
try {
Element e = builder.build(file).getRootElement();
java.util.List l = RetrievalToolkit.xpathQuery(e, "//VisualDescriptor", null);
for (Iterator iterator = l.iterator(); iterator.hasNext();) {
Element desc = (Element) iterator.next();
if (desc.getAttributeValue("type", xsi).equals("ColorLayoutType")) {
ColorLayout cl = new ColorLayout(desc);
colorLayoutList.add(cl);
} else if (desc.getAttributeValue("type", xsi).equals("ScalableColorType")) {
ScalableColor sc = new ScalableColor(desc);
scalableColorList.add(sc);
} else if (desc.getAttributeValue("type", xsi).equals("EdgeHistogramType")) {
try {
EdgeHistogram sc = new EdgeHistogram(desc);
edgeHistogramList.add(sc);
} catch (Exception e1) {
e1.printStackTrace();
}
}
}
l = RetrievalToolkit.xpathQuery(e, "//CreationCoordinates/Date/TimePoint", null);
if (l.size()>0) {
Element date = (Element) l.get(0);
Calendar cal = Calendar.getInstance();
// 2004-06-30T10:05:48
String[] tmp = date.getTextTrim().split("T");
StringTokenizer st = new StringTokenizer(tmp[0], "-");
int year = Integer.parseInt(st.nextToken());
int month = Integer.parseInt(st.nextToken());
int day = Integer.parseInt(st.nextToken());
st = new StringTokenizer(tmp[1], ":");
int hour = Integer.parseInt(st.nextToken());
int minute = Integer.parseInt(st.nextToken());
int second = Integer.parseInt(st.nextToken());
cal.set(year, month, day, hour, minute, second);
dateList.add(cal.getTime());
}
} catch (JDOMException e1) {
e1.printStackTrace();
} catch (IOException e1) {
e1.printStackTrace();
}
}
FastmapDistanceMatrix matrixFastmap;
matrixFastmap = new ArrayFastmapDistanceMatrix(colorLayoutList, new VisualDescriptorDistanceCalculator());
// matrixFastmapFastmap = new ArrayFastmapDistanceMatrix(dateList, new DateDistanceCalculator());
// matrixFastmapFastmap = new ArrayFastmapDistanceMatrix(edgeHistogramList, new VisualDescriptorDistanceCalculator());
// matrixFastmapFastmap = new ArrayFastmapDistanceMatrix(scalableColorList, new VisualDescriptorDistanceCalculator());
// System.out.println(matrixFastmapFastmap.getSimilarity(0,1));
assertTrue(matrixFastmap.getDistance(0, 0) == 0f);
// DecimalFormat df = (DecimalFormat) DecimalFormat.getInstance();
// df.setMaximumFractionDigits(2);
// df.setMinimumIntegerDigits(2);
// df.setMinimumFractionDigits(2);
//
// for (int i = 0; i< matrixFastmapFastmap.getDimension(); i++) {
// for (int j = 0; j<matrixFastmapFastmap.getDimension(); j++) {
// System.out.print(df.format(matrixFastmapFastmap.getSimilarity(i,j)) + "\t");
// }
// System.out.println("");
// }
//
int max = matrixFastmap.getMaximumDistance(4);
System.out.println("Maximale distanz in der 4. reihe: " + matrixFastmap.getDistance(max, 4) + " (" + max + ")");
int newmax = matrixFastmap.getMaximumDistance(max);
System.out.println("Maximale distanz zum Maximum: " + matrixFastmap.getDistance(newmax, max) + " (" + newmax + ")");
FastMap fm = new FastMap(matrixFastmap, 2);
long ms = System.currentTimeMillis();
fm.run();
ms = System.currentTimeMillis() - ms;
System.out.println("Time for " + matrixFastmap.getDimension() + " images: " + ms + " ms");
// float[][] points = fm.getPoints();
// for (int i = 0; i < points.length; i++) {
// float[] point = points[i];
// System.out.println("("+point[0]+","+point[1]+")");
// }
JFrame frame = new JFrame("Test FastMap");
frame.setSize(800, 600);
frame.addWindowListener(new WindowAdapter() {
public void windowClosing(WindowEvent e) {
System.exit(0);
}
});
frame.getContentPane().add(new PointPanel(fm.getPoints(), matrixFastmap));
frame.setVisible(true);
}
public static void main(String[] args) throws IOException, JDOMException {
FastMapTest t = new FastMapTest();
t.testMatrix();
}
}