mcify.py

import argparse
import os
from tran import model_to_minecraft, ROTATE_ANGLE, PITCH
# example: python mcify.py model.obj --game_path path --rotate 45,0,0 --pitch 0.5 --no-wool --no-glass

def parse_tuple(tuple_str):
    try:
        x, y, z = map(float, tuple_str.split(','))
        return (x, y, z)
    except:
        raise argparse.ArgumentTypeError("--rotate must be in the format x,y,z")

def main():
    parser = argparse.ArgumentParser(description='将3D模型转换为Minecraft方块')
    
    # 必需参数
    parser.add_argument('obj_file', help='Model file path | 模型文件路径')

    # 可选参数
    parser.add_argument('--game_path', default='.', help='Minecraft game path | Minecraft 游戏主目录')
    parser.add_argument('--rotate', type=parse_tuple, default=ROTATE_ANGLE,
                       help='rotate angle rx,ry,rz | 旋转角度 rx,ry,rz')
    parser.add_argument('--pitch', type=float, default=PITCH,
                       help='voxel pitch | 体素大小')
    
    # 方块类型选项
    parser.add_argument('--no-wool', action='store_false', dest='wool',
                       help='don\'t use wool blocks | 不使用羊毛方块')
    parser.add_argument('--no-concrete', action='store_false', dest='concrete',
                       help='don\'t use concrete blocks | 不使用混凝土方块')
    parser.add_argument('--no-terracotta', action='store_false', dest='terracotta',
                       help='don\'t use terracotta blocks | 不使用陶瓦方块')
    parser.add_argument('--no-glass', action='store_false', dest='glass',
                       help='don\'t use glass blocks | 不使用玻璃方块')
    
    args = parser.parse_args()

    if args.game_path == '.':
        save_path = os.path.join(os.getcwd(), 'output')
    else:
        save_path = os.path.join(args.game_path, 'config', 'mybuilds')
    if not os.path.exists(save_path):
        os.makedirs(save_path)
    
    model_to_minecraft(
        obj_file=args.obj_file,
        save_path=save_path,
        pitch=args.pitch,
        rotate_angle=args.rotate,
        wool=args.wool,
        concrete=args.concrete,
        terracotta=args.terracotta,
        glass=args.glass
    )

if __name__ == '__main__':
    main()