X-Git-Url: https://git.sesse.net/?p=movit;a=blobdiff_plain;f=diffusion_effect.cpp;h=145eaf695860608040afec87bc0120eb08188c6d;hp=4e996b83ed999693ea37119375d3d01811ffcc8e;hb=09c983894685554b41f622dadd40ac1a4efc527d;hpb=34121ba75f6274c25c2cee53169819ae96e77d66

diff --git a/diffusion_effect.cpp b/diffusion_effect.cpp
index 4e996b8..145eaf6 100644
--- a/diffusion_effect.cpp
+++ b/diffusion_effect.cpp
@@ -1,33 +1,37 @@
-#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"
 
+using namespace std;
+
+namespace movit {
+
 DiffusionEffect::DiffusionEffect()
 	: blur(new BlurEffect),
 	  overlay_matte(new OverlayMatteEffect)
 {
 }
 
-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->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) {
+bool DiffusionEffect::set_float(const string &key, float value) {
 	if (key == "blurred_mix_amount") {
 		return overlay_matte->set_float(key, value);
 	}
@@ -40,7 +44,9 @@ OverlayMatteEffect::OverlayMatteEffect()
 	register_float("blurred_mix_amount", &blurred_mix_amount);
 }
 
-std::string OverlayMatteEffect::output_fragment_shader()
+string OverlayMatteEffect::output_fragment_shader()
 {
 	return read_file("overlay_matte_effect.frag");
 }
+
+}  // namespace movit