Class GridPositionExtractor

src/scope_refine.py:683–750

Extract grid position information from Manim code

class GridPositionExtractor:
    """Extract grid position information from Manim code"""

    def __init__(self):
        # Match place_at_grid and place_in_area methods
        self.grid_patterns = [
            r'self\.place_at_grid\(\s*([^,]+),\s*[\'"]([A-F][1-6])[\'"](?:,\s*scale_factor=([0-9.]+))?\s*\)',
            r'self\.place_in_area\(\s*([^,]+),\s*[\'"]([A-F][1-6])[\'"],\s*[\'"]([A-F][1-6])[\'"](?:,\s*scale_factor=([0-9.]+))?\s*\)',
        ]

    def extract_grid_positions(self, code: str) -> List[GridPosition]:
        """Extract all grid position information from the code"""
        positions = []
        lines = code.split("\n")

        for line_num, line in enumerate(lines, 1):
            # Check place_at_grid
            match = re.search(self.grid_patterns[0], line)
            if match:
                obj_name = match.group(1).strip()
                grid_pos = match.group(2)
                scale = float(match.group(3)) if match.group(3) else None

                positions.append(
                    GridPosition(
                        object_name=obj_name,
                        method="place_at_grid",
                        position=grid_pos,
                        scale_factor=scale,
                        line_number=line_num,
                        original_code=line.strip(),
                    )
                )

            # Check place_in_area
            match = re.search(self.grid_patterns[1], line)
            if match:
                obj_name = match.group(1).strip()
                start_pos = match.group(2)
                end_pos = match.group(3)
                scale = float(match.group(4)) if match.group(4) else None

                positions.append(
                    GridPosition(
                        object_name=obj_name,
                        method="place_in_area",
                        position=f"{start_pos}-{end_pos}",
                        scale_factor=scale,
                        line_number=line_num,
                        original_code=line.strip(),
                    )
                )

        return positions

    def generate_position_table(self, positions: List[GridPosition]) -> str:
        """Generate a position table for MLLM analysis"""
        if not positions: