X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=diffusion_effect.cpp;h=a202f8c4d017c60828dccbf7624572f19a069a0d;hp=cf3188bf6013bfe162148a71cc122c0ee10bcd59;hb=f270222986351338627231f5d082f884b5d08101;hpb=05ae48a62f4a507c1eef75b9220f88f2b9fda563

diff --git a/diffusion_effect.cpp b/diffusion_effect.cpp
index cf3188b..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,19 +12,19 @@ DiffusionEffect::DiffusionEffect()
 {
 }
 
-std::string DiffusionEffect::output_fragment_shader()
+void DiffusionEffect::rewrite_graph(EffectChain *graph, Node *self)
 {
-	return read_file("sandbox_effect.frag");
-}
+	assert(self->incoming_links.size() == 1);
+	Node *input = self->incoming_links[0];
 
-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);
+	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->get_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) {