com.jennifer.ui.chart.brush.ColumnBrush.java Source code

Introduction

Here is the source code for com.jennifer.ui.chart.brush.ColumnBrush.java

Source

/*
 * Copyright (C) 2014 (JenniferSoft Inc.)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package com.jennifer.ui.chart.brush;

import com.jennifer.ui.chart.ChartBuilder;
import com.jennifer.ui.chart.grid.Grid;

import com.jennifer.ui.util.dom.Transform;
import org.json.JSONArray;
import org.json.JSONObject;

import java.text.DecimalFormat;

import static com.jennifer.ui.util.DomUtil.el;

/**
 * Created by Jayden on 2014-10-27.
 */
public class ColumnBrush extends Brush {
    private Transform root;
    private int outerPadding;
    private int innerPadding;
    private Grid x;
    private Grid y;
    private double zeroY;
    private int count;
    private double width;
    private double half_width;
    private JSONArray target;
    private double columnWidth;
    private boolean minCheck;

    public ColumnBrush(ChartBuilder chart, JSONObject options) {
        super(chart, options);
    }

    @Override
    public void drawBefore() {

        root = el("g").translate(chart.area("x"), chart.area("y"));

        outerPadding = options.optInt("outerPadding", 2);
        innerPadding = options.optInt("innerPadding", 1);
        minCheck = options.optBoolean("minCheck", false);

        x = (Grid) options.get("x");
        y = (Grid) options.get("y");

        zeroY = y.get(0);
        count = chart.data().length();

        target = options.getJSONArray("target");

        width = x.rangeBand();
        half_width = width - outerPadding * 2;
        columnWidth = (half_width - (target.length() - 1) * innerPadding) / target.length();

    }

    @Override
    public Object draw() {
        Transform[] t = new Transform[target.length()];
        double[] maxValue = new double[target.length()];
        for (int i = 0; i < maxValue.length; i++) {
            maxValue[i] = Double.MIN_VALUE;
        }

        for (int i = 0; i < count; i++) {
            double startX = x.get(i) - half_width / 2;

            Transform group = root.group();

            for (int j = 0, len = target.length(); j < len; j++) {

                double valueValue = chart.dataDouble(i, target.getString(j));

                double startY = y.get(((minCheck && valueValue == 0) ? 1 : valueValue));
                double h = Math.abs(zeroY - startY);
                String _color = this.color(j);

                JSONObject o = new JSONObject().put("x", startX).put("height", h).put("width", columnWidth)
                        .put("fill", _color);
                if (startY <= zeroY) {
                    o.put("y", startY);
                } else {
                    o.put("y", zeroY);
                }

                group.rect(o);

                // display max value
                if (valueValue > maxValue[j]) {
                    maxValue[j] = valueValue;
                    t[j] = createMaxElement(startX + half_width / 2, startY, formatNumber(valueValue), _color);
                }

                startX = startX + (columnWidth + innerPadding);
            }

        }

        // Max Element 
        for (int i = 0; i < t.length; i++) {
            root.append(t[i]);
        }

        return new JSONObject().put("root", root);
    }
}