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

Introduction

Here is the source code for com.jennifer.ui.chart.brush.EqualizerBrush.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 static com.jennifer.ui.util.DomUtil.el;

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

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

    @Override
    public void drawBefore() {
        root = el("g").translate(chart.area("x"), chart.area("y"));

        innerPadding = options.optInt("innerPadding", 5);
        outerPadding = options.optInt("outerPadding", 15);
        unit = options.optInt("unit", 5);
        gap = options.optInt("gap", 5);

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

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

        width = x.rangeBand();
        target = options.getJSONArray("target");
        half_width = (width - outerPadding * 2) / 2;
        barWidth = (width - outerPadding * 2 - (target.length() - 1) * innerPadding) / target.length();

    }

    @Override
    public Object draw() {
        for (int i = 0; i < count; i++) {
            double startX = x.get(i) - half_width;

            for (int j = 0, jLen = target.length(); j < jLen; j++) {
                Transform barGroup = root.group();

                double startY = y.get(chart.dataDouble(i, target.getString(j)));
                double padding = 1.5;
                double eY = zeroY;
                int eIndex = 0;

                JSONObject o = new JSONObject().put("x", startX).put("width", barWidth).put("fill",
                        color((int) Math.floor(eIndex / gap)));

                if (startY <= zeroY) {
                    while (eY > startY) {
                        double unitHeight = (eY - unit < startY) ? Math.abs(eY - startY) : unit;

                        o.put("y", eY - unitHeight).put("height", unitHeight);

                        eY -= unitHeight + padding;
                        eIndex++;
                    }
                } else {
                    while (eY < startY) {
                        double unitHeight = (eY + unit > startY) ? Math.abs(eY - startY) : unit;

                        o.put("y", eY).put("height", unitHeight);

                        eY += unitHeight + padding;
                        eIndex++;

                    }
                }

                barGroup.rect(o);

                startX += barWidth + innerPadding;
            }
        }

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