Subversion Repositories modular_oscilloscope

#include <QtGui>
#include <QSvgGenerator>
#include <QFileDialog>
#include "rvioscilloscope.h"
#include <qwt_counter.h>
#include <qwt_plot_zoomer.h>
#include "data_plot.h"
#include "ui_rvioscilloscope.h"
#include <qwt_plot_picker.h>
#include <qwt_picker_machine.h>
 
 
class Zoomer: public QwtPlotZoomer
{
public:
    Zoomer(int xAxis, int yAxis, QwtPlotCanvas *canvas):
        QwtPlotZoomer(xAxis, yAxis, canvas)
    {
        setTrackerMode(QwtPicker::AlwaysOff);
        setRubberBand(QwtPicker::NoRubberBand);
 
        // RightButton: zoom out by 1
        // Ctrl+RightButton: zoom out to full size
 
        setMousePattern(QwtEventPattern::MouseSelect2,
            Qt::RightButton, Qt::ControlModifier);
 
        setMousePattern(QwtEventPattern::MouseSelect3,
            Qt::RightButton);
 
    }
};
 
 
RVIOscilloscope::RVIOscilloscope(QWidget *parent)
    : QMainWindow(parent), ui(new Ui::RVIOscilloscope)
{
    ui->setupUi(this);
 
    extraTools = new RVIBoardTools(this);
    ui->toolsLayout->addWidget(extraTools);
 
    createPlot();
 
 
 
    // zoom
    d_zoomer[0] = new Zoomer( QwtPlot::xBottom, QwtPlot::yLeft,
    plot->canvas());
    d_zoomer[0]->setRubberBand(QwtPicker::RectRubberBand);
    d_zoomer[0]->setRubberBandPen(QColor(Qt::green));
    d_zoomer[0]->setTrackerMode(QwtPicker::ActiveOnly);
    d_zoomer[0]->setTrackerPen(QColor(Qt::white));
 
    d_zoomer[1] = new Zoomer(QwtPlot::xTop, QwtPlot::yRight,
    plot->canvas());
 
    d_panner = new QwtPlotPanner(plot->canvas());
    d_panner->setMouseButton(Qt::MidButton);
 
 
 
    enableZoomMode(false);
 
 
    // picker
    QwtPlotPicker *selectLine = new QwtPlotPicker( QwtPlot::xBottom, QwtPlot::yLeft ,
                                    QwtPlotPicker::VLineRubberBand,
                                    QwtPlotPicker::AlwaysOff,
                                    plot->canvas());
 
    //selectLine->setRubberBand(QwtPlotPicker::VLineRubberBand);
        selectLine->setStateMachine(new QwtPickerClickPointMachine);
    selectLine->setAxis(QwtPlot::xBottom, QwtPlot::yLeft);
    selectLine->setRubberBandPen(QPen(Qt::yellow, 0, Qt::SolidLine));
    selectLine->setEnabled(true);
 
    //selectLine->begin();
    //selectLine->append(QPoint(0,0));
    selectLine->setTrackerPen(QColor(Qt::white));
 
 
    connect(ui->actionDockControls,SIGNAL(triggered(bool)), this, SLOT(setDockedControlsDock(bool)));
    connect(ui->controlsDock,SIGNAL(topLevelChanged(bool)), ui->actionDockControls, SLOT(toggle()));
    connect(ui->actionShowControls, SIGNAL(toggled(bool)), ui->controlsDock, SLOT(setVisible(bool)));
    connect(ui->controlsDock, SIGNAL(visibilityChanged(bool)), ui->actionShowControls, SLOT(setChecked(bool)));
 
    connect(ui->actionZoom, SIGNAL(triggered(bool)), this,  SLOT(enableZoomMode(bool)));
    connect(ui->actionSaveImage, SIGNAL(triggered()), SLOT(exportSVG()));
    connect(ui->actionPause, SIGNAL(toggled(bool)), plot, SLOT(setPaused(bool)));
    connect(ui->actionPause, SIGNAL(toggled(bool)), extraTools, SLOT(setPaused(bool)));
 
    connect(ui->actionEnChAGrid, SIGNAL(triggered(bool)), plot, SLOT(curveAShowGrid(bool)));
    connect(ui->actionEnChBGrid, SIGNAL(triggered(bool)), plot, SLOT(curveBShowGrid(bool)));
    connect(ui->actionEnTimeGrid, SIGNAL(triggered(bool)), plot, SLOT(timeShowGrid(bool)));
 
    connect(extraTools, SIGNAL(statusChanged(QString)) ,ui->statusBar, SLOT(showMessage(QString)));
    connect(extraTools, SIGNAL(channelAEnabled(bool)), plot, SLOT(curveAShow(bool)));
    connect(extraTools, SIGNAL(channelBEnabled(bool)), plot, SLOT(curveBShow(bool)));
    connect(extraTools, SIGNAL(channelAColorChanged(QColor)),plot,SLOT(setCurveAColor(QColor)));
    connect(extraTools, SIGNAL(channelBColorChanged(QColor)),plot,SLOT(setCurveBColor(QColor)));
    connect(extraTools, SIGNAL(channelAData(QVector<double> , QVector<double> )),plot,SLOT(curveAUpdate(QVector<double>,QVector<double>)));
    connect(extraTools, SIGNAL(channelBData(QVector<double> , QVector<double> )),plot,SLOT(curveBUpdate(QVector<double> , QVector<double> )));
    connect(plot, SIGNAL(aScaleDivChanged(double)), extraTools, SLOT(setChannelADiv(double)));
    connect(plot, SIGNAL(tScaleDivChanged(double)), extraTools, SLOT(setTimeDiv(double)));
    //connect(extraTools, SIGNAL(channelAChangeDiv(double,double)),plot,SLOT(curveASetLimits(double,double)));
    connect(extraTools, SIGNAL(channelAChangeDiv(int)), plot, SLOT(curveAZoom(int)));
    connect(extraTools, SIGNAL(channelAMove(int)), plot, SLOT(curveAMove(int)));
    connect(extraTools, SIGNAL(channelAResetPos()), plot, SLOT(curveAResetPos()));
    connect(extraTools, SIGNAL(channelBChangeDiv(int)), plot, SLOT(curveBZoom(int)));
    connect(extraTools, SIGNAL(channelBMove(int)), plot, SLOT(curveBMove(int)));
    connect(extraTools, SIGNAL(channelBResetPos()), plot, SLOT(curveBResetPos()));
    connect(extraTools, SIGNAL(timeChangeDiv(int)), plot, SLOT(timeZoom(int)));
    connect(extraTools, SIGNAL(timeMove(int)), plot, SLOT(timeMove(int)));
    connect(extraTools, SIGNAL(timeChangeLimits(double,double)), plot, SLOT(timeSetLimits(double,double)));
    connect(extraTools, SIGNAL(timeChangeLimits(double,double)), this, SLOT(updateZoomLimits()));
 
    connect(extraTools, SIGNAL(showTriggerLine(bool, int)),plot,SLOT(enableTriggerLine(bool, int)));
    connect(extraTools, SIGNAL(changeTriggerValue(double)),plot,SLOT(setTriggerLineValue(double)));
 
    extraTools->setDefaultValues();
    plot->updateDivs();
}
 
RVIOscilloscope::~RVIOscilloscope()
{
    delete ui;
}
 
 
void RVIOscilloscope::createPlot()
{
 
    plot = new DataPlot(this);
    //setCentralWidget(plot);
    ui->mainLayout->addWidget(plot);
    plot->setMinimumHeight(200);
    plot->setMinimumWidth(200);
    plot->setCanvasBackground(QColor(Qt::black));
 
    plot->setCurveAColor(Qt::red);
    plot->setCurveBColor(Qt::yellow);
//    QSizePolicy sizePolicy(QSizePolicy::Preferred, QSizePolicy::Preferred);
//    sizePolicy.setVerticalStretch(30);
//    plot->setSizePolicy(sizePolicy);
    //plot->setMargin(12);
 
}
 
 
void RVIOscilloscope::enableZoomMode(bool on)
{
    d_panner->setEnabled(on);
 
    d_zoomer[0]->setEnabled(on);
   // d_zoomer[0]->zoom(0);
 
    d_zoomer[1]->setEnabled(on);
   // d_zoomer[1]->zoom(0);
}
 
 
void RVIOscilloscope::updateZoomLimits()
{
    d_zoomer[0]->setZoomBase();
    d_zoomer[1]->setZoomBase();
}
 
 
void RVIOscilloscope::on_actionBackgroundColor_triggered()
{
    QColorDialog *colorChooser = new QColorDialog(plot->canvasBackground());
    plot->setCanvasBackground(colorChooser->getColor());
 
}
 
void RVIOscilloscope::setDockedControlsDock(bool set)
{
    ui->controlsDock->setFloating(!(set));
}
 
void RVIOscilloscope::exportSVG()
{
    QString fileName = "plot.svg";
 
    fileName = QFileDialog::getSaveFileName(
        this, "Export File Name", QString(),
        "SVG Documents (*.svg)");
 
    if ( !fileName.isEmpty() )
    {
        QSvgGenerator generator;
        generator.setFileName(fileName);
        generator.setSize(QSize(800, 600));
 
        plot->print(generator);
    }
 
}
 
void RVIOscilloscope::on_actionSaveData_triggered()
{
    QString fileName = "data";
 
    fileName = QFileDialog::getSaveFileName(
        this, "File name:", fileName,
        "Text file (*.txt);;All files (*.*)");
 
    if ( !fileName.isEmpty() )
    {
        QFile file(fileName);
 
        if (file.open(QFile::WriteOnly | QFile::Truncate))
        {
             QTextStream out(&file);
             out << qSetFieldWidth(15) << right << "Time"  << "Channel A" << "Channel B" << endl;
             for (int i = 0; i < extraTools->getData(RVIBoardTools::Time).size(); i++)
             {
                 out << qSetFieldWidth(15) << right  \
                     << extraTools->getData(RVIBoardTools::Time).value(i, 0.0)\
                     << extraTools->getData(RVIBoardTools::ChannelA).value(i, 0.0) \
                     << extraTools->getData(RVIBoardTools::ChannelB).value(i, 0.0) << endl;
             }
 
        }
    }
}