View Javadoc

1   /**
2    * Copyright (c) 2011, The University of Southampton and the individual contributors.
3    * All rights reserved.
4    *
5    * Redistribution and use in source and binary forms, with or without modification,
6    * are permitted provided that the following conditions are met:
7    *
8    *   * 	Redistributions of source code must retain the above copyright notice,
9    * 	this list of conditions and the following disclaimer.
10   *
11   *   *	Redistributions in binary form must reproduce the above copyright notice,
12   * 	this list of conditions and the following disclaimer in the documentation
13   * 	and/or other materials provided with the distribution.
14   *
15   *   *	Neither the name of the University of Southampton nor the names of its
16   * 	contributors may be used to endorse or promote products derived from this
17   * 	software without specific prior written permission.
18   *
19   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
20   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
21   * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
22   * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
23   * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
24   * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25   * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
26   * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27   * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
28   * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29   */
30  package org.openimaj.tools.imagecollection.collection.config;
31  
32  import java.io.File;
33  import java.io.IOException;
34  
35  import org.kohsuke.args4j.Option;
36  import org.openimaj.image.MBFImage;
37  import org.openimaj.tools.imagecollection.metamapper.ConsoleMetaMapper;
38  import org.openimaj.tools.imagecollection.metamapper.FileMetaMapper;
39  import org.openimaj.tools.imagecollection.metamapper.MetaMapper;
40  import org.openimaj.tools.imagecollection.processor.DirectoryImageProcessor;
41  import org.openimaj.tools.imagecollection.processor.ImageCollectionProcessor;
42  
43  public enum MetaMapperMode {
44  	CONSOLE{
45  		@Override
46  		public MetaMapper mapper(ImageCollectionProcessor<MBFImage> processor) {
47  			return new ConsoleMetaMapper();
48  		}
49  		
50  	},
51  	FILE{
52  		@Option(name="--meta-output-name", aliases="-mn", required=false, usage="The name of the meta output file", metaVar="STRING")
53  		private String outputName = "meta.txt";
54  		
55  		@Option(name="--directory-mode-root", aliases="-mdr", required=false, usage="If the output mode is directory, put the meta file in the directory", metaVar="STRING")
56  		private boolean directoryModeRoot = true;
57  		
58  		@Option(name="--meta-output-root", aliases="-mor", required=false, usage="If not using the directory mode root, the root folder", metaVar="STRING")
59  		private String actualRoot= "./metaout";
60  		@Override
61  		public MetaMapper mapper(ImageCollectionProcessor<MBFImage> processor) throws IOException {
62  			String actualOutDirectory = actualRoot;
63  			if(directoryModeRoot && processor instanceof DirectoryImageProcessor){
64  				DirectoryImageProcessor<?> dproc = ((DirectoryImageProcessor<?>)processor);
65  				actualOutDirectory = dproc.getDirectoryFile().getAbsolutePath();
66  			}
67  			
68  			File outFile = new File(actualOutDirectory,outputName);
69  			
70  			return new FileMetaMapper(outFile);
71  			
72  		}
73  		
74  	};
75  	public abstract MetaMapper mapper(ImageCollectionProcessor<MBFImage> processor) throws IOException;
76  }