Script: Add hat predicate code

Author: adazem009

include/scratchcpp/script.h

@@ -31,6 +31,9 @@ class LIBSCRATCHCPP_EXPORT Script
         ExecutableCode *code() const;
         void setCode(std::shared_ptr<ExecutableCode> code);
 
+        ExecutableCode *hatPredicateCode() const;
+        void setHatPredicateCode(std::shared_ptr<ExecutableCode> code);
+
         bool runHatPredicate(Target *target);
 
         std::shared_ptr<Thread> start();

src/engine/script.cpp

@@ -42,6 +42,18 @@ void Script::setCode(std::shared_ptr<ExecutableCode> code)
     impl->code = code;
 }
 
+/*! Returns the executable code of the hat predicate. */
+ExecutableCode *Script::hatPredicateCode() const
+{
+    return impl->hatPredicateCode.get();
+}
+
+/*! Sets the executable code of the hat predicate. */
+void Script::setHatPredicateCode(std::shared_ptr<ExecutableCode> code)
+{
+    impl->hatPredicateCode = code;
+}
+
 /*!
  * Runs the edge-activated hat predicate as the given target and returns the reported value.
  * \note If there isn't any predicate, nothing will happen and the returned value will be false.

src/engine/script_p.h

@@ -22,6 +22,7 @@ struct ScriptPrivate
         ScriptPrivate(const ScriptPrivate &) = delete;
 
         std::shared_ptr<ExecutableCode> code;
+        std::shared_ptr<ExecutableCode> hatPredicateCode;
 
         Target *target = nullptr;
         std::shared_ptr<Block> topBlock;

test/script/script_test.cpp

@@ -41,6 +41,16 @@ TEST_F(ScriptTest, Code)
     ASSERT_EQ(script.code(), code.get());
 }
 
+TEST_F(ScriptTest, HatPredicateCode)
+{
+    Script script(nullptr, nullptr, nullptr);
+    ASSERT_EQ(script.hatPredicateCode(), nullptr);
+
+    auto code = std::make_shared<ExecutableCodeMock>();
+    script.setHatPredicateCode(code);
+    ASSERT_EQ(script.hatPredicateCode(), code.get());
+}
+
 TEST_F(ScriptTest, Start)
 {
     Script script1(nullptr, nullptr, nullptr);