feat: added a feature for shapes in draw clipart

Author: samruddhi-Rahegaonkar

lib/providers/draw_badge_provider.dart

@@ -2,24 +2,66 @@ import 'package:badgemagic/badge_animation/ani_left.dart';
 import 'package:badgemagic/badge_animation/animation_abstract.dart';
 import 'package:flutter/material.dart';
 
+// Optional: shape enum, doesn't affect freehand
+enum DrawShape { freehand, square, rectangle, circle, triangle }
+
 class DrawBadgeProvider extends ChangeNotifier {
-  //List that contains the state of each cell of the badge for draw view
+  // 11x44 LED grid, default all false (off)
   List<List<bool>> _drawViewGrid =
       List.generate(11, (i) => List.generate(44, (j) => false));
 
-  //getter for the drawViewGrid
+  // Drawing mode (true = draw, false = erase)
+  bool isDrawing = true;
+
+  // Currently selected shape
+  DrawShape _selectedShape = DrawShape.freehand;
+
+  // Animation used in badge (not part of drawing)
+  BadgeAnimation currentAnimation = LeftAnimation();
+
+  // Return the current grid
   List<List<bool>> getDrawViewGrid() => _drawViewGrid;
 
-  //setter for the drawViewGrid
-  void setDrawViewGrid(int row, int col) {
-    _drawViewGrid[row][col] = isDrawing;
+  // Return current drawing mode
+  bool getIsDrawing() => isDrawing;
+
+  // Return selected shape
+  DrawShape get selectedShape => _selectedShape;
+
+  // Toggle between drawing and erasing
+  void toggleIsDrawing(bool drawing) {
+    isDrawing = drawing;
     notifyListeners();
   }
 
-  BadgeAnimation currentAnimation = LeftAnimation();
+  // Set selected shape
+  void setShape(DrawShape shape) {
+    _selectedShape = shape;
+    notifyListeners();
+  }
+
+  // Set a single LED (used by gesture drawing)
+  void setDrawViewGrid(int row, int col) {
+    // Only allow grid update for freehand drawing
+    if (_selectedShape == DrawShape.freehand) {
+      if (row >= 0 &&
+          row < _drawViewGrid.length &&
+          col >= 0 &&
+          col < _drawViewGrid[0].length) {
+        _drawViewGrid[row][col] = isDrawing;
+        notifyListeners();
+      }
+    }
+  }
 
+  // Reset the grid to all OFF
+  void resetDrawViewGrid() {
+    _drawViewGrid = List.generate(11, (i) => List.generate(44, (j) => false));
+    notifyListeners();
+  }
+
+  // Load a grid into the draw view
   void updateDrawViewGrid(List<List<bool>> badgeData) {
-    //copy the badgeData to the drawViewGrid and all the drawViewGrid after badgeData will remain unchanged
     for (int i = 0; i < _drawViewGrid.length; i++) {
       for (int j = 0; j < _drawViewGrid[0].length; j++) {
         if (j < badgeData[0].length) {
@@ -31,22 +73,4 @@ class DrawBadgeProvider extends ChangeNotifier {
     }
     notifyListeners();
   }
-
-  //function to reset the state of the cell
-  void resetDrawViewGrid() {
-    _drawViewGrid = List.generate(11, (i) => List.generate(44, (j) => false));
-    notifyListeners();
-  }
-
-  //boolean variable to check for isDrawing on Draw badge screen
-  bool isDrawing = true;
-
-  //function to toggle the isDrawing variable
-  void toggleIsDrawing(bool drawing) {
-    isDrawing = drawing;
-    notifyListeners();
-  }
-
-  //function to get the isDrawing variable
-  bool getIsDrawing() => isDrawing;
 }