package kg.apc.jmeter.vizualizers;

 import java.util.HashSet;

 import java.util.Iterator;

 import kg.apc.charting.AbstractGraphRow;

 import kg.apc.charting.DateTimeRenderer;

 import kg.apc.jmeter.JMeterPluginsUtils;

 import kg.apc.jmeter.graphs.AbstractOverTimeVisualizer;

 import kg.apc.jmeter.graphs.CompositeNotifierInterface;

 import org.apache.jmeter.reporters.ResultCollector;

 import org.apache.jmeter.samplers.SampleResult;

 import org.apache.jmeter.testelement.TestElement;

 import org.apache.jmeter.testelement.property.CollectionProperty;

 import org.apache.jmeter.testelement.property.JMeterProperty;

 import org.apache.jmeter.testelement.property.NullProperty;

 import org.apache.jmeter.testelement.property.PropertyIterator;

 public class CompositeGraphGui extends AbstractOverTimeVisualizer {

     private JCompositeRowsSelectorPanel compositeRowsSelectorPanel;

     public CompositeModel compositeModel;

     private long lastUpdate = 0;

     private static String CONFIG_PROPERTY = "COMPOSITE_CFG";

     public CompositeGraphGui() {

         graphPanel.getGraphObject().setDisplayPrecision(false);

         compositeModel = new CompositeModel();

         compositeRowsSelectorPanel = new JCompositeRowsSelectorPanel(compositeModel, this);

         compositeModel.setNotifier((CompositeNotifierInterface) compositeRowsSelectorPanel);

         graphPanel.replaceRowTab("Graphs", compositeRowsSelectorPanel, "Select graphs/rows to display");

         graphPanel.getGraphObject().setxAxisLabelRenderer(new DateTimeRenderer("HH:mm:ss"));

         graphPanel.getGraphObject().setReSetColors(true);

         graphPanel.getGraphObject().setxAxisLabel("Elapsed time");

         graphPanel.getGraphObject().setYAxisLabel("Scaled values");

         graphPanel.getGraphObject().getChartSettings().setExpendRows(true);

         CompositeResultCollector compositeResultCollector = new CompositeResultCollector();

         compositeResultCollector.setCompositeModel(compositeModel);

         setModel(compositeResultCollector);

         hideFilePanel();

         enableMaximize(false);

     }

     @Override

     protected JSettingsPanel createSettingsPanel() {

         return new JSettingsPanel(this,

                 JSettingsPanel.GRADIENT_OPTION

                 | JSettingsPanel.FINAL_ZEROING_OPTION

                 | JSettingsPanel.LIMIT_POINT_OPTION

                 | JSettingsPanel.MAXY_OPTION

                 | JSettingsPanel.RELATIVE_TIME_OPTION

                 | JSettingsPanel.AUTO_EXPAND_OPTION

                 | JSettingsPanel.MARKERS_OPTION);

     }

     @Override

     public String getLabelResource() {

         return this.getClass().getSimpleName();

     }

     @Override

     public String getStaticLabel() {

         return JMeterPluginsUtils.prefixLabel("Composite Graph");

     }

     @Override

     public TestElement createTestElement() {

         ResultCollector modelNew = getModel();

         if (modelNew == null) {

             modelNew = new CompositeResultCollector();

             ((CompositeResultCollector) modelNew).setCompositeModel(compositeModel);

             setModel(modelNew);

         }

         modifyTestElement(modelNew);

         modelNew.setComment(JMeterPluginsUtils.getWikiLinkText("CompositeGraph"));

         return modelNew;

     }

     @Override

     public void configure(TestElement te) {

         //log.info("Configure");

         super.configure(te);

         ((CompositeResultCollector) te).setCompositeModel(compositeModel);

         JMeterProperty data = te.getProperty(CONFIG_PROPERTY);

         if (!(data instanceof NullProperty)) {

             setConfig((CollectionProperty) data);

         }

     }

     @Override

     public void modifyTestElement(TestElement c) {

         super.modifyTestElement(c);

         c.setProperty(getConfig());

     }

     private CollectionProperty getConfig() {

         CollectionProperty ret = new CollectionProperty();

         CollectionProperty testplans = new CollectionProperty();

         CollectionProperty rows = new CollectionProperty();

         ret.setName(CONFIG_PROPERTY);

         Iterator<String[]> iter = compositeRowsSelectorPanel.getItems();

         while (iter.hasNext()) {

             String[] item = iter.next();

             testplans.addItem(item[0]);

             rows.addItem(item[1]);

         }

         ret.addItem(testplans);

         ret.addItem(rows);

         return ret;

     }

     private void setConfig(CollectionProperty properties) {

         PropertyIterator iter = properties.iterator();

         CollectionProperty testplans = (CollectionProperty) iter.next();

         CollectionProperty rows = (CollectionProperty) iter.next();

         if (rows.size() > 0) {

             PropertyIterator iterTestplans = testplans.iterator();

             PropertyIterator iterRows = rows.iterator();

             while (iterTestplans.hasNext() && iterRows.hasNext()) {

                 String testplan = iterTestplans.next().getStringValue();

                 String row = iterRows.next().getStringValue();

                 compositeRowsSelectorPanel.addItemsToComposite(testplan, row);

             }

         }

     }

     @Override

     public void updateGui() {

         Iterator<String[]> iter = compositeRowsSelectorPanel.getItems();

         HashSet<String> validRows = new HashSet<String>();

         while (iter.hasNext()) {

             String[] item = iter.next();

             AbstractGraphRow row = compositeModel.getRow(item[0], item[1]);

             if (row != null) {

                 String rowName = item[0] + " > " + item[1];

                 validRows.add(rowName);

                 if (!model.containsKey(rowName)) {

                     model.put(rowName, row);

                     // handle relative start times for JTL reload

                     if (relativeStartTime == 0 || relativeStartTime > row.getFirstTime()) {

                         relativeStartTime = row.getFirstTime();

                         handleRelativeStartTime();

                     }

                 }

             }

         }

         //remove invalid rows

         Iterator<String> iterModelRows = model.keySet().iterator();

         while (iterModelRows.hasNext()) {

             String rowName = iterModelRows.next();

             if (!validRows.contains(rowName)) {

                 iterModelRows.remove();

             }

         }

         super.updateGui();

     }

     @Override

     public void add(SampleResult sr) {

         super.add(sr);

         long time = System.currentTimeMillis();

         if (time > lastUpdate + 1000) {

             lastUpdate = time;

             updateGui();

         }

     }

     @Override

     public String getWikiPage() {

         return "CompositeGraph";

     }

     @Override

     public void clearData() {

         super.clearData();

         updateGui();

     }

 }