/*
 * Copyright (c) 2014-2017 Paul Scherrer Institute. All rights reserved.
 */

import ch.psi.pshell.device.DescStatsDouble;
import ch.psi.pshell.epics.Epics;
import ch.psi.pshell.plot.LinePlotErrorSeries;
import ch.psi.pshell.plot.LinePlotJFree;
import ch.psi.pshell.plot.Plot;
import ch.psi.pshell.scan.Scan;
import ch.psi.pshell.scan.ScanListener;
import ch.psi.pshell.scan.ScanRecord;
import ch.psi.pshell.ui.Panel;
import ch.psi.utils.State;
import ch.psi.utils.swing.SwingUtils;
import java.util.ArrayList;
import javax.swing.JSpinner;

/**
 *
 */
public class GunScan extends Panel {

    LinePlotErrorSeries seriesEnergy = new LinePlotErrorSeries("Energy");
    LinePlotErrorSeries seriesEnergySpread = new LinePlotErrorSeries("Energy Spread");
    double dispersion;
    double energy0;

    public GunScan() {
        initComponents();
        plot.setStyle(LinePlotJFree.Style.ErrorY);
        plot.addSeries(seriesEnergy);
        plot.addSeries(seriesEnergySpread);
        plot.getAxis(Plot.AxisId.X).setLabel("Gun Phase");
        plot.getAxis(Plot.AxisId.Y).setLabel("Energy (MeV)");
        plot.getAxis(Plot.AxisId.Y2).setLabel("Energy Spread (MeV)");
        plot.setLegendVisible(true);
        setPersistedComponents(SwingUtils.getComponentsByType(panelScanPars, JSpinner.class));
    }

    //Overridable callbacks
    @Override
    public void onInitialize(int runCount) {
    }

    @Override
    public void onStateChange(State state, State former) {
        buttonAbort.setEnabled(state.isProcessing());
        buttonStart.setEnabled(state == State.Ready);
        spinnerStart.setEnabled(buttonStart.isEnabled());
        spinnerStop.setEnabled(buttonStart.isEnabled());
        spinnerStep.setEnabled(buttonStart.isEnabled());
        spinnerSamples.setEnabled(buttonStart.isEnabled());
        spinnerLatency.setEnabled(buttonStart.isEnabled());
    }

    @Override
    public void onExecutedFile(String fileName, Object result) {
    }

    //Callback to perform update - in event thread
    @Override
    protected void doUpdate() {
    }

    @Override
    public void onStart() {
        super.onStart();
        getContext().addScanListener(scanListener);
    }

    @Override
    public void onStop() {
        getContext().removeScanListener(scanListener);
        super.onStop();
    }

    ScanListener scanListener = new ScanListener() {
        @Override
        public void onScanStarted(Scan scan, String plotTitle) {
            if ("GunScan".equals(getContext().getExecutionPars().getName())) {
                seriesEnergy.clear();
                seriesEnergySpread.clear();
            }
        }

        @Override
        public void onNewRecord(Scan scan, ScanRecord record) {
            if ("GunScan".equals(getContext().getExecutionPars().getName())) {
                double phase = (Double) record.getPositions()[0];
                double energy_mean = ((DescStatsDouble) record.getValues()[0]).getMean();
                double energy_std = ((DescStatsDouble) record.getValues()[0]).getStdev();
                double energy_spread_mean = ((DescStatsDouble) record.getValues()[1]).getMean();
                double energy_spread_std = ((DescStatsDouble) record.getValues()[1]).getStdev();
                seriesEnergy.appendData(phase,
                        energy0 * (1 + energy_mean / 1e6 / dispersion),
                        energy0 * (energy_std / 1e6 / dispersion)
                );
                seriesEnergySpread.appendData(phase,
                        energy0 * (energy_spread_mean / 1e6 / dispersion),
                        energy0 * (energy_spread_std / 1e6 / dispersion)
                );
            }
        }

        @Override
        public void onScanEnded(Scan scan, Exception ex) {
        }
    };

    @SuppressWarnings("unchecked")
    // <editor-fold defaultstate="collapsed" desc="Generated Code">//GEN-BEGIN:initComponents
    private void initComponents() {

        panelScanPars = new javax.swing.JPanel();
        labelStart = new javax.swing.JLabel();
        labelStop = new javax.swing.JLabel();
        spinnerStep = new javax.swing.JSpinner();
        labelStep = new javax.swing.JLabel();
        spinnerStart = new javax.swing.JSpinner();
        labelSamples = new javax.swing.JLabel();
        spinnerSamples = new javax.swing.JSpinner();
        labelLatency = new javax.swing.JLabel();
        spinnerLatency = new javax.swing.JSpinner();
        spinnerStop = new javax.swing.JSpinner();
        panelBeamPars = new javax.swing.JPanel();
        spinnerDispersion = new javax.swing.JSpinner();
        labelEnergy = new javax.swing.JLabel();
        spinnerEnergy = new javax.swing.JSpinner();
        labelDispersion = new javax.swing.JLabel();
        plot = new ch.psi.pshell.plot.LinePlotJFree();
        buttonStart = new javax.swing.JButton();
        buttonAbort = new javax.swing.JButton();

        setPreferredSize(new java.awt.Dimension(737, 445));

        panelScanPars.setBorder(javax.swing.BorderFactory.createTitledBorder("Scan Parameters"));

        labelStart.setText("Start:");

        labelStop.setText("Stop:");

        spinnerStep.setModel(new javax.swing.SpinnerNumberModel(5.0d, 1.0d, 90.0d, 1.0d));
        spinnerStep.setPreferredSize(new java.awt.Dimension(64, 20));

        labelStep.setText("Step:");

        spinnerStart.setModel(new javax.swing.SpinnerNumberModel(0.0d, -180.0d, 360.0d, 1.0d));
        spinnerStart.setPreferredSize(new java.awt.Dimension(64, 20));

        labelSamples.setText("Nb Samples:");

        spinnerSamples.setModel(new javax.swing.SpinnerNumberModel(1.0d, 1.0d, 50.0d, 1.0d));
        spinnerSamples.setPreferredSize(new java.awt.Dimension(64, 20));

        labelLatency.setText("Latency:");

        spinnerLatency.setModel(new javax.swing.SpinnerNumberModel(0.1d, 0.0d, 5.0d, 0.01d));
        spinnerLatency.setPreferredSize(new java.awt.Dimension(64, 20));

        spinnerStop.setModel(new javax.swing.SpinnerNumberModel(180.0d, -180.0d, 360.0d, 1.0d));
        spinnerStop.setMinimumSize(new java.awt.Dimension(53, 20));
        spinnerStop.setPreferredSize(new java.awt.Dimension(64, 20));

        javax.swing.GroupLayout panelScanParsLayout = new javax.swing.GroupLayout(panelScanPars);
        panelScanPars.setLayout(panelScanParsLayout);
        panelScanParsLayout.setHorizontalGroup(
            panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(panelScanParsLayout.createSequentialGroup()
                .addContainerGap()
                .addGroup(panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING)
                    .addGroup(panelScanParsLayout.createSequentialGroup()
                        .addComponent(labelLatency)
                        .addGap(2, 2, 2)
                        .addComponent(spinnerLatency, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
                    .addGroup(panelScanParsLayout.createSequentialGroup()
                        .addComponent(labelStop)
                        .addGap(2, 2, 2)
                        .addComponent(spinnerStop, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
                    .addGroup(panelScanParsLayout.createSequentialGroup()
                        .addComponent(labelStart)
                        .addGap(2, 2, 2)
                        .addComponent(spinnerStart, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
                    .addGroup(panelScanParsLayout.createSequentialGroup()
                        .addComponent(labelSamples)
                        .addGap(2, 2, 2)
                        .addComponent(spinnerSamples, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
                    .addGroup(panelScanParsLayout.createSequentialGroup()
                        .addComponent(labelStep)
                        .addGap(2, 2, 2)
                        .addComponent(spinnerStep, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)))
                .addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
        );
        panelScanParsLayout.setVerticalGroup(
            panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(panelScanParsLayout.createSequentialGroup()
                .addContainerGap()
                .addGroup(panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(spinnerStart, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(labelStart))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addGroup(panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(spinnerStop, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(labelStop))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addGroup(panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(spinnerStep, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(labelStep))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addGroup(panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(spinnerSamples, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(labelSamples))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addGroup(panelScanParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(spinnerLatency, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(labelLatency))
                .addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
        );

        panelBeamPars.setBorder(javax.swing.BorderFactory.createTitledBorder("Beam Parameters"));

        spinnerDispersion.setModel(new javax.swing.SpinnerNumberModel(-0.32d, -100.0d, 100.0d, 0.01d));
        spinnerDispersion.setPreferredSize(new java.awt.Dimension(64, 20));

        labelEnergy.setText("Energy:");

        spinnerEnergy.setModel(new javax.swing.SpinnerNumberModel(7.1d, 0.0d, 100.0d, 0.1d));
        spinnerEnergy.setPreferredSize(new java.awt.Dimension(64, 20));

        labelDispersion.setText("Dispersion:");

        javax.swing.GroupLayout panelBeamParsLayout = new javax.swing.GroupLayout(panelBeamPars);
        panelBeamPars.setLayout(panelBeamParsLayout);
        panelBeamParsLayout.setHorizontalGroup(
            panelBeamParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(javax.swing.GroupLayout.Alignment.TRAILING, panelBeamParsLayout.createSequentialGroup()
                .addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE)
                .addGroup(panelBeamParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING)
                    .addGroup(panelBeamParsLayout.createSequentialGroup()
                        .addComponent(labelEnergy)
                        .addGap(0, 0, 0)
                        .addComponent(spinnerEnergy, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
                    .addGroup(panelBeamParsLayout.createSequentialGroup()
                        .addComponent(labelDispersion)
                        .addGap(0, 0, 0)
                        .addComponent(spinnerDispersion, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)))
                .addContainerGap())
        );
        panelBeamParsLayout.setVerticalGroup(
            panelBeamParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(panelBeamParsLayout.createSequentialGroup()
                .addContainerGap()
                .addGroup(panelBeamParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(spinnerDispersion, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(labelDispersion))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addGroup(panelBeamParsLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(spinnerEnergy, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                    .addComponent(labelEnergy))
                .addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
        );

        plot.setTitle("");

        buttonStart.setText("Start");
        buttonStart.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                buttonStartActionPerformed(evt);
            }
        });

        buttonAbort.setText("Abort");
        buttonAbort.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                buttonAbortActionPerformed(evt);
            }
        });

        javax.swing.GroupLayout layout = new javax.swing.GroupLayout(this);
        this.setLayout(layout);
        layout.setHorizontalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING, false)
                    .addGroup(layout.createSequentialGroup()
                        .addComponent(buttonStart)
                        .addGap(34, 34, 34)
                        .addComponent(buttonAbort))
                    .addComponent(panelScanPars, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE)
                    .addComponent(panelBeamPars, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addComponent(plot, javax.swing.GroupLayout.DEFAULT_SIZE, 531, Short.MAX_VALUE)
                .addContainerGap())
        );
        layout.setVerticalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
                    .addComponent(plot, javax.swing.GroupLayout.DEFAULT_SIZE, 419, Short.MAX_VALUE)
                    .addGroup(layout.createSequentialGroup()
                        .addComponent(panelScanPars, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                        .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                        .addComponent(panelBeamPars, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
                        .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED, javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE)
                        .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                            .addComponent(buttonStart)
                            .addComponent(buttonAbort))))
                .addGap(14, 14, 14))
        );

        panelScanPars.getAccessibleContext().setAccessibleDescription("");
    }// </editor-fold>//GEN-END:initComponents

    private void buttonStartActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_buttonStartActionPerformed
        ArrayList parameters = new ArrayList();
        dispersion = (Double) spinnerDispersion.getValue();
        energy0 = (Double) spinnerEnergy.getValue();
        parameters.add(spinnerStart.getValue());
        parameters.add(spinnerStop.getValue());
        parameters.add(spinnerStep.getValue());
        parameters.add(spinnerSamples.getValue());
        parameters.add(spinnerLatency.getValue());
        parameters.add(dispersion);
        parameters.add(energy0);
        try {
            runAsync("RFscan/GunScan", parameters).handle((ret, ex) -> {
                if (ex != null) {
                    getLogger().info("Exception executing scan: " + ex);
                } else {

                }
                return ret;
            });
        } catch (Exception ex) {
            showException(ex);
        }
    }//GEN-LAST:event_buttonStartActionPerformed

    private void buttonAbortActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_buttonAbortActionPerformed
        try {
            abort();
        } catch (Exception ex) {
            showException(ex);
        }
    }//GEN-LAST:event_buttonAbortActionPerformed

    // Variables declaration - do not modify//GEN-BEGIN:variables
    private javax.swing.JButton buttonAbort;
    private javax.swing.JButton buttonStart;
    private javax.swing.JLabel labelDispersion;
    private javax.swing.JLabel labelEnergy;
    private javax.swing.JLabel labelLatency;
    private javax.swing.JLabel labelSamples;
    private javax.swing.JLabel labelStart;
    private javax.swing.JLabel labelStep;
    private javax.swing.JLabel labelStop;
    private javax.swing.JPanel panelBeamPars;
    private javax.swing.JPanel panelScanPars;
    private ch.psi.pshell.plot.LinePlotJFree plot;
    private javax.swing.JSpinner spinnerDispersion;
    private javax.swing.JSpinner spinnerEnergy;
    private javax.swing.JSpinner spinnerLatency;
    private javax.swing.JSpinner spinnerSamples;
    private javax.swing.JSpinner spinnerStart;
    private javax.swing.JSpinner spinnerStep;
    private javax.swing.JSpinner spinnerStop;
    // End of variables declaration//GEN-END:variables
}