fop/contrib/plan/src/org/apache/fop/plan/SimplePlanDrawer.java

/* $Id$
 * Copyright (C) 2001 The Apache Software Foundation. All rights reserved.
 * For details on use and redistribution please refer to the
 * LICENSE file included with these sources.
 */

package org.apache.fop.plan;

import org.apache.fop.svg.SVGUtilities;

import java.util.*;
import java.text.*;
import java.awt.*;
import java.awt.geom.AffineTransform;
import java.awt.geom.Rectangle2D;
import java.awt.font.FontRenderContext;

import org.w3c.dom.*;
import org.w3c.dom.svg.*;
import org.w3c.dom.css.*;

import org.apache.batik.dom.svg.SVGDOMImplementation;

public class SimplePlanDrawer implements PlanDrawer {
    final static String svgNS = SVGDOMImplementation.SVG_NAMESPACE_URI;
    float fontSize;
    HashMap hints;
    java.awt.Font font = null;
    boolean bord = false;
    float lSpace = 15;
    float width;
    float height;
    float topEdge;
    float rightEdge;

    String[] colours;
    String[] darkcolours;

    Date startDate;
    Date endDate;

    public void setStartDate(Date sd) {
        startDate = sd;
    }

    public void setEndDate(Date ed) {
        endDate = ed;
    }

    public Document createDocument(EventList data, float w, float h,
                                   HashMap hints) {
        this.width = w;
        this.height = h;
        this.hints = hints;
        fontSize = ((Float) hints.get(PlanHints.FONT_SIZE)).floatValue();
        bord = ((Boolean) hints.get(PlanHints.PLAN_BORDER)).booleanValue();

        String title = "";

        DOMImplementation impl =
          SVGDOMImplementation.getDOMImplementation();
        Document doc = impl.createDocument(svgNS, "svg", null);

        Element svgRoot = doc.getDocumentElement();
        svgRoot.setAttributeNS(null, "width", "" + width);
        svgRoot.setAttributeNS(null, "height", "" + height);
        svgRoot.setAttributeNS(null, "style",
                               "font-size:" + 8 + ";font-family:" +
                               hints.get(PlanHints.FONT_FAMILY));

        font = new java.awt.Font( (String) hints.get(PlanHints.FONT_FAMILY),
                                  java.awt.Font.PLAIN, (int) fontSize);

        if (bord) {
            Element border =
              SVGUtilities.createRect(doc, 0, 0, width, height);
            border.setAttributeNS(null, "style", "stroke:black;fill:none");
            svgRoot.appendChild(border);
        }

        float strwidth = SVGUtilities.getStringWidth(title, font);

        Element text;
        float pos = (float)(80 - strwidth / 2.0);
        if (pos < 5) {
            pos = 5;
        }
        text = SVGUtilities.createText(doc, pos, 18, title);
        text.setAttributeNS(null, "style", "font-size:14");
        svgRoot.appendChild(text);

        topEdge = SVGUtilities.getStringHeight(title, font) + 5;

        // add the actual pie chart
        addPlan(doc, svgRoot, data);
        //addLegend(doc, svgRoot, data);

        return doc;
    }

    protected void addPlan(Document doc, Element svgRoot, EventList data) {
        Date currentDate = new Date();

        Date lastWeek = startDate;
        Date future = endDate;
        Calendar lw = Calendar.getInstance();
        if(lastWeek == null || future == null) {
            int dow = lw.get(Calendar.DAY_OF_WEEK);
            lw.add(Calendar.DATE, -dow - 6);
            lastWeek = lw.getTime();
            lw.add(Calendar.DATE, 5 * 7);
            future = lw.getTime();
        }
        long totalDays = (long)((future.getTime() - lastWeek.getTime() + 43200000) / 86400000);
        lw.setTime(lastWeek);
        int startDay = lw.get(Calendar.DAY_OF_WEEK);

        float graphTop = topEdge;
        Element g;
        Element line;
        line = SVGUtilities.createLine(doc, 0, topEdge, width, topEdge);
        line.setAttributeNS(null, "style", "fill:none;stroke:black");
        svgRoot.appendChild(line);

        Element clip1 = SVGUtilities.createClip(doc,
                                                SVGUtilities.createPath(doc,
                                                                        "m0 0l126 0l0 " + height + "l-126 0z"), "clip1");
        Element clip2 = SVGUtilities.createClip(doc,
                                                SVGUtilities.createPath(doc,
                                                                        "m130 0l66 0l0 " + height + "l-66 0z"), "clip2");
        Element clip3 = SVGUtilities.createClip(doc,
                                                SVGUtilities.createPath(doc,
                                                                        "m200 0l" + (width - 200) + " 0l0 " + height + "l-" +
                                                                        (width - 200) + " 0z"), "clip3");
        svgRoot.appendChild(clip1);
        svgRoot.appendChild(clip2);
        svgRoot.appendChild(clip3);

        DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT);
        Element text;
        text = SVGUtilities.createText(doc, 201, topEdge - 1,
                                       df.format(lastWeek));
        svgRoot.appendChild(text);

        text = SVGUtilities.createText(doc, width, topEdge - 1,
                                       df.format(future));
        text.setAttributeNS(null, "text-anchor", "end");
        svgRoot.appendChild(text);

        line = SVGUtilities.createLine(doc, 128, topEdge, 128, height);
        line.setAttributeNS(null, "style", "fill:none;stroke:rgb(150,150,150)");
        svgRoot.appendChild(line);
        int offset = 0;
        for (int count = startDay; count < startDay + totalDays - 1; count++) {
            offset++;
            if (count % 7 == 0 || count % 7 == 1) {
                Element rect = SVGUtilities.createRect(doc,
                                                       200 + (offset - 1) * (width - 200) / (totalDays - 2),
                                                       (float)(topEdge + 0.5), (width - 200) / (totalDays - 3),
                                                       height - 1 - topEdge);
                rect.setAttributeNS(null, "style", "stroke:none;fill:rgb(230,230,230)");
                svgRoot.appendChild(rect);
            }
            line = SVGUtilities.createLine(doc,
                                           200 + (offset - 1) * (width - 200) / (totalDays - 2),
                                           (float)(topEdge + 0.5),
                                           200 + (offset - 1) * (width - 200) / (totalDays - 2),
                                           (float)(height - 0.5));
            line.setAttributeNS(null, "style", "fill:none;stroke:rgb(200,200,200)");
            svgRoot.appendChild(line);
        }


        for (int count = 0; count < data.getSize(); count++) {
            GroupInfo gi = data.getGroupInfo(count);
            g = SVGUtilities.createG(doc);
            text = SVGUtilities.createText(doc, 1, topEdge + 12,
                                           gi.getName());
            text.setAttributeNS(null, "style", "clip-path:url(#clip1)");
            g.appendChild(text);
            if (count > 0) {
                line = SVGUtilities.createLine(doc, 0, topEdge + 2,
                                               width, topEdge + 2);
                line.setAttributeNS(null, "style", "fill:none;stroke:rgb(100,100,100)");
                g.appendChild(line);
            }

            float lastTop = topEdge;
            topEdge += 14;
            boolean showing = false;
            for (int count1 = 0; count1 < gi.getSize(); count1++) {
                ActionInfo act = gi.getActionInfo(count1);
                String name = act.getOwner();
                String label = act.getLabel();
                text = SVGUtilities.createText(doc, 8, topEdge + 12, label);
                text.setAttributeNS(null, "style", "clip-path:url(#clip1)");
                g.appendChild(text);

                text = SVGUtilities.createText(doc, 130, topEdge + 12,
                                               name);
                text.setAttributeNS(null, "style", "clip-path:url(#clip2)");
                g.appendChild(text);
                int type = act.getType();
                Date start = act.getStartDate();
                Date end = act.getEndDate();
                if (end.after(lastWeek) && start.before(future)) {
                    showing = true;
                    int left = 200;
                    int right = 500;

                    int daysToStart = (int)((start.getTime() -
                                             lastWeek.getTime() + 43200000) / 86400000);
                    int days = (int)((end.getTime() - start.getTime() +
                                      43200000) / 86400000);
                    int daysFromEnd =
                      (int)((future.getTime() - end.getTime() +
                             43200000) / 86400000);
                    Element taskGraphic;
                    switch (type) {
                        case ActionInfo.TASK:
                            taskGraphic = SVGUtilities.createRect(doc,
                                                                  left + daysToStart * 300 / (totalDays - 2),
                                                                  topEdge + 2, days * 300 / (totalDays - 2), 10);
                            taskGraphic.setAttributeNS(null, "style", "stroke:black;fill:blue;stroke-width:1;clip-path:url(#clip3)");
                            g.appendChild(taskGraphic);
                            break;
                        case ActionInfo.MILESTONE:
                            taskGraphic = SVGUtilities.createPath(doc,
                                                                  "m " + (left +
                                                                          daysToStart * 300 / (totalDays - 2) - 6) +
                                                                  " " + (topEdge + 6) + "l6 6l6-6l-6-6z");
                            taskGraphic.setAttributeNS(null, "style", "stroke:black;fill:black;stroke-width:1;clip-path:url(#clip3)");
                            g.appendChild(taskGraphic);
                            text = SVGUtilities.createText(doc,
                                                           left + daysToStart * 300 / (totalDays - 2) + 8,
                                                           topEdge + 9, df.format(start));
                            g.appendChild(text);

                            break;
                        case ActionInfo.GROUPING:
                            taskGraphic = SVGUtilities.createPath(doc,
                                                                  "m " + (left +
                                                                          daysToStart * 300 / (totalDays - 2) - 6) +
                                                                  " " + (topEdge + 6) + "l6 -6l" +
                                                                  (days * 300 / (totalDays - 2)) +
                                                                  " 0l6 6l-6 6l-4-4l" + -
                                                                  (days * 300 / (totalDays - 2) - 8) + " 0l-4 4l-6-6z");
                            taskGraphic.setAttributeNS(null, "style", "stroke:black;fill:black;stroke-width:1;clip-path:url(#clip3)");
                            g.appendChild(taskGraphic);
                            break;
                        default:
                            break;
                    }
                }

                topEdge += 14;
            }
            if (showing) {
                svgRoot.appendChild(g);
            } else {
                topEdge = lastTop;
            }
        }
        int currentDays =
          (int)((currentDate.getTime() - lastWeek.getTime() +
                 43200000) / 86400000);

        text = SVGUtilities.createText(doc,
                                       (float)(200 + (currentDays + 0.5) * 300 / 35),
                                       graphTop - 1, df.format(currentDate));
        text.setAttributeNS(null, "text-anchor", "middle");
        text.setAttributeNS(null, "style", "stroke:rgb(100,100,100)");
        svgRoot.appendChild(text);

        line = SVGUtilities.createLine(doc,
                                       (float)(200 + (currentDays + 0.5) * 300 / 35), graphTop,
                                       (float)(200 + (currentDays + 0.5) * 300 / 35), height);
        line.setAttributeNS(null, "style", "fill:none;stroke:rgb(200,50,50);stroke-dasharray:5 5");
        svgRoot.appendChild(line);


    }
}