mirror of https://github.com/lalbornoz/roar.git
151 lines
6.4 KiB
Python
151 lines
6.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
#
|
||
|
# SAUCEToMiRCART.py -- convert SAUCE-encoded ANSi to mIRC art file (for spoke)
|
||
|
# Copyright (c) 2019 Lucio Andrés Illanes Albornoz <lucio@lucioillanes.de>
|
||
|
# This project is licensed under the terms of the MIT licence.
|
||
|
#
|
||
|
|
||
|
import os, re, struct, sys
|
||
|
|
||
|
AnsiBgToMiRCARTColours = {
|
||
|
107: 0, # Bright White
|
||
|
40: 1, # Black
|
||
|
104: 2, # Blue
|
||
|
42: 3, # Green
|
||
|
101: 4, # Red
|
||
|
41: 5, # Light Red
|
||
|
45: 6, # Pink
|
||
|
43: 7, # Yellow
|
||
|
103: 8, # Light Yellow
|
||
|
102: 9, # Light Green
|
||
|
46: 10, # Cyan
|
||
|
106: 11, # Light Cyan
|
||
|
44: 12, # Light Blue
|
||
|
105: 13, # Light Pink
|
||
|
100: 14, # Grey
|
||
|
47: 15, # Light Grey
|
||
|
};
|
||
|
|
||
|
AnsiFgToMiRCARTColours = {
|
||
|
97: 0, # Bright White
|
||
|
30: 1, # Black
|
||
|
94: 2, # Blue
|
||
|
32: 3, # Green
|
||
|
91: 4, # Red
|
||
|
31: 5, # Light Red
|
||
|
35: 6, # Pink
|
||
|
33: 7, # Yellow
|
||
|
93: 8, # Light Yellow
|
||
|
92: 9, # Light Green
|
||
|
36: 10, # Cyan
|
||
|
96: 11, # Light Cyan
|
||
|
34: 12, # Light Blue
|
||
|
95: 13, # Light Pink
|
||
|
90: 14, # Grey
|
||
|
37: 15, # Light Grey
|
||
|
};
|
||
|
|
||
|
AnsiFgBoldToMiRCARTColours = {
|
||
|
97: 0, # Bright White
|
||
|
30: 14, # Grey
|
||
|
94: 12, # Light Blue
|
||
|
32: 9, # Light Green
|
||
|
91: 4, # Light Red
|
||
|
31: 4, # Light Red
|
||
|
35: 13, # Light Pink
|
||
|
33: 8, # Light Yellow
|
||
|
93: 8, # Light Yellow
|
||
|
92: 9, # Light Green
|
||
|
36: 11, # Light Cyan
|
||
|
96: 11, # Light Cyan
|
||
|
34: 12, # Light Blue
|
||
|
95: 13, # Light Pink
|
||
|
90: 14, # Grey
|
||
|
37: 0, # Bright White
|
||
|
};
|
||
|
|
||
|
def SAUCEToMiRCART(inPathName, outPathName):
|
||
|
with open(inPathName, "rb") as inFile:
|
||
|
inFileStat = os.stat(inPathName)
|
||
|
inFile.seek(inFileStat.st_size - 128, 0)
|
||
|
inFile.seek(5 + 2 + 35 + 20 + 20 + 8 + 4, 1)
|
||
|
if (inFile.read(1) != b'\x01') \
|
||
|
or (inFile.read(1) != b'\x01'):
|
||
|
print("error: only character based ANSi SAUCE files are supported.", file=sys.stderr)
|
||
|
return 1
|
||
|
else:
|
||
|
width, height = struct.unpack("H", inFile.read(2))[0], struct.unpack("H", inFile.read(2))[0]
|
||
|
with open(outPathName, "w+") as outFile:
|
||
|
inFile.seek(0, 0)
|
||
|
inFileData, row, rowChars = inFile.read(inFileStat.st_size - 128).decode("cp437"), "", 0
|
||
|
inFileChar, inFileCharMax = 0, len(inFileData)
|
||
|
curBg, curFg = 1, 15; curBgAnsi, curBoldAnsi, curFgAnsi = 30, False, 37;
|
||
|
while True:
|
||
|
if inFileChar >= inFileCharMax:
|
||
|
break
|
||
|
else:
|
||
|
m = re.match('\x1b\[((?:\d{1,3};?)+)m', inFileData[inFileChar:])
|
||
|
if m:
|
||
|
newBg, newFg = -1, -1
|
||
|
for ansiCode in m[1].split(";"):
|
||
|
ansiCode = int(ansiCode)
|
||
|
if ansiCode == 0:
|
||
|
curBgAnsi, curBoldAnsi, curFgAnsi = 30, False, 37; newBg, newFg = 1, 15;
|
||
|
elif ansiCode == 1:
|
||
|
curBoldAnsi, newFg = True, AnsiFgBoldToMiRCARTColours[curFgAnsi]
|
||
|
elif ansiCode == 2:
|
||
|
curBoldAnsi, newFg = False, AnsiFgToMiRCARTColours[curFgAnsi]
|
||
|
elif ansiCode == 7:
|
||
|
newBg, newFg = curFg, curBg; curBgAnsi, curFgAnsi = curFgAnsi, curBgAnsi;
|
||
|
elif ansiCode in AnsiBgToMiRCARTColours:
|
||
|
curBgAnsi, newBg = ansiCode, AnsiBgToMiRCARTColours[ansiCode]
|
||
|
elif ansiCode in AnsiFgToMiRCARTColours:
|
||
|
if curBoldAnsi:
|
||
|
newFg = AnsiFgBoldToMiRCARTColours[ansiCode]
|
||
|
else:
|
||
|
newFg = AnsiFgToMiRCARTColours[ansiCode]
|
||
|
curFgAnsi = ansiCode
|
||
|
elif ansiCode in AnsiFgBoldToMiRCARTColours:
|
||
|
curFgAnsi, newFg = ansiCode, AnsiFgBoldToMiRCARTColours[ansiCode]
|
||
|
if ((newBg != -1) and (newFg != -1)) \
|
||
|
and ((newBg == curFg) and (newFg == curBg)):
|
||
|
row += "\u0016"; curBg, curFg = newBg, newFg;
|
||
|
elif ((newBg != -1) and (newFg != -1)) \
|
||
|
and ((newBg != curBg) and (newFg != curFg)):
|
||
|
row += "\u0003{},{}".format(newFg, newBg); curBg, curFg = newBg, newFg;
|
||
|
elif (newBg != -1) and (newBg != curBg):
|
||
|
row += "\u0003{},{}".format(curFg, newBg); curBg = newBg;
|
||
|
elif (newFg != -1) and (newFg != curFg):
|
||
|
row += "\u0003{}".format(newFg); curFg = newFg;
|
||
|
inFileChar += len(m[0])
|
||
|
else:
|
||
|
m = re.match('\x1b\[(\d+)C', inFileData[inFileChar:])
|
||
|
if m:
|
||
|
row += m[0]; inFileChar += len(m[0]); rowChars += int(m[1]);
|
||
|
elif inFileData[inFileChar:inFileChar+2] == "\r\n":
|
||
|
inFileChar += 2; rowChars = width;
|
||
|
elif inFileData[inFileChar] == "\r" \
|
||
|
or inFileData[inFileChar] == "\n":
|
||
|
inFileChar += 1; rowChars = width;
|
||
|
else:
|
||
|
row += inFileData[inFileChar]; inFileChar += 1; rowChars += 1;
|
||
|
if rowChars >= width:
|
||
|
print(row, file=outFile); row = ""; rowChars = 0;
|
||
|
if (curBg != 1) and (curFg != 15):
|
||
|
row += "\u0003{},{}".format(curFg, curBg);
|
||
|
elif curBg != 1:
|
||
|
row += "\u0003{},{}".format(curFg, curBg);
|
||
|
elif curFg != 1:
|
||
|
row += "\u0003{}".format(curFg);
|
||
|
#
|
||
|
# Entry point
|
||
|
def main(*argv):
|
||
|
SAUCEToMiRCART(argv[1], argv[2])
|
||
|
if __name__ == "__main__":
|
||
|
if (len(sys.argv) - 1) != 2:
|
||
|
print("usage: {} <SAUCE input file pathname> <mIRC art output file pathname>".format(sys.argv[0]), file=sys.stderr)
|
||
|
else:
|
||
|
main(*sys.argv)
|
||
|
|
||
|
# vim:expandtab foldmethod=marker sw=4 ts=4 tw=120
|