Change stretchpanel to QToolBar

2022-04-18 10:20:19 +02:00
parent ce836a8ff3
commit ea834ebd16
6 changed files with 29 additions and 44 deletions
@@ -0,0 +1,65 @@
+#include "stretchtoolbar.h"
+#include <QVBoxLayout>
+#include <QDebug>
+#include <QToolButton>
+#include "imageringlist.h"
+
+const float BLACK_POINT_SIGMA = -2.8f;
+const float MAD_TO_SIGMA = 1.4826f;
+const float TARGET_BACKGROUND = 0.25f;
+
+float MTF(float x, float m)
+{
+    if(x < 0)return 0;
+    if(x > 1)return 1;
+    return ((m - 1) * x) / ((2 * m - 1) * x - m);
+}
+
+StretchToolbar::StretchToolbar(QWidget *parent) : QToolBar(tr("Stretch toolbar"), parent)
+{
+    setObjectName("stretchtoolbar");
+    m_stfSlider = new STFSlider(this);
+    addWidget(m_stfSlider);
+    connect(m_stfSlider, SIGNAL(paramChanged(float, float, float)), this, SIGNAL(paramChanged(float,float,float)));
+
+    QAction *autoStretchButton = addAction(QIcon(":/nuke.png"), tr("Auto Stretch F12"));
+    autoStretchButton->setShortcut(Qt::Key_F12);
+    connect(autoStretchButton, SIGNAL(triggered()), this, SIGNAL(autoStretch()));
+
+    QAction *resetButton = addAction(style()->standardIcon(QStyle::SP_DialogResetButton), tr("Reset Screen Transfer Function F11"));
+    resetButton->setShortcut(Qt::Key_F11);
+    connect(resetButton, &QAction::triggered, this, &StretchToolbar::resetMTF);
+
+    QAction *invertButton = addAction(QIcon(":/invert.png"), tr("Invert colors"));
+    invertButton->setCheckable(true);
+    connect(invertButton, SIGNAL(toggled(bool)), this, SIGNAL(invert(bool)));
+
+    QAction *superPixelButton = addAction(QIcon(":/bayer.png"), tr("Superpixel CFA draw 2x2 pixel as one"));
+    superPixelButton->setCheckable(true);
+    connect(superPixelButton, SIGNAL(toggled(bool)), this, SIGNAL(superPixel(bool)));
+}
+
+void StretchToolbar::stretchImage(Image *img)
+{
+    if(img)
+    {
+        if(img->rawImage())
+        {
+            double median, mad;
+            img->rawImage()->imageStats(nullptr, nullptr, &median, nullptr, nullptr, &mad);
+            median /= img->rawImage()->norm();
+            mad /= img->rawImage()->norm();
+            float bp = median + mad * BLACK_POINT_SIGMA * MAD_TO_SIGMA;
+            float mid = MTF(median - bp, TARGET_BACKGROUND);
+            m_stfSlider->setMTFParams(bp, mid, 1.0f);
+            emit paramChanged(m_stfSlider->blackPoint(), m_stfSlider->midPoint(), 1.0f);
+        }
+    }
+}
+
+void StretchToolbar::resetMTF()
+{
+    m_stfSlider->setMTFParams(0, 0.5, 1);
+    emit paramChanged(0, 0.5, 1);
+}
+