Function main

Scripts/generate_subclasses.py:112–139

()

"""

def main():
    for category, info in DATA.items():
        base = info["Base"]
        classes = info["Classes"]
        
        pub_dir = os.path.join(PUBLIC_ROOT, category)
        priv_dir = os.path.join(PRIVATE_ROOT, category)
        
        os.makedirs(pub_dir, exist_ok=True)
        os.makedirs(priv_dir, exist_ok=True)
        
        for class_name, enum_val in classes.items():
            # Header
            h_path = os.path.join(pub_dir, f"{class_name}.h")
            h_content = HEADER_TEMPLATE.format(Base=base, ClassName=class_name, Enum=enum_val)
            
            with open(h_path, "w") as f:
                f.write(h_content)
                print(f"Created {h_path}")
            
            # Cpp
            custom_cpp = CPP_TEMPLATE.replace("{Category}", category) # To handle path insertion
            cpp_path = os.path.join(priv_dir, f"{class_name}.cpp")
            cpp_content = custom_cpp.format(Base=base, ClassName=class_name, Enum=enum_val)
            
            with open(cpp_path, "w") as f:
                f.write(cpp_content)
                print(f"Created {cpp_path}")

if __name__ == "__main__":
    main()