/*

  Copyright © 2017 Hasan Yavuz Özderya

  This file is part of serialplot.

  serialplot is free software: you can redistribute it and/or modify

  it under the terms of the GNU General Public License as published by

  the Free Software Foundation, either version 3 of the License, or

  (at your option) any later version.

  serialplot is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License

  along with serialplot.  If not, see <http://www.gnu.org/licenses/>.

*/

#include "abstractreader.h"

AbstractReader::AbstractReader(QIODevice* device, ChannelManager* channelMan,

                               DataRecorder* recorder, QObject* parent) :

    QObject(parent)

{

    _device = device;

    _channelMan = channelMan;

    _recorder = recorder;

    recording = false;

    // initialize sps counter

    sampleCount = 0;

    samplesPerSecond = 0;

    QObject::connect(&spsTimer, &QTimer::timeout,

                     this, &AbstractReader::spsTimerTimeout);

    // TODO: start sps timer when reader is enabled

    spsTimer.start(SPS_UPDATE_TIMEOUT * 1000);

}

void AbstractReader::spsTimerTimeout()

{

    unsigned currentSps = samplesPerSecond;

    samplesPerSecond = (sampleCount/numOfChannels())/SPS_UPDATE_TIMEOUT;

    if (currentSps != samplesPerSecond)

    {

        emit samplesPerSecondChanged(samplesPerSecond);

    }

    sampleCount = 0;

}

void AbstractReader::addData(double* samples, unsigned length)

{

    _channelMan->addData(samples, length);

    if (recording)

    {

        _recorder->addData(samples, length, numOfChannels());

    }

    sampleCount += length;

}

  serialplot is free software: you can redistribute it and/or modify

  it under the terms of the GNU General Public License as published by

  the Free Software Foundation, either version 3 of the License, or

  (at your option) any later version.

  serialplot is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License

  along with serialplot.  If not, see <http://www.gnu.org/licenses/>.

*/

#include "datarecorder.h"

#include <QtDebug>

DataRecorder::DataRecorder(QObject *parent) :

    QObject(parent),

    fileStream(&file)

{

    lastNumChannels = 0;

}

bool DataRecorder::startRecording(QString fileName, QStringList channelNames)

{

    Q_ASSERT(!file.isOpen());

    // open file

    file.setFileName(fileName);

    if (!file.open(QIODevice::WriteOnly))

    {

        qCritical() << "Opening file " << fileName

                    << " for recording failed with error: " << file.error();

        return false;

    }

    // write header line

    if (!channelNames.isEmpty())

    {

        fileStream << channelNames.join(",");

        fileStream << "\n";

        lastNumChannels = channelNames.length();

    }

    return true;

}

void DataRecorder::addData(double* data, unsigned length, unsigned numOfChannels)

{

    Q_ASSERT(length > 0);

    Q_ASSERT(length % numOfChannels == 0);

    if (lastNumChannels != 0 && numOfChannels != lastNumChannels)

    {

        qWarning() << "Number of channels changed from " << lastNumChannels

                   << " to " << numOfChannels <<

            " during recording, CSV file is corrupted but no data will be lost.";

    }

    lastNumChannels = numOfChannels;

    unsigned numOfSamples = length / numOfChannels; // per channel

    for (unsigned int i = 0; i < numOfSamples; i++)

    {

        for (unsigned ci = 0; ci < numOfChannels; ci++)

        {

            fileStream << data[ci * numOfSamples + i];

            if (ci != numOfChannels-1) fileStream << ",";

        }

        fileStream << '\n';

    }

}

void DataRecorder::stopRecording()

{

    Q_ASSERT(file.isOpen());

    file.close();

    lastNumChannels = 0;

}

  (at your option) any later version.

  serialplot is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License

  along with serialplot.  If not, see <http://www.gnu.org/licenses/>.

*/

#ifndef DATARECORDER_H

#define DATARECORDER_H

#include <QObject>

#include <QFile>

#include <QTextStream>

class DataRecorder : public QObject

{

    Q_OBJECT

public:

    explicit DataRecorder(QObject *parent = 0);

    /**

     * @brief Starts recording data to a file in CSV format.

     *

     * File is opened and header line (names of channels) is written.

     *

     * @param fileName name of the recording file

     * @param channelNames names of the channels for header line, if empty no header line is written

     * @return false if file operation fails (read only etc.)

     */

    bool startRecording(QString fileName, QStringList channelNames);

    /**

     * @brief Adds data to a channel.

     *

     * Multiple rows of data can be added at a time. Each channels

     * data should be ordered consecutively in the `data` array:

     *

     * [CH0_SMP0, CH0_SMP1 ... CH0_SMPN, CH1_SMP0, CH1_SMP1, ... , CHN_SMPN]

     *

     * If `numOfChannels` changes during recording, no data will be

     * lost (ie. it will be written to the file) but this will produce

     * an invalid CSV file. An error message will be written to the

     * console.

     *

RecordPanel::RecordPanel(DataRecorder* recorder, QWidget *parent) :

    QWidget(parent),

    ui(new Ui::RecordPanel),

    recordToolBar(tr("Record Toolbar")),

    recordAction(QIcon::fromTheme("media-record"), tr("Record"), this)

{

    overwriteSelected = false;

    _recorder = recorder;

    ui->setupUi(this);

    recordToolBar.setObjectName("tbRecord");

    recordAction.setCheckable(true);

    recordToolBar.addAction(&recordAction);

    ui->pbRecord->setDefaultAction(&recordAction);

    connect(ui->pbBrowse, &QPushButton::clicked,

            this, &RecordPanel::selectFile);

    connect(&recordAction, &QAction::triggered,

            this, &RecordPanel::onRecord);

    connect(ui->cbRecordPaused, SIGNAL(toggled(bool)),

            this, SIGNAL(recordPausedChanged(bool)));

}

RecordPanel::~RecordPanel()

{

    delete ui;

}

QToolBar* RecordPanel::toolbar()

{

    return &recordToolBar;

}

bool RecordPanel::isRecording()

{

    return recordAction.isChecked();

}

bool RecordPanel::recordPaused()

{

    return ui->cbRecordPaused->isChecked();

}

bool RecordPanel::selectFile()

{

     <item>

      <widget class="QCheckBox" name="cbAutoIncrement">

       <property name="toolTip">

        <string>Increments file name automatically everytime a new recording starts</string>

       </property>

       <property name="text">

        <string>Auto increment file name</string>

       </property>

       <property name="checked">

        <bool>true</bool>

       </property>

      </widget>

     </item>

     <item>

      <widget class="QCheckBox" name="cpStopOnClose">

       <property name="text">

        <string>Stop recording when port closed</string>

       </property>

       <property name="checked">

        <bool>true</bool>

       </property>

      </widget>

     </item>

     <item>

      <spacer name="verticalSpacer">

       <property name="orientation">

        <enum>Qt::Vertical</enum>

       </property>

       <property name="sizeHint" stdset="0">

        <size>

         <width>20</width>

         <height>40</height>

        </size>

       </property>

      </spacer>

     </item>

    </layout>

   </item>

   <item>

    <layout class="QVBoxLayout" name="verticalLayout_2">

     <item>

      <widget class="QToolButton" name="pbRecord">

       <property name="minimumSize">

        <size>

         <width>85</width>

         <height>50</height>

        </size>

       </property>