manim-chess 0.0.33__tar.gz → 0.0.35__tar.gz

{manim_chess-0.0.33 → manim_chess-0.0.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: manim-chess
-Version: 0.0.33
+Version: 0.0.35
 Summary: A plugin for animating chessboards. Includes additional features such as PGN conversion, arrows, etc.
 Author: swoyer2
 Author-email: swoyer.logan@gmail.com

{manim_chess-0.0.33 → manim_chess-0.0.35}/manim_chess/board.py

@@ -21,6 +21,14 @@ class Board(Mobject):
         A list of coordinates of squares that are currently highlighted.
     arrows : list
         A list of arrow objects currently drawn on the board.
+    color_dark : ManimColor
+        The manim color of the dark squares 
+    color_light : ManimColor
+        The manim color of the light squares
+    color_highlight_light : ManimColor
+        The manim color of the highlight on dark squares 
+    color_highlight_dark : ManimColor
+        The manim color of the highlight on light squares 
 
     Methods:
     -------
@@ -66,11 +74,15 @@ class Board(Mobject):
         Returns the piece at a given coordinate, if any.
     """
 
-    def __init__(self) -> None:
+    def __init__(self, color_dark='#769656', color_light='#eeeed2', color_highlight_light='#F7F769', color_highlight_dark='#BBCB2B') -> None:
         """
         Initializes the Board object.
         """
         super().__init__()
+        self.color_dark = ManimColor(color_dark)
+        self.color_light = ManimColor(color_light)
+        self.color_highlight_light = ManimColor(color_highlight_light)
+        self.color_highlight_dark = ManimColor(color_highlight_dark)
         self.size_of_board = 8
         self.cell_size = 0.8  # Size of each square in the board
         self.squares = {}  # squares[coordinate] = square
@@ -90,8 +102,8 @@ class Board(Mobject):
 
         for row in range(self.size_of_board):
             for col in range(self.size_of_board):
-                CHESS_GREEN = ManimColor('#769656')
-                CHESS_WHITE = ManimColor('#eeeed2')
+                CHESS_GREEN = self.color_dark
+                CHESS_WHITE = self.color_light
                 color = CHESS_WHITE if (row + col) % 2 else CHESS_GREEN
                 # Create a square for each cell in the board
                 square = Square(side_length=self.cell_size)
@@ -123,12 +135,9 @@ class Board(Mobject):
         number : str
             The number to be displayed on the square.
         """
-        CHESS_GREEN = ManimColor('#769656')
-        CHESS_WHITE = ManimColor('#eeeed2')
-
         offset = np.array([self.cell_size / 8, -self.cell_size / 6, 0])
 
-        number_color = CHESS_WHITE if square.fill_color == CHESS_GREEN else CHESS_GREEN
+        number_color = self.color_light if square.fill_color == self.color_dark else self.color_dark
         number = Text(f'{number}', color=number_color, font_size=14 * self.cell_size, font="Arial")
         square_top_left = square.get_center() + np.array([-self.cell_size / 2, self.cell_size / 2, 0])
         number.move_to(square_top_left + offset)
@@ -145,12 +154,10 @@ class Board(Mobject):
         letter : str
             The letter to be displayed on the square.
         """
-        CHESS_GREEN = ManimColor('#769656')
-        CHESS_WHITE = ManimColor('#eeeed2')
 
         offset = np.array([-self.cell_size / 8, self.cell_size / 6, 0])
 
-        letter_color = CHESS_WHITE if square.fill_color == CHESS_GREEN else CHESS_GREEN
+        letter_color = self.color_light if square.fill_color == self.color_dark else self.color_dark
         letter = Text(f'{letter}', color=letter_color, font_size=14 * self.cell_size, font="Arial")
         square_bot_right = square.get_center() + np.array([self.cell_size / 2, -self.cell_size / 2, 0])
         letter.move_to(square_bot_right + offset)
@@ -326,13 +333,11 @@ class Board(Mobject):
         coordinate : str
             The coordinate of the square to be unmarked.
         """
-        CHESS_GREEN = ManimColor('#769656')
-        CHESS_WHITE = ManimColor('#eeeed2')
 
         if self.is_light_square(coordinate):
-            self.squares[coordinate].set_fill(CHESS_WHITE)
+            self.squares[coordinate].set_fill(self.color_light)
         else:
-            self.squares[coordinate].set_fill(CHESS_GREEN)
+            self.squares[coordinate].set_fill(self.color_dark)
 
         # Add back text on square if needed
         if coordinate[1] == '1':
@@ -349,13 +354,11 @@ class Board(Mobject):
         coordinate : str
             The coordinate of the square to be highlighted.
         """
-        LIGHT_HIGHLIGHT_COLOR = ManimColor('#F7F769')
-        DARK_HIGHLIGHT_COLOR = ManimColor('#BBCB2B')
 
         if self.is_light_square(coordinate):
-            self.squares[coordinate].set_fill(LIGHT_HIGHLIGHT_COLOR)
+            self.squares[coordinate].set_fill(self.color_highlight_light)
         else:
-            self.squares[coordinate].set_fill(DARK_HIGHLIGHT_COLOR)
+            self.squares[coordinate].set_fill(self.color_highlight_dark)
 
         # Add back text on square if needed
         if coordinate[1] == '1':
@@ -425,9 +428,7 @@ class Board(Mobject):
             self.add(finished_arrow)
         else:
             arrow0 = Line(stroke_width=15, stroke_opacity=.8, fill_color=ARROW_COLOR, stroke_color=ARROW_COLOR)
-            arrow0.reset_endpoints_based_on_tip = lambda *args: None
             arrow1 = Line(stroke_width=15, stroke_opacity=.8, fill_color=ARROW_COLOR, stroke_color=ARROW_COLOR)
-            arrow1.reset_endpoints_based_on_tip = lambda *args: None
 
             if dir_y > 0:
                 buffer_y = np.array([0, 0.25, 0])
@@ -448,7 +449,7 @@ class Board(Mobject):
                 tip.move_to(tip_position + buffer_x)
             else:
                 arrow0.set_points_as_corners([end_position - buffer_x, np.array([tip_position[0], end_position[1], 0])])
-                arrow1.set_points_as_corners([np.array([tip_position[0]-tip_buffer, end_position[1], 0]), tip_position + buffer_y])
+                arrow1.set_points_as_corners([np.array([tip_position[0], end_position[1]+tip_buffer, 0]), tip_position + buffer_y])
                 tip = arrow1.create_tip()
                 tip.move_to(tip_position + buffer_y)
 

{manim_chess-0.0.33 → manim_chess-0.0.35}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "manim-chess"
-version = "0.0.33"
+version = "0.0.35"
 description = "A plugin for animating chessboards. Includes additional features such as PGN conversion, arrows, etc."
 authors = [
     {name = "swoyer2",email = "swoyer.logan@gmail.com"}