001 //$HeadURL: svn+ssh://jwilden@svn.wald.intevation.org/deegree/base/branches/2.5_testing/src/org/deegree/tools/raster/Rescaler.java $
002 /*----------------------------------------------------------------------------
003 This file is part of deegree, http://deegree.org/
004 Copyright (C) 2001-2009 by:
005 Department of Geography, University of Bonn
006 and
007 lat/lon GmbH
008
009 This library is free software; you can redistribute it and/or modify it under
010 the terms of the GNU Lesser General Public License as published by the Free
011 Software Foundation; either version 2.1 of the License, or (at your option)
012 any later version.
013 This library is distributed in the hope that it will be useful, but WITHOUT
014 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
015 FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
016 details.
017 You should have received a copy of the GNU Lesser General Public License
018 along with this library; if not, write to the Free Software Foundation, Inc.,
019 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
020
021 Contact information:
022
023 lat/lon GmbH
024 Aennchenstr. 19, 53177 Bonn
025 Germany
026 http://lat-lon.de/
027
028 Department of Geography, University of Bonn
029 Prof. Dr. Klaus Greve
030 Postfach 1147, 53001 Bonn
031 Germany
032 http://www.geographie.uni-bonn.de/deegree/
033
034 e-mail: info@deegree.org
035 ----------------------------------------------------------------------------*/
036
037 package org.deegree.tools.raster;
038
039 import java.awt.image.BufferedImage;
040 import java.awt.image.renderable.ParameterBlock;
041 import java.io.File;
042 import java.net.URI;
043 import java.util.ArrayList;
044 import java.util.List;
045 import java.util.Properties;
046
047 import javax.media.jai.Interpolation;
048 import javax.media.jai.InterpolationBilinear;
049 import javax.media.jai.JAI;
050 import javax.media.jai.RenderedOp;
051
052 import org.deegree.datatypes.QualifiedName;
053 import org.deegree.framework.log.ILogger;
054 import org.deegree.framework.log.LoggerFactory;
055 import org.deegree.framework.util.ConvenienceFileFilter;
056 import org.deegree.framework.util.ImageUtils;
057 import org.deegree.io.dbaseapi.DBaseFile;
058 import org.deegree.model.coverage.grid.WorldFile;
059 import org.deegree.ogcbase.CommonNamespaces;
060
061 /**
062 *
063 *
064 * @author <a href="mailto:poth@lat-lon.de">Andreas Poth</a>
065 * @author last edited by: $Author: mschneider $
066 *
067 * @version $Revision: 18195 $, $Date: 2009-06-18 17:55:39 +0200 (Do, 18 Jun 2009) $
068 */
069 public class Rescaler {
070
071 private static final ILogger LOG = LoggerFactory.getLogger( Rescaler.class );
072
073 private static final URI DEEGREEAPP = CommonNamespaces.buildNSURI( "http://www.deegree.org/app" );
074
075 private static final String APP_PREFIX = "app";
076
077 private String outDir;
078
079 private String format;
080
081 private Interpolation interpolation = new InterpolationBilinear();
082
083 private List<String> fileList;
084
085 private double resolution = 0;
086
087 /**
088 *
089 * @param resolution
090 * @param rootDir
091 * @param outDir
092 * @param format
093 * @param subDirs
094 */
095 Rescaler( double resolution, String rootDir, String outDir, String format, boolean subDirs ) {
096 this.resolution = resolution;
097 this.outDir = outDir;
098 this.format = format;
099 fileList = getFileList( rootDir, subDirs );
100 }
101
102 /**
103 *
104 * @param resolution
105 * @param rootDir
106 * @param outDir
107 * @param dBase
108 * @param format
109 * @param fileColumn
110 * @throws Exception
111 */
112 Rescaler( double resolution, String rootDir, String outDir, String dBase, String format, String fileColumn )
113 throws Exception {
114 this.resolution = resolution;
115 this.outDir = outDir;
116 this.format = format;
117 fileList = getFileList( dBase, fileColumn, rootDir );
118 }
119
120 /**
121 * returns the list of image map files to consider read from a dbase file defined by the dbase parameter
122 *
123 * @param dbaseFile
124 * name of the dbase file
125 * @param fileColumn
126 * name of the column containing the image map files names
127 * @param baseDir
128 * name of the directory where the image map files are stored if this parameter is <code>null</code> it
129 * is assumed that the image map files are full referenced within the dbase
130 * @return the list of image map files to consider read from a dbase file defined by the dbase parameter
131 * @throws Exception
132 */
133 private static List<String> getFileList( String dBaseFile, String fileColumn, String baseDir )
134 throws Exception {
135
136 // handle dbase file extension and file location/reading problems
137 if ( dBaseFile.endsWith( ".dbf" ) ) {
138 dBaseFile = dBaseFile.substring( 0, dBaseFile.lastIndexOf( "." ) );
139 }
140 DBaseFile dbf = new DBaseFile( dBaseFile );
141
142 // sort dbase file contents chronologicaly (oldest first)
143 int cnt = dbf.getRecordNum();
144
145 Object[] mapItems = new Object[cnt];
146 QualifiedName fileC = new QualifiedName( APP_PREFIX, fileColumn.toUpperCase(), DEEGREEAPP );
147
148 for ( int i = 0; i < cnt; i++ ) {
149 // name of map file
150 mapItems[i] = dbf.getFRow( i + 1 ).getDefaultProperty( fileC ).getValue();
151 }
152
153 // extract names of image files from dBase file and attach them to rootDir
154 if ( baseDir == null ) {
155 baseDir = "";
156 } else if ( !baseDir.endsWith( "/" ) && !baseDir.endsWith( "\\" ) ) {
157 baseDir = baseDir + '/';
158 }
159 List<String> imageFiles = new ArrayList<String>( mapItems.length );
160 for ( int i = 0; i < mapItems.length; i++ ) {
161 if ( mapItems[i] != null ) {
162 imageFiles.add( baseDir + mapItems[i] );
163 }
164 }
165
166 return imageFiles;
167 }
168
169 /**
170 * returns the list of image map files to consider read from a defined root directory.
171 *
172 * @param rootDir
173 * root directory where to read image map files
174 * @param subdirs
175 * true if subdirectories of the root directory shall be parsed for image maps too
176 * @return the list of image map files to consider read from a defined root directory.
177 */
178 private static List<String> getFileList( String rootDir, boolean subdirs ) {
179 List<String> list = new ArrayList<String>( 10000 );
180 File file = new File( rootDir );
181 List<String> extensions = new ArrayList<String>();
182 extensions.add( "JPEG" );
183 extensions.add( "JPG" );
184 extensions.add( "BMP" );
185 extensions.add( "PNG" );
186 extensions.add( "GIF" );
187 extensions.add( "TIF" );
188 extensions.add( "TIFF" );
189 extensions.add( "GEOTIFF" );
190 ConvenienceFileFilter cff = new ConvenienceFileFilter( extensions, true );
191 String[] entries = file.list( cff );
192 if ( entries != null ) {
193 for ( int i = 0; i < entries.length; i++ ) {
194 File entry = new File( rootDir + '/' + entries[i] );
195 if ( entry.isDirectory() && subdirs ) {
196 list = readSubDirs( entry, list, cff );
197 } else {
198 list.add( rootDir + '/' + entries[i] );
199 }
200 }
201 }
202 return list;
203 }
204
205 /**
206 *
207 * @param file
208 * @param list
209 * @return a list of strings
210 */
211 private static List<String> readSubDirs( File file, List<String> list, ConvenienceFileFilter cff ) {
212
213 String[] entries = file.list( cff );
214 if ( entries != null ) {
215 for ( int i = 0; i < entries.length; i++ ) {
216 File entry = new File( file.getAbsolutePath() + '/' + entries[i] );
217 if ( entry.isDirectory() ) {
218 list = readSubDirs( entry, list, cff );
219 } else {
220 list.add( file.getAbsolutePath() + '/' + entries[i] );
221 }
222 }
223 }
224 return list;
225 }
226
227 /**
228 *
229 * @throws Exception
230 */
231 public void process()
232 throws Exception {
233 for ( int i = 0; i < fileList.size(); i++ ) {
234 System.out.print( fileList.get( i ) + "\r" );
235 File file = new File( fileList.get( i ) );
236 BufferedImage image = ImageUtils.loadImage( file );
237 WorldFile wf = WorldFile.readWorldFile( fileList.get( i ), WorldFile.TYPE.CENTER, image.getWidth(),
238 image.getHeight() );
239 float qx = (float) ( wf.getResx() / resolution );
240 float qy = (float) ( wf.getResy() / resolution );
241
242 ParameterBlock pb = new ParameterBlock();
243 pb.addSource( image );
244 pb.add( qx ); // The xScale
245 pb.add( qy ); // The yScale
246 pb.add( 0.0F ); // The x translation
247 pb.add( 0.0F ); // The y translation
248 pb.add( interpolation ); // The interpolation
249 // Create the scale operation
250 RenderedOp ro = JAI.create( "scale", pb, null );
251 try {
252 image = ro.getAsBufferedImage();
253 } catch ( Exception e ) {
254 e.printStackTrace();
255 }
256
257 wf = new WorldFile( resolution, resolution, 0, 0, wf.getEnvelope() );
258 int p = file.getName().lastIndexOf( '.' );
259 String fileBaseName = file.getName().substring( 0, p );
260 ImageUtils.saveImage( image, outDir + fileBaseName + '.' + format, 1 );
261 WorldFile.writeWorldFile( wf, outDir + fileBaseName );
262 System.gc();
263 }
264 }
265
266 private static void printHelp() {
267 System.out.println();
268 System.out.println( "Parameter description for RasterSplitter:" );
269 System.out.println( "-res : desired raster resolution" );
270 System.out.println( "-format : desired image format of result images (mandatory)" );
271 System.out.println( "-outDir : directory where result images shall be stored (mandatory)" );
272
273 System.out.println( "-rootDir : directory from where images to split will be read (mandatory)" );
274 System.out.println( "-subDirs : (true|false). If 'true' all sub directories of the 'rootDir' " );
275 System.out.println( " will be searched for images too (optional; default = false)" );
276 }
277
278 private static boolean validate( Properties map ) {
279 if ( map.getProperty( "-res" ) == null ) {
280 System.out.println( "-res must be set!" );
281 return false;
282 }
283 if ( map.getProperty( "-format" ) == null ) {
284 System.out.println( "-format must be set!" );
285 return false;
286 }
287 if ( map.getProperty( "-outDir" ) == null ) {
288 System.out.println( "-outDir must be set!" );
289 return false;
290 }
291 if ( map.getProperty( "-rootDir" ) == null ) {
292 System.out.println( "-rootDir must be set!" );
293 return false;
294 }
295 if ( map.getProperty( "-subDirs" ) != null && !"true".equals( map.getProperty( "-subDirs" ) )
296 && !"false".equals( map.getProperty( "-subDirs" ) ) ) {
297 System.out.println( "if -subDirs is set it must be true or false!" );
298 return false;
299 }
300 return true;
301 }
302
303 /**
304 * @param args
305 * @throws Exception
306 */
307 public static void main( String[] args )
308 throws Exception {
309 Properties map = new Properties();
310 for ( int i = 0; i < args.length; i += 2 ) {
311 map.put( args[i], args[i + 1] );
312 }
313 if ( !validate( map ) ) {
314 printHelp();
315 return;
316 }
317
318 String format = map.getProperty( "-format" );
319 String outDir = map.getProperty( "-outDir" );
320 double res = Double.parseDouble( map.getProperty( "-res" ) );
321 Rescaler rescaler = null;
322 if ( map.get( "-dbaseFile" ) != null ) {
323 String dBaseFile = map.getProperty( "-dbaseFile" );
324 String fileColum = map.getProperty( "-fileColumn" );
325 String baseDir = map.getProperty( "-baseDir" );
326 if ( baseDir == null ) {
327 baseDir = map.getProperty( "-rootDir" );
328 }
329 rescaler = new Rescaler( res, baseDir, outDir, dBaseFile, format, fileColum );
330 } else if ( map.get( "-rootDir" ) != null ) {
331 String rootDir = map.getProperty( "-rootDir" );
332 boolean subDirs = "true".equals( map.get( "-subDirs" ) );
333 rescaler = new Rescaler( res, rootDir, outDir, format, subDirs );
334 } else {
335 LOG.logInfo( map.toString() );
336 System.out.println( "-rootDir or -dbaseFile parameter must be defined" );
337 return;
338 }
339 rescaler.process();
340
341 }
342
343 }