Implement motion_setrotationstyle block

Author: adazem009

src/blocks/motionblocks.cpp

@@ -5,6 +5,7 @@
 #include <scratchcpp/compilerconstant.h>
 #include <scratchcpp/sprite.h>
 #include <scratchcpp/input.h>
+#include <scratchcpp/field.h>
 #include <scratchcpp/value.h>
 #include <scratchcpp/executioncontext.h>
 #include <scratchcpp/thread.h>
@@ -52,6 +53,7 @@ void MotionBlocks::registerBlocks(IEngine *engine)
     engine->addCompileFunction(this, "motion_changeyby", &compileChangeYBy);
     engine->addCompileFunction(this, "motion_sety", &compileSetY);
     engine->addCompileFunction(this, "motion_ifonedgebounce", &compileIfOnEdgeBounce);
+    engine->addCompileFunction(this, "motion_setrotationstyle", &compileSetRotationStyle);
 }
 
 CompilerValue *MotionBlocks::compileMoveSteps(Compiler *compiler)
@@ -299,6 +301,32 @@ CompilerValue *MotionBlocks::compileIfOnEdgeBounce(Compiler *compiler)
     return nullptr;
 }
 
+CompilerValue *MotionBlocks::compileSetRotationStyle(Compiler *compiler)
+{
+    Target *target = compiler->target();
+
+    if (target->isStage())
+        return nullptr;
+
+    Sprite *sprite = static_cast<Sprite *>(target);
+    Field *field = compiler->field("STYLE");
+
+    if (!field)
+        return nullptr;
+
+    std::string option = field->value().toString();
+    sprite->setRotationStyle(field->value().toString());
+
+    if (option == "left-right")
+        compiler->addTargetFunctionCall("motion_set_left_right_style");
+    else if (option == "don't rotate")
+        compiler->addTargetFunctionCall("motion_set_do_not_rotate_style");
+    else if (option == "all around")
+        compiler->addTargetFunctionCall("motion_set_all_around_style");
+
+    return nullptr;
+}
+
 extern "C" void motion_movesteps(Sprite *sprite, double steps)
 {
     double dir = sprite->direction();
@@ -667,6 +695,21 @@ extern "C" void motion_ifonedgebounce(Sprite *sprite)
     sprite->setPosition(fencedX, fencedY);
 }
 
+extern "C" void motion_set_left_right_style(Sprite *sprite)
+{
+    sprite->setRotationStyle(Sprite::RotationStyle::LeftRight);
+}
+
+extern "C" void motion_set_do_not_rotate_style(Sprite *sprite)
+{
+    sprite->setRotationStyle(Sprite::RotationStyle::DoNotRotate);
+}
+
+extern "C" void motion_set_all_around_style(Sprite *sprite)
+{
+    sprite->setRotationStyle(Sprite::RotationStyle::AllAround);
+}
+
 extern "C" double motion_xposition(Sprite *sprite)
 {
     return sprite->x();

src/blocks/motionblocks.h

@@ -31,6 +31,7 @@ class MotionBlocks : public IExtension
         static CompilerValue *compileChangeYBy(Compiler *compiler);
         static CompilerValue *compileSetY(Compiler *compiler);
         static CompilerValue *compileIfOnEdgeBounce(Compiler *compiler);
+        static CompilerValue *compileSetRotationStyle(Compiler *compiler);
 };
 
 } // namespace libscratchcpp

test/blocks/motion_blocks_test.cpp

@@ -1822,3 +1822,83 @@ TEST_F(MotionBlocksTest, IfOnEdgeBounce)
         builder.run();
     }
 }
+
+TEST_F(MotionBlocksTest, SetRotationStyle)
+{
+    // left-right
+    {
+        auto sprite = std::make_shared<Sprite>();
+        ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+
+        builder.addBlock("motion_setrotationstyle");
+        builder.addDropdownField("STYLE", "left-right");
+
+        builder.build();
+        builder.run();
+        ASSERT_EQ(sprite->rotationStyle(), Sprite::RotationStyle::LeftRight);
+    }
+
+    m_engine->clear();
+
+    // don't rotate
+    {
+        auto sprite = std::make_shared<Sprite>();
+        ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+
+        builder.addBlock("motion_setrotationstyle");
+        builder.addDropdownField("STYLE", "don't rotate");
+
+        builder.build();
+        builder.run();
+        ASSERT_EQ(sprite->rotationStyle(), Sprite::RotationStyle::DoNotRotate);
+    }
+
+    m_engine->clear();
+
+    // all around
+    {
+        auto sprite = std::make_shared<Sprite>();
+        ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+
+        builder.addBlock("motion_setrotationstyle");
+        builder.addDropdownField("STYLE", "all around");
+        builder.build();
+
+        sprite->setRotationStyle(Sprite::RotationStyle::DoNotRotate);
+        builder.run();
+        ASSERT_EQ(sprite->rotationStyle(), Sprite::RotationStyle::AllAround);
+    }
+
+    m_engine->clear();
+
+    // invalid
+    {
+        auto sprite = std::make_shared<Sprite>();
+        ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+
+        builder.addBlock("motion_setrotationstyle");
+        builder.addDropdownField("STYLE", "invalid");
+        builder.build();
+
+        sprite->setRotationStyle(Sprite::RotationStyle::LeftRight);
+        builder.run();
+        ASSERT_EQ(sprite->rotationStyle(), Sprite::RotationStyle::LeftRight);
+
+        sprite->setRotationStyle(Sprite::RotationStyle::AllAround);
+        builder.run();
+        ASSERT_EQ(sprite->rotationStyle(), Sprite::RotationStyle::AllAround);
+    }
+
+    m_engine->clear();
+
+    {
+        auto stage = std::make_shared<Stage>();
+        ScriptBuilder builder(m_extension.get(), m_engine, stage);
+
+        builder.addBlock("motion_setrotationstyle");
+        builder.addDropdownField("STYLE", "all around");
+
+        builder.build();
+        builder.run();
+    }
+}