source: trunk/src/org/expeditee/items/widgets/charts/AbstractCategory.java@ 919

/**
 *    AbstractCategory.java
 *    Copyright (C) 2010 New Zealand Digital Library, http://expeditee.org
 *
 *    This program is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.expeditee.items.widgets.charts;

import java.awt.Color;
import java.util.Collection;

import org.expeditee.gui.AttributeValuePair;
import org.expeditee.items.Text;
import org.jfree.data.category.DefaultCategoryDataset;

public abstract class AbstractCategory extends AbstractValueAxis {

        private DefaultCategoryDataset _data;

        protected DefaultCategoryDataset getChartData() {
                if (_data == null)
                        _data = new DefaultCategoryDataset();
                return _data;
        }

        public AbstractCategory(Text source, String[] args) {
                super(source, args);
        }

        @Override
        protected void clearData() {
                _data.clear();
        }

        @Override
        protected boolean addCategoryData(String categoryName,
                        Collection<Text> items, boolean swap) {
                boolean foundData = false;
                Color newColor = null;
                for (Text i : items) {
                        try {
                                if (!i.isLineEnd()) {
                                        Text t = (Text) i;
                                        AttributeValuePair avp = new AttributeValuePair(t.getText());
                                        if (avp != null) {
                                                if (swap) {
                                                        String attribute = avp.getAttribute();
                                                        _data.setValue(avp.getDoubleValue(), attribute,
                                                                        categoryName);
                                                        if (_paints.get(attribute) == null) {
                                                                _paints.put(attribute, i.getBackgroundColor());
                                                        }
                                                } else {
                                                        _data.setValue(avp.getDoubleValue(), categoryName,
                                                                        avp.getAttribute());
                                                }
                                                foundData = true;
                                                if (newColor == null)
                                                        newColor = i.getBackgroundColor();
                                        }
                                }
                        } catch (Exception e) {

                        }
                }
                if (foundData && !swap) {
                        _paints.put(categoryName, newColor);
                }
                return foundData;
        }
}