X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fcolorcorrection%2Fwaveformgenerator.h;h=d974ee9c2f173054309063d2bcdd52f1016359d5;hb=82160d2172d631c106edb75e7c946aa810cf47d0;hp=d3abd6047b2b8b673fc07381358780d5f7a87cd1;hpb=b540689a832244cfc6f4c5e62f2b01644f5bc437;p=kdenlive

diff --git a/src/colorcorrection/waveformgenerator.h b/src/colorcorrection/waveformgenerator.h
index d3abd604..d974ee9c 100644
--- a/src/colorcorrection/waveformgenerator.h
+++ b/src/colorcorrection/waveformgenerator.h
@@ -12,21 +12,25 @@
 #define WAVEFORMGENERATOR_H
 
 #include <QObject>
-#include <QImage>
-#include <QSize>
+class QImage;
+class QSize;
 
 class WaveformGenerator : public QObject
 {
     Q_OBJECT
 
 public:
+    enum PaintMode { PaintMode_Green, PaintMode_Yellow, PaintMode_White };
+    enum Rec { Rec_601, Rec_709 };
+
     WaveformGenerator();
     ~WaveformGenerator();
 
-    QImage calculateWaveform(const QSize &waveformSize, const QImage &image, const bool &drawAxis, const uint &accelFactor = 1);
+    QImage calculateWaveform(const QSize &waveformSize, const QImage &image, WaveformGenerator::PaintMode paintMode,
+                             bool drawAxis, const WaveformGenerator::Rec rec, uint accelFactor = 1);
 
-signals:
-    void signalCalculationFinished(QImage image, const uint &ms);
+//signals:
+    //void signalCalculationFinished(QImage image, const uint &ms);
 
 };