Implement motion_direction block

Author: adazem009

src/blocks/motionblocks.cpp

@@ -56,6 +56,7 @@ void MotionBlocks::registerBlocks(IEngine *engine)
     engine->addCompileFunction(this, "motion_setrotationstyle", &compileSetRotationStyle);
     engine->addCompileFunction(this, "motion_xposition", &compileXPosition);
     engine->addCompileFunction(this, "motion_yposition", &compileYPosition);
+    engine->addCompileFunction(this, "motion_direction", &compileDirection);
 }
 
 CompilerValue *MotionBlocks::compileMoveSteps(Compiler *compiler)
@@ -345,6 +346,14 @@ CompilerValue *MotionBlocks::compileYPosition(Compiler *compiler)
         return compiler->addTargetFunctionCall("motion_yposition", Compiler::StaticType::Number);
 }
 
+CompilerValue *MotionBlocks::compileDirection(Compiler *compiler)
+{
+    if (compiler->target()->isStage())
+        return compiler->addConstValue(90);
+    else
+        return compiler->addTargetFunctionCall("motion_direction", Compiler::StaticType::Number);
+}
+
 extern "C" void motion_movesteps(Sprite *sprite, double steps)
 {
     double dir = sprite->direction();
@@ -737,3 +746,8 @@ extern "C" double motion_yposition(Sprite *sprite)
 {
     return sprite->y();
 }
+
+extern "C" double motion_direction(Sprite *sprite)
+{
+    return sprite->direction();
+}

src/blocks/motionblocks.h

@@ -34,6 +34,7 @@ class MotionBlocks : public IExtension
         static CompilerValue *compileSetRotationStyle(Compiler *compiler);
         static CompilerValue *compileXPosition(Compiler *compiler);
         static CompilerValue *compileYPosition(Compiler *compiler);
+        static CompilerValue *compileDirection(Compiler *compiler);
 };
 
 } // namespace libscratchcpp

test/blocks/motion_blocks_test.cpp

@@ -1979,3 +1979,41 @@ TEST_F(MotionBlocksTest, YPosition)
         ASSERT_EQ(Value(list->data()[0]).toDouble(), 0);
     }
 }
+
+TEST_F(MotionBlocksTest, Direction)
+{
+    {
+        auto sprite = std::make_shared<Sprite>();
+        ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+
+        builder.addBlock("motion_direction");
+        builder.captureBlockReturnValue();
+        builder.build();
+
+        sprite->setX(5.2);
+        sprite->setY(-0.25);
+        sprite->setDirection(-61.42);
+        builder.run();
+
+        List *list = builder.capturedValues();
+        ASSERT_EQ(list->size(), 1);
+        ASSERT_EQ(Value(list->data()[0]).toDouble(), -61.42);
+    }
+
+    m_engine->clear();
+
+    {
+        auto stage = std::make_shared<Stage>();
+        ScriptBuilder builder(m_extension.get(), m_engine, stage);
+
+        builder.addBlock("motion_direction");
+        builder.captureBlockReturnValue();
+
+        builder.build();
+        builder.run();
+
+        List *list = builder.capturedValues();
+        ASSERT_EQ(list->size(), 1);
+        ASSERT_EQ(Value(list->data()[0]).toDouble(), 90);
+    }
+}