def rebuild_png_with_mpng(input_path, output_path, streams, recompress_level=6, replace_map=None): """Rebuild PNG with modified MPNG streams""" with open(input_path, 'rb') as fin, open(output_path, 'wb') as fout: fout.write(fin.read(8)) # PNG signature
PNG_SIGNATURE = b'\x89PNG\r\n\x1a\n' CHUNK_TYPE_MPNG = b'mPNg' # Custom MPNG chunk CHUNK_TYPE_IDAT = b'IDAT' CHUNK_TYPE_IEND = b'IEND' xtool -mpng+reflate
def write_chunk(f, chunk_type, data): """Write PNG chunk with CRC""" f.write(struct.pack('>I', len(data))) f.write(chunk_type) f.write(data) crc = zlib.crc32(chunk_type + data) & 0xffffffff f.write(struct.pack('>I', crc)) 'rb') as fin
if extract_only: for idx, comp in streams: raw = zlib.decompress(comp) out_file = f"{Path(input_png).stem}_stream_{idx}.raw" Path(out_file).write_bytes(raw) print(f"Extracted stream {idx} to {out_file}") return crc)) if extract_only: for idx