Merge pull request #491 from atsampson/nroptions

Add options for NTSC noise reduction levels
happycube · Jun 1, 2020 · d0474eb · d0474eb
2 parents fd9f66f + d14602c
commit d0474eb
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/tools/ld-chroma-decoder/main.cpp b/tools/ld-chroma-decoder/main.cpp
@@ -181,6 +181,18 @@ int main(int argc, char *argv[])
                                         QCoreApplication::translate("main", "NTSC: Use 75% white-point (default 100%)"));
     parser.addOption(whitePointOption);
 
+    // Option to set the chroma noise reduction level
+    QCommandLineOption chromaNROption(QStringList() << "chroma-nr",
+                                      QCoreApplication::translate("main", "NTSC: Chroma noise reduction level in dB (default 0.0)"),
+                                      QCoreApplication::translate("main", "number"));
+    parser.addOption(chromaNROption);
+
+    // Option to set the luma noise reduction level
+    QCommandLineOption lumaNROption(QStringList() << "luma-nr",
+                                    QCoreApplication::translate("main", "NTSC: Luma noise reduction level in dB (default 1.0)"),
+                                    QCoreApplication::translate("main", "number"));
+    parser.addOption(lumaNROption);
+
     // -- PAL decoder options --
 
     // Option to use Simple PAL UV filter
@@ -313,6 +325,26 @@ int main(int argc, char *argv[])
         combConfig.showOpticalFlowMap = true;
     }
 
+    if (parser.isSet(chromaNROption)) {
+        combConfig.cNRLevel = parser.value(chromaNROption).toDouble();
+
+        if (combConfig.cNRLevel < 0.0) {
+            // Quit with error
+            qCritical("Chroma noise reduction cannot be negative");
+            return -1;
+        }
+    }
+
+    if (parser.isSet(lumaNROption)) {
+        combConfig.yNRLevel = parser.value(lumaNROption).toDouble();
+
+        if (combConfig.yNRLevel < 0.0) {
+            // Quit with error
+            qCritical("Luma noise reduction cannot be negative");
+            return -1;
+        }
+    }
+
     if (parser.isSet(transformModeOption)) {
         const QString name = parser.value(transformModeOption);