manim-chess 0.0.33__py3-none-any.whl → 0.0.35__py3-none-any.whl

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.dist-info → manim_chess-0.0.35.dist-info}/METADATA

@@ -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.dist-info → manim_chess-0.0.35.dist-info}/RECORD

@@ -1,5 +1,5 @@
 manim_chess/__init__.py,sha256=SEtbIVW1Lnh-ZF4_mMYWpI-UFQLOdnGkOD7W0iqK0uc,242
-manim_chess/board.py,sha256=jOA4fvU9zDD67uDdRhrBvpyfk9afJBDMLDu4aMCj_RI,20003
+manim_chess/board.py,sha256=Ljq_9AS8GKejOuvtCHdXdN6hZ2BTMMAwCmpKJRLA67Y,20219
 manim_chess/evaluation_bar.py,sha256=ELjhyFPi5x_Q4Gw3U0FLVhQqN55k8rYCu-lpkru4nPo,3655
 manim_chess/game_player.py,sha256=fESDu5eksZK9ic47CwQNgeq1rpkcMKt0O7_X7-E5lp0,31102
 manim_chess/piece_svgs/bB.svg,sha256=d-nev6XLgFKglq_9SdXU2EIxPM1RlNWoXji4p4MYhb8,683
@@ -15,7 +15,7 @@ manim_chess/piece_svgs/wP.svg,sha256=BZa3zm9xUXgAtRA_SrA3DUeebWbmnBQR8nSQ0gdy6NY
 manim_chess/piece_svgs/wQ.svg,sha256=3SCkD3QlZH7SwQ435sYe8uP3Dx9zy_xxdSy5_DbdwzU,1102
 manim_chess/piece_svgs/wR.svg,sha256=6ela3ONk9djAPG3E2pqfhPVewWSyCRAtRvfm9Ycspcc,481
 manim_chess/pieces.py,sha256=890HggU4vGv_PgLyUPnJj2p3ZdivoqG5BukjqsS6MeU,7634
-manim_chess-0.0.33.dist-info/LICENSE.txt,sha256=B-V-3Tb7v93ub_E_3quknwdeUDtveYnceR0DcdAgolc,1067
-manim_chess-0.0.33.dist-info/METADATA,sha256=S2ryW_FLYh6nLN58xy-EYrcUe9iry4Jb6sKEllPCKV0,7154
-manim_chess-0.0.33.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-manim_chess-0.0.33.dist-info/RECORD,,
+manim_chess-0.0.35.dist-info/LICENSE.txt,sha256=B-V-3Tb7v93ub_E_3quknwdeUDtveYnceR0DcdAgolc,1067
+manim_chess-0.0.35.dist-info/METADATA,sha256=0YyaH5DPxrg6CUMHXeJj0qGtKckaC5sq3j31cUaMaPY,7154
+manim_chess-0.0.35.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+manim_chess-0.0.35.dist-info/RECORD,,