Implement pen_setPenColorParamTo block

Author: adazem009

src/blocks/penblocks.cpp

@@ -35,6 +35,7 @@ void PenBlocks::registerBlocks(IEngine *engine)
     engine->addCompileFunction(this, "pen_penUp", &compilePenUp);
     engine->addCompileFunction(this, "pen_setPenColorToColor", &compileSetPenColorToColor);
     engine->addCompileFunction(this, "pen_changePenColorParamBy", &compileChangePenColorParamBy);
+    engine->addCompileFunction(this, "pen_setPenColorParamTo", &compileSetPenColorParamTo);
     engine->addCompileFunction(this, "pen_changePenSizeBy", &compileChangePenSizeBy);
     engine->addCompileFunction(this, "pen_setPenSizeTo", &compileSetPenSizeTo);
     engine->addCompileFunction(this, "pen_changePenShadeBy", &compileChangePenShadeBy);
@@ -101,6 +102,35 @@ void PenBlocks::compileChangePenColorParamBy(libscratchcpp::Compiler *compiler)
     }
 }
 
+void PenBlocks::compileSetPenColorParamTo(Compiler *compiler)
+{
+    Input *input = compiler->input(COLOR_PARAM);
+
+    if (input->type() != Input::Type::ObscuredShadow) {
+        assert(input->pointsToDropdownMenu());
+        std::string value = input->selectedMenuItem();
+        BlockFunc f = nullptr;
+
+        if (value == "color")
+            f = &setPenColorTo;
+        else if (value == "saturation")
+            f = &setPenSaturationTo;
+        else if (value == "brightness")
+            f = &setPenBrightnessTo;
+        else if (value == "transparency")
+            f = &setPenTransparencyTo;
+
+        if (f) {
+            compiler->addInput(VALUE);
+            compiler->addFunctionCall(f);
+        }
+    } else {
+        compiler->addInput(input);
+        compiler->addInput(VALUE);
+        compiler->addFunctionCall(&setPenColorParamTo);
+    }
+}
+
 void PenBlocks::compileChangePenSizeBy(libscratchcpp::Compiler *compiler)
 {
     compiler->addInput(SIZE);
@@ -343,6 +373,62 @@ unsigned int PenBlocks::changePenTransparencyBy(VirtualMachine *vm)
     return 1;
 }
 
+unsigned int PenBlocks::setPenColorParamTo(VirtualMachine *vm)
+{
+    SpriteModel *model = getSpriteModel(vm);
+
+    if (model) {
+        const auto it = COLOR_PARAM_MAP.find(vm->getInput(0, 2)->toString());
+
+        if (it == COLOR_PARAM_MAP.cend())
+            return 2;
+
+        setOrChangeColorParam(it->second, vm->getInput(1, 2)->toDouble(), model->penState(), false);
+    }
+
+    return 2;
+}
+
+unsigned int PenBlocks::setPenColorTo(VirtualMachine *vm)
+{
+    SpriteModel *model = getSpriteModel(vm);
+
+    if (model)
+        setOrChangeColorParam(ColorParam::COLOR, vm->getInput(0, 1)->toDouble(), model->penState(), false);
+
+    return 1;
+}
+
+unsigned int PenBlocks::setPenSaturationTo(VirtualMachine *vm)
+{
+    SpriteModel *model = getSpriteModel(vm);
+
+    if (model)
+        setOrChangeColorParam(ColorParam::SATURATION, vm->getInput(0, 1)->toDouble(), model->penState(), false);
+
+    return 1;
+}
+
+unsigned int PenBlocks::setPenBrightnessTo(VirtualMachine *vm)
+{
+    SpriteModel *model = getSpriteModel(vm);
+
+    if (model)
+        setOrChangeColorParam(ColorParam::BRIGHTNESS, vm->getInput(0, 1)->toDouble(), model->penState(), false);
+
+    return 1;
+}
+
+unsigned int PenBlocks::setPenTransparencyTo(VirtualMachine *vm)
+{
+    SpriteModel *model = getSpriteModel(vm);
+
+    if (model)
+        setOrChangeColorParam(ColorParam::TRANSPARENCY, vm->getInput(0, 1)->toDouble(), model->penState(), false);
+
+    return 1;
+}
+
 SpriteModel *PenBlocks::getSpriteModel(libscratchcpp::VirtualMachine *vm)
 {
     Target *target = vm->target();
@@ -373,6 +459,10 @@ void PenBlocks::setOrChangeColorParam(ColorParam param, double value, PenState &
         case ColorParam::TRANSPARENCY:
             penState.transparency = std::clamp(value + (change ? penState.transparency : 0), COLOR_PARAM_MIN, COLOR_PARAM_MAX);
             break;
+
+        default:
+            assert(false);
+            return;
     }
 
     penState.updateColor();

src/blocks/penblocks.h

@@ -33,6 +33,7 @@ class PenBlocks : public libscratchcpp::IBlockSection
         static void compilePenUp(libscratchcpp::Compiler *compiler);
         static void compileSetPenColorToColor(libscratchcpp::Compiler *compiler);
         static void compileChangePenColorParamBy(libscratchcpp::Compiler *compiler);
+        static void compileSetPenColorParamTo(libscratchcpp::Compiler *compiler);
         static void compileChangePenSizeBy(libscratchcpp::Compiler *compiler);
         static void compileSetPenSizeTo(libscratchcpp::Compiler *compiler);
         static void compileChangePenShadeBy(libscratchcpp::Compiler *compiler);
@@ -51,6 +52,12 @@ class PenBlocks : public libscratchcpp::IBlockSection
         static unsigned int changePenBrightnessBy(libscratchcpp::VirtualMachine *vm);
         static unsigned int changePenTransparencyBy(libscratchcpp::VirtualMachine *vm);
 
+        static unsigned int setPenColorParamTo(libscratchcpp::VirtualMachine *vm);
+        static unsigned int setPenColorTo(libscratchcpp::VirtualMachine *vm);
+        static unsigned int setPenSaturationTo(libscratchcpp::VirtualMachine *vm);
+        static unsigned int setPenBrightnessTo(libscratchcpp::VirtualMachine *vm);
+        static unsigned int setPenTransparencyTo(libscratchcpp::VirtualMachine *vm);
+
         static unsigned int changePenSizeBy(libscratchcpp::VirtualMachine *vm);
         static unsigned int setPenSizeTo(libscratchcpp::VirtualMachine *vm);
         static unsigned int changePenShadeBy(libscratchcpp::VirtualMachine *vm);