X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=unsharp_mask_effect.cpp;fp=unsharp_mask_effect.cpp;h=db8ccd21d767367c8b72d30b793310885ad11b59;hb=6da2b27aecd1e3ccea21abc6f57e219cd811aab6;hp=0000000000000000000000000000000000000000;hpb=ebe6d6ba984b2fb77b365aeacd1f30d4da59964b;p=movit

diff --git a/unsharp_mask_effect.cpp b/unsharp_mask_effect.cpp
new file mode 100644
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);
+}