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

diff --git a/src/colorcorrection/waveformgenerator.h b/src/colorcorrection/waveformgenerator.h
index d3abd604..b4f41553 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,
+                             const bool &drawAxis, const WaveformGenerator::Rec rec, const uint &accelFactor = 1);
 
-signals:
-    void signalCalculationFinished(QImage image, const uint &ms);
+//signals:
+    //void signalCalculationFinished(QImage image, const uint &ms);
 
 };