Add an unsharp mask effect, to complement the more expensive deconvolution sharpening.

[movit] / unsharp_mask_effect.cpp

diff --git a/unsharp_mask_effect.cpp b/unsharp_mask_effect.cpp
new file mode 100644 (file)
index 0000000..db8ccd2
--- /dev/null
+++ b/unsharp_mask_effect.cpp
@@ -0,0 +1,40 @@
+#include <math.h>
+#include <assert.h>
+
+#include "unsharp_mask_effect.h"
+#include "blur_effect.h"
+#include "mix_effect.h"
+#include "effect_chain.h"
+#include "util.h"
+
+UnsharpMaskEffect::UnsharpMaskEffect()
+       : blur(new BlurEffect),
+         mix(new MixEffect)
+{
+       mix->set_float("strength_first", 1.0f);
+       mix->set_float("strength_second", -0.3f);
+}
+
+void UnsharpMaskEffect::rewrite_graph(EffectChain *graph, Node *self)
+{
+       assert(self->incoming_links.size() == 1);
+       Node *input = self->incoming_links[0];
+
+       Node *blur_node = graph->add_node(blur);
+       Node *mix_node = graph->add_node(mix);
+       graph->replace_receiver(self, mix_node);
+       graph->connect_nodes(input, blur_node);
+       graph->connect_nodes(blur_node, mix_node);
+       graph->replace_sender(self, mix_node);
+
+       self->disabled = true;
+}
+
+bool UnsharpMaskEffect::set_float(const std::string &key, float value) {
+       printf("%s = %f\n", key.c_str(), value);
+       if (key == "amount") {
+               bool ok = mix->set_float("strength_first", 1.0f + value);
+               return ok && mix->set_float("strength_second", -value);
+       }
+       return blur->set_float(key, value);
+}