PlottablesRenderingUtils.cpp
120 lines
| 4.0 KiB
| text/x-c
|
CppLexer
Alexandre Leroux
|
r917 | #include "Visualization/PlottablesRenderingUtils.h" | ||
|
Alexandre Leroux
|
r918 | #include <Common/ColorUtils.h> | ||
|
Alexandre Leroux
|
r917 | #include <Data/ScalarSeries.h> | ||
|
Alexandre Leroux
|
r921 | #include <Data/SpectrogramSeries.h> | ||
|
Alexandre Leroux
|
r917 | #include <Data/VectorSeries.h> | ||
| #include <Visualization/qcustomplot.h> | ||||
| namespace { | ||||
|
Alexandre Leroux
|
r921 | /// Default gradient used for colormap | ||
| const auto DEFAULT_COLORMAP_GRADIENT = QCPColorGradient::gpJet; | ||||
|
Alexandre Leroux
|
r917 | /** | ||
| * Delegate used to set plottables properties | ||||
| */ | ||||
| template <typename T, typename Enabled = void> | ||||
| struct PlottablesSetter { | ||||
| static void setProperties(T &, PlottablesMap &) | ||||
| { | ||||
| // Default implementation does nothing | ||||
| } | ||||
| }; | ||||
| /** | ||||
| * Specialization of PlottablesSetter for scalars and vectors | ||||
| * @sa ScalarSeries | ||||
| * @sa VectorSeries | ||||
| */ | ||||
| template <typename T> | ||||
| struct PlottablesSetter<T, typename std::enable_if_t<std::is_base_of<ScalarSeries, T>::value | ||||
| or std::is_base_of<VectorSeries, T>::value> > { | ||||
| static void setProperties(T &dataSeries, PlottablesMap &plottables) | ||||
| { | ||||
|
Alexandre Leroux
|
r918 | // Gets the number of components of the data series | ||
| dataSeries.lockRead(); | ||||
| auto componentCount = dataSeries.valuesData()->componentCount(); | ||||
| dataSeries.unlock(); | ||||
| // Generates colors for each component | ||||
| auto colors = ColorUtils::colors(Qt::blue, Qt::red, componentCount); | ||||
| // For each component of the data series, creates a QCPGraph to add to the plot | ||||
| for (auto i = 0; i < componentCount; ++i) { | ||||
| auto graph = plottables.at(i); | ||||
| graph->setPen(QPen{colors.at(i)}); | ||||
| } | ||||
|
Alexandre Leroux
|
r917 | } | ||
| }; | ||||
|
Alexandre Leroux
|
r921 | /** | ||
| * Specialization of PlottablesSetter for spectrograms | ||||
| * @sa SpectrogramSeries | ||||
| */ | ||||
| template <typename T> | ||||
| struct PlottablesSetter<T, | ||||
| typename std::enable_if_t<std::is_base_of<SpectrogramSeries, T>::value> > { | ||||
| static void setProperties(T &, PlottablesMap &plottables) | ||||
| { | ||||
| // Checks that for a spectrogram there is only one plottable, that is a colormap | ||||
| if (plottables.size() != 1) { | ||||
| return; | ||||
| } | ||||
| if (auto colormap = dynamic_cast<QCPColorMap *>(plottables.begin()->second)) { | ||||
| colormap->setInterpolate(false); // No value interpolation | ||||
| colormap->setTightBoundary(true); | ||||
| // Finds color scale in the colormap's plot to associate with it | ||||
| auto plot = colormap->parentPlot(); | ||||
| auto plotElements = plot->plotLayout()->elements(false); | ||||
| for (auto plotElement : plotElements) { | ||||
| if (auto colorScale = dynamic_cast<QCPColorScale *>(plotElement)) { | ||||
| colormap->setColorScale(colorScale); | ||||
| } | ||||
| } | ||||
| // Sets gradient used for color scale | ||||
| colormap->setGradient(DEFAULT_COLORMAP_GRADIENT); | ||||
| colormap->rescaleDataRange(); | ||||
| } | ||||
| } | ||||
| }; | ||||
|
Alexandre Leroux
|
r917 | /** | ||
| * Default implementation of IPlottablesHelper, which takes data series to set plottables properties | ||||
| * @tparam T the data series' type | ||||
| */ | ||||
| template <typename T> | ||||
| struct PlottablesHelper : public IPlottablesHelper { | ||||
| explicit PlottablesHelper(T &dataSeries) : m_DataSeries{dataSeries} {} | ||||
| void setProperties(PlottablesMap &plottables) override | ||||
| { | ||||
| PlottablesSetter<T>::setProperties(m_DataSeries, plottables); | ||||
| } | ||||
| T &m_DataSeries; | ||||
| }; | ||||
| } // namespace | ||||
| std::unique_ptr<IPlottablesHelper> | ||||
| IPlottablesHelperFactory::create(std::shared_ptr<IDataSeries> dataSeries) noexcept | ||||
| { | ||||
| if (auto scalarSeries = std::dynamic_pointer_cast<ScalarSeries>(dataSeries)) { | ||||
| return std::make_unique<PlottablesHelper<ScalarSeries> >(*scalarSeries); | ||||
| } | ||||
|
Alexandre Leroux
|
r921 | else if (auto spectrogramSeries = std::dynamic_pointer_cast<SpectrogramSeries>(dataSeries)) { | ||
| return std::make_unique<PlottablesHelper<SpectrogramSeries> >(*spectrogramSeries); | ||||
| } | ||||
|
Alexandre Leroux
|
r917 | else if (auto vectorSeries = std::dynamic_pointer_cast<VectorSeries>(dataSeries)) { | ||
| return std::make_unique<PlottablesHelper<VectorSeries> >(*vectorSeries); | ||||
| } | ||||
| else { | ||||
| return std::make_unique<PlottablesHelper<IDataSeries> >(*dataSeries); | ||||
| } | ||||
| } | ||||