Add output parser for dssim algo. (#33)

Co-authored-by: Jonathan Leonard <johanatan@gmail.com>
oliyh · Nov 14, 2020 · b2251fe · b2251fe
1 parent 9e73a86
commit b2251fe
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/clj/kamera/core.clj b/src/clj/kamera/core.clj
@@ -116,6 +116,11 @@
              (update :actual #(.getAbsolutePath %))))
        normalisation-chain))
 
+(def ^:private default-compare-parser #(last (re-find #"all: .* \((.*)\)" %1)))
+(def ^:private compare-parsers
+  {"mae" default-compare-parser
+   "dssim" #(last (re-find #"all: (.*)" %1))})
+
 (defn compare-images [^File expected
                       ^File actual
                       {:keys [metric screenshot-directory normalisations]}
@@ -147,7 +152,7 @@
                     (.getAbsolutePath actual)
                     (.getAbsolutePath difference)]
                    opts)
-           mean-absolute-error (when-let [e (last (re-find #"all: .* \((.*)\)" stderr))]
+           metric (when-let [e ((get compare-parsers metric default-compare-parser) stderr)]
                                  (read-string e))]
 
        (merge-with concat
@@ -160,8 +165,8 @@
                      {:errors [(format "Error comparing images - ImageMagick exited with code %s \n stdout: %s \n stderr: %s"
                                        exit-code stdout stderr)]})
 
-                   (if mean-absolute-error
-                     {:metric mean-absolute-error}
+                   (if metric
+                     {:metric metric}
                      {:errors [(format "Could not parse ImageMagick output\n stdout: %s \n stderr: %s"
                                        stdout stderr)]}))))))