luc
9 years ago
parent
ec04d27473
commit
8da20718aa
@ -0,0 +1,267 @@
|
||||
import java.io.File;
|
||||
import java.io.FileFilter;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.net.URL;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.Map.Entry;
|
||||
|
||||
import org.apache.commons.io.filefilter.FileFileFilter;
|
||||
|
||||
import net.yacy.cora.util.ConcurrentLog;
|
||||
import net.yacy.peers.graphics.EncodedImage;
|
||||
import net.yacy.server.serverObjects;
|
||||
|
||||
// ViewImageTest.java
|
||||
// -----------------------
|
||||
// part of YaCy
|
||||
// (C) by Michael Peter Christen; mc@yacy.net
|
||||
// first published on http://www.anomic.de
|
||||
// Frankfurt, Germany, 2006
|
||||
// created 03.04.2006
|
||||
//
|
||||
// This program 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.
|
||||
//
|
||||
// This program 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 this program; if not, write to the Free Software
|
||||
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
|
||||
/**
|
||||
* Test rendering of one or more image files by ViewImage
|
||||
*
|
||||
* @author luc
|
||||
*
|
||||
*/
|
||||
public class ViewImageTest {
|
||||
|
||||
/** Default image */
|
||||
private static final String DEFAULT_IMG_RESOURCES = "/viewImageTest/test";
|
||||
|
||||
/** Default output encoding format */
|
||||
private static final String DEFAULT_OUT_EXT = "png";
|
||||
|
||||
/**
|
||||
* @param testFile
|
||||
* file to load
|
||||
* @return testFile content as a bytes array
|
||||
* @throws IOException
|
||||
* when an error occured while loading
|
||||
*/
|
||||
private static byte[] getBytes(File testFile) throws IOException {
|
||||
InputStream inStream = new FileInputStream(testFile);
|
||||
byte[] res = new byte[inStream.available()];
|
||||
try {
|
||||
inStream.read(res);
|
||||
} finally {
|
||||
inStream.close();
|
||||
}
|
||||
return res;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param args
|
||||
* main parameters. first item may contain input file or folder
|
||||
* URL
|
||||
* @return file or folder to be used : specified as first in args or default
|
||||
* one
|
||||
*/
|
||||
private static File getInputURL(String args[]) {
|
||||
String fileURL;
|
||||
if (args != null && args.length > 0) {
|
||||
fileURL = args[0];
|
||||
} else {
|
||||
URL defaultURL = ViewImageTest.class.getResource(DEFAULT_IMG_RESOURCES);
|
||||
if (defaultURL == null) {
|
||||
throw new IllegalArgumentException("File not found : " + DEFAULT_IMG_RESOURCES);
|
||||
}
|
||||
fileURL = defaultURL.getFile();
|
||||
}
|
||||
return new File(fileURL);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param args
|
||||
* main parameters. args[2] main contains directory url for
|
||||
* rendered images files
|
||||
* @return output directory to use
|
||||
* @throws IllegalArgumentException
|
||||
* when args[2] is not set and default is not found
|
||||
*/
|
||||
private static File getOuputDir(String[] args) {
|
||||
File outDir;
|
||||
if (args.length > 2) {
|
||||
outDir = new File(args[2]);
|
||||
} else {
|
||||
String tmpDir = System.getProperty("java.io.tmpdir");
|
||||
if (tmpDir == null) {
|
||||
throw new IllegalArgumentException("No destination dir specified, and default not found");
|
||||
}
|
||||
outDir = new File(tmpDir + File.separator + ViewImageTest.class.getCanonicalName());
|
||||
}
|
||||
return outDir;
|
||||
}
|
||||
|
||||
/**
|
||||
* Build post parameters to use with ViewImage
|
||||
*
|
||||
* @param args
|
||||
* main parameters : args[3] and args[4] may respectively contain
|
||||
* max width and max height
|
||||
* @return a serverObjects instance
|
||||
*/
|
||||
private static serverObjects makePostParams(String args[]) {
|
||||
serverObjects post = new serverObjects();
|
||||
if (args != null && args.length > 3) {
|
||||
int maxWidth = Integer.parseInt(args[3]);
|
||||
post.put("maxwidth", String.valueOf(maxWidth));
|
||||
}
|
||||
|
||||
if (args != null && args.length > 4) {
|
||||
int maxHeight = Integer.parseInt(args[4]);
|
||||
post.put("maxheight", String.valueOf(maxHeight));
|
||||
}
|
||||
|
||||
return post;
|
||||
}
|
||||
|
||||
/**
|
||||
*
|
||||
* @param args
|
||||
* main parameters : fourth item may contain extension
|
||||
* @return extension to use for encoding
|
||||
*/
|
||||
private static String getEncodingExt(String args[]) {
|
||||
String ext = DEFAULT_OUT_EXT;
|
||||
if (args != null && args.length > 3) {
|
||||
ext = args[3];
|
||||
}
|
||||
return ext;
|
||||
}
|
||||
|
||||
/**
|
||||
* Display detailed results. All parametrers required not to be null.
|
||||
*
|
||||
* @param inFiles
|
||||
* input image files
|
||||
* @param failures
|
||||
* map input file url which failed with eventual cause exception
|
||||
*/
|
||||
private static void displayResults(File[] inFiles, Map<String, Exception> failures) {
|
||||
if (failures.size() > 0) {
|
||||
if (failures.size() == inFiles.length) {
|
||||
System.out.println("No input files could be processed :");
|
||||
} else {
|
||||
System.out.println("Some input files could not be processed :");
|
||||
}
|
||||
for (Entry<String, Exception> entry : failures.entrySet()) {
|
||||
System.out.println(entry.getKey());
|
||||
if (entry.getValue() != null) {
|
||||
System.out.println("cause : " + entry.getValue());
|
||||
}
|
||||
}
|
||||
} else {
|
||||
System.out.println("All input files were successfully processed.");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Test image(s) (default : JPEG_example_JPG_RIP_100.jpg) are parsed and
|
||||
* rendered to an output foler. Result can then be checked with program of
|
||||
* your choice.
|
||||
*
|
||||
* @param args
|
||||
* may be empty or contain parameters to override defaults :
|
||||
* <ul>
|
||||
* <li>args[0] : input image file URL or folder containing image
|
||||
* files URL. Default :
|
||||
* viewImageTest/test/JPEG_example_JPG_RIP_100.jpg</li>
|
||||
* <li>args[1] : output format name (for example : "jpg") for
|
||||
* rendered image</li>
|
||||
* <li>args[2] : ouput folder URL</li>
|
||||
* <li>args[3] : max width (in pixels) for rendered image.
|
||||
* Default : no value.</li>
|
||||
* <li>args[4] : max height (in pixels) for rendered image.
|
||||
* Default : no value.</li>
|
||||
* </ul>
|
||||
* @throws IOException
|
||||
* when a read/write error occured
|
||||
*/
|
||||
public static void main(String args[]) throws IOException {
|
||||
File inURL = getInputURL(args);
|
||||
String ext = getEncodingExt(args);
|
||||
File outDir = getOuputDir(args);
|
||||
serverObjects post = makePostParams(args);
|
||||
outDir.mkdirs();
|
||||
|
||||
File[] inFiles;
|
||||
if (inURL.isFile()) {
|
||||
inFiles = new File[1];
|
||||
inFiles[0] = inURL;
|
||||
System.out.println("Testing ViewImage rendering with input file : " + inURL.getAbsolutePath()
|
||||
+ " encoded To : " + ext);
|
||||
} else if (inURL.isDirectory()) {
|
||||
FileFilter filter = FileFileFilter.FILE;
|
||||
inFiles = inURL.listFiles(filter);
|
||||
System.out.println("Testing ViewImage rendering with input files in folder : " + inURL.getAbsolutePath()
|
||||
+ " encoded To : " + ext);
|
||||
} else {
|
||||
inFiles = new File[0];
|
||||
}
|
||||
if (inFiles.length == 0) {
|
||||
throw new IllegalArgumentException(inURL.getAbsolutePath() + " is not a valid file or folder url.");
|
||||
}
|
||||
System.out.println("Rendered images will be written in dir : " + outDir.getAbsolutePath());
|
||||
|
||||
Map<String, Exception> failures = new HashMap<String, Exception>();
|
||||
try {
|
||||
for (File inFile : inFiles) {
|
||||
/* Delete eventual previous result file */
|
||||
File outFile = new File(outDir, inFile.getName() + "." + ext);
|
||||
if (outFile.exists()) {
|
||||
outFile.delete();
|
||||
}
|
||||
|
||||
byte[] resourceb = getBytes(inFile);
|
||||
String urlString = inFile.getAbsolutePath();
|
||||
EncodedImage img = null;
|
||||
Exception error = null;
|
||||
try {
|
||||
img = ViewImage.parseAndScale(post, true, urlString, ext, false, resourceb);
|
||||
} catch (Exception e) {
|
||||
error = e;
|
||||
}
|
||||
|
||||
if (img == null) {
|
||||
failures.put(urlString, error);
|
||||
} else {
|
||||
FileOutputStream outFileStream = null;
|
||||
try {
|
||||
outFileStream = new FileOutputStream(outFile);
|
||||
img.getImage().writeTo(outFileStream);
|
||||
} finally {
|
||||
if (outFileStream != null) {
|
||||
outFileStream.close();
|
||||
}
|
||||
img.getImage().close();
|
||||
}
|
||||
}
|
||||
}
|
||||
displayResults(inFiles, failures);
|
||||
} finally {
|
||||
ConcurrentLog.shutdown();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in new issue