Musicreater/msctspt/bdxOpera_CP.py

import os
import brotli

'''感谢由 Charlie_Ping “查理平” 带来的bdx转换代码'''


class BdxConverter:
    __header = "BD@"
    __bin_header = b"BDX"
    __generator_author = b"&Musicreater"

    keys = {
        # x--, x++, addSmallX(-128~127), addX(-32768~32767), addBigX(-2147483648~2147483647)
        "x": [b"\x0f", b"\x0e", b"\x1c", b"\x14", b"\x15"],
        "y": [b"\x11", b"\x10", b"\x1d", b"\x16", b"\x17"],
        "z": [b"\x13", b"\x12", b"\x1e", b"\x18", b"\x19"],
        "end": b"\x58",
        "isSigned": b"\x5a",
        "placeCommandBlockWithData": b"\x1b",
        "placeBlock": b"\x07"
    }

    def __init__(self, file_path: str, author: str, blocks):
        self.author = author
        self.blocks = blocks
        self.file_path = file_path
        self.direction = [0, 0, 0]
        self.block_type = self.get_block_type
        self.__file = self.create_and_upload_file

    @property
    def get_block_type(self):
        """
        blocks
        [
            {
                "direction": [x: int, y: int, z: int],
                block_name: str,
                particular_value: int,
            }
        ]
        :return: list 给出的所有方块种类名称
        """
        block_type = set()
        for block in self.blocks:
            block_type.add(block["block_name"])
        block_type = list(block_type)
        return block_type

    @property
    def create_and_upload_file(self):
        """
        创建一个bdx文件
        :return: 一个文件对象
        """
        _dir = os.path.dirname(self.file_path)
        if not os.path.isdir(_dir):
            os.makedirs(_dir)
        _bytes = self.__bin_header
        _bytes += b"\x00"
        _bytes += self.author.encode("utf-8") + self.__generator_author
        for i in self.block_type:
            _bytes += b"\x00\x01"
            _bytes += bytes(i, encoding="utf-8")
        _bytes += b"\x00"
        _bytes += self.upload_blocks()
        _bytes += b"X"
        with open(self.file_path, "w+") as f:
            f.write("BD@")
            f.close()
        with open(self.file_path, "ab+") as f:
            f.write(brotli.compress(_bytes))
            f.close()
        return open(self.file_path,'a+')
    
    def upload_blocks(self):
        """
        计算差值
        写入移动过程
        写入方块
        更新差值
        :return:
        """
        _types = b""


        for block in self.blocks:
            # print(f"当前方块：{block['block_name']}, 位置： {block['direction']}]")
            diff = self.move_pointer(self.direction, block["direction"])
            _types += diff
            if block["block_name"] in ["command_block",
                                       "chain_command_block",
                                       "repeating_command_block"]:
                _types += self.obtain_command_block(block)
            else:
                _types += self.obtain_universal_block(block)
            self.direction = block["direction"]

        
        return _types

    def move_pointer(self, direction: list, new_direction):
        """
        给出 两个[x, y, z]坐标，返回pointer的移动过程
        :param direction: 坐标 1
        :param new_direction: 坐标 2
        :return: bytes
        """
        _bytes = b""
        for i, sign in enumerate(["x", "y", "z"]):
            # print(f"<{sign}> 新-旧={new_direction[i]-direction[i]}")
            distance = new_direction[i] - direction[i]
            if distance == 0:
                # print("距离是0？跳过了")
                continue
            _bytes += self.obtain_pointer_type(distance, sign)
            # print(f"向 {sign} 运动了 {distance} 格子")
        return _bytes

    @classmethod
    def obtain_pointer_type(cls, num: int, coordinate: str):
        """

        用于确定辅助玩家以某一数据类型走指定长度

        -1 -> 0
        1 -> 1
        [128, 127] -> 2
        [-32768, 32767] -> 3
        [-2147483648, 2147483647] -> 4
        :param num:
        :param coordinate: 坐标轴种类，x y 或 z
        :return:
        """
        if num == 0:
            return
        pointer = 0
        condition = (num != -1,  # byte=0, pointer=1
                     num < -1 or num > 1,  # byte=1, pointer=2
                     num < -128 or num > 127,  # byte=2, pointer=3
                     num < -32768 or num > 32767,  # byte=4, pointer=4
                     )
        for i in condition:
            if i:
                pointer += 1
        pointer_type = cls.keys[coordinate][pointer]

        byte_len = 2 ** (pointer - 2)
        if byte_len >= 1:
            num_byte = num.to_bytes(byte_len, byteorder="big", signed=True)
            return pointer_type + num_byte
        return pointer_type

    def obtain_universal_block(self, block):
        """
        给定一个方块， 返回此方块在这个bdx中的id和方块data
        :param block: {block_name: str,particular_value: int}
        :return: bytes
        """
        block_id = b"\x07" + self.block_type.index(block["block_name"]).to_bytes(2, byteorder="big", signed=False)
        particular_value = block["particular_value"].to_bytes(2, byteorder="big", signed=False)
        block_header = block_id + particular_value
        return block_header

    def obtain_command_block(self, block):
        """
        给定一个命令方块，返回命令方块各种数据
        :param block: {
            "direction": [x: int, y: int, z: int]
            "block_name": str,
            "particular_value": int,
            "impluse": int,  # unsigned_int32
            "command": str,
            "customName": str,
            "lastOutput": str,  # 没特殊要求写个\x00就得了
            "tickdelay": int,  # int32
            "executeOnFirstTick": int,  # 1 bytes
            "trackOutput": int,  # 1 bytes
            "conditional": int,  # 1 bytes
            "needRedstone": int  # 1 bytes
        }
        :return: bytes of command_block
        """

        block_id = b"\x1b" + self.block_type.index(block["block_name"]).to_bytes(2, byteorder="big", signed=False)
        particular_value = block["particular_value"].to_bytes(2, byteorder="big", signed=False)
        block_header = block_id + particular_value
        for i in [
            block["impluse"].to_bytes(4, byteorder="big", signed=False),
            bytes(block["command"], encoding="utf-8") + b"\x00",
            bytes(block["customName"], encoding="utf-8") + b"\x00",
            bytes(block["lastOutput"], encoding="utf-8") + b"\x00",
            block["tickdelay"].to_bytes(4, byteorder="big", signed=True),
            block["executeOnFirstTick"].to_bytes(1, byteorder="big"),
            block["trackOutput"].to_bytes(1, byteorder="big"),
            block["conditional"].to_bytes(1, byteorder="big"),
            block["needRedstone"].to_bytes(1, byteorder="big")
        ]:
            block_header += i
        return block_header

if __name__ == '__main__':
    block = [{"direction": [-1, -1, -1], "block_name": "concrete", "particular_value": 5},
             {"direction": [1, 5, 1], "block_name": "stained_glass", "particular_value": 7},
             {"direction": [2, 4, 1], "block_name": "command_block", "particular_value": 3,
              "impluse": 0,
              "command": "say A generator test",
              "customName": "test",
              "lastOutput": "",
              "tickdelay": 24,
              "executeOnFirstTick": 0,
              "trackOutput": 0,
              "conditional": 0,
              "needRedstone": 1
              },
             {"direction": [3, 4, 1], "block_name": "concrete", "particular_value": 6},
             {"direction": [-123412133, 4, 1], "block_name": "concrete", "particular_value": 7}]
    bdx = BdxConverter("./test02.bdx", "Charlie_Ping",block)