/*
* SVG Salamander
* Copyright (c) 2004, Mark McKay
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or
* without modification, are permitted provided that the following
* conditions are met:
*
* - Redistributions of source code must retain the above
* copyright notice, this list of conditions and the following
* disclaimer.
* - Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following
* disclaimer in the documentation and/or other materials
* provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
* OF THE POSSIBILITY OF SUCH DAMAGE.
*
* Mark McKay can be contacted at mark@kitfox.com. Salamander and other
* projects can be found at http://www.kitfox.com
*
* Created on February 18, 2004, 11:43 PM
*/
package com.kitfox.svg;
import com.kitfox.svg.app.beans.SVGIcon;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.beans.PropertyChangeListener;
import java.beans.PropertyChangeSupport;
import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.Reader;
import java.io.Serializable;
import java.lang.ref.SoftReference;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLConnection;
import java.net.URLStreamHandler;
import java.util.HashMap;
import java.util.Iterator;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.zip.GZIPInputStream;
import javax.imageio.ImageIO;
import org.xml.sax.EntityResolver;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.SAXParseException;
import org.xml.sax.XMLReader;
import org.xml.sax.helpers.XMLReaderFactory;
/**
* Many SVG files can be loaded at one time. These files will quite likely need
* to reference one another. The SVG universe provides a container for all these
* files and the means for them to relate to each other.
*
* @author Mark McKay
* @author <a href="mailto:mark@kitfox.com">Mark McKay</a>
*/
public class SVGUniverse implements Serializable
{
public static final long serialVersionUID = 0;
transient private PropertyChangeSupport changes = new PropertyChangeSupport(this);
/**
* Maps document URIs to their loaded SVG diagrams. Note that URIs for
* documents loaded from URLs will reflect their URLs and URIs for documents
* initiated from streams will have the scheme <i>svgSalamander</i>.
*/
final HashMap loadedDocs = new HashMap();
final HashMap loadedFonts = new HashMap();
final HashMap loadedImages = new HashMap();
public static final String INPUTSTREAM_SCHEME = "svgSalamander";
/**
* Current time in this universe. Used for resolving attributes that are
* influenced by track information. Time is in milliseconds. Time 0
* coresponds to the time of 0 in each member diagram.
*/
protected double curTime = 0.0;
private boolean verbose = false;
//Cache reader for efficiency
XMLReader cachedReader;
/**
* Creates a new instance of SVGUniverse
*/
public SVGUniverse()
{
}
public void addPropertyChangeListener(PropertyChangeListener l)
{
changes.addPropertyChangeListener(l);
}
public void removePropertyChangeListener(PropertyChangeListener l)
{
changes.removePropertyChangeListener(l);
}
/**
* Release all loaded SVG document from memory
*/
public void clear()
{
loadedDocs.clear();
loadedFonts.clear();
loadedImages.clear();
}
/**
* Returns the current animation time in milliseconds.
*/
public double getCurTime()
{
return curTime;
}
public void setCurTime(double curTime)
{
double oldTime = this.curTime;
this.curTime = curTime;
changes.firePropertyChange("curTime", new Double(oldTime), new Double(curTime));
}
/**
* Updates all time influenced style and presentation attributes in all SVG
* documents in this universe.
*/
public void updateTime() throws SVGException
{
for (Iterator it = loadedDocs.values().iterator(); it.hasNext();)
{
SVGDiagram dia = (SVGDiagram) it.next();
dia.updateTime(curTime);
}
}
/**
* Called by the Font element to let the universe know that a font has been
* loaded and is available.
*/
void registerFont(Font font)
{
loadedFonts.put(font.getFontFace().getFontFamily(), font);
}
public Font getDefaultFont()
{
for (Iterator it = loadedFonts.values().iterator(); it.hasNext();)
{
return (Font) it.next();
}
return null;
}
public Font getFont(String fontName)
{
return (Font) loadedFonts.get(fontName);
}
URL registerImage(URI imageURI)
{
String scheme = imageURI.getScheme();
if (scheme.equals("data"))
{
String path = imageURI.getRawSchemeSpecificPart();
int idx = path.indexOf(';');
String mime = path.substring(0, idx);
String content = path.substring(idx + 1);
if (content.startsWith("base64"))
{
content = content.substring(6);
try
{
byte[] buf = new sun.misc.BASE64Decoder().decodeBuffer(content);
ByteArrayInputStream bais = new ByteArrayInputStream(buf);
BufferedImage img = ImageIO.read(bais);
URL url;
int urlIdx = 0;
while (true)
{
url = new URL("inlineImage", "localhost", "img" + urlIdx);
if (!loadedImages.containsKey(url))
{
break;
}
urlIdx++;
}
SoftReference ref = new SoftReference(img);
loadedImages.put(url, ref);
return url;
} catch (IOException ex)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not decode inline image", ex);
}
}
return null;
} else
{
try
{
URL url = imageURI.toURL();
registerImage(url);
return url;
} catch (MalformedURLException ex)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Bad url", ex);
}
return null;
}
}
void registerImage(URL imageURL)
{
if (loadedImages.containsKey(imageURL))
{
return;
}
SoftReference ref;
try
{
String fileName = imageURL.getFile();
if (".svg".equals(fileName.substring(fileName.length() - 4).toLowerCase()))
{
SVGIcon icon = new SVGIcon();
icon.setSvgURI(imageURL.toURI());
BufferedImage img = new BufferedImage(icon.getIconWidth(), icon.getIconHeight(), BufferedImage.TYPE_INT_ARGB);
Graphics2D g = img.createGraphics();
icon.paintIcon(null, g, 0, 0);
g.dispose();
ref = new SoftReference(img);
} else
{
BufferedImage img = ImageIO.read(imageURL);
ref = new SoftReference(img);
}
loadedImages.put(imageURL, ref);
} catch (Exception e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not load image: " + imageURL, e);
}
}
BufferedImage getImage(URL imageURL)
{
SoftReference ref = (SoftReference) loadedImages.get(imageURL);
if (ref == null)
{
return null;
}
BufferedImage img = (BufferedImage) ref.get();
//If image was cleared from memory, reload it
if (img == null)
{
try
{
img = ImageIO.read(imageURL);
} catch (Exception e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not load image", e);
}
ref = new SoftReference(img);
loadedImages.put(imageURL, ref);
}
return img;
}
/**
* Returns the element of the document at the given URI. If the document is
* not already loaded, it will be.
*/
public SVGElement getElement(URI path)
{
return getElement(path, true);
}
public SVGElement getElement(URL path)
{
try
{
URI uri = new URI(path.toString());
return getElement(uri, true);
} catch (Exception e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not parse url " + path, e);
}
return null;
}
/**
* Looks up a href within our universe. If the href refers to a document
* that is not loaded, it will be loaded. The URL #target will then be
* checked against the SVG diagram's index and the coresponding element
* returned. If there is no coresponding index, null is returned.
*/
public SVGElement getElement(URI path, boolean loadIfAbsent)
{
try
{
//Strip fragment from URI
URI xmlBase = new URI(path.getScheme(), path.getSchemeSpecificPart(), null);
SVGDiagram dia = (SVGDiagram) loadedDocs.get(xmlBase);
if (dia == null && loadIfAbsent)
{
//System.err.println("SVGUnivserse: " + xmlBase.toString());
//javax.swing.JOptionPane.showMessageDialog(null, xmlBase.toString());
URL url = xmlBase.toURL();
loadSVG(url, false);
dia = (SVGDiagram) loadedDocs.get(xmlBase);
if (dia == null)
{
return null;
}
}
String fragment = path.getFragment();
return fragment == null ? dia.getRoot() : dia.getElement(fragment);
} catch (Exception e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not parse path " + path, e);
return null;
}
}
public SVGDiagram getDiagram(URI xmlBase)
{
return getDiagram(xmlBase, true);
}
/**
* Returns the diagram that has been loaded from this root. If diagram is
* not already loaded, returns null.
*/
public SVGDiagram getDiagram(URI xmlBase, boolean loadIfAbsent)
{
if (xmlBase == null)
{
return null;
}
SVGDiagram dia = (SVGDiagram) loadedDocs.get(xmlBase);
if (dia != null || !loadIfAbsent)
{
return dia;
}
//Load missing diagram
try
{
URL url;
if ("jar".equals(xmlBase.getScheme()) && xmlBase.getPath() != null && !xmlBase.getPath().contains("!/"))
{
//Workaround for resources stored in jars loaded by Webstart.
//https://bugs.openjdk.java.net/browse/JDK-6753651
url = SVGUniverse.class.getResource("xmlBase.getPath()");
}
else
{
url = xmlBase.toURL();
}
loadSVG(url, false);
dia = (SVGDiagram) loadedDocs.get(xmlBase);
return dia;
} catch (Exception e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not parse", e);
}
return null;
}
/**
* Wraps input stream in a BufferedInputStream. If it is detected that this
* input stream is GZIPped, also wraps in a GZIPInputStream for inflation.
*
* @param is Raw input stream
* @return Uncompressed stream of SVG data
* @throws java.io.IOException
*/
private InputStream createDocumentInputStream(InputStream is) throws IOException
{
BufferedInputStream bin = new BufferedInputStream(is);
bin.mark(2);
int b0 = bin.read();
int b1 = bin.read();
bin.reset();
//Check for gzip magic number
if ((b1 << 8 | b0) == GZIPInputStream.GZIP_MAGIC)
{
GZIPInputStream iis = new GZIPInputStream(bin);
return iis;
} else
{
//Plain text
return bin;
}
}
public URI loadSVG(URL docRoot)
{
return loadSVG(docRoot, false);
}
/**
* Loads an SVG file and all the files it references from the URL provided.
* If a referenced file already exists in the SVG universe, it is not
* reloaded.
*
* @param docRoot - URL to the location where this SVG file can be found.
* @param forceLoad - if true, ignore cached diagram and reload
* @return - The URI that refers to the loaded document
*/
public URI loadSVG(URL docRoot, boolean forceLoad)
{
try
{
URI uri = new URI(docRoot.toString());
if (loadedDocs.containsKey(uri) && !forceLoad)
{
return uri;
}
InputStream is = docRoot.openStream();
return loadSVG(uri, new InputSource(createDocumentInputStream(is)));
} catch (URISyntaxException ex)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not parse", ex);
} catch (IOException e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not parse", e);
}
return null;
}
public URI loadSVG(InputStream is, String name) throws IOException
{
return loadSVG(is, name, false);
}
public URI loadSVG(InputStream is, String name, boolean forceLoad) throws IOException
{
URI uri = getStreamBuiltURI(name);
if (uri == null)
{
return null;
}
if (loadedDocs.containsKey(uri) && !forceLoad)
{
return uri;
}
return loadSVG(uri, new InputSource(createDocumentInputStream(is)));
}
public URI loadSVG(Reader reader, String name)
{
return loadSVG(reader, name, false);
}
/**
* This routine allows you to create SVG documents from data streams that
* may not necessarily have a URL to load from. Since every SVG document
* must be identified by a unique URL, Salamander provides a method to fake
* this for streams by defining it's own protocol - svgSalamander - for SVG
* documents without a formal URL.
*
* @param reader - A stream containing a valid SVG document
* @param name - <p>A unique name for this document. It will be used to
* construct a unique URI to refer to this document and perform resolution
* with relative URIs within this document.</p> <p>For example, a name of
* "/myScene" will produce the URI svgSalamander:/myScene.
* "/maps/canada/toronto" will produce svgSalamander:/maps/canada/toronto.
* If this second document then contained the href "../uk/london", it would
* resolve by default to svgSalamander:/maps/uk/london. That is, SVG
* Salamander defines the URI scheme svgSalamander for it's own internal use
* and uses it for uniquely identfying documents loaded by stream.</p> <p>If
* you need to link to documents outside of this scheme, you can either
* supply full hrefs (eg, href="url(http://www.kitfox.com/index.html)") or
* put the xml:base attribute in a tag to change the defaultbase URIs are
* resolved against</p> <p>If a name does not start with the character '/',
* it will be automatically prefixed to it.</p>
* @param forceLoad - if true, ignore cached diagram and reload
*
* @return - The URI that refers to the loaded document
*/
public URI loadSVG(Reader reader, String name, boolean forceLoad)
{
//System.err.println(url.toString());
//Synthesize URI for this stream
URI uri = getStreamBuiltURI(name);
if (uri == null)
{
return null;
}
if (loadedDocs.containsKey(uri) && !forceLoad)
{
return uri;
}
return loadSVG(uri, new InputSource(reader));
}
/**
* Synthesize a URI for an SVGDiagram constructed from a stream.
*
* @param name - Name given the document constructed from a stream.
*/
public URI getStreamBuiltURI(String name)
{
if (name == null || name.length() == 0)
{
return null;
}
if (name.charAt(0) != '/')
{
name = '/' + name;
}
try
{
//Dummy URL for SVG documents built from image streams
return new URI(INPUTSTREAM_SCHEME, name, null);
} catch (Exception e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not parse", e);
return null;
}
}
private XMLReader getXMLReaderCached() throws SAXException
{
if (cachedReader == null)
{
cachedReader = XMLReaderFactory.createXMLReader();
}
return cachedReader;
}
protected URI loadSVG(URI xmlBase, InputSource is)
{
// Use an instance of ourselves as the SAX event handler
SVGLoader handler = new SVGLoader(xmlBase, this, verbose);
//Place this docment in the universe before it is completely loaded
// so that the load process can refer to references within it's current
// document
loadedDocs.put(xmlBase, handler.getLoadedDiagram());
try
{
// Parse the input
XMLReader reader = getXMLReaderCached();
reader.setEntityResolver(
new EntityResolver()
{
public InputSource resolveEntity(String publicId, String systemId)
{
//Ignore all DTDs
return new InputSource(new ByteArrayInputStream(new byte[0]));
}
});
reader.setContentHandler(handler);
reader.parse(is);
handler.getLoadedDiagram().updateTime(curTime);
return xmlBase;
} catch (SAXParseException sex)
{
System.err.println("Error processing " + xmlBase);
System.err.println(sex.getMessage());
loadedDocs.remove(xmlBase);
return null;
} catch (Throwable e)
{
Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
"Could not load SVG " + xmlBase, e);
}
return null;
}
// public static void main(String argv[])
// {
// try
// {
// URL url = new URL("svgSalamander", "localhost", -1, "abc.svg",
// new URLStreamHandler()
// {
// protected URLConnection openConnection(URL u)
// {
// return null;
// }
// }
// );
//// URL url2 = new URL("svgSalamander", "localhost", -1, "abc.svg");
//
// //Investigate URI resolution
// URI uriA, uriB, uriC, uriD, uriE;
//
// uriA = new URI("svgSalamander", "/names/mySpecialName", null);
//// uriA = new URI("http://www.kitfox.com/salamander");
//// uriA = new URI("svgSalamander://mySpecialName/grape");
// System.err.println(uriA.toString());
// System.err.println(uriA.getScheme());
//
// uriB = uriA.resolve("#begin");
// System.err.println(uriB.toString());
//
// uriC = uriA.resolve("tree#boing");
// System.err.println(uriC.toString());
//
// uriC = uriA.resolve("../tree#boing");
// System.err.println(uriC.toString());
// }
// catch (Exception e)
// {
// Logger.getLogger(SVGConst.SVG_LOGGER).log(Level.WARNING,
// "Could not parse", e);
// }
// }
public boolean isVerbose()
{
return verbose;
}
public void setVerbose(boolean verbose)
{
this.verbose = verbose;
}
/**
* Uses serialization to duplicate this universe.
*/
public SVGUniverse duplicate() throws IOException, ClassNotFoundException
{
ByteArrayOutputStream bs = new ByteArrayOutputStream();
ObjectOutputStream os = new ObjectOutputStream(bs);
os.writeObject(this);
os.close();
ByteArrayInputStream bin = new ByteArrayInputStream(bs.toByteArray());
ObjectInputStream is = new ObjectInputStream(bin);
SVGUniverse universe = (SVGUniverse) is.readObject();
is.close();
return universe;
}
}