Modeling/csvlogger.cpp at master · wuzzapcom/Modeling · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#include "csvlogger.h"

CsvLogger::CsvLogger()
{
}

void CsvLogger::startLogging(int numberOfMatPoints)
{
    time_t timestamp = time(NULL);
    currentOpenedFile.setFileName(QString::number((uintmax_t)timestamp).append(".csv"));
    currentOpenedFile.open(QFile::Append | QFile::Text);

    QTextStream writeStream(&currentOpenedFile);

    for (int i = 0; i < numberOfMatPoints; i++)
    {
        QString varNumber = QString::number(i);
        writeStream << "x_" << varNumber << ", ";
        writeStream << "vx_" << varNumber << ", ";
        writeStream << "y_" << varNumber << ", ";
        writeStream << "vy_" << varNumber << ", ";
        writeStream << "phi_" << varNumber << ", ";
        writeStream << "vphi_" << varNumber << ", ";
    }

    writeStream << "Kinetic energy, PotentionalEnergy\r\n";
}

void CsvLogger::log(std::valarray<double> values, double W, double E)
{
    QTextStream writeStream(&currentOpenedFile);
    for (size_t i = 0; i < values.size(); i++)
    {
        writeStream << values[i] << ",";
    }
    writeStream << W << ",";
    writeStream << E << "\r\n";
}

void CsvLogger::stopLogging()
{
    currentOpenedFile.close();
}

void CsvLogger::close()
{
    currentOpenedFile.close();
}

void CsvLogger::logLengths(double L0, double x1, double y1, double x2, double y2)
{
    QFile file("lengths.csv");
    file.open(QFile::Append | QFile::Text);

    QTextStream stream(&file);

    stream << L0 << ", ";
    stream << x1 << ", ";
    stream << y1 << ", ";
    stream << x2 << ", ";
    stream << y2 << "\r\n";

    file.close();
}

CsvLogger::~CsvLogger()
{
//    if (currentOpenedFile != nullptr)
//    {
//        close();
//    }
}