X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=diffusion_effect.cpp;h=a202f8c4d017c60828dccbf7624572f19a069a0d;hp=8d2903bb8e17e08e571a13da8e47a1789fa8fbd9;hb=f270222986351338627231f5d082f884b5d08101;hpb=1e2d23133575fc1cfeb14652e4b383883bf24512

diff --git a/diffusion_effect.cpp b/diffusion_effect.cpp
index 8d2903b..a202f8c 100644
--- a/diffusion_effect.cpp
+++ b/diffusion_effect.cpp
@@ -1,8 +1,8 @@
-#include <math.h>
 #include <assert.h>
+#include <vector>
 
-#include "diffusion_effect.h"
 #include "blur_effect.h"
+#include "diffusion_effect.h"
 #include "effect_chain.h"
 #include "util.h"
 
@@ -12,14 +12,19 @@ DiffusionEffect::DiffusionEffect()
 {
 }
 
-void DiffusionEffect::add_self_to_effect_chain(EffectChain *chain, const std::vector<Effect *> &inputs) {
-	assert(inputs.size() == 1);
-	blur->add_self_to_effect_chain(chain, inputs);
+void DiffusionEffect::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 *overlay_matte_node = graph->add_node(overlay_matte);
+	graph->replace_receiver(self, overlay_matte_node);
+	graph->connect_nodes(input, blur_node);
+	graph->connect_nodes(blur_node, overlay_matte_node);
+	graph->replace_sender(self, overlay_matte_node);
 
-	std::vector<Effect *> overlay_matte_inputs;
-	overlay_matte_inputs.push_back(inputs[0]);
-	overlay_matte_inputs.push_back(chain->last_added_effect());  // FIXME
-	overlay_matte->add_self_to_effect_chain(chain, overlay_matte_inputs);
+	self->disabled = true;
 }
 
 bool DiffusionEffect::set_float(const std::string &key, float value) {