mirror of
https://github.com/TriM-Organization/Musicreater.git
synced 2024-11-14 19:37:49 +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
|