mirror of
https://github.com/TriM-Organization/Musicreater.git
synced 2024-11-14 11:27:42 +08:00
349 lines
9.6 KiB
Python
349 lines
9.6 KiB
Python
import math
|
||
import os
|
||
import brotli
|
||
|
||
key = {
|
||
"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"],
|
||
}
|
||
"""key存储了方块移动指令的数据，其中可以用key[x|y|z][0|1]来表示xyz的减或增
|
||
而key[][2+]是用来增加指定数目的"""
|
||
|
||
x = "x"
|
||
y = "y"
|
||
z = "z"
|
||
|
||
|
||
def move(axis: str, value: int):
|
||
if value == 0:
|
||
return b''
|
||
if abs(value) == 1:
|
||
return key[axis][0 if value == -1 else 1]
|
||
|
||
pointer = sum(
|
||
[
|
||
1 if i else 0
|
||
for i in (
|
||
value != -1,
|
||
value < -1 or value > 1,
|
||
value < -128 or value > 127,
|
||
value < -32768 or value > 32767,
|
||
)
|
||
]
|
||
)
|
||
|
||
return key[axis][pointer] + value.to_bytes(2 ** (pointer - 2), 'big', signed=True)
|
||
|
||
|
||
def makeZip(sourceDir, outFilename, compression=8, exceptFile=None):
|
||
"""使用compression指定的算法打包目录为zip文件\n
|
||
默认算法为DEFLATED(8),可用算法如下：\n
|
||
STORED = 0\n
|
||
DEFLATED = 8\n
|
||
BZIP2 = 12\n
|
||
LZMA = 14\n
|
||
"""
|
||
import zipfile
|
||
|
||
zipf = zipfile.ZipFile(outFilename, "w", compression)
|
||
pre_len = len(os.path.dirname(sourceDir))
|
||
for parent, dirnames, filenames in os.walk(sourceDir):
|
||
for filename in filenames:
|
||
if filename == exceptFile:
|
||
continue
|
||
pathfile = os.path.join(parent, filename)
|
||
arcname = pathfile[pre_len:].strip(os.path.sep) # 相对路径
|
||
zipf.write(pathfile, arcname)
|
||
zipf.close()
|
||
|
||
|
||
def formCMDblk(
|
||
command: str,
|
||
particularValue: int,
|
||
impluse: int = 0,
|
||
condition: bool = False,
|
||
needRedstone: bool = True,
|
||
tickDelay: int = 0,
|
||
customName: str = "",
|
||
executeOnFirstTick: bool = False,
|
||
trackOutput: bool = True,
|
||
):
|
||
"""
|
||
使用指定项目返回指定的指令方块放置指令项
|
||
:param command: `str`
|
||
指令
|
||
:param particularValue:
|
||
方块特殊值，即朝向
|
||
:0 下 无条件
|
||
:1 上 无条件
|
||
:2 z轴负方向 无条件
|
||
:3 z轴正方向 无条件
|
||
:4 x轴负方向 无条件
|
||
:5 x轴正方向 无条件
|
||
:6 下 无条件
|
||
:7 下 无条件
|
||
|
||
:8 下 有条件
|
||
:9 上 有条件
|
||
:10 z轴负方向 有条件
|
||
:11 z轴正方向 有条件
|
||
:12 x轴负方向 有条件
|
||
:13 x轴正方向 有条件
|
||
:14 下 有条件
|
||
:14 下 有条件
|
||
注意！此处特殊值中的条件会被下面condition参数覆写
|
||
:param impluse: `int 0|1|2`
|
||
方块类型
|
||
0脉冲 1循环 2连锁
|
||
:param condition: `bool`
|
||
是否有条件
|
||
:param needRedstone: `bool`
|
||
是否需要红石
|
||
:param tickDelay: `int`
|
||
执行延时
|
||
:param customName: `str`
|
||
悬浮字
|
||
lastOutput: `str`
|
||
上次输出字符串，注意此处需要留空
|
||
:param executeOnFirstTick: `bool`
|
||
执行第一个已选项(循环指令方块是否激活后立即执行，若为False，则从激活时起延迟后第一次执行)
|
||
:param trackOutput: `bool`
|
||
是否输出
|
||
|
||
:return:str
|
||
"""
|
||
block = b"\x24" + particularValue.to_bytes(2, byteorder="big", signed=False)
|
||
|
||
for i in [
|
||
impluse.to_bytes(4, byteorder="big", signed=False),
|
||
bytes(command, encoding="utf-8") + b"\x00",
|
||
bytes(customName, encoding="utf-8") + b"\x00",
|
||
bytes("", encoding="utf-8") + b"\x00",
|
||
tickDelay.to_bytes(4, byteorder="big", signed=True),
|
||
executeOnFirstTick.to_bytes(1, byteorder="big"),
|
||
trackOutput.to_bytes(1, byteorder="big"),
|
||
condition.to_bytes(1, byteorder="big"),
|
||
needRedstone.to_bytes(1, byteorder="big"),
|
||
]:
|
||
block += i
|
||
return block
|
||
|
||
|
||
def __fillSquareSideLength(total: int, maxHeight: int):
|
||
"""给定总方块数量和最大高度，返回所构成的图形外切正方形的边长
|
||
:param total: 总方块数量
|
||
:param maxHeight: 最大高度
|
||
:return: 外切正方形的边长 int"""
|
||
return math.ceil(math.sqrt(math.ceil(total / maxHeight)))
|
||
|
||
|
||
axisParticularValue = {
|
||
x: {
|
||
True: 5,
|
||
False: 4,
|
||
},
|
||
y: {
|
||
True: 1,
|
||
False: 0,
|
||
},
|
||
z: {
|
||
True: 3,
|
||
False: 2,
|
||
},
|
||
}
|
||
|
||
|
||
def toLineBDXbytes(
|
||
commands: list,
|
||
axis: str,
|
||
forward: bool,
|
||
):
|
||
_bytes = b''
|
||
for cmd, condition in commands:
|
||
_bytes += formCMDblk(
|
||
cmd,
|
||
axisParticularValue[axis][forward],
|
||
impluse=2,
|
||
condition=condition,
|
||
needRedstone=False,
|
||
tickDelay=0,
|
||
customName="",
|
||
executeOnFirstTick=False,
|
||
trackOutput=True,
|
||
) + move(axis, 1 if forward else -1)
|
||
return _bytes
|
||
|
||
|
||
def toBDXbytes(
|
||
commands: list,
|
||
maxheight: int = 64,
|
||
):
|
||
"""
|
||
:param commands: 指令列表(指令, 条件)
|
||
:param maxheight: 生成结构最大高度
|
||
:return 成功与否，成功返回(True,未经过压缩的源,结构占用大小)，失败返回(False,str失败原因)
|
||
"""
|
||
|
||
_sideLength = __fillSquareSideLength(len(commands), maxheight)
|
||
_bytes = b''
|
||
|
||
yforward = True
|
||
zforward = True
|
||
|
||
nowy = 0
|
||
nowz = 0
|
||
nowx = 0
|
||
|
||
for cmd, condition in commands:
|
||
_bytes += formCMDblk(
|
||
cmd,
|
||
(1 if yforward else 0)
|
||
if (
|
||
((nowy != 0) and (not yforward))
|
||
or ((yforward) and (nowy != (maxheight - 1)))
|
||
)
|
||
else (3 if zforward else 2)
|
||
if (
|
||
((nowz != 0) and (not zforward))
|
||
or ((zforward) and (nowz != _sideLength))
|
||
)
|
||
else 5,
|
||
impluse=2,
|
||
condition=condition,
|
||
needRedstone=False,
|
||
tickDelay=0,
|
||
customName="",
|
||
executeOnFirstTick=False,
|
||
trackOutput=True,
|
||
)
|
||
|
||
nowy += 1 if yforward else -1
|
||
|
||
if ((nowy >= maxheight) and (yforward)) or ((nowy < 0) and (not yforward)):
|
||
nowy -= 1 if yforward else -1
|
||
|
||
yforward = not yforward
|
||
|
||
nowz += 1 if zforward else -1
|
||
|
||
if ((nowz > _sideLength) and (zforward)) or ((nowz < 0) and (not zforward)):
|
||
nowz -= 1 if zforward else -1
|
||
zforward = not zforward
|
||
_bytes += key[x][1]
|
||
nowx += 1
|
||
else:
|
||
|
||
_bytes += key[z][int(zforward)]
|
||
|
||
else:
|
||
|
||
_bytes += key[y][int(yforward)]
|
||
|
||
return (
|
||
_bytes,
|
||
[nowx + 1, maxheight if nowx or nowz else nowy, _sideLength if nowx else nowz],
|
||
[nowx, nowy, nowz],
|
||
)
|
||
|
||
|
||
|
||
|
||
def toBDXfile(
|
||
funcList: list,
|
||
author: str = "Eilles",
|
||
maxheight: int = 64,
|
||
outfile: str = "./test.bdx",
|
||
):
|
||
"""
|
||
:funcList list: 指令集列表： 指令系统[ 指令集[ 单个指令( str指令, bool条件性 ), ], ]
|
||
:param author: 作者名称
|
||
:param maxheight: 生成结构最大高度
|
||
:outfile: str 输出文件
|
||
:return 成功与否，指令总长度，指令总延迟，指令结构总大小，画笔最终坐标
|
||
"""
|
||
|
||
with open(os.path.abspath(outfile), "w+", encoding="utf-8") as f:
|
||
f.write("BD@")
|
||
|
||
_bytes = (
|
||
b"BDX\x00" + author.encode("utf-8") + b" & Musicreater\x00\x01command_block\x00"
|
||
)
|
||
totalSize = {x: 0, y: 0, z: 0}
|
||
totalLen = 0
|
||
for func in funcList:
|
||
totalLen += len(func)
|
||
cmdBytes, size, finalPos = toBDXbytes(func, maxheight)
|
||
_bytes += cmdBytes
|
||
_bytes += move(x, 2)
|
||
_bytes += move(y, -finalPos[1])
|
||
_bytes += move(z, -finalPos[2])
|
||
totalSize[x] += size[0] + 2
|
||
totalSize[y] = max(totalSize[y], size[1])
|
||
totalSize[z] = max(totalSize[z], size[2])
|
||
|
||
with open(
|
||
os.path.abspath(outfile),
|
||
"ab+",
|
||
) as f:
|
||
f.write(brotli.compress(_bytes + b"XE"))
|
||
|
||
return (True, totalLen, 0, list(totalSize.values()), finalPos)
|
||
|
||
|
||
|
||
def toLineBDXfile(
|
||
funcList: list,
|
||
axis_: str,
|
||
forward_: bool,
|
||
author: str = "Eilles",
|
||
outfile: str = "./test.bdx",
|
||
):
|
||
"""
|
||
:funcList list: 指令集列表： 指令系统[ 指令集[ 单个指令( str指令, bool条件性 ), ], ]
|
||
:param author: 作者名称
|
||
:param maxheight: 生成结构最大高度
|
||
:outfile: str 输出文件
|
||
:return 成功与否，指令总长度，指令总延迟，指令结构总大小，画笔最终坐标
|
||
"""
|
||
|
||
with open(os.path.abspath(outfile), "w+", encoding="utf-8") as f:
|
||
f.write("BD@")
|
||
|
||
_bytes = (
|
||
b"BDX\x00" + author.encode("utf-8") + b" & Musicreater\x00\x01command_block\x00"
|
||
)
|
||
totalSize = {x: 0, y: 0, z: 0}
|
||
totalLen = 0
|
||
for func in funcList:
|
||
totalLen += len(func)
|
||
_bytes += toLineBDXbytes(func, axis_, forward_)
|
||
_bytes += move(z if axis_ == x else x, 2)
|
||
|
||
totalSize[z if axis_ == x else x] += 2
|
||
totalSize[axis_] = max(totalSize[axis_], len(func))
|
||
|
||
with open(
|
||
os.path.abspath(outfile),
|
||
"ab+",
|
||
) as f:
|
||
f.write(brotli.compress(_bytes + b"XE"))
|
||
|
||
return (True, totalLen, 0, list(totalSize.values()))
|
||
|
||
def formatipt(notice: str, fun, errnote: str = "", *extraArg):
|
||
'''循环输入，以某种格式
|
||
notice: 输入时的提示
|
||
fun: 格式函数
|
||
errnote: 输入不符格式时的提示
|
||
*extraArg: 对于函数的其他参数'''
|
||
while True:
|
||
result = input(notice)
|
||
try:
|
||
funresult = fun(result, *extraArg)
|
||
break
|
||
except:
|
||
print(errnote)
|
||
continue
|
||
return result, funresult
|