mirror of
https://github.com/yacy/yacy_search_server.git
synced 2024-09-19 00:01:41 +02:00
268 lines
8.0 KiB
Java
Executable File
268 lines
8.0 KiB
Java
Executable File
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 : second 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 > 1) {
|
|
ext = args[1];
|
|
}
|
|
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();
|
|
}
|
|
|
|
}
|
|
|
|
}
|