From 5affc3769980c160d4e74b908d694c280d83f23e Mon Sep 17 00:00:00 2001
From: itsmattkc <itsmattkc@gmail.com>
Date: Mon, 9 Dec 2019 23:46:50 +1100
Subject: [PATCH] added custom category and data types to transition base class

---
 app/node/block/transition/transition.cpp | 7 +++++++
 app/node/block/transition/transition.h   | 1 +
 2 files changed, 8 insertions(+)

diff --git a/app/node/block/transition/transition.cpp b/app/node/block/transition/transition.cpp
index cf417953c..451ebe1d4 100644
--- a/app/node/block/transition/transition.cpp
+++ b/app/node/block/transition/transition.cpp
@@ -3,9 +3,11 @@
 TransitionBlock::TransitionBlock()
 {
   out_block_input_ = new NodeInput("out_block_in");
+  out_block_input_->set_data_type(NodeParam::kBuffer);
   AddInput(out_block_input_);
 
   in_block_input_ = new NodeInput("in_block_in");
+  in_block_input_->set_data_type(NodeParam::kBuffer);
   AddInput(in_block_input_);
 }
 
@@ -14,6 +16,11 @@ Block::Type TransitionBlock::type() const
   return kTransition;
 }
 
+QString TransitionBlock::Category() const
+{
+  return tr("Transition");
+}
+
 NodeInput *TransitionBlock::out_block_input() const
 {
   return out_block_input_;
diff --git a/app/node/block/transition/transition.h b/app/node/block/transition/transition.h
index dc2213fb5..14c6a3645 100644
--- a/app/node/block/transition/transition.h
+++ b/app/node/block/transition/transition.h
@@ -9,6 +9,7 @@ public:
   TransitionBlock();
 
   virtual Type type() const override;
+  virtual QString Category() const override;
 
   NodeInput* out_block_input() const;