import aspose.cad as cad
import os
from aspose.cad.fileformats.cad import CadImage
from aspose.cad.fileformats.cad.cadobjects import CadInsertObject

def set_license():
    """
    Tries to find and apply an Aspose.CAD license.
    """
    try:
        from aspose.cad import License
        license = License()
        # The SetLicense method expects the name of the license file.
        # Common names are Aspose.CAD.lic or Aspose.Total.lic.
        # It should be placed in the folder with the script, or you can specify a full path.
        # As we don't have the license file, this will likely fail silently
        # or work if the user has placed it correctly.
        license.set_license("Aspose.CAD.lic")
        print("License set successfully (if found).")
    except Exception as e:
        print(f"License not set or error: {e}. Running in evaluation mode.")

def find_block_coordinates(dwg_path, printable_blocks):
    """
    Finds specified blocks in a DWG file and returns their coordinates and layout.

    :param dwg_path: Path to the DWG file.
    :param printable_blocks: A dictionary of block names to find.
    :return: A list of dictionaries, each containing block info.
    """
    if not os.path.exists(dwg_path):
        print(f"Error: File not found at {dwg_path}")
        return []

    found_blocks = []
    
    with cad.Image.load(dwg_path) as image:
        # Based on the file `extract_lines_correct.py`, the correct object is returned directly.
        # The issue might be specific to the environment or the DWG file itself.
        # Let's use the exact loading method from the working example.
        cad_image = cad.fileformats.cad.CadImage.load(dwg_path)
        
        try:
            print(f"Successfully loaded image. Type: {type(cad_image).__name__}")
            
            # A DWG's paper space layouts are stored in blocks with names like "*Paper_Space"
            # Model space is in "*Model_Space"
            # We iterate through these special blocks to find the block insertions.
            if hasattr(cad_image, 'blocks'):
                print("Iterating through cad_image.blocks...")
                for block in cad_image.blocks:
                    # Layout blocks often start with '*'
                    if block.name.startswith('*'):
                        layout_name = block.name
                        for entity in block.entities:
                            if isinstance(entity, CadInsertObject):
                                block_name = entity.name
                                if block_name in printable_blocks:
                                    min_point = entity.min_point
                                    max_point = entity.max_point
                                    
                                    block_info = {
                                        "block_name": block_name,
                                        "layout": layout_name,
                                        "min_point": (min_point.x, min_point.y),
                                        "max_point": (max_point.x, max_point.y)
                                    }
                                    found_blocks.append(block_info)
                                    print(f"  Found block '{block_name}' in layout/block '{layout_name}'")
            else:
                print("Attribute 'blocks' not found on CadImage object.")
        finally:
            # Manually dispose of the image if CadImage.load was used
            cad_image.dispose()

    return found_blocks

if __name__ == "__main__":
    # It's a good practice to set the license at the start of the application
    set_license()
    
    # Define the blocks we are interested in, based on 思路.MD
    PRINTABLE_BLOCKS = {
        "A$C2EB80DB8": ("A1", "portrait"),
        "A$C1CC9093B": ("A1", "landscape"),
        "A$C6D564680": ("A2", "portrait"),
        "新块": ("A3", "landscape"),
        "新块1": ("A4", "portrait"),
    }

    # Path to the test DWG file
    dwg_file_path = os.path.join("04_Test_Files", "图签测试.dwg")
    
    print(f"Loading DWG file: {dwg_file_path}")
    
    blocks_data = find_block_coordinates(dwg_file_path, PRINTABLE_BLOCKS)
    
    print("\n--- Found Blocks Summary ---")
    if blocks_data:
        for block in blocks_data:
            print(f"Block: {block['block_name']}")
            print(f"  Layout: {block['layout']}")
            print(f"  Min Point: {block['min_point']}")
            print(f"  Max Point: {block['max_point']}")
            print("-" * 20)
    else:
        print("No specified blocks were found in the DWG file.")