pyscreeps-arena 0.5.7b0__py3-none-any.whl → 0.5.7.2__py3-none-any.whl

pyscreeps_arena/ui/qmapv/test_array_drag.py

@@ -0,0 +1,191 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Test script for array format drag and drop functionality
+"""
+
+import sys
+import json
+from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, 
+                             QHBoxLayout, QPushButton, QLabel, QListWidget)
+from PyQt6.QtCore import Qt, QMimeData, pyqtSignal
+from PyQt6.QtGui import QDrag, QPixmap
+
+# Add the parent directory to the path to import our modules
+sys.path.insert(0, 'f:\\Python\\Python312\\Lib\\site-packages\\pyscreeps_arena\\ui\\qmapv')
+
+from qco import QPSACellObject
+from qcinfo import QPSACellInfo
+
+
+class DragSourceWidget(QWidget):
+    """Widget that can initiate drag operations with array format."""
+    
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._init_ui()
+        
+    def _init_ui(self):
+        """Initialize UI."""
+        layout = QVBoxLayout()
+        
+        # Create test data
+        self.test_array = [
+            {
+                "x": 10,
+                "y": 20,
+                "type": "Creep",
+                "method": "move",
+                "id": "creep1",
+                "name": "Worker1"
+            },
+            {
+                "x": 15,
+                "y": 25,
+                "type": "Structure",
+                "method": "build",
+                "id": "struct1",
+                "name": "Spawn1"
+            },
+            {
+                "x": 30,
+                "y": 40,
+                "type": "Resource",
+                "method": "harvest",
+                "id": "res1",
+                "name": "Energy1"
+            }
+        ]
+        
+        self.test_single = {
+            "x": 50,
+            "y": 60,
+            "type": "Tower",
+            "method": "attack",
+            "id": "tower1",
+            "name": "Defense1"
+        }
+        
+        # Add buttons to trigger drag operations
+        btn_array = QPushButton("Drag Array [{}]")
+        btn_array.clicked.connect(lambda: self._start_drag(self.test_array))
+        layout.addWidget(btn_array)
+        
+        btn_single = QPushButton("Drag Single {}")
+        btn_single.clicked.connect(lambda: self._start_drag(self.test_single))
+        layout.addWidget(btn_single)
+        
+        # Add labels to show what will be dragged
+        label_array = QLabel(f"Array: {json.dumps(self.test_array, indent=2)}")
+        label_array.setWordWrap(True)
+        layout.addWidget(label_array)
+        
+        label_single = QLabel(f"Single: {json.dumps(self.test_single, indent=2)}")
+        label_single.setWordWrap(True)
+        layout.addWidget(label_single)
+        
+        self.setLayout(layout)
+        
+    def _start_drag(self, data):
+        """Start a drag operation with the given data."""
+        print(f"[DEBUG] Starting drag with data: {data}")
+        
+        # Create mime data
+        mime_data = QMimeData()
+        json_str = json.dumps(data)
+        mime_data.setText(json_str)
+        mime_data.setData("application/json", json_str.encode('utf-8'))
+        
+        # Create drag object
+        drag = QDrag(self)
+        drag.setMimeData(mime_data)
+        
+        # Start drag operation
+        result = drag.exec(Qt.DropAction.CopyAction)
+        print(f"[DEBUG] Drag result: {result}")
+
+
+class TestWindow(QMainWindow):
+    """Main test window."""
+    
+    def __init__(self):
+        super().__init__()
+        self.setWindowTitle("Array Format Drag & Drop Test")
+        self.setGeometry(100, 100, 800, 600)
+        self._init_ui()
+        
+    def _init_ui(self):
+        """Initialize UI."""
+        central_widget = QWidget()
+        layout = QHBoxLayout()
+        
+        # Left side - drag source
+        drag_source = DragSourceWidget()
+        layout.addWidget(drag_source)
+        
+        # Right side - drop targets
+        right_layout = QVBoxLayout()
+        
+        # QPSACellObject drop target
+        cell_object_label = QLabel("QPSACellObject (Drop Here):")
+        right_layout.addWidget(cell_object_label)
+        
+        self.cell_object = QPSACellObject()
+        self.cell_object.setMinimumHeight(200)
+        right_layout.addWidget(self.cell_object)
+        
+        # QPSACellInfo drop target
+        cell_info_label = QLabel("QPSACellInfo (Drop Here):")
+        right_layout.addWidget(cell_info_label)
+        
+        self.cell_info = QPSACellInfo()
+        right_layout.addWidget(self.cell_info)
+        
+        # Status label
+        self.status_label = QLabel("Status: Ready")
+        right_layout.addWidget(self.status_label)
+        
+        # Clear buttons
+        clear_layout = QHBoxLayout()
+        btn_clear_object = QPushButton("Clear QPSACellObject")
+        btn_clear_object.clicked.connect(self.cell_object.clear_objects)
+        clear_layout.addWidget(btn_clear_object)
+        
+        btn_clear_info = QPushButton("Clear QPSACellInfo")
+        btn_clear_info.clicked.connect(self._clear_cell_info)
+        clear_layout.addWidget(btn_clear_info)
+        
+        right_layout.addLayout(clear_layout)
+        
+        layout.addLayout(right_layout)
+        central_widget.setLayout(layout)
+        self.setCentralWidget(central_widget)
+        
+        # Connect signals
+        self.cell_object.objectAdded.connect(self._on_object_added)
+        self.cell_object.objectRemoved.connect(self._on_object_removed)
+        
+    def _clear_cell_info(self):
+        """Clear QPSACellInfo objects."""
+        self.cell_info.objects = []
+        self.status_label.setText("Status: QPSACellInfo cleared")
+        
+    def _on_object_added(self, obj_data):
+        """Handle object added to QPSACellObject."""
+        self.status_label.setText(f"Status: Object added - {obj_data.get('name', 'Unknown')}")
+        
+    def _on_object_removed(self, obj_id):
+        """Handle object removed from QPSACellObject."""
+        self.status_label.setText(f"Status: Object removed - {obj_id}")
+
+
+def main():
+    """Main function."""
+    app = QApplication(sys.argv)
+    window = TestWindow()
+    window.show()
+    sys.exit(app.exec())
+
+
+if __name__ == "__main__":
+    main()

pyscreeps_arena/ui/qmapv/test_drag.py

@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Test script for drag functionality in QPSAMapViewer
+"""
+
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
+
+from PyQt6.QtWidgets import QApplication, QWidget, QVBoxLayout, QHBoxLayout, QTextEdit, QLabel
+from PyQt6.QtCore import Qt
+from PyQt6.QtGui import QDropEvent
+from ui.qmapv.qmapv import QPSAMapViewer, CellInfo
+
+class DropTarget(QWidget):
+    """A simple drop target to test drag functionality."""
+    
+    def __init__(self):
+        super().__init__()
+        self.setAcceptDrops(True)
+        self.text_edit = QTextEdit()
+        self.text_edit.setReadOnly(True)
+        self.label = QLabel("Drop target - drag data will appear here")
+        
+        layout = QVBoxLayout()
+        layout.addWidget(self.label)
+        layout.addWidget(self.text_edit)
+        self.setLayout(layout)
+        
+        self.setWindowTitle("Drag Test Target")
+        self.resize(400, 300)
+    
+    def dragEnterEvent(self, event):
+        """Accept drag enter events."""
+        print(f"[TEST] Drag enter event received: hasText={event.mimeData().hasText()}")
+        if event.mimeData().hasText():
+            event.acceptProposedAction()
+    
+    def dropEvent(self, event):
+        """Handle drop events."""
+        print(f"[TEST] Drop event received")
+        if event.mimeData().hasText():
+            text = event.mimeData().text()
+            self.text_edit.setPlainText(text)
+            print(f"[TEST] Received text: {text}")
+            event.acceptProposedAction()
+
+def test_drag_functionality():
+    """Test the drag functionality."""
+    app = QApplication(sys.argv)
+    
+    # Create main window
+    main_window = QWidget()
+    main_window.setWindowTitle("Drag Test")
+    main_window.resize(1200, 800)
+    
+    # Create layout
+    layout = QHBoxLayout()
+    
+    # Create map viewer
+    map_viewer = QPSAMapViewer()
+    
+    # Create a simple test map with some cells
+    # For testing, we'll manually add some cell info
+    test_cells = [
+        CellInfo(10, 10),
+        CellInfo(15, 15),  # Empty cell for point testing
+        CellInfo(20, 20)
+    ]
+    
+    # Add objects to test cells
+    test_cells[0]._objects = [
+        {"type": "Source", "method": "Source", "id": "source1", "name": "Source 1"},
+        {"type": "Mineral", "method": "Mineral", "id": "mineral1", "name": "Mineral 1"}
+    ]
+    test_cells[2]._objects = [
+        {"type": "Controller", "method": "Controller", "id": "ctrl1", "name": "Controller 1"}
+    ]
+    
+    # Set up the map viewer with test data
+    # Note: This is a simplified test - in real usage you'd load an actual map
+    
+    # Create drop target
+    drop_target = DropTarget()
+    
+    # Add widgets to layout
+    layout.addWidget(map_viewer, 2)
+    layout.addWidget(drop_target, 1)
+    
+    main_window.setLayout(layout)
+    
+    # Show windows
+    main_window.show()
+    drop_target.show()
+    
+    print("[TEST] Drag test started!")
+    print("[TEST] Instructions:")
+    print("[TEST] 1. Hold Ctrl or Alt key")
+    print("[TEST] 2. Click and drag on the map")
+    print("[TEST] 3. Drop on the right panel")
+    print("[TEST] 4. Check the debug output")
+    
+    sys.exit(app.exec())
+
+if __name__ == "__main__":
+    test_drag_functionality()

pyscreeps_arena/ui/qmapv/test_qcinfo.py

@@ -0,0 +1,169 @@
+# -*- coding: utf-8 -*-
+"""
+Test script for QPSACellInfo component
+"""
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))))
+
+from PyQt6.QtWidgets import QApplication, QVBoxLayout, QWidget, QPushButton, QHBoxLayout
+from PyQt6.QtCore import Qt
+from pyscreeps_arena.ui.qmapv.qcinfo import QPSACellInfo
+from pyscreeps_arena.ui.qmapv.qmapv import CellInfo
+
+
+class TestWindow(QWidget):
+    def __init__(self):
+        super().__init__()
+        self.setWindowTitle("QPSACellInfo Component Test")
+        self.setGeometry(100, 100, 400, 500)
+        self._init_ui()
+        
+    def _init_ui(self):
+        """Initialize test UI."""
+        layout = QVBoxLayout()
+        
+        # Create the cell info component
+        self.cell_info = QPSACellInfo()
+        layout.addWidget(self.cell_info)
+        
+        # Add test buttons
+        button_layout = QHBoxLayout()
+        
+        # Test button 1 - Plain terrain with objects
+        btn1 = QPushButton("Test Plain Cell")
+        btn1.clicked.connect(self.test_plain_cell)
+        button_layout.addWidget(btn1)
+        
+        # Test button 2 - Swamp terrain
+        btn2 = QPushButton("Test Swamp Cell")
+        btn2.clicked.connect(self.test_swamp_cell)
+        button_layout.addWidget(btn2)
+        
+        # Test button 3 - Wall terrain
+        btn3 = QPushButton("Test Wall Cell")
+        btn3.clicked.connect(self.test_wall_cell)
+        button_layout.addWidget(btn3)
+        
+        # Test button 4 - Empty cell
+        btn4 = QPushButton("Test Empty Cell")
+        btn4.clicked.connect(self.test_empty_cell)
+        button_layout.addWidget(btn4)
+        
+        layout.addLayout(button_layout)
+        
+        # Connect signals for testing
+        self.cell_info.itemSelected.connect(self.on_item_selected)
+        self.cell_info.itemCancelSelected.connect(self.on_item_cancel_selected)
+        self.cell_info.itemSelectChanged.connect(self.on_item_select_changed)
+        
+        self.setLayout(layout)
+        
+    def test_plain_cell(self):
+        """Test with plain terrain cell."""
+        print("[DEBUG] Testing plain cell")  # 调试输出
+        
+        # Create test cell info
+        cell_data = CellInfo(15, 25)
+        cell_data.terrain = '2'  # Plain
+        
+        # Add some test objects
+        cell_data.add_object({
+            'id': 'obj1',
+            'type': 'StructureRoad',
+            'name': 'Road'
+        })
+        cell_data.add_object({
+            'id': 'obj2', 
+            'type': 'StructureSpawn',
+            'name': 'Spawn'
+        })
+        cell_data.add_object({
+            'id': 'obj3',
+            'type': 'Creep',
+            'name': 'Worker'
+        })
+        
+        # Set cell data
+        self.cell_info.data = cell_data
+        
+        # Test selection states
+        self.cell_info.selects = {
+            'obj1': True,
+            'obj2': False,
+            'obj3': True
+        }
+        
+    def test_swamp_cell(self):
+        """Test with swamp terrain cell."""
+        print("[DEBUG] Testing swamp cell")  # 调试输出
+        
+        # Create test cell info
+        cell_data = CellInfo(30, 40)
+        cell_data.terrain = 'A'  # Swamp
+        
+        # Add some test objects
+        cell_data.add_object({
+            'id': 'swamp_obj1',
+            'type': 'StructureRoad',
+            'name': 'Swamp Road'
+        })
+        
+        # Set cell data
+        self.cell_info.data = cell_data
+        
+    def test_wall_cell(self):
+        """Test with wall terrain cell."""
+        print("[DEBUG] Testing wall cell")  # 调试输出
+        
+        # Create test cell info
+        cell_data = CellInfo(50, 60)
+        cell_data.terrain = 'X'  # Wall
+        
+        # Add some test objects
+        cell_data.add_object({
+            'id': 'wall_obj1',
+            'type': 'StructureWall',
+            'name': 'Wall'
+        })
+        
+        # Set cell data
+        self.cell_info.data = cell_data
+        
+    def test_empty_cell(self):
+        """Test with empty cell."""
+        print("[DEBUG] Testing empty cell")  # 调试输出
+        
+        # Create test cell info
+        cell_data = CellInfo(75, 85)
+        cell_data.terrain = '2'  # Plain
+        
+        # Set cell data (no objects)
+        self.cell_info.data = cell_data
+        
+    def on_item_selected(self, obj_id):
+        """Handle item selected signal."""
+        print(f"[DEBUG] Item selected: {obj_id}")  # 调试输出
+        
+    def on_item_cancel_selected(self, obj_id):
+        """Handle item cancel selected signal."""
+        print(f"[DEBUG] Item cancel selected: {obj_id}")  # 调试输出
+        
+    def on_item_select_changed(self, obj_id, selected):
+        """Handle item select changed signal."""
+        print(f"[DEBUG] Item select changed: {obj_id} -> {selected}")  # 调试输出
+
+
+def main():
+    """Main function."""
+    app = QApplication(sys.argv)
+    
+    # Create and show test window
+    window = TestWindow()
+    window.show()
+    
+    sys.exit(app.exec())
+
+
+if __name__ == "__main__":
+    main()

pyscreeps_arena/ui/qmapv/test_qco_drag.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Test script for QPSACellObject drag and drop functionality
+"""
+import sys
+from PyQt6.QtWidgets import Q

pyscreeps_arena/ui/qmapv/test_qmapv.py

@@ -0,0 +1,224 @@
+# -*- coding: utf-8 -*-
+"""
+QPSA Map Viewer Test Application
+"""
+import sys
+import os
+from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, 
+                             QHBoxLayout, QLabel, QTextEdit, QSplitter, QPushButton)
+from PyQt6.QtCore import Qt, pyqtSlot
+from PyQt6.QtGui import QFont
+
+# Add parent directory to path for imports
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
+
+from ui.qmapv import QPSAMapViewer, CellInfo
+from ui.qmapv.qcinfo import QPSACellInfo
+from ui.qmapv.qco import QPSACellObject
+from pyscreeps_arena import config
+
+
+class MapViewerDemo(QMainWindow):
+    """Demo application for QPSAMapViewer."""
+    
+    def __init__(self):
+        super().__init__()
+        self._init_ui()
+        
+    def _init_ui(self):
+        """Initialize the UI."""
+        self.setWindowTitle("QPSA Map Viewer Demo")
+        self.setGeometry(100, 100, 1400, 800)
+        
+        # Central widget
+        central_widget = QWidget()
+        self.setCentralWidget(central_widget)
+        
+        # Main layout
+        main_layout = QHBoxLayout(central_widget)
+        
+        # Splitter for resizable panels
+        splitter = QSplitter(Qt.Orientation.Horizontal)
+        
+        # Left panel - Map viewer
+        self._map_viewer = QPSAMapViewer()
+        self._map_viewer.currentChanged.connect(self._on_current_changed)
+        self._map_viewer.selectChanged.connect(self._on_select_changed)
+        self._map_viewer.rightClicked.connect(self._on_right_clicked)
+        splitter.addWidget(self._map_viewer)
+        
+        # Right panel - Info panel with QPSACellInfo components
+        info_panel = QWidget()
+        info_layout = QVBoxLayout(info_panel)
+        
+        # Current cell info (hover)
+        current_group = QWidget()
+        current_layout = QVBoxLayout(current_group)
+        
+        current_label = QLabel("当前单元格 (鼠标悬停)")
+        current_label.setFont(QFont("Arial", 12, QFont.Weight.Bold))
+        current_layout.addWidget(current_label)
+        
+        self._current_cell_info = QPSACellInfo()
+        self._current_cell_info.setMaximumHeight(180)  # Updated height constraint
+        current_layout.addWidget(self._current_cell_info)
+        
+        info_layout.addWidget(current_group)
+        
+        # Selected cell info (click)
+        selected_group = QWidget()
+        selected_layout = QVBoxLayout(selected_group)
+        
+        selected_label = QLabel("选中单元格 (鼠标点击)")
+        selected_label.setFont(QFont("Arial", 12, QFont.Weight.Bold))
+        selected_layout.addWidget(selected_label)
+        
+        self._selected_cell_info = QPSACellInfo()
+        self._selected_cell_info.setMaximumHeight(200)
+        selected_layout.addWidget(self._selected_cell_info)
+        
+        info_layout.addWidget(selected_group)
+        
+        # Right-clicked cell info
+        right_click_group = QWidget()
+        right_click_layout = QVBoxLayout(right_click_group)
+        
+        right_click_label = QLabel("右键点击单元格")
+        right_click_label.setFont(QFont("Arial", 12, QFont.Weight.Bold))
+        right_click_layout.addWidget(right_click_label)
+        
+        self._right_click_cell_info = QPSACellInfo()
+        self._right_click_cell_info.setMaximumHeight(200)
+        right_click_layout.addWidget(self._right_click_cell_info)
+        
+        info_layout.addWidget(right_click_group)
+        
+        # Language toggle button
+        self._lang_button = QPushButton("切换语言 (CN/EN)")
+        self._lang_button.clicked.connect(self._toggle_language)
+        info_layout.addWidget(self._lang_button)
+        
+        # Instructions
+        instructions = QLabel(
+            "使用说明:\n"
+            "1. 点击'打开地图'按钮选择JSON文件\n"
+            "2. 鼠标悬停查看单元格信息\n"
+            "3. 鼠标点击选择单元格\n"
+            "4. 鼠标右键点击触发右键事件\n"
+            "5. 使用鼠标滚轮缩放地图\n"
+            "6. 拖拽地图进行平移\n"
+            "\n"
+            "注意: GameObject和Structure基础类型已被过滤，\n"
+            "但它们的子类(如StructureSpawn等)会正常显示"
+        )
+        instructions.setWordWrap(True)
+        instructions.setStyleSheet("font-size: 11px; color: #666; padding: 10px;")
+        # Cell objects component for drag and drop test
+        objects_group = QWidget()
+        objects_layout = QVBoxLayout(objects_group)
+        
+        objects_label = QLabel("拖拽对象列表")
+        objects_label.setFont(QFont("Arial", 12, QFont.Weight.Bold))
+        objects_layout.addWidget(objects_label)
+        
+        self._cell_objects = QPSACellObject()
+        self._cell_objects.setMinimumHeight(200)
+        objects_layout.addWidget(self._cell_objects)
+        
+        info_layout.addWidget(objects_group)
+        
+        info_layout.addWidget(instructions)
+        
+        info_layout.addStretch()
+        
+        splitter.addWidget(info_panel)
+        splitter.setSizes([1024, 376])
+        
+        main_layout.addWidget(splitter)
+        
+        # Try to load default map if available
+        default_map = os.path.join(os.path.dirname(__file__), '..', '..', 'map.json')
+        if os.path.exists(default_map):
+            try:
+                self._map_viewer.load_map(default_map)
+            except Exception as e:
+                print(f"[DEBUG] Failed to load default map: {e}")  # 调试输出
+    
+    @pyqtSlot(object)
+    def _on_current_changed(self, cell_info):
+        """Handle current cell change."""
+        self._current_cell_info.data = cell_info
+        # Set map image for object icons
+        self._current_cell_info.image = self._map_viewer.image
+        print(f"[DEBUG] Current cell updated: {cell_info}")  # 调试输出
+    
+    @pyqtSlot(object)
+    def _on_select_changed(self, cell_info):
+        """Handle selected cell change."""
+        self._selected_cell_info.data = cell_info
+        # Set map image for object icons
+        self._selected_cell_info.image = self._map_viewer.image
+        print(f"[DEBUG] Selected cell updated: {cell_info}")  # 调试输出
+    
+    @pyqtSlot(object)
+    def _on_right_clicked(self, cell_info):
+        """Handle right-click events."""
+        self._right_click_cell_info.data = cell_info
+        # Set map image for object icons
+        self._right_click_cell_info.image = self._map_viewer.image
+        print(f"[DEBUG] Right-click cell updated: {cell_info}")  # 调试输出
+    
+    @pyqtSlot(str)
+    def _on_item_selected(self, obj_id):
+        """Handle object selection in QPSACellInfo."""
+        print(f"[DEBUG] Object selected: {obj_id}")  # 调试输出
+    
+    @pyqtSlot(str)
+    def _on_item_cancel_selected(self, obj_id):
+        """Handle object deselection in QPSACellInfo."""
+        print(f"[DEBUG] Object deselected: {obj_id}")  # 调试输出
+    
+    @pyqtSlot(str, bool)
+    def _on_item_select_changed(self, obj_id, selected):
+        """Handle object selection change in QPSACellInfo."""
+        print(f"[DEBUG] Object selection changed: {obj_id} -> {selected}")  # 调试输出
+    
+    @pyqtSlot()
+    def _toggle_language(self):
+        """Toggle language between CN and EN."""
+        config.language = 'en' if config.language == 'cn' else 'cn'
+        print(f"[DEBUG] Language switched to: {config.language}")  # 调试输出：查看语言切换
+        
+        # Update button text
+        self._lang_button.setText("切换语言 (CN/EN)" if config.language == 'cn' else "Switch Language (CN/EN)")
+        
+        # Reload current map to refresh UI text
+        current_map = self._map_viewer._current_map_path if hasattr(self._map_viewer, '_current_map_path') else None
+        if current_map and os.path.exists(current_map):
+            self._map_viewer.load_map(current_map)
+        else:
+            # Reset to default state to show new language
+            self._map_viewer._reset_to_default_state()
+    
+    def closeEvent(self, event):
+        """Clean up when closing."""
+        self._map_viewer.cleanup()
+        event.accept()
+
+
+def main():
+    """Main function."""
+    app = QApplication(sys.argv)
+    
+    # Set application style
+    app.setStyle('Fusion')
+    
+    # Create and show demo
+    demo = MapViewerDemo()
+    demo.show()
+    
+    sys.exit(app.exec())
+
+
+if __name__ == '__main__':
+    main()