001/**
002 * Copyright (c) 2011, The University of Southampton and the individual contributors.
003 * All rights reserved.
004 *
005 * Redistribution and use in source and binary forms, with or without modification,
006 * are permitted provided that the following conditions are met:
007 *
008 *   *  Redistributions of source code must retain the above copyright notice,
009 *      this list of conditions and the following disclaimer.
010 *
011 *   *  Redistributions in binary form must reproduce the above copyright notice,
012 *      this list of conditions and the following disclaimer in the documentation
013 *      and/or other materials provided with the distribution.
014 *
015 *   *  Neither the name of the University of Southampton nor the names of its
016 *      contributors may be used to endorse or promote products derived from this
017 *      software without specific prior written permission.
018 *
019 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
020 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
021 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
022 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
023 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
024 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
025 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
026 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
027 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
028 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
029 */
030package org.openimaj.tools.localfeature;
031
032import java.io.File;
033import java.io.IOException;
034import java.util.ArrayList;
035import java.util.List;
036
037import org.openimaj.feature.local.list.LocalFeatureList;
038import org.openimaj.feature.local.list.MemoryLocalFeatureList;
039import org.openimaj.image.feature.local.keypoints.Keypoint;
040import org.openimaj.image.feature.local.keypoints.SIFTGeoKeypoint;
041import org.openimaj.io.IOUtils;
042import org.openimaj.util.function.Operation;
043import org.openimaj.util.parallel.Parallel;
044
045/**
046 * Simple tool to batch convert files in siftgeo format to lowe's sift format.
047 * 
048 * @author Jonathon Hare (jsh2@ecs.soton.ac.uk)
049 * 
050 */
051public class SIFTGeoConverter {
052        private static void getInputs(File file, List<File> files) {
053                if (file.isDirectory()) {
054                        for (final File f : file.listFiles()) {
055                                getInputs(f, files);
056                        }
057                } else {
058                        if (file.getName().endsWith(".siftgeo")) {
059                                files.add(file);
060                        }
061                }
062        }
063
064        /**
065         * @param args
066         */
067        public static void main(String[] args) {
068                // final File inDir = new
069                // File("/Volumes/My Book/Data/ukbench/features/hesaff");
070                // final File outDir = new
071                // File("/Volumes/My Book/Data/ukbench/features/hesaff-sift");
072                final File inDir = new File(args[0]);
073                final File outDir = new File(args[1]);
074
075                outDir.mkdirs();
076
077                final List<File> input = new ArrayList<File>();
078                getInputs(inDir, input);
079
080                Parallel.forEach(input, new Operation<File>() {
081                        @Override
082                        public void perform(File file) {
083                                try {
084                                        System.out.println(file);
085
086                                        final LocalFeatureList<SIFTGeoKeypoint> sgkeys = SIFTGeoKeypoint.read(file);
087                                        final LocalFeatureList<Keypoint> keys = new MemoryLocalFeatureList<Keypoint>(128, sgkeys.size());
088
089                                        for (final SIFTGeoKeypoint sg : sgkeys) {
090                                                final Keypoint k = new Keypoint();
091                                                k.ivec = sg.descriptor;
092                                                k.x = sg.location.x;
093                                                k.y = sg.location.y;
094                                                k.ori = sg.location.orientation;
095                                                k.scale = sg.location.scale;
096                                                keys.add(k);
097                                        }
098
099                                        final File path = new File(file.getAbsolutePath()
100                                                        .replace(inDir.getAbsolutePath(), outDir.getAbsolutePath())
101                                                        .replace(".siftgeo", ".sift"));
102                                        path.getParentFile().mkdirs();
103                                        IOUtils.writeBinary(path, keys);
104                                } catch (final IOException e) {
105                                        e.printStackTrace();
106                                }
107                        }
108                });
109        }
110}