AmdaResultParserHelper.cpp
423 lines
| 15.2 KiB
| text/x-c
|
CppLexer
Alexandre Leroux
|
r945 | #include "AmdaResultParserHelper.h" | ||
|
Alexandre Leroux
|
r947 | #include <Common/DateUtils.h> | ||
|
Alexandre Leroux
|
r953 | #include <Common/SortUtils.h> | ||
|
Alexandre Leroux
|
r947 | |||
|
Alexandre Leroux
|
r980 | #include <Data/DataSeriesUtils.h> | ||
|
Alexandre Leroux
|
r947 | #include <Data/ScalarSeries.h> | ||
|
Alexandre Leroux
|
r953 | #include <Data/SpectrogramSeries.h> | ||
|
Alexandre Leroux
|
r946 | #include <Data/Unit.h> | ||
|
Alexandre Leroux
|
r947 | #include <Data/VectorSeries.h> | ||
| #include <QtCore/QDateTime> | ||||
| #include <QtCore/QRegularExpression> | ||||
|
Alexandre Leroux
|
r946 | |||
|
Alexandre Leroux
|
r954 | #include <functional> | ||
|
Alexandre Leroux
|
r945 | Q_LOGGING_CATEGORY(LOG_AmdaResultParserHelper, "AmdaResultParserHelper") | ||
|
Alexandre Leroux
|
r946 | namespace { | ||
|
Alexandre Leroux
|
r947 | // ///////// // | ||
| // Constants // | ||||
| // ///////// // | ||||
| /// Separator between values in a result line | ||||
| const auto RESULT_LINE_SEPARATOR = QRegularExpression{QStringLiteral("\\s+")}; | ||||
| /// Format for dates in result files | ||||
| const auto DATE_FORMAT = QStringLiteral("yyyy-MM-ddThh:mm:ss.zzz"); | ||||
|
Alexandre Leroux
|
r946 | // /////// // | ||
| // Methods // | ||||
| // /////// // | ||||
| /** | ||||
| * Checks that the properties contain a specific unit and that this unit is valid | ||||
| * @param properties the properties map in which to search unit | ||||
| * @param key the key to search for the unit in the properties | ||||
| * @param errorMessage the error message to log in case the unit is invalid | ||||
| * @return true if the unit is valid, false it it's invalid or was not found in the properties | ||||
| */ | ||||
| bool checkUnit(const Properties &properties, const QString &key, const QString &errorMessage) | ||||
| { | ||||
| auto unit = properties.value(key).value<Unit>(); | ||||
| if (unit.m_Name.isEmpty()) { | ||||
| qCWarning(LOG_AmdaResultParserHelper()) << errorMessage; | ||||
| return false; | ||||
| } | ||||
| return true; | ||||
| } | ||||
|
Alexandre Leroux
|
r947 | QDateTime dateTimeFromString(const QString &stringDate) noexcept | ||
| { | ||||
| #if QT_VERSION >= QT_VERSION_CHECK(5, 8, 0) | ||||
| return QDateTime::fromString(stringDate, Qt::ISODateWithMs); | ||||
| #else | ||||
| return QDateTime::fromString(stringDate, DATE_FORMAT); | ||||
| #endif | ||||
| } | ||||
| /// Converts a string date to a double date | ||||
| /// @return a double that represents the date in seconds, NaN if the string date can't be converted | ||||
| double doubleDate(const QString &stringDate) noexcept | ||||
| { | ||||
| // Format: yyyy-MM-ddThh:mm:ss.zzz | ||||
| auto dateTime = dateTimeFromString(stringDate); | ||||
| dateTime.setTimeSpec(Qt::UTC); | ||||
| return dateTime.isValid() ? DateUtils::secondsSinceEpoch(dateTime) | ||||
| : std::numeric_limits<double>::quiet_NaN(); | ||||
| } | ||||
| /** | ||||
| * Reads a line from the AMDA file and tries to extract a x-axis data and value data from it | ||||
| * @param xAxisData the vector in which to store the x-axis data extracted | ||||
| * @param valuesData the vector in which to store the value extracted | ||||
| * @param line the line to read to extract the property | ||||
| * @param valuesIndexes indexes of insertion of read values. For example, if the line contains three | ||||
| * columns of values, and valuesIndexes are {2, 0, 1}, the value of the third column will be read | ||||
| * and inserted first, then the value of the first column, and finally the value of the second | ||||
| * column. | ||||
| * @param fillValue value that tags an invalid data. For example, if fillValue is -1 and a read | ||||
| * value is -1, then this value is considered as invalid and converted to NaN | ||||
| */ | ||||
| void tryReadResult(std::vector<double> &xAxisData, std::vector<double> &valuesData, | ||||
| const QString &line, const std::vector<int> &valuesIndexes, | ||||
| double fillValue = std::numeric_limits<double>::quiet_NaN()) | ||||
| { | ||||
| auto lineData = line.split(RESULT_LINE_SEPARATOR, QString::SkipEmptyParts); | ||||
| // Checks that the line contains expected number of values + x-axis value | ||||
|
Alexandre Leroux
|
r961 | if (static_cast<size_t>(lineData.size()) == valuesIndexes.size() + 1) { | ||
|
Alexandre Leroux
|
r947 | // X : the data is converted from date to double (in secs) | ||
| auto x = doubleDate(lineData.at(0)); | ||||
| // Adds result only if x is valid. Then, if value is invalid, it is set to NaN | ||||
| if (!std::isnan(x)) { | ||||
| xAxisData.push_back(x); | ||||
| // Values | ||||
| for (auto valueIndex : valuesIndexes) { | ||||
| bool valueOk; | ||||
| // we use valueIndex + 1 to skip column 0 (x-axis value) | ||||
| auto value = lineData.at(valueIndex + 1).toDouble(&valueOk); | ||||
| if (!valueOk) { | ||||
| qCWarning(LOG_AmdaResultParserHelper()) | ||||
| << QObject::tr( | ||||
| "Value from (line %1, column %2) is invalid and will be " | ||||
| "converted to NaN") | ||||
| .arg(line, valueIndex); | ||||
| value = std::numeric_limits<double>::quiet_NaN(); | ||||
| } | ||||
| // Handles fill value | ||||
| if (!std::isnan(fillValue) && !std::isnan(value) && fillValue == value) { | ||||
| value = std::numeric_limits<double>::quiet_NaN(); | ||||
| } | ||||
| valuesData.push_back(value); | ||||
| } | ||||
| } | ||||
| else { | ||||
| qCWarning(LOG_AmdaResultParserHelper()) | ||||
| << QObject::tr("Can't retrieve results from line %1: x is invalid").arg(line); | ||||
| } | ||||
| } | ||||
| else { | ||||
| qCWarning(LOG_AmdaResultParserHelper()) | ||||
| << QObject::tr("Can't retrieve results from line %1: invalid line").arg(line); | ||||
| } | ||||
| } | ||||
|
Alexandre Leroux
|
r946 | /** | ||
| * Reads a line from the AMDA file and tries to extract a property from it | ||||
| * @param properties the properties map in which to put the property extracted from the line | ||||
| * @param key the key to which the property is added in the properties map | ||||
| * @param line the line to read to extract the property | ||||
| * @param regex the expected regex to extract the property. If the line matches this regex, the | ||||
| * property is generated | ||||
| * @param fun the function used to generate the property | ||||
| * @return true if the property could be generated, false if the line does not match the regex, or | ||||
| * if a property has already been generated for the key | ||||
| */ | ||||
| template <typename GeneratePropertyFun> | ||||
| bool tryReadProperty(Properties &properties, const QString &key, const QString &line, | ||||
| const QRegularExpression ®ex, GeneratePropertyFun fun) | ||||
| { | ||||
| if (properties.contains(key)) { | ||||
| return false; | ||||
| } | ||||
| auto match = regex.match(line); | ||||
| if (match.hasMatch()) { | ||||
| properties.insert(key, fun(match)); | ||||
| } | ||||
| return match.hasMatch(); | ||||
| } | ||||
|
Alexandre Leroux
|
r985 | /** | ||
| * Reads a line from the AMDA file and tries to extract a data from it. Date is converted to double | ||||
| * @sa tryReadProperty() | ||||
| */ | ||||
| bool tryReadDate(Properties &properties, const QString &key, const QString &line, | ||||
| const QRegularExpression ®ex, bool timeUnit = false) | ||||
| { | ||||
| return tryReadProperty(properties, key, line, regex, [timeUnit](const auto &match) { | ||||
| return QVariant::fromValue(doubleDate(match.captured(1))); | ||||
| }); | ||||
| } | ||||
|
Alexandre Leroux
|
r951 | /** | ||
| * Reads a line from the AMDA file and tries to extract a double from it | ||||
| * @sa tryReadProperty() | ||||
| */ | ||||
| bool tryReadDouble(Properties &properties, const QString &key, const QString &line, | ||||
| const QRegularExpression ®ex) | ||||
| { | ||||
| return tryReadProperty(properties, key, line, regex, [](const auto &match) { | ||||
| bool ok; | ||||
| // If the value can't be converted to double, it is set to NaN | ||||
| auto doubleValue = match.captured(1).toDouble(&ok); | ||||
| if (!ok) { | ||||
| doubleValue = std::numeric_limits<double>::quiet_NaN(); | ||||
| } | ||||
| return QVariant::fromValue(doubleValue); | ||||
| }); | ||||
| } | ||||
| /** | ||||
| * Reads a line from the AMDA file and tries to extract a vector of doubles from it | ||||
| * @param sep the separator of double values in the line | ||||
| * @sa tryReadProperty() | ||||
| */ | ||||
| bool tryReadDoubles(Properties &properties, const QString &key, const QString &line, | ||||
| const QRegularExpression ®ex, const QString &sep = QStringLiteral(",")) | ||||
| { | ||||
| return tryReadProperty(properties, key, line, regex, [sep](const auto &match) { | ||||
| std::vector<double> doubleValues{}; | ||||
| // If the value can't be converted to double, it is set to NaN | ||||
| auto values = match.captured(1).split(sep); | ||||
| for (auto value : values) { | ||||
| bool ok; | ||||
| auto doubleValue = value.toDouble(&ok); | ||||
| if (!ok) { | ||||
| doubleValue = std::numeric_limits<double>::quiet_NaN(); | ||||
| } | ||||
| doubleValues.push_back(doubleValue); | ||||
| } | ||||
| return QVariant::fromValue(doubleValues); | ||||
| }); | ||||
| } | ||||
|
Alexandre Leroux
|
r946 | /** | ||
| * Reads a line from the AMDA file and tries to extract a unit from it | ||||
| * @sa tryReadProperty() | ||||
| */ | ||||
| bool tryReadUnit(Properties &properties, const QString &key, const QString &line, | ||||
| const QRegularExpression ®ex, bool timeUnit = false) | ||||
| { | ||||
| return tryReadProperty(properties, key, line, regex, [timeUnit](const auto &match) { | ||||
| return QVariant::fromValue(Unit{match.captured(1), timeUnit}); | ||||
| }); | ||||
| } | ||||
| } // namespace | ||||
|
Alexandre Leroux
|
r945 | // ////////////////// // | ||
| // ScalarParserHelper // | ||||
| // ////////////////// // | ||||
| bool ScalarParserHelper::checkProperties() | ||||
| { | ||||
|
Alexandre Leroux
|
r946 | return checkUnit(m_Properties, X_AXIS_UNIT_PROPERTY, | ||
| QObject::tr("The x-axis unit could not be found in the file")); | ||||
|
Alexandre Leroux
|
r945 | } | ||
| std::shared_ptr<IDataSeries> ScalarParserHelper::createSeries() | ||||
| { | ||||
|
Alexandre Leroux
|
r947 | return std::make_shared<ScalarSeries>(std::move(m_XAxisData), std::move(m_ValuesData), | ||
| m_Properties.value(X_AXIS_UNIT_PROPERTY).value<Unit>(), | ||||
| m_Properties.value(VALUES_UNIT_PROPERTY).value<Unit>()); | ||||
|
Alexandre Leroux
|
r945 | } | ||
| void ScalarParserHelper::readPropertyLine(const QString &line) | ||||
| { | ||||
|
Alexandre Leroux
|
r946 | tryReadUnit(m_Properties, X_AXIS_UNIT_PROPERTY, line, DEFAULT_X_AXIS_UNIT_REGEX, true); | ||
|
Alexandre Leroux
|
r945 | } | ||
| void ScalarParserHelper::readResultLine(const QString &line) | ||||
| { | ||||
|
Alexandre Leroux
|
r947 | tryReadResult(m_XAxisData, m_ValuesData, line, valuesIndexes()); | ||
| } | ||||
| std::vector<int> ScalarParserHelper::valuesIndexes() const | ||||
| { | ||||
| // Only one value to read | ||||
| static auto result = std::vector<int>{0}; | ||||
| return result; | ||||
|
Alexandre Leroux
|
r945 | } | ||
|
Alexandre Leroux
|
r949 | // /////////////////////// // | ||
| // SpectrogramParserHelper // | ||||
| // /////////////////////// // | ||||
| bool SpectrogramParserHelper::checkProperties() | ||||
| { | ||||
|
Alexandre Leroux
|
r951 | // Generates y-axis data from bands extracted (take the middle of the intervals) | ||
| auto minBands = m_Properties.value(MIN_BANDS_PROPERTY).value<std::vector<double> >(); | ||||
| auto maxBands = m_Properties.value(MAX_BANDS_PROPERTY).value<std::vector<double> >(); | ||||
|
Alexandre Leroux
|
r997 | if (minBands.size() < 2 || minBands.size() != maxBands.size()) { | ||
|
Alexandre Leroux
|
r951 | qCWarning(LOG_AmdaResultParserHelper()) << QObject::tr( | ||
| "Can't generate y-axis data from bands extracted: bands intervals are invalid"); | ||||
| return false; | ||||
| } | ||||
|
Alexandre Leroux
|
r952 | std::transform( | ||
| minBands.begin(), minBands.end(), maxBands.begin(), std::back_inserter(m_YAxisData), | ||||
| [](const auto &minValue, const auto &maxValue) { return (minValue + maxValue) / 2.; }); | ||||
|
Alexandre Leroux
|
r953 | // Generates values indexes, i.e. the order in which each value will be retrieved (in ascending | ||
| // order of the associated bands) | ||||
| m_ValuesIndexes = SortUtils::sortPermutation(m_YAxisData, std::less<double>()); | ||||
| // Sorts y-axis data accoding to the ascending order | ||||
| m_YAxisData = SortUtils::sort(m_YAxisData, 1, m_ValuesIndexes); | ||||
| // Sets fill value | ||||
| m_FillValue = m_Properties.value(FILL_VALUE_PROPERTY).value<double>(); | ||||
|
Alexandre Leroux
|
r951 | return true; | ||
|
Alexandre Leroux
|
r949 | } | ||
| std::shared_ptr<IDataSeries> SpectrogramParserHelper::createSeries() | ||||
| { | ||||
|
Alexandre Leroux
|
r980 | // Before creating the series, we handle its data holes | ||
| handleDataHoles(); | ||||
|
Alexandre Leroux
|
r953 | return std::make_shared<SpectrogramSeries>( | ||
| std::move(m_XAxisData), std::move(m_YAxisData), std::move(m_ValuesData), | ||||
| Unit{"t", true}, // x-axis unit is always a time unit | ||||
| m_Properties.value(Y_AXIS_UNIT_PROPERTY).value<Unit>(), | ||||
|
Alexandre Leroux
|
r987 | m_Properties.value(VALUES_UNIT_PROPERTY).value<Unit>(), | ||
| m_Properties.value(MIN_SAMPLING_PROPERTY).value<double>()); | ||||
|
Alexandre Leroux
|
r949 | } | ||
| void SpectrogramParserHelper::readPropertyLine(const QString &line) | ||||
| { | ||||
|
Alexandre Leroux
|
r951 | // Set of functions to test on the line to generate a property. If a function is valid (i.e. a | ||
| // property has been generated for the line), the line is treated as processed and the other | ||||
| // functions are not called | ||||
| std::vector<std::function<bool()> > functions{ | ||||
| // values unit | ||||
| [&] { | ||||
| return tryReadUnit(m_Properties, VALUES_UNIT_PROPERTY, line, | ||||
| SPECTROGRAM_VALUES_UNIT_REGEX); | ||||
| }, | ||||
| // y-axis unit | ||||
| [&] { | ||||
| return tryReadUnit(m_Properties, Y_AXIS_UNIT_PROPERTY, line, | ||||
| SPECTROGRAM_Y_AXIS_UNIT_REGEX); | ||||
| }, | ||||
| // min sampling | ||||
| [&] { | ||||
| return tryReadDouble(m_Properties, MIN_SAMPLING_PROPERTY, line, | ||||
| SPECTROGRAM_MIN_SAMPLING_REGEX); | ||||
| }, | ||||
| // max sampling | ||||
| [&] { | ||||
| return tryReadDouble(m_Properties, MAX_SAMPLING_PROPERTY, line, | ||||
| SPECTROGRAM_MAX_SAMPLING_REGEX); | ||||
| }, | ||||
| // fill value | ||||
| [&] { | ||||
| return tryReadDouble(m_Properties, FILL_VALUE_PROPERTY, line, | ||||
| SPECTROGRAM_FILL_VALUE_REGEX); | ||||
| }, | ||||
| // min bounds of each band | ||||
| [&] { | ||||
| return tryReadDoubles(m_Properties, MIN_BANDS_PROPERTY, line, | ||||
| SPECTROGRAM_MIN_BANDS_REGEX); | ||||
| }, | ||||
| // max bounds of each band | ||||
| [&] { | ||||
| return tryReadDoubles(m_Properties, MAX_BANDS_PROPERTY, line, | ||||
| SPECTROGRAM_MAX_BANDS_REGEX); | ||||
|
Alexandre Leroux
|
r985 | }, | ||
| // start time of data | ||||
| [&] { | ||||
| return tryReadDate(m_Properties, START_TIME_PROPERTY, line, | ||||
| SPECTROGRAM_START_TIME_REGEX); | ||||
| }, | ||||
| // end time of data | ||||
| [&] { | ||||
| return tryReadDate(m_Properties, END_TIME_PROPERTY, line, SPECTROGRAM_END_TIME_REGEX); | ||||
|
Alexandre Leroux
|
r951 | }}; | ||
| for (auto function : functions) { | ||||
| // Stops at the first function that is valid | ||||
| if (function()) { | ||||
| return; | ||||
| } | ||||
| } | ||||
|
Alexandre Leroux
|
r949 | } | ||
| void SpectrogramParserHelper::readResultLine(const QString &line) | ||||
| { | ||||
|
Alexandre Leroux
|
r953 | tryReadResult(m_XAxisData, m_ValuesData, line, m_ValuesIndexes, m_FillValue); | ||
|
Alexandre Leroux
|
r949 | } | ||
|
Alexandre Leroux
|
r980 | void SpectrogramParserHelper::handleDataHoles() | ||
| { | ||||
| // Fills data holes according to the max resolution found in the AMDA file | ||||
| auto resolution = m_Properties.value(MAX_SAMPLING_PROPERTY).value<double>(); | ||||
| auto fillValue = m_Properties.value(FILL_VALUE_PROPERTY).value<double>(); | ||||
|
Alexandre Leroux
|
r984 | auto minBound = m_Properties.value(START_TIME_PROPERTY).value<double>(); | ||
| auto maxBound = m_Properties.value(END_TIME_PROPERTY).value<double>(); | ||||
|
Alexandre Leroux
|
r980 | |||
|
Alexandre Leroux
|
r984 | DataSeriesUtils::fillDataHoles(m_XAxisData, m_ValuesData, resolution, fillValue, minBound, | ||
| maxBound); | ||||
|
Alexandre Leroux
|
r980 | } | ||
|
Alexandre Leroux
|
r945 | // ////////////////// // | ||
| // VectorParserHelper // | ||||
| // ////////////////// // | ||||
| bool VectorParserHelper::checkProperties() | ||||
| { | ||||
|
Alexandre Leroux
|
r946 | return checkUnit(m_Properties, X_AXIS_UNIT_PROPERTY, | ||
| QObject::tr("The x-axis unit could not be found in the file")); | ||||
|
Alexandre Leroux
|
r945 | } | ||
| std::shared_ptr<IDataSeries> VectorParserHelper::createSeries() | ||||
| { | ||||
|
Alexandre Leroux
|
r947 | return std::make_shared<VectorSeries>(std::move(m_XAxisData), std::move(m_ValuesData), | ||
| m_Properties.value(X_AXIS_UNIT_PROPERTY).value<Unit>(), | ||||
| m_Properties.value(VALUES_UNIT_PROPERTY).value<Unit>()); | ||||
|
Alexandre Leroux
|
r945 | } | ||
| void VectorParserHelper::readPropertyLine(const QString &line) | ||||
| { | ||||
|
Alexandre Leroux
|
r946 | tryReadUnit(m_Properties, X_AXIS_UNIT_PROPERTY, line, DEFAULT_X_AXIS_UNIT_REGEX, true); | ||
|
Alexandre Leroux
|
r945 | } | ||
| void VectorParserHelper::readResultLine(const QString &line) | ||||
| { | ||||
|
Alexandre Leroux
|
r947 | tryReadResult(m_XAxisData, m_ValuesData, line, valuesIndexes()); | ||
| } | ||||
| std::vector<int> VectorParserHelper::valuesIndexes() const | ||||
| { | ||||
| // 3 values to read, in order in the file (x, y, z) | ||||
| static auto result = std::vector<int>{0, 1, 2}; | ||||
| return result; | ||||
|
Alexandre Leroux
|
r945 | } | ||