mirror of
https://github.com/TriM-Organization/Musicreater.git
synced 2024-11-14 11:27:42 +08:00
216 lines
7.7 KiB
Python
216 lines
7.7 KiB
Python
|
import os
|
|||
|
|
import brotli
|
|||
|
|
|
|||
|
|
'''感谢由 Charlie_Ping “查理平” 带来的bdx转换代码'''
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BdxConverter:
|
|||
|
|
__header = "BD@"
|
|||
|
|
__bin_header = b"BDX"
|
|||
|
|
__generator_author = b"&Charlie_Ping"
|
|||
|
|
|
|||
|
|
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):
|
|||
|
|
"""
|
|||
|
|
（瞎用property？ 害怕
|
|||
|
|
创建一个bdx文件
|
|||
|
|
要close！
|
|||
|
|
: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
|
|||
|
|
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)
|