package kg.apc.jmeter.vizualizers;

 import java.awt.BorderLayout;

 import java.awt.Dimension;

 import java.io.File;

 import java.io.FileNotFoundException;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.text.DecimalFormat;

 import java.text.Format;

 import java.util.ArrayList;

 import java.util.Collection;

 import java.util.Map;

 import java.util.concurrent.ConcurrentHashMap;

 import javax.swing.*;

 import javax.swing.border.Border;

 import javax.swing.border.EmptyBorder;

 import javax.swing.table.TableCellRenderer;

 import kg.apc.charting.GraphPanelChart;

 import kg.apc.jmeter.graphs.AbstractGraphPanelVisualizer;

 import org.apache.jmeter.gui.util.HeaderAsPropertyRenderer;

 import org.apache.jmeter.samplers.SampleResult;

 import org.apache.jmeter.save.CSVSaveService;

 import org.apache.jmeter.testelement.TestElement;

 import org.apache.jmeter.util.JMeterUtils;

 import org.apache.jmeter.visualizers.SamplingStatCalculator;

 import org.apache.jmeter.visualizers.StatGraphVisualizer;

 import org.apache.jmeter.visualizers.StatVisualizer;

 import org.apache.jorphan.gui.NumberRenderer;

 import org.apache.jorphan.gui.ObjectTableModel;

 import org.apache.jorphan.gui.RateRenderer;

 import org.apache.jorphan.gui.RendererUtils;

 import org.apache.jorphan.logging.LoggingManager;

 import org.apache.jorphan.reflect.Functor;

 import org.apache.log.Logger;

 /**

  * OMG, I had to copy full contents of StatVisualizer just to make its data

  * visible to JMeterPluginsCMD. This class is just invisible in menu;

  *

  * @see StatVisualizer

  */

 public class AggregateReportGui extends AbstractGraphPanelVisualizer {

     private Collection<String> emptyCollection = new ArrayList<String>();

     private static final long serialVersionUID = 241L;

     private static final Logger log = LoggingManager.getLoggerForClass();

     private static final String USE_GROUP_NAME = "useGroupName";

     private static final String SAVE_HEADERS = "saveHeaders";

     private static final String[] COLUMNS = {

         "sampler_label",

         "aggregate_report_count",

         "average",

         "aggregate_report_median",

         "aggregate_report_90%_line",

         "aggregate_report_min",

         "aggregate_report_max",

         "aggregate_report_error%",

         "aggregate_report_rate",

         "aggregate_report_bandwidth",

         "aggregate_report_stddev",};

     private final String TOTAL_ROW_LABEL = JMeterUtils.getResString("aggregate_report_total_label");

     private JTable myJTable;

     private JScrollPane myScrollPane;

     private final JButton saveTable =

             new JButton(JMeterUtils.getResString("aggregate_graph_save_table"));

     private final JCheckBox saveHeaders = // should header be saved with the data?

             new JCheckBox(JMeterUtils.getResString("aggregate_graph_save_table_header"), true);

     private final JCheckBox useGroupName =

             new JCheckBox(JMeterUtils.getResString("aggregate_graph_use_group_name"));

     private transient ObjectTableModel statModel;

     private final Map<String, SamplingStatCalculator> tableRows =

             new ConcurrentHashMap<String, SamplingStatCalculator>();

     public AggregateReportGui() {

         super();

         statModel = new ObjectTableModel(COLUMNS,

                 SamplingStatCalculator.class,

                 new Functor[]{

                     new Functor("getLabel"),

                     new Functor("getCount"),

                     new Functor("getMeanAsNumber"),

                     new Functor("getMedian"),

                     new Functor("getPercentPoint",

                     new Object[]{new Float(.900)}),

                     new Functor("getMin"),

                     new Functor("getMax"),

                     new Functor("getErrorPercentage"),

                     new Functor("getRate"),

                     new Functor("getKBPerSecond"),

                     new Functor("getStandardDeviation"),},

                 new Functor[]{null, null, null, null, null, null, null, null, null, null, null},

                 new Class[]{String.class, Long.class, Long.class, Long.class, Long.class,

                     Long.class, Long.class, String.class, String.class, String.class, String.class});

         clearData();

         init();

     }

     //do not insert this vizualiser in any JMeter menu

     @Override

     public Collection<String> getMenuCategories() {

         return emptyCollection;

     }

     // Column renderers

     private static final TableCellRenderer[] RENDERERS =

         new TableCellRenderer[]{

             null, // Label

             null, // count

             null, // Mean

             null, // median

             null, // 90%

             null, // Min

             null, // Max

             new NumberRenderer("#0.00%"), // Error %age

             new RateRenderer("#.0"), // Throughput

             new NumberRenderer("#.0"), // pageSize

             new NumberRenderer("#0.00"), // Std Dev.

         };

     // Column formats

     static final Format[] FORMATS =

         new Format[]{

             null, // Label

             null, // count

             null, // Mean

             null, // median

             null, // 90%

             null, // Min

             null, // Max

             new DecimalFormat("#0.00%"), // Error %age

             new DecimalFormat("#.0"), // Throughput

             new DecimalFormat("#.0"), // pageSize

             new DecimalFormat("#0.00"), // Std Dev.

         };

     @Override

     public String getLabelResource() {

         return "aggregate_report";

     }

     @Override

     public void add(SampleResult res) {

         if (!isSampleIncluded(res)) {

             return;

         }

         SamplingStatCalculator row;

         final String sampleLabel = res.getSampleLabel(useGroupName.isSelected());

         synchronized (tableRows) {

             row = tableRows.get(sampleLabel);

             if (row == null) {

                 row = new SamplingStatCalculator(sampleLabel);

                 tableRows.put(row.getLabel(), row);

                 statModel.insertRow(row, statModel.getRowCount() - 1);

             }

         }

         /*

          * Synch is needed because multiple threads can update the counts.

          */

         synchronized (row) {

             row.addSample(res);

         }

         SamplingStatCalculator tot = tableRows.get(TOTAL_ROW_LABEL);

         synchronized (tot) {

             tot.addSample(res);

         }

         //statModel.fireTableDataChanged();

     }

     /**

      * Clears this visualizer and its model, and forces a repaint of the table.

      */

     @Override

     public final void clearData() {

         synchronized (tableRows) {

             statModel.clearData();

             tableRows.clear();

             tableRows.put(TOTAL_ROW_LABEL, new SamplingStatCalculator(TOTAL_ROW_LABEL));

             statModel.addRow(tableRows.get(TOTAL_ROW_LABEL));

         }

     }

     /**

      * Main visualizer setup.

      */

     private void init() {

         this.setLayout(new BorderLayout());

         // MAIN PANEL

         JPanel mainPanel = new JPanel();

         Border margin = new EmptyBorder(10, 10, 5, 10);

         mainPanel.setBorder(margin);

         mainPanel.setLayout(new BoxLayout(mainPanel, BoxLayout.Y_AXIS));

         mainPanel.add(makeTitlePanel());

         // SortFilterModel mySortedModel =

         // new SortFilterModel(myStatTableModel);

         myJTable = new JTable(statModel);

         myJTable.getTableHeader().setDefaultRenderer(new HeaderAsPropertyRenderer());

         myJTable.setPreferredScrollableViewportSize(new Dimension(500, 70));

         RendererUtils.applyRenderers(myJTable, RENDERERS);

         myScrollPane = new JScrollPane(myJTable);

         this.add(mainPanel, BorderLayout.NORTH);

         this.add(myScrollPane, BorderLayout.CENTER);

         JPanel opts = new JPanel();

         opts.add(useGroupName, BorderLayout.WEST);

         opts.add(saveTable, BorderLayout.CENTER);

         opts.add(saveHeaders, BorderLayout.EAST);

         this.add(opts, BorderLayout.SOUTH);

     }

     @Override

     public void modifyTestElement(TestElement c) {

         super.modifyTestElement(c);

         c.setProperty(USE_GROUP_NAME, useGroupName.isSelected(), false);

         c.setProperty(SAVE_HEADERS, saveHeaders.isSelected(), true);

     }

     @Override

     public void configure(TestElement el) {

         super.configure(el);

         useGroupName.setSelected(el.getPropertyAsBoolean(USE_GROUP_NAME, false));

         saveHeaders.setSelected(el.getPropertyAsBoolean(SAVE_HEADERS, true));

     }

     @Override

     protected JSettingsPanel createSettingsPanel() {

         return new JSettingsPanel(this, 0);

     }

     @Override

     public String getWikiPage() {

         return "JMeterPluginsCMD";

     }

     @Override

     public GraphPanelChart getGraphPanelChart() {

         return new FakeGraphPanelChart();

     }

     @Override

     public String getStaticLabel() {

         return "Nobody never should not see this. No, no, no.";

     }

     private class FakeGraphPanelChart extends GraphPanelChart {

         public FakeGraphPanelChart() {

             super(false);

         }

         @Override

         public void saveGraphToCSV(File file) throws IOException {

             log.info("Saving CSV to " + file.getAbsolutePath());

             FileWriter writer = null;

             try {

                 writer = new FileWriter(file);

                 CSVSaveService.saveCSVStats(SynthesisReportGui.getAllTableData(statModel, FORMATS)

                                 , writer, saveHeaders.isSelected() ? COLUMNS : null);

             } catch (FileNotFoundException e) {

                 log.warn(e.getMessage());

             } catch (IOException e) {

                 log.warn(e.getMessage());

             } finally {

                 try {

                     writer.close();

                 } catch (IOException ex) {

                     log.warn("There was problem closing file stream", ex);

                 }

             }

         }

         @Override

         public void saveGraphToPNG(File file, int w, int h) throws IOException {

             throw new UnsupportedOperationException("This plugin type cannot be saved as image");

         }

     }

 }