Changeset 213 for trunk/src/org/expeditee/items/widgets/charts/AbstractChart.java

Timestamp:

08/11/08 18:03:36 (16 years ago)

Author:

ra33

Message:

Added Charts

File:

• trunk/src/org/expeditee/items/widgets/charts/AbstractChart.java (modified) (4 diffs)

trunk/src/org/expeditee/items/widgets/charts/AbstractChart.java

@@ -1 +1 @@
 package org.expeditee.items.widgets.charts;
 
+import java.awt.Color;
 import java.awt.Font;
+import java.awt.event.KeyEvent;
+import java.awt.event.KeyListener;
+import java.util.Collection;
 
 import javax.swing.JPanel;
 
-import org.expeditee.gui.AttributeUtils;
-import org.expeditee.gui.AttributeValuePair;
+import org.expeditee.gui.ColorUtils;
 import org.expeditee.gui.Frame;
+import org.expeditee.gui.FrameIO;
+import org.expeditee.gui.FrameKeyboardActions;
+import org.expeditee.gui.FunctionKey;
+import org.expeditee.items.Item;
 import org.expeditee.items.Text;
-import org.expeditee.items.widgets.InteractiveWidget;
+import org.expeditee.items.widgets.DataFrameWidget;
 import org.jfree.chart.ChartPanel;
 import org.jfree.chart.JFreeChart;
-import org.jfree.chart.labels.PieSectionLabelGenerator;
-import org.jfree.chart.plot.PiePlot;
 import org.jfree.chart.plot.Plot;
+import org.jfree.chart.plot.PlotOrientation;
+import org.jfree.chart.plot.ValueAxisPlot2;
 import org.jfree.chart.title.LegendTitle;
 import org.jfree.chart.title.TextTitle;
-import org.jfree.data.general.AbstractDataset;
-import org.jfree.data.general.DefaultPieDataset;
-
-public abstract class AbstractChart extends InteractiveWidget {
+
+public abstract class AbstractChart extends DataFrameWidget {
 
         protected static final String DEFAULT_TITLE = "";
+
+        protected static final String DEFAULT_XAXIS = "X";
+
+        protected static final String DEFAULT_YAXIS = "Y";
 
         private JFreeChart _chart = null;
@@ -51 +60 @@
                 ChartPanel cp = new ChartPanel(_chart);
                 cp.setPopupMenu(null);
+                final Item firstItem = getItems().get(0);
+                cp.addKeyListener(new KeyListener() {
+                        public void keyPressed(KeyEvent e) {
+                                int index = e.getKeyCode() - KeyEvent.VK_F1 + 1;
+                                // Make sure the key is within range
+                                if (index < 0 || index >= FunctionKey.values().length)
+                                        return;
+
+                                FunctionKey key = FunctionKey.values()[index];
+                                switch (key) {
+                                case SizeUp:
+                                        FrameKeyboardActions.SetSize(firstItem, 1, false, true);
+                                        break;
+                                case SizeDown:
+                                        FrameKeyboardActions.SetSize(firstItem, -1, false, true);
+                                        break;
+                                case ChangeColor:
+                                        if (e.isControlDown()) {
+                                                Color newColor = ColorUtils.getNextColor(getSource()
+                                                                .getFillColor(), Frame.COLOR_WHEEL, null);
+                                                if (e.isShiftDown()) {
+                                                        newColor = null;
+                                                        _chart.getPlot().setBackgroundPaint(
+                                                                        Plot.DEFAULT_BACKGROUND_PAINT);
+                                                } else {
+                                                        _chart.getPlot().setBackgroundPaint(newColor);
+                                                }
+                                                getSource().setFillColor(newColor);
+                                        } else {
+                                                Color newColor = ColorUtils.getNextColor(getSource()
+                                                                .getBackgroundColor(), Frame.COLOR_WHEEL, null);
+                                                if (e.isShiftDown()) {
+                                                        newColor = null;
+                                                        _chart
+                                                                        .setBackgroundPaint(JFreeChart.DEFAULT_BACKGROUND_PAINT);
+                                                } else {
+                                                        _chart.setBackgroundPaint(newColor);
+                                                }
+                                                getSource().setBackgroundColor(newColor);
+
+                                        }
+                                        break;
+                                }
+                        }
+
+                        public void keyReleased(KeyEvent e) {
+                        }
+
+                        public void keyTyped(KeyEvent e) {
+                        }
+                });
                 super._swingComponent = cp;
+                Color c = getSource().getFillColor();
+                if (c != null) {
+                        _chart.getPlot().setBackgroundPaint(c);
+                }
+                c = getSource().getBackgroundColor();
+                if (c != null) {
+                        _chart.setBackgroundPaint(c);
+                }
         }
 
@@ -61 +129 @@
                 // Get the data from the linked frame
                 Frame dataFrame = getDataFrame();
-                
+
                 _chart.clearSubtitles();
 
@@ -92 +160 @@
          * @param dataFrame
          */
-        protected abstract void refreshPlot(Frame dataFrame, Plot plot);
+        protected void refreshPlot(Frame dataFrame, Plot simplePlot) {
+                if (this.fixedAxis())
+                        return;
+                ValueAxisPlot2 plot = (ValueAxisPlot2) simplePlot;
+                if (dataFrame.hasAnnotation("xaxis")) {
+                        plot.getDomainAxis()
+                                        .setLabel(dataFrame.getAnnotationValue("xaxis"));
+                }
+                if (dataFrame.hasAnnotation("yaxis")) {
+                        plot.getRangeAxis().setLabel(dataFrame.getAnnotationValue("yaxis"));
+                }
+                if (dataFrame.hasAnnotation("horizontal")) {
+                        plot.setOrientation(PlotOrientation.HORIZONTAL);
+                } else {
+                        plot.setOrientation(PlotOrientation.VERTICAL);
+                }
+        }
+
+        private boolean fixedAxis() {
+                return !(this instanceof ValueAxisPlot2);
+        }
 
         /**
          * @param dataFrame
          */
-        protected abstract void refreshData(Frame dataFrame);
+        protected void refreshData(Frame dataFrame) {
+                clearSubjects();
+                boolean swap = false;
+                if (dataFrame.hasAnnotation("swap")) {
+                        swap = true;
+                }
+                boolean foundData = false;
+                // First do a pass to see if there are any line ends
+                Collection<Text> textItems = dataFrame.getNonAnnotationText(true);
+                textItems.remove(dataFrame.getTitleItem());
+
+                // Search for line ends
+                for (Text text : textItems) {
+                        if (!text.isLineEnd()) {
+                                continue;
+                        }
+                        foundData |= addCategoryData(text.getText(), text
+                                        .getEnclosedNonAnnotationText(), swap);
+                }
+
+                // As a second option look for linked items on the frame
+                if (!foundData) {
+                        for (Text category : textItems) {
+                                if (category.isLineEnd() || category.getLink() == null)
+                                        continue;
+                                Frame linkFrame = FrameIO.LoadFrame(category.getAbsoluteLink());
+                                if (linkFrame == null)
+                                        continue;
+
+                                if (foundData |= addCategoryData(category.getText(), linkFrame
+                                                .getNonAnnotationText(true), swap)) {
+                                        addSubject(linkFrame);
+                                }
+                        }
+                }
+
+                // If there were no groupings in boxes then just get all the values on
+                // the frame
+                if (!foundData) {
+                        foundData |= addCategoryData("", textItems, swap);
+                }
+                if (foundData) {
+                        addSubject(dataFrame);
+                }
+        }
+
+        protected boolean addCategoryData(String categoryName,
+                        Collection<Text> items, boolean swap) {
+                return true;
+        }
+
+        @Override
+        public void setLink(String link) {
+                super.setLink(link);
+
+                Frame oldDataFrame = getDataFrame();
+                if (oldDataFrame != null)
+                        clearSubjects();
+                setDataFrame(null);
+        }
 
         @Override