git.sesse.net Git - movit/blob - unsharp_mask_effect.cpp

   1 #include <assert.h>
   2 #include <vector>
   3
   4 #include "blur_effect.h"
   5 #include "effect_chain.h"
   6 #include "mix_effect.h"
   7 #include "unsharp_mask_effect.h"
   8 #include "util.h"
   9
  10 using namespace std;
  11
  12 namespace movit {
  13
  14 UnsharpMaskEffect::UnsharpMaskEffect()
  15         : blur(new BlurEffect),
  16           mix(new MixEffect)
  17 {
  18         CHECK(mix->set_float("strength_first", 1.0f));
  19         CHECK(mix->set_float("strength_second", -0.3f));
  20 }
  21
  22 void UnsharpMaskEffect::rewrite_graph(EffectChain *graph, Node *self)
  23 {
  24         assert(self->incoming_links.size() == 1);
  25         Node *input = self->incoming_links[0];
  26
  27         Node *blur_node = graph->add_node(blur);
  28         Node *mix_node = graph->add_node(mix);
  29         graph->replace_receiver(self, mix_node);
  30         graph->connect_nodes(input, blur_node);
  31         graph->connect_nodes(blur_node, mix_node);
  32         graph->replace_sender(self, mix_node);
  33
  34         self->disabled = true;
  35 }
  36
  37 bool UnsharpMaskEffect::set_float(const string &key, float value) {
  38         if (key == "amount") {
  39                 bool ok = mix->set_float("strength_first", 1.0f + value);
  40                 return ok && mix->set_float("strength_second", -value);
  41         }
  42         return blur->set_float(key, value);
  43 }
  44
  45 }  // namespace movit