design/designer-base/src/main/java/com/fr/design/data/datapane/SummaryMethodComboBox.java

package com.fr.design.data.datapane;

import com.fr.data.util.function.AbstractDataFunction;
import com.fr.data.util.function.AverageFunction;
import com.fr.data.util.function.CountFunction;
import com.fr.data.util.function.MaxFunction;
import com.fr.data.util.function.MinFunction;
import com.fr.data.util.function.NoneFunction;
import com.fr.data.util.function.SumFunction;
import com.fr.design.gui.icombobox.UIComboBox;
import com.fr.design.i18n.Toolkit;
import com.fr.general.ComparatorUtils;
import com.fr.log.FineLoggerFactory;
import com.fr.plugin.chart.base.FirstFunction;

/**
 * 图表数据汇总方式下拉框
 *
 * 支持首个，最后一个，求和，平均，最大值，最小值，个数
 *
 */
public class SummaryMethodComboBox extends UIComboBox {
    public static final String[] CALCULATE_ARRAY = {Toolkit.i18nText("Fine-Design_Chart_Data_Function_First"), Toolkit.i18nText("Fine-Design_Chart_Data_Function_Last"),
            Toolkit.i18nText("Fine-Design_Chart_Data_Function_Sum"), Toolkit.i18nText("Fine-Design_Chart_Data_Function_Average"),
            Toolkit.i18nText("Fine-Design_Chart_Data_Function_Max"), Toolkit.i18nText("Fine-Design_Chart_Data_Function_Min"),
            Toolkit.i18nText("Fine-Design_Chart_Data_Function_Count")};
    public static final Class[] CLASS_ARRAY = {FirstFunction.class, NoneFunction.class, SumFunction.class, AverageFunction.class,
            MaxFunction.class, MinFunction.class, CountFunction.class};

    public SummaryMethodComboBox() {
        super(CALCULATE_ARRAY);
        setSelectedIndex(2);
    }

    public void reset() {
        this.setSelectedItem(Toolkit.i18nText("Fine-Design_Chart_Data_Function_Sum"));
    }

    /**
     * 更新公式选择.
     */
    public void populateBean(AbstractDataFunction function) {
        if (function != null) {
            for (int i = 0; i < CLASS_ARRAY.length; i++) {
                if (this.getModel() != null && this.getModel().getSize() > i
                        && ComparatorUtils.equals(function.getClass(), CLASS_ARRAY[i])) {
                    setSelectedIndex(i);
                    break;
                }
            }
        }
    }

    /**
     * 返回当前选择的公式
     */
    public AbstractDataFunction updateBean() {
        try {
            int selectIndex = getSelectedIndex();
            if (selectIndex >= 0 && selectIndex < CLASS_ARRAY.length) {
                return (AbstractDataFunction) CLASS_ARRAY[selectIndex].newInstance();
            }
        } catch (InstantiationException e) {
            FineLoggerFactory.getLogger().error("Function Error");
            return null;
        } catch (IllegalAccessException e) {
            FineLoggerFactory.getLogger().error("Function Error");
            return null;
        }

        return null;
    }
}