/*****************************************************************************
* Copyright (C) The Apache Software Foundation. All rights reserved. *
* ------------------------------------------------------------------------- *
* This software is published under the terms of the Apache Software License *
* version 1.1, a copy of which has been included with this distribution in *
* the LICENSE file. *
*****************************************************************************/
package org.apache.batik.apps.rasterizer;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.net.URL;
import java.net.MalformedURLException;
import org.apache.batik.transcoder.TranscoderInput;
import org.apache.batik.transcoder.TranscoderOutput;
import org.apache.batik.transcoder.image.ImageTranscoder;
import org.apache.batik.transcoder.image.PNGTranscoder;
import org.apache.batik.transcoder.image.JPEGTranscoder;
import org.xml.sax.InputSource;
/**
* A simple class that can generate images from svg documents.
*
* @author <a href="mailto:Thierry.Kormann@sophia.inria.fr">Thierry Kormann</a>
* @version $Id: Main.java,v 1.8 2001/04/18 23:18:16 vhardy Exp $
*/
public class Main {
public static void writeImage(ImageTranscoder transcoder,
String uri, String output) {
try {
System.out.println("Converting "+uri+" to "+output);
OutputStream ostream =
new BufferedOutputStream(new FileOutputStream(output));
transcoder.transcode(new TranscoderInput(uri),
new TranscoderOutput(ostream));
ostream.flush();
ostream.close();
} catch(Exception ex) {
System.out.println("Error while writing "+uri+" to "+output);
}
}
public static void error(String msg) {
System.err.println(msg);
}
public static void usage(PrintStream out) {
out.println("usage: rasterizer [options] [@files]");
out.println("-d <directory> Destination directory for output files");
out.println("-m <mimetype> Mime type for output files");
out.println("-w <width> Width of the output image");
out.println("-h <height> Height of the output image");
}
public static void main(String [] args) {
String mimeType = "image/png";
String directory = null;
List svgFiles = new LinkedList();
int i=0;
float width = Float.NaN;
float height = Float.NaN;
while (i < args.length) {
if (args[i].equals("-d")) {
if (i+1 < args.length) {
i++;
directory = args[i++];
continue;
} else {
error("option -d requires an argument");
usage(System.err);
System.exit(1);
}
} else if (args[i].equals("-m")) {
if (i+1 < args.length) {
i++;
mimeType = args[i++];
continue;
} else {
error("option -m requires an argument");
usage(System.err);
System.exit(1);
}
} else if (args[i].equals("-w")) {
if (i+1 < args.length) {
i++;
try{
width = Float.parseFloat(args[i++]);
}catch(NumberFormatException e){
usage(System.err);
System.exit(1);
}
continue;
} else {
error("option -w requires an argument");
usage(System.err);
System.exit(1);
}
} else if (args[i].equals("-h")) {
if (i+1 < args.length) {
i++;
try{
width = Float.parseFloat(args[i++]);
}catch(NumberFormatException e){
usage(System.err);
System.exit(1);
}
continue;
} else {
error("option -h requires an argument");
usage(System.err);
System.exit(1);
}
} else if (args[i].equals("-help")) {
usage(System.out);
System.exit(0);
} else {
svgFiles.add(args[i++]);
continue;
}
}
/*TranscoderFactory factory =
ConcreteTranscoderFactory.getTranscoderFactoryImplementation();
*/
ImageTranscoder t = null;
if (mimeType.equals("image/jpg") ||
mimeType.equals("image/jpeg") ||
mimeType.equals("image/jpe")) {
t = new JPEGTranscoder();
t.addTranscodingHint(JPEGTranscoder.KEY_XML_PARSER_CLASSNAME,
"org.apache.crimson.parser.XMLReaderImpl");
} else if (mimeType.equals("image/png")) {
t = new PNGTranscoder();
t.addTranscodingHint(PNGTranscoder.KEY_XML_PARSER_CLASSNAME,
"org.apache.crimson.parser.XMLReaderImpl");
}
if (t == null) {
error("No transcoder found for mime type : "+mimeType);
System.exit(1);
}
if(!Float.isNaN(width)){
t.addTranscodingHint(ImageTranscoder.KEY_WIDTH,
new Float(width));
}
if(!Float.isNaN(height)){
t.addTranscodingHint(ImageTranscoder.KEY_HEIGHT,
new Float(height));
}
for (Iterator iter = svgFiles.iterator(); iter.hasNext();) {
String s = (String) iter.next();
URL url = getSVGURL(s);
if(url != null){
String uri = url.getFile();
int j = uri.lastIndexOf('/');
if(j > 0){
uri = uri.substring(j);
}
if (uri.endsWith(".svg")) {
uri = uri.substring(0, uri.lastIndexOf(".svg"));
int k = mimeType.lastIndexOf('/');
if (k > 0) {
String ext = mimeType.substring(k+1);
if (ext.length() > 0) {
uri += "."+ext;
}
} else {
uri += "."+mimeType;
}
}
if (directory == null) {
directory = getDirectory(s);
}
if (directory != null) {
File output = new File(directory, uri);
writeImage((ImageTranscoder)t,
url.toString(),
output.getAbsolutePath());
}
else{
error("No valid output directory for : " + s);
}
}
}
System.exit(0);
}
public static URL getSVGURL(String s) {
URL url = null;
try{
File f = new File(s);
if(f.exists()){
url = f.toURL();
}
else{
url = new URL(s);
}
}catch(MalformedURLException e){
error("Bad svg file: " + s);
}
return url;
}
public static String getDirectory(String s){
File f = new File(s);
if(f.exists()){
return f.getParent();
}
else{
return null;
}
}
}