TrendGraph.java

Go to the documentation of this file.

/*
 * Copyright (c) 2008, AIST, the University of Tokyo and General Robotix Inc.
 * All rights reserved. This program is made available under the terms of the
 * Eclipse Public License v1.0 which accompanies this distribution, and is
 * available at http://www.eclipse.org/legal/epl-v10.html
 * Contributors:
 * General Robotix Inc.
 * National Institute of Advanced Industrial Science and Technology (AIST) 
 */
package com.generalrobotix.ui.view.graph;

import java.util.*;
import org.eclipse.swt.graphics.RGB;
import com.generalrobotix.ui.grxui.Activator;


public class TrendGraph {

    // -----------------------------------------------------------------
    private static final double MAX_DIV = 5;
    private static final double LOG10 = Math.log(10);

    public static final int SUCCEEDED = 0; 
    public static final int NOT_MATCHED = 1;
    public static final int NOT_SUPPORTED = 2;
    
    private static final String[] colorTable_ = {"green", "yellow", "cyan", "magenta", "red", "blue" };
      
    private static final int numColors_;
    static {
        numColors_ = colorTable_.length;
    }
    public RGB getGraphColor(int index) {
        int tmpcolorCounter_ = colorCounter_;
        tmpcolorCounter_ = tmpcolorCounter_ + index;
        if (tmpcolorCounter_ >= numColors_) {
            tmpcolorCounter_ = 0;
        }
        return Activator.getDefault().getColor(colorTable_[tmpcolorCounter_]).getRGB();
    }

    // -----------------------------------------------------------------
    private int colorCounter_;

    private DataKind dataKind_;
    private XYLineGraph graph_;
    private TrendGraphModel model_;
    private AxisInfo yAxisInfo_;
    private AxisInfo xAxisInfo_;

    private HashMap<String, DataItemInfo> dataItemInfoMap_;

    private String nodeName_;
    //private boolean projectRead_;

//    private DataItemNode[] dataItemNodeArray_;
//    private int dataItemNodeCount_;

    // -----------------------------------------------------------------
    public TrendGraph(TrendGraphModel model, String node) {
        model_ = model;
        nodeName_ = node;

        xAxisInfo_ = model_.getTimeAxisInfo();
        yAxisInfo_ = new AxisInfo(0, 1);
        yAxisInfo_.unitFont = Activator.getDefault().getFont( "dialog12" );
        yAxisInfo_.unitXOfs = 5;
        yAxisInfo_.unitYOfs = 15;

        dataKind_ = null;

        colorCounter_ = 0;

        dataItemInfoMap_ = new HashMap<String, DataItemInfo>();
   }

    // -----------------------------------------------------------------

    public void setGraph(XYLineGraph graph, LegendPanel legend) {
        graph_ = graph;
        graph_.setAxisInfo(
            XYLineGraph.AXIS_BOTTOM,
            xAxisInfo_
        );

        graph_.setLegend(legend);
        /*
        graph_.setAxisInfo( 
            XYLineGraph.AXIS_LEFT,
            yAxisInfo_
        );
        */
    }

    public XYLineGraph getGraph() {
        return graph_;
    }

    public void repaint() {
        graph_.redraw();
    }

    /*
    public void setDataKind(
        String dataKind,
        double base,
        double extent
    ) {
        dataKind_ = GraphProperties.getDataKindFromName(dataKind);
        yAxisInfo_ = new AxisInfo(base, extent);
        _updateDiv();
        yAxisInfo_.unitFont = new Font("dialog", Font.PLAIN, 12);
        yAxisInfo_.unitXOfs = 5;
        yAxisInfo_.unitYOfs = 7;
        yAxisInfo_.unitLabel = dataKind_.unitLabel;
        yAxisInfo_.factor = dataKind_.factor;
        graph_.setAxisInfo(
            XYLineGraph.AXIS_LEFT,
            yAxisInfo_
        );
    }
    */

    public int addDataItem(
                DataItemInfo di
    ) {
        //projectRead_ = false;

        DataKind dataKind = GraphProperties.getDataKindFromAttr(di.dataItem.getFullAttributeName());
        if (dataKind == null) { 
            return NOT_SUPPORTED;
        }
        if (dataKind_ != null) {
            if (!dataKind.equals(dataKind_)) {
                return NOT_MATCHED;
            }
        } else {
            dataKind_ = dataKind;
            //yAxisInfo_ = new AxisInfo(dataKind_.base, dataKind_.extent);
            //yAxisInfo_.unitFont = new Font("dialog", Font.PLAIN, 12);
            //yAxisInfo_.unitXOfs = 5;
            //yAxisInfo_.unitYOfs = 7;
            yAxisInfo_.base = dataKind_.base;
            yAxisInfo_.extent = dataKind_.extent;
            yAxisInfo_.unitLabel = dataKind_.unitLabel;
            yAxisInfo_.factor = dataKind_.factor;
            _updateDiv();
            graph_.setAxisInfo(
                XYLineGraph.AXIS_LEFT,
                yAxisInfo_
            );
//            SEString dk = new SEString(dataKind_.name);
            //System.out.println(nodeName_ + "." + GraphNode.DATA_KIND + "=" + dk.toString());
//            world_.updateAttribute(nodeName_ + "." + GraphNode.DATA_KIND + "=" + dk.toString());
//            SEDoubleArray vr = new SEDoubleArray(new double[]{dataKind_.base, dataKind_.base + dataKind_.extent});
//            world_.updateAttribute(nodeName_ + "." + GraphNode.V_RANGE + "=" + vr.toString());
        }
        if(di.color==null)
                di.color = Activator.getDefault().getColor(colorTable_[colorCounter_]).getRGB();
        if(di.legend==null)
            di.legend = di.dataItem.toString();
        if(_addDataItem(di))
                if (++colorCounter_ >= numColors_)
            colorCounter_ = 0;
 
        return SUCCEEDED;
    }

    public void removeDataItem(
        DataItemInfo dataItemInfo
    ) {
        model_.removeDataItem(dataItemInfo.dataItem);
        String key = dataItemInfo.dataItem.toString();

        graph_.removeDataSeries(dataItemInfo.dataSeries);

        dataItemInfoMap_.remove(key);
        if (dataItemInfoMap_.isEmpty()) {
            dataKind_ = null;
            //yAxisInfo_ = null; 
            graph_.setAxisInfo(
                XYLineGraph.AXIS_LEFT,
                null
            );
//            SEString dk = new SEString("");
//            world_.updateAttribute(nodeName_ + "." + GraphNode.DATA_KIND + "=" + dk.toString());
//            SEDoubleArray vr = new SEDoubleArray(new double[]{0, 1});
//            world_.updateAttribute(nodeName_ + "." + GraphNode.V_RANGE + "=" + vr.toString());
        }
//        for (int i = 0; i < dataItemListenerList_.size(); i++) {
//            DataItemListener listener = dataItemListenerList_.getListener(i);
//            listener.dataItemRemoved(dii);
//        }
    }

    public int getNumDataItems() {
        return dataItemInfoMap_.size();
    }

    public DataItemInfo[] getDataItemInfoList() {
        return dataItemInfoMap_.values().toArray(new DataItemInfo[0]);
    }

   public void setDataItemInfo(
        DataItemInfo dii
    ) {
//        DataSeries ds = (DataSeries)dataSeriesMap_.get(dii.dataItem.toString());
//        Color pc = graph_.getStyle(ds);
//        String pl = graph_.getLegendLabel(ds);
//        String diNodeName = _getDataItemNodeName(dii.dataItem);
        _setDataItemInfo(dii);
 /*       if (!pc.equals(dii.color)) { 
           world_.updateAttribute(
                diNodeName + "." + DataItemNode.COLOR + "=" + (String)revColorMap_.get(dii.color)
            );
        }
        if (!pl.equals(dii.legend)) {
            world_.updateAttribute(
                diNodeName + "." + DataItemNode.LEGEND + "=" + dii.legend
            );
        }*/
    }

    private void _setDataItemInfo(
        DataItemInfo dii
    ) {
        graph_.setStyle(dii.dataSeries, dii.color);
        graph_.setLegendLabel(dii.dataSeries, dii.legend);
    }

    public void setRange(
        double base,
        double extent
    ) {
        _setRange(base, extent);
//        SEDoubleArray vr = new SEDoubleArray(new double[]{base, base + extent});
//        world_.updateAttribute(nodeName_ + "." + GraphNode.V_RANGE + "=" + vr.toString());
    }

    public String getUnitLabel() {
        return yAxisInfo_.unitLabel;
    }

    public double getBase() {
        return yAxisInfo_.base;
    }

    public double getExtent() {
        return yAxisInfo_.extent;
    }

    public void setTimeRangeAndPos(
        double timeRange,
        double markerPos
    ) {
        _setTimeRangeAndPos(timeRange, markerPos);
//        SEDoubleArray tr = new SEDoubleArray(new double[]{timeRange, markerPos});
        //SEDouble hr = new SEDouble(timeRange);
        //SEDouble mp = new SEDouble(markerPos);
//        world_.updateAttribute(nodeName_ + "." + GraphNode.TIME_RANGE + "=" + tr.toString());
    }

//    public void addDataItemListener(DataItemListener listener) {
//        dataItemListenerList_.addListener(listener);
//    }

  /*  public void removeDataItemListener(DataItemListener listener) {
        dataItemListenerList_.removeListener(listener);
    }*/

    // -----------------------------------------------------------------
    // WorldReplaceListener
/*    public void replaceWorld(SimulationWorld world) {
        world_ = world;
        while (dataItemList_.size() > 0) {
            DataItem di = (DataItem)dataItemList_.get(0);
            model_.removeDataItem(di, false);
            String key = di.toString();
            dataItemInfoMap_.remove(key);
            DataSeries ds = (DataSeries)dataSeriesMap_.get(key);
            graph_.removeDataSeries(ds);
            dataSeriesMap_.remove(key);
            dataItemList_.remove(0);
        }

        //projectRead_ = true;

        dataKind_ = null;

        world_.addAttributeListener(this);
        world_.addNodeListener(this);
    }
*/
    // -----------------------------------------------------------------
    // AttributeListener
 /*   public void attributeChanged(
        SimulationNode node,
        String attribute,
        StringExchangeable value
    ) {
        String nodeType = node.getNodeName();
        String name = node.getName();
        if (nodeType.equals(GraphNode.NODE_TYPE)
            && name.equals(nodeName_)
        ) {
            if (attribute.equals(GraphNode.TIME_RANGE)) {
                // timeRange
                //System.out.println("set Time Range: " + value);
                SEDoubleArray tr = (SEDoubleArray)value;
                _setTimeRangeAndPos(tr.doubleValue(0), tr.doubleValue(1));
            } else if (attribute.equals(GraphNode.DATA_KIND)) {
                // dataKind
                _setDataKind(value.toString());
            } else if (attribute.equals(GraphNode.H_RANGE)) {
                // hRange
                _setTimeRange(((SEDouble)value).doubleValue());
                //System.out.println("set HRange: " + value);
            } else if (attribute.equals(GraphNode.MARKER_POS)) {
                // markerPos
                _setMarkerPos(((SEDouble)value).doubleValue());
            } else if (attribute.equals(GraphNode.V_RANGE)) {
                // vRange
                SEDoubleArray vr = (SEDoubleArray)value;
                _setRange(
                    vr.doubleValue(0),
                    vr.doubleValue(1) - vr.doubleValue(0)
                );
            }
        } else if (nodeType.equals(DataItemNode.NODE_TYPE)
            && name.startsWith(nodeName_)
        ) {
            //System.out.println("### data item attribute changed = " + attribute);
            DataItemNode din = (DataItemNode)node;
            if (attribute.equals("color")) {
                String diname = din.getDataItemName();
                if (diname == null) {
                    //System.out.println("color $$$$ Data item name is not decided.");
                    return;
                }
                DataItemInfo dii = (DataItemInfo)dataItemInfoMap_.get(din.getDataItemName());
                if (dii == null) {
                    //System.out.println("color $$$$ Data item info is not found. (" + diname + ")");
                    return;
                }
                dii.color = (Color)colorMap_.get(value.getValue());
                _setDataItemInfo(dii);
                                //
                graph_.repaint();
                graph_.getLegendPanel().repaint();
            }
            if (attribute.equals("legend")) {
                String diname = din.getDataItemName();
                if (diname == null) {
                    //System.out.println("legend $$$$ Data item name is not decided.");
                    return;
                }
                DataItemInfo dii = (DataItemInfo)dataItemInfoMap_.get(din.getDataItemName());
                if (dii == null) {
                    //System.out.println("legend $$$$ Data item info is not found. (" + diname + ")");
                    return;
                }
                dii.legend = value.toString();
                _setDataItemInfo(dii);
                graph_.repaint();
                graph_.getLegendPanel().repaint();
            }
        }
    }*/

    // -----------------------------------------------------------------
    private boolean _addDataItem(
        DataItemInfo dataItemInfo
    ) {
        DataItem dataItem = dataItemInfo.dataItem;
        String key = dataItem.toString();

        DataSeries ds = model_.addDataItem(dataItem);
        if (ds == null) {
            return false;
        }
        dataItemInfo.dataSeries = ds;

        if (dataItemInfoMap_.containsKey(key)) {
            return false;
        }

        dataItemInfoMap_.put(key, dataItemInfo);

 
        graph_.addDataSeries(
            ds, xAxisInfo_, yAxisInfo_,
            dataItemInfo.color,
            dataItemInfo.legend
        );

        return true;
    }

/*  
     private void _setDataKind( String dataKind ) {
        if (dataKind_ != null) {
            if (dataKind.equals(dataKind_.name)) {
                return;
            }
        }
        dataKind_ = GraphProperties.getDataKindFromName(dataKind);
        if (dataKind_ != null) {
            //yAxisInfo_.base = dataKind_.base;
            //yAxisInfo_.extent = dataKind_.extent;
            yAxisInfo_.unitLabel = dataKind_.unitLabel;
            yAxisInfo_.factor = dataKind_.factor;
            _updateDiv();
            graph_.setAxisInfo(
                XYLineGraph.AXIS_LEFT,
                yAxisInfo_
            );
        }
    }
*/
    private void _setRange(
        double base,
        double extent
    ) {
        yAxisInfo_.base = base;
        yAxisInfo_.extent = extent;
        _updateDiv();
    }

    String _getDataItemNodeName(DataItem di) {
        StringBuffer sb = new StringBuffer(nodeName_);
        sb.append('.');
        sb.append(nodeName_);
        if (di.object != null) {
            sb.append('_');
            sb.append(di.object);
        }
        sb.append('_');
        sb.append(di.node);
        sb.append('_');
        sb.append(di.attribute);
        if (di.index >= 0) {
            sb.append('_');
            sb.append(di.index);
        }

        return sb.toString();
    }
    
    String getNodeName() {
        return nodeName_;
    }

    private void _setTimeRangeAndPos(
        double timeRange,
        double markerPos
    ) {
        model_.setRangeAndPos(timeRange, markerPos);
    }

    private void _updateDiv() {
        double sMin = yAxisInfo_.extent / MAX_DIV;
        int eMin = (int)Math.floor(Math.log(sMin) / LOG10);
        double step = 0;
        String format = "0";
        int e = eMin;
        boolean found = false;
        while (!found) {
            int m = 1;
            for (int i = 1; i <= 3; i++) {
                step = m * Math.pow(10.0, e);
                if (sMin <= step) { // && step <= sMax
                    if (e < 0) {
                        char[] c = new char[-e + 2];
                        c[0] = '0';
                        c[1] = '.';
                        for (int j = 0; j < -e; j++) {
                            c[j + 2] = '0';
                        }
                        format = new String(c);
                    }
                    found = true;
                    break;
                }
                m += (2 * i - 1);
            }
            e++;
        }
        yAxisInfo_.tickEvery = step;
        yAxisInfo_.labelEvery = step;
        yAxisInfo_.gridEvery = step;
        yAxisInfo_.labelFormat = format;
    }

        public void clearDataItem() {
                DataItemInfo[] info = getDataItemInfoList();
                for (int j = 0; j < info.length; j++)
                        removeDataItem(info[j]);
        }
}