package timeflow.views;

import timeflow.model.*;
import timeflow.views.ListView.LinkIt;
import timeflow.data.db.*;
import timeflow.data.time.*;

import javax.swing.*;

import timeflow.util.*;

import java.awt.*;
import java.awt.event.*;
import java.util.*;

public class BarGraphView extends AbstractView
{

        BarGraph graph = new BarGraph();
        JPanel controls;
        ArrayList<BarData> bars;

        enum Aggregate
        {

                TOTAL, AVERAGE, COUNT
        };
        Aggregate agg;
        JComboBox splitFieldChoice, numFieldChoice;

        public BarGraphView(TFModel model)
        {
                super(model);

                setLayout(new BorderLayout());
                controls = new JPanel();
                add(controls, BorderLayout.NORTH);
                controls.setLayout(null);
                controls.setBackground(Color.white);

                JScrollPane scrollPane = new JScrollPane(graph);
                add(scrollPane, BorderLayout.CENTER);

                makeTop();
        }

        protected JComponent _getControls()
        {
                return controls;
        }

        void makeTop()
        {
                int x = 10, y = 10;
                int ch = 25, pad = 5, cw = 160;

                controls.removeAll();
                TFModel model = getModel();
                if (model.getDB() == null || model.getDB().size() == 0)
                {
                        JLabel empty = new JLabel("Empty database");
                        controls.add(empty);
                        empty.setBounds(x, y, cw, ch);
                        return;
                }

                JLabel top = new JLabel("For each value of");
                controls.add(top);
                top.setBounds(x, y, cw, ch);
                y += ch + pad;

                splitFieldChoice = new JComboBox();
                String splitSelection = null;
                for (Field f : DBUtils.categoryFields(model.getDB()))
                {
                        splitFieldChoice.addItem(f.getName());
                        if (f == graph.splitField)
                        {
                                splitSelection = f.getName();
                        }
                }
                controls.add(splitFieldChoice);
                splitFieldChoice.setBounds(x, y, cw, ch);
                y += ch + 3 * pad;

                if (splitSelection != null)
                {
                        splitFieldChoice.setSelectedItem(splitSelection);
                } else if (getModel().getColorField() != null)
                {
                        splitFieldChoice.setSelectedItem(getModel().getColorField().getName());
                }
                splitFieldChoice.addActionListener(new ActionListener()
                {

                        @Override
                        public void actionPerformed(ActionEvent e)
                        {
                                graph.redo();
                        }
                });

                JLabel showLabel = new JLabel("show");
                controls.add(showLabel);
                showLabel.setBounds(x, y, cw, ch);
                y += ch + pad;

                numFieldChoice = new JComboBox();
                numFieldChoice.addItem("Number of events");
                final ArrayList<Field> valueFields = new ArrayList<Field>();
                for (Field f : model.getDB().getFields(Double.class))
                {
                        numFieldChoice.addItem("Total: " + f.getName());
                        numFieldChoice.addItem("Average: " + f.getName());
                        valueFields.add(f);
                }
                controls.add(numFieldChoice);
                numFieldChoice.setBounds(x, y, cw, ch);

                boolean chosen = false;
                for (int i = 0; i < numFieldChoice.getItemCount(); i++)
                {
                        if (numFieldChoice.getItemAt(i).equals(graph.lastValueMenuChoice))
                        {
                                numFieldChoice.setSelectedIndex(i);
                                chosen = true;
                        }
                }
                if (!chosen)
                {
                        Field size = getModel().getDB().getField(VirtualField.SIZE);
                        if (size != null)
                        {
                                numFieldChoice.setSelectedItem("Total: " + size.getName());
                        }
                }
                numFieldChoice.addActionListener(new ActionListener()
                {

                        @Override
                        public void actionPerformed(ActionEvent e)
                        {

                                graph.redo();
                        }
                });
                revalidate();
                repaint();
        }

        void reset()
        {
                makeTop();
                graph.redo();
                revalidate();
                repaint();
        }

        @Override
        protected void _note(TFEvent e)
        {
                if (e.affectsSchema())
                {
                        reset();
                } else
                {
                        graph.redo();
                }
                repaint();
        }

        @Override
        public String getName()
        {
                return "Bar Graph";
        }

        @Override
        protected void onscreen(boolean majorChange)
        {
                reset();
        }

        class BarData
        {

                Object thing;
                double num;

                BarData(Object thing, double num)
                {
                        this.thing = thing;
                        this.num = num;
                }
        }

        class BarGraph extends JPanel
        {

                int numVals = 0;
                int rowHeight = 30;
                int barHeight = 20;
                int labelX = 10, barLeft = 300, barRight;
                int topY = 45;
                int numX = 210;
                Field splitField, valueField;
                String lastValueMenuChoice;
                double min, max;

                void redo()
                {
                        bars = new ArrayList<BarData>();
                        splitField = getModel().getDB().getField((String) splitFieldChoice.getSelectedItem());
                        if (splitField != null)
                        {
                                int n = numFieldChoice.getSelectedIndex();

                                if (n == 0)
                                {
                                        agg = Aggregate.COUNT;
                                } else
                                {
                                        agg = n % 2 == 1 ? Aggregate.TOTAL : Aggregate.AVERAGE;
                                }

                                if (agg == Aggregate.COUNT)
                                {
                                        Bag<String> bag = DBUtils.countValues(getModel().getActs(), splitField);
                                        for (String s : bag.list())
                                        {
                                                bars.add(new BarData(s, bag.num(s)));
                                        }
                                } else
                                {
                                        lastValueMenuChoice = (String) numFieldChoice.getSelectedItem();
                                        int colon = lastValueMenuChoice.indexOf(':');
                                        valueField = getModel().getDB().getField(lastValueMenuChoice.substring(colon + 2));
                                        DoubleBag<String> bag = new DoubleBag<String>();
                                        for (Act a : getModel().getActs())
                                        {
                                                if (splitField.getType() == String.class)
                                                {
                                                        bag.add(a.getString(splitField), a.getValue(valueField));
                                                } else
                                                {
                                                        String[] tags = a.getTextList(splitField);
                                                        for (String tag : tags)
                                                        {
                                                                bag.add(tag, a.getValue(valueField));
                                                        }
                                                }
                                        }
                                        boolean isSum = agg == Aggregate.TOTAL;
                                        for (String s : bag.list(isSum))
                                        {
                                                bars.add(new BarData(s, isSum ? bag.num(s) : bag.average(s)));
                                        }
                                }
                        }
                        revalidate();
                        repaint();
                }

                public void paintComponent(Graphics g1)
                {
                        Graphics2D g = (Graphics2D) g1;
                        int w = getSize().width, h = getSize().height;
                        g.setColor(Color.white);
                        g.fillRect(0, 0, w, h);
                        TFModel model = getModel();
                        Display display = model.getDisplay();

                        if (display.emptyMessage(g, model))
                        {
                                return;
                        }

                        if (bars == null)
                        {
                                return;
                        }

                        if (bars.size() == 0)
                        {
                                g.setColor(Color.gray);
                                g.drawString("(No data selected.)", 10, 30);
                                return;
                        }

                        int n = bars.size();
                        max = bars.get(0).num;
                        min = Math.min(0, bars.get(n - 1).num);


                        barRight = w - 30;

                        int zero = scaleX(0);
                        boolean isInColor = (splitField != null && getModel().getColorField() == splitField);

                        // draw header
                        int titleY = topY - 15;
                        g.setColor(Color.black);
                        g.setFont(display.big());
                        g.drawString(splitField.getName().toUpperCase(), labelX, titleY);
                        String aggLabel = agg.toString();
                        if (agg != Aggregate.COUNT)
                        {
                                aggLabel += " " + valueField.getName().toUpperCase();
                        }
                        g.drawString(aggLabel, barLeft, titleY);
                        g.setFont(display.plain());
                        FontMetrics fm = display.plainFontMetrics();
                        // draw bars

                        for (int i = 0; i < n; i++)
                        {
                                int y = topY + i * rowHeight;
                                int ty = y + barHeight;
                                BarData data = bars.get(i);

                                Color c = null;

                                g.setColor(Color.gray);

                                // label value
                                boolean missing = data.thing == null || (data.thing.toString().length() == 0);
                                String label = missing ? "[missing]"
                                        : display.format(display.toString(data.thing), 25, false);
                                if (isInColor)
                                {
                                        g.setColor(missing ? Color.gray : display.makeColor(data.thing.toString()));
                                        display.makeColor(label);
                                }
                                g.drawString(label, labelX, ty);

                                // label number
                                String numLabel = display.format(data.num);
                                g.drawString(numLabel, numX + 70 - fm.stringWidth(numLabel), ty);

                                // draw bar.
                                g.setColor(missing ? Color.lightGray : (isInColor ? c : Display.barColor));
                                int x = scaleX(data.num);
                                int a = Math.min(x, zero);
                                int b = Math.max(x, zero);
                                g.fillRect(a, y + 5, b - a, barHeight);
                        }
                }

                int scaleX(double x)
                {
                        if (max == min)
                        {
                                return barLeft;
                        }
                        return (int) (barLeft + (barRight - barLeft) * (x - min) / (max - min));
                }

                public Dimension getPreferredSize()
                {
                        return new Dimension(400, 100 + rowHeight * (bars == null ? 0 : bars.size()));
                }
        }
}