mirror of
https://github.com/freedoom/freedoom.git
synced 2025-09-02 16:25:47 -04:00
Blacken all Python files
Using the black code reformatter, pass it over all our Python files. This allows for a consistent style across the code base. Exception: lumps/dmxgus/stats.py, for readability.
This commit is contained in:
Mike Swanson
parent
6b486b6332
commit
4701d8f351
30 changed files with 2528 additions and 2102 deletions
|
|
@ -14,171 +14,181 @@ INSTR_DATA_LEN = 36
|
|||
INSTR_NAME_LEN = 32
|
||||
|
||||
FLAG_FIXED_PITCH = 0x0001
|
||||
FLAG_TWO_VOICE = 0x0004
|
||||
FLAG_TWO_VOICE = 0x0004
|
||||
|
||||
KSL_MASK = 0xc0
|
||||
VOLUME_MASK = 0x3f
|
||||
KSL_MASK = 0xC0
|
||||
VOLUME_MASK = 0x3F
|
||||
|
||||
# Order of fields in GENMIDI data structures.
|
||||
|
||||
GENMIDI_FIELDS = [
|
||||
"m_am_vibrato_eg",
|
||||
"m_attack_decay",
|
||||
"m_sustain_release",
|
||||
"m_waveform",
|
||||
"m_ksl",
|
||||
"m_volume",
|
||||
"feedback_fm",
|
||||
"c_am_vibrato_eg",
|
||||
"c_attack_decay",
|
||||
"c_sustain_release",
|
||||
"c_waveform",
|
||||
"c_ksl",
|
||||
"c_volume",
|
||||
"null",
|
||||
"note_offset"
|
||||
"m_am_vibrato_eg",
|
||||
"m_attack_decay",
|
||||
"m_sustain_release",
|
||||
"m_waveform",
|
||||
"m_ksl",
|
||||
"m_volume",
|
||||
"feedback_fm",
|
||||
"c_am_vibrato_eg",
|
||||
"c_attack_decay",
|
||||
"c_sustain_release",
|
||||
"c_waveform",
|
||||
"c_ksl",
|
||||
"c_volume",
|
||||
"null",
|
||||
"note_offset",
|
||||
]
|
||||
|
||||
# Encode a single voice of an instrument to binary.
|
||||
|
||||
|
||||
def encode_voice(data, offset):
|
||||
result = dict(data)
|
||||
result = dict(data)
|
||||
|
||||
result["m_ksl"] = data["m_ksl_volume"] & KSL_MASK
|
||||
result["m_volume"] = data["m_ksl_volume"] & VOLUME_MASK
|
||||
result["c_ksl"] = data["c_ksl_volume"] & KSL_MASK
|
||||
result["c_volume"] = data["c_ksl_volume"] & VOLUME_MASK
|
||||
result["m_ksl"] = data["m_ksl_volume"] & KSL_MASK
|
||||
result["m_volume"] = data["m_ksl_volume"] & VOLUME_MASK
|
||||
result["c_ksl"] = data["c_ksl_volume"] & KSL_MASK
|
||||
result["c_volume"] = data["c_ksl_volume"] & VOLUME_MASK
|
||||
|
||||
result["null"] = 0
|
||||
result["note_offset"] = offset
|
||||
result["null"] = 0
|
||||
result["note_offset"] = offset
|
||||
|
||||
return struct.pack(
|
||||
"<BBBBBBBBBBBBBBh", *map(lambda key: result[key], GENMIDI_FIELDS)
|
||||
)
|
||||
|
||||
return struct.pack("<BBBBBBBBBBBBBBh",
|
||||
*map(lambda key: result[key], GENMIDI_FIELDS))
|
||||
|
||||
# Encode an instrument to binary.
|
||||
|
||||
|
||||
def encode_instrument(instrument):
|
||||
flags = 0
|
||||
flags = 0
|
||||
|
||||
instr1_data = encode_voice(instrument.voice1, instrument.offset1)
|
||||
instr1_data = encode_voice(instrument.voice1, instrument.offset1)
|
||||
|
||||
if instrument.voice2 is not None:
|
||||
flags |= FLAG_TWO_VOICE
|
||||
instr2_data = encode_voice(instrument.voice2,
|
||||
instrument.offset2)
|
||||
else:
|
||||
instr2_data = encode_voice(NullInstrument.voice1, 0)
|
||||
if instrument.voice2 is not None:
|
||||
flags |= FLAG_TWO_VOICE
|
||||
instr2_data = encode_voice(instrument.voice2, instrument.offset2)
|
||||
else:
|
||||
instr2_data = encode_voice(NullInstrument.voice1, 0)
|
||||
|
||||
if instrument.fixed_note is not None:
|
||||
flags |= FLAG_FIXED_PITCH
|
||||
fixed_note = instrument.fixed_note
|
||||
else:
|
||||
fixed_note = 0
|
||||
if instrument.fixed_note is not None:
|
||||
flags |= FLAG_FIXED_PITCH
|
||||
fixed_note = instrument.fixed_note
|
||||
else:
|
||||
fixed_note = 0
|
||||
|
||||
header = struct.pack("<hBB", flags, 128, fixed_note)
|
||||
header = struct.pack("<hBB", flags, 128, fixed_note)
|
||||
|
||||
return header + instr1_data + instr2_data
|
||||
|
||||
return header + instr1_data + instr2_data
|
||||
|
||||
def encode_instruments(instruments):
|
||||
result = []
|
||||
result = []
|
||||
|
||||
for instrument in instruments:
|
||||
result.append(encode_instrument(instrument))
|
||||
for instrument in instruments:
|
||||
result.append(encode_instrument(instrument))
|
||||
|
||||
return b"".join(result)
|
||||
|
||||
return b"".join(result)
|
||||
|
||||
def encode_instrument_names(instruments):
|
||||
result = []
|
||||
result = []
|
||||
|
||||
for instrument in instruments:
|
||||
instr_name = instrument.voice1["name"].encode("ascii")
|
||||
result.append(struct.pack("32s", instr_name))
|
||||
for instrument in instruments:
|
||||
instr_name = instrument.voice1["name"].encode("ascii")
|
||||
result.append(struct.pack("32s", instr_name))
|
||||
|
||||
return b"".join(result)
|
||||
|
||||
return b"".join(result)
|
||||
|
||||
def write(filename, instruments):
|
||||
header = struct.pack("%is" % len(GENMIDI_HEADER),
|
||||
GENMIDI_HEADER.encode("ascii"))
|
||||
header = struct.pack(
|
||||
"%is" % len(GENMIDI_HEADER), GENMIDI_HEADER.encode("ascii")
|
||||
)
|
||||
|
||||
with open(filename, "wb") as f:
|
||||
f.write(header)
|
||||
f.write(encode_instruments(instruments))
|
||||
f.write(encode_instrument_names(instruments))
|
||||
|
||||
with open(filename, "wb") as f:
|
||||
f.write(header)
|
||||
f.write(encode_instruments(instruments))
|
||||
f.write(encode_instrument_names(instruments))
|
||||
|
||||
def decode_voice(data, name):
|
||||
|
||||
fields = struct.unpack("<BBBBBBBBBBBBBBh", data)
|
||||
fields = struct.unpack("<BBBBBBBBBBBBBBh", data)
|
||||
|
||||
result = {}
|
||||
for i in range(len(GENMIDI_FIELDS)):
|
||||
result[GENMIDI_FIELDS[i]] = fields[i]
|
||||
result = {}
|
||||
for i in range(len(GENMIDI_FIELDS)):
|
||||
result[GENMIDI_FIELDS[i]] = fields[i]
|
||||
|
||||
result["m_ksl_volume"] = result["m_ksl"] | result["m_volume"]
|
||||
result["c_ksl_volume"] = result["c_ksl"] | result["c_volume"]
|
||||
result["name"] = name.decode("ascii").rstrip("\0")
|
||||
result["m_ksl_volume"] = result["m_ksl"] | result["m_volume"]
|
||||
result["c_ksl_volume"] = result["c_ksl"] | result["c_volume"]
|
||||
result["name"] = name.decode("ascii").rstrip("\0")
|
||||
|
||||
return result
|
||||
|
||||
return result
|
||||
|
||||
def decode_instrument(data, name):
|
||||
flags, finetune, fixed_note = struct.unpack("<hBB", data[0:4])
|
||||
flags, finetune, fixed_note = struct.unpack("<hBB", data[0:4])
|
||||
|
||||
voice1 = decode_voice(data[4:20], name)
|
||||
offset1 = voice1["note_offset"]
|
||||
voice1 = decode_voice(data[4:20], name)
|
||||
offset1 = voice1["note_offset"]
|
||||
|
||||
# Second voice?
|
||||
# Second voice?
|
||||
|
||||
if (flags & FLAG_TWO_VOICE) != 0:
|
||||
voice2 = decode_voice(data[20:], name)
|
||||
offset2 = voice2["note_offset"]
|
||||
else:
|
||||
voice2 = None
|
||||
offset2 = 0
|
||||
if (flags & FLAG_TWO_VOICE) != 0:
|
||||
voice2 = decode_voice(data[20:], name)
|
||||
offset2 = voice2["note_offset"]
|
||||
else:
|
||||
voice2 = None
|
||||
offset2 = 0
|
||||
|
||||
# Null out fixed_note if the fixed pitch flag isn't set:
|
||||
# Null out fixed_note if the fixed pitch flag isn't set:
|
||||
|
||||
if (flags & FLAG_FIXED_PITCH) == 0:
|
||||
fixed_note = None
|
||||
if (flags & FLAG_FIXED_PITCH) == 0:
|
||||
fixed_note = None
|
||||
|
||||
return Instrument(
|
||||
voice1, voice2, off1=offset1, off2=offset2, note=fixed_note
|
||||
)
|
||||
|
||||
return Instrument(voice1, voice2,
|
||||
off1=offset1, off2=offset2,
|
||||
note=fixed_note)
|
||||
|
||||
def read(filename):
|
||||
with open(filename, "rb") as f:
|
||||
data = f.read()
|
||||
with open(filename, "rb") as f:
|
||||
data = f.read()
|
||||
|
||||
# Check header:
|
||||
# Check header:
|
||||
|
||||
header = data[0:len(GENMIDI_HEADER)]
|
||||
if header.decode("ascii") != GENMIDI_HEADER:
|
||||
raise Exception("Incorrect header for GENMIDI lump")
|
||||
header = data[0 : len(GENMIDI_HEADER)]
|
||||
if header.decode("ascii") != GENMIDI_HEADER:
|
||||
raise Exception("Incorrect header for GENMIDI lump")
|
||||
|
||||
body = data[len(GENMIDI_HEADER):]
|
||||
instr_data = body[0:NUM_INSTRUMENTS * INSTR_DATA_LEN]
|
||||
instr_names = body[NUM_INSTRUMENTS * INSTR_DATA_LEN:]
|
||||
result = []
|
||||
body = data[len(GENMIDI_HEADER) :]
|
||||
instr_data = body[0 : NUM_INSTRUMENTS * INSTR_DATA_LEN]
|
||||
instr_names = body[NUM_INSTRUMENTS * INSTR_DATA_LEN :]
|
||||
result = []
|
||||
|
||||
for i in range(NUM_INSTRUMENTS):
|
||||
data = instr_data[i * INSTR_DATA_LEN:(i+1) * INSTR_DATA_LEN]
|
||||
name = instr_names[i * INSTR_NAME_LEN:(i+1) * INSTR_NAME_LEN]
|
||||
for i in range(NUM_INSTRUMENTS):
|
||||
data = instr_data[i * INSTR_DATA_LEN : (i + 1) * INSTR_DATA_LEN]
|
||||
name = instr_names[i * INSTR_NAME_LEN : (i + 1) * INSTR_NAME_LEN]
|
||||
|
||||
result.append(decode_instrument(data, name))
|
||||
result.append(decode_instrument(data, name))
|
||||
|
||||
return result
|
||||
|
||||
return result
|
||||
|
||||
if __name__ == "__main__":
|
||||
for filename in sys.argv[1:]:
|
||||
instruments = read(filename)
|
||||
for i in range(len(instruments)):
|
||||
instrument = instruments[i]
|
||||
fixed_note = instrument.fixed_note
|
||||
for filename in sys.argv[1:]:
|
||||
instruments = read(filename)
|
||||
for i in range(len(instruments)):
|
||||
instrument = instruments[i]
|
||||
fixed_note = instrument.fixed_note
|
||||
|
||||
if fixed_note is not None:
|
||||
print("%i (fixed note: %i):" % (i, fixed_note))
|
||||
else:
|
||||
print("%i:" % i)
|
||||
|
||||
print("\tVoice 1: %s" % instrument.voice1)
|
||||
if instrument.voice2 is not None:
|
||||
print("\tVoice 2: %s" % instrument.voice2)
|
||||
if fixed_note is not None:
|
||||
print("%i (fixed note: %i):" % (i, fixed_note))
|
||||
else:
|
||||
print("%i:" % i)
|
||||
|
||||
print("\tVoice 1: %s" % instrument.voice1)
|
||||
if instrument.voice2 is not None:
|
||||
print("\tVoice 2: %s" % instrument.voice2)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue