com.twosigma.beaker.chart.xychart.CombinedPlot.java Source code

Introduction

Here is the source code for com.twosigma.beaker.chart.xychart.CombinedPlot.java

Source

/*
 *  Copyright 2014 TWO SIGMA OPEN SOURCE, LLC
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package com.twosigma.beaker.chart.xychart;

import org.apache.commons.lang3.SerializationUtils;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

/**
 * CombinedPlot
 *
 */
public class CombinedPlot implements Cloneable, Serializable {
    private int initWidth = 640;
    private int initHeight = 480;
    private String title;
    private String xLabel;
    private List<XYChart> subplots = new ArrayList<>();
    private List<Integer> weights = new ArrayList<>();

    public CombinedPlot setInitWidth(int w) {
        this.initWidth = w;
        return this;
    }

    public Integer getInitWidth() {
        return this.initWidth;
    }

    public CombinedPlot setInitHeight(int h) {
        this.initHeight = h;
        return this;
    }

    public Integer getInitHeight() {
        return this.initHeight;
    }

    public CombinedPlot setTitle(String title) {
        this.title = title;
        return this;
    }

    public String getTitle() {
        return this.title;
    }

    public CombinedPlot setXLabel(String xLabel) {
        this.xLabel = xLabel;
        return this;
    }

    public CombinedPlot setxLabel(String xLabel) {
        this.xLabel = xLabel;
        return this;
    }

    public String getXLabel() {
        return this.xLabel;
    }

    public CombinedPlot add(XYChart plot, int weight) {
        this.subplots.add(plot);
        this.weights.add(weight);
        return this;
    }

    public CombinedPlot add(XYChart plot) {
        this.subplots.add(plot);
        this.weights.add(1);
        return this;
    }

    public CombinedPlot leftShift(Object obj) {
        if (obj instanceof XYChart) {
            this.add((XYChart) obj, 1);
        } else if (obj instanceof List && ((List) obj).size() == 2) {
            List list = (List) obj;
            XYChart plot = (XYChart) list.get(0);
            int weight = ((Number) list.get(1)).intValue();

            this.add(plot, weight);
        } else {
            throw new IllegalArgumentException("leftShift takes XYChart or List that hold a XYChart and weight");
        }
        return this;
    }

    public List<XYChart> getSubplots() {
        return this.subplots;
    }

    public List<Integer> getWeights() {
        return this.weights;
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
        return SerializationUtils.clone(this);
    }
}