com.xmlcalabash.extensions.DiTAA Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xmlcalabash1-ditaa Show documentation
XML Calabash extension step for DiTAA diagrams
There is a newer version: 1.2.1
package com.xmlcalabash.extensions;

import com.xmlcalabash.core.XMLCalabash;
import com.xmlcalabash.core.XProcConstants;
import com.xmlcalabash.core.XProcException;
import com.xmlcalabash.core.XProcRuntime;
import com.xmlcalabash.io.ReadablePipe;
import com.xmlcalabash.io.WritablePipe;
import com.xmlcalabash.library.DefaultStep;
import com.xmlcalabash.runtime.XAtomicStep;
import com.xmlcalabash.util.Base64;
import com.xmlcalabash.util.TreeWriter;
import com.xmlcalabash.util.XProcURIResolver;
import net.sf.saxon.s9api.QName;
import net.sf.saxon.s9api.SaxonApiException;
import net.sf.saxon.s9api.XdmNode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.stathissideris.ascii2image.core.ConversionOptions;
import org.stathissideris.ascii2image.graphics.BitmapRenderer;
import org.stathissideris.ascii2image.graphics.Diagram;
import org.stathissideris.ascii2image.text.TextGrid;
import org.xml.sax.InputSource;

import javax.imageio.ImageIO;
import javax.xml.transform.Source;
import javax.xml.transform.TransformerException;
import javax.xml.transform.URIResolver;
import javax.xml.transform.sax.SAXSource;
import java.awt.image.RenderedImage;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.ArrayList;

/**
 * Created by IntelliJ IDEA.
 * User: ndw
 * Date: Oct 8, 2008
 * Time: 7:44:07 AM
 * To change this template use File | Settings | File Templates.
 */

@XMLCalabash(
        name = "cx:ditaa",
        type = "{http://xmlcalabash.com/ns/extensions}ditaa")

public class DiTAA extends DefaultStep {
    private static final QName _shadows = new QName("", "shadows");
    private static final QName _antialias = new QName("", "antialias");
    private static final QName _corners = new QName("", "corners");
    private static final QName _separation = new QName("", "separation");
    private static final QName _scale = new QName("", "scale");
    private static final QName _html = new QName("", "html");
    private static final QName h_img = new QName("", "http://www.w3.org/1999/xhtml", "img");
    private static final QName _src = new QName("", "src");
    private static final String library_xpl = "http://xmlcalabash.com/extension/steps/ditaa.xpl";
    private static final String library_uri = "/com/xmlcalabash/extensions/ditaa/library.xpl";

    private ReadablePipe source = null;
    private WritablePipe result = null;
    private ConversionOptions options = null;

    /**
     * Creates a new instance of Identity
     * @param runtime The XProc runtime
     * @param step The atomic step
     */
    public DiTAA(XProcRuntime runtime, XAtomicStep step) {
        super(runtime, step);
    }

    public void setInput(String port, ReadablePipe pipe) {
        source = pipe;
    }

    public void setOutput(String port, WritablePipe pipe) {
        result = pipe;
    }

    public void reset() {
        source.resetReader();
        result.resetWriter();
    }

    public void run() throws SaxonApiException {
        super.run();

        String base64 = null;
        boolean html = false;

        try {
            options = new ConversionOptions();

            boolean shadows = getOption(_shadows, true);
            boolean antialias = getOption(_antialias, true);
            boolean roundedCorners = "rounded".equals(getOption(_corners, "rounded"));
            boolean separation = getOption(_separation, true);
            html = getOption(_html, false);

            String scaleStr = getOption(_scale, (String) null);
            if (scaleStr != null) {
                float scale = Float.parseFloat(scaleStr);
                options.renderingOptions.setScale(scale);
            }

            options.processingOptions.setCharacterEncoding("utf-8");
            options.renderingOptions.setDropShadows(shadows);
            options.processingOptions.setAllCornersAreRound(roundedCorners);
            options.processingOptions.setPerformSeparationOfCommonEdges(separation);
            options.renderingOptions.setAntialias(antialias);


            TextGrid grid = new TextGrid();
            if(options.processingOptions.getCustomShapes() != null){
                grid.addToMarkupTags(options.processingOptions.getCustomShapes().keySet());
            }

            XdmNode doc = source.read();
            String text = doc.getStringValue();

            ArrayList lines = new ArrayList();
            String[] linesArray = text.split("\n");
            for (int i = 0; i  < linesArray.length; i++) {
                lines.add(new StringBuffer(linesArray[i]));
            }
            grid.initialiseWithLines(lines, options.processingOptions);
            Diagram diagram = new Diagram(grid, options);
            RenderedImage image = new BitmapRenderer().renderToImage(diagram, options.renderingOptions);

            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            ImageIO.write(image, "png", baos);
            base64 = Base64.encodeBytes(baos.toByteArray(), 0, baos.size());
        } catch (Exception e) {
            throw new XProcException(e);
        }

        TreeWriter tree = new TreeWriter(runtime);
        tree.startDocument(step.getNode().getBaseURI());

        if (html) {
            tree.addStartElement(h_img);
            tree.addAttribute(_src, "data:image/png;base64," + base64);
            tree.startContent();
            tree.addEndElement();
        } else {
            tree.addStartElement(XProcConstants.c_data);
            tree.startContent();
            tree.addText("data:image/png;base64," + base64);
            tree.addEndElement();
        }

        tree.endDocument();
        result.write(tree.getResult());
    }

    public static void configureStep(XProcRuntime runtime) {
        XProcURIResolver resolver = runtime.getResolver();
        URIResolver uriResolver = resolver.getUnderlyingURIResolver();
        URIResolver myResolver = new StepResolver(uriResolver);
        resolver.setUnderlyingURIResolver(myResolver);
    }

    private static class StepResolver implements URIResolver {
        Logger logger = LoggerFactory.getLogger(DiTAA.class);
        URIResolver nextResolver = null;

        public StepResolver(URIResolver next) {
            nextResolver = next;
        }

        @Override
        public Source resolve(String href, String base) throws TransformerException {
            try {
                URI baseURI = new URI(base);
                URI xpl = baseURI.resolve(href);
                if (library_xpl.equals(xpl.toASCIIString())) {
                    URL url = DiTAA.class.getResource(library_uri);
                    logger.debug("Reading library.xpl for cx:ditaa from " + url);
                    InputStream s = DiTAA.class.getResourceAsStream(library_uri);
                    if (s != null) {
                        SAXSource source = new SAXSource(new InputSource(s));
                        return source;
                    } else {
                        logger.info("Failed to read " + library_uri + " for cx:ditaa");
                    }
                }
            } catch (URISyntaxException e) {
                // nevermind
            }

            if (nextResolver != null) {
                return nextResolver.resolve(href, base);
            } else {
                return null;
            }
        }
    }
}