319 lines
9.6 KiB
Python
Executable File
319 lines
9.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
#
|
|
# Copyright 2016 Ville Rantanen
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU Lesser General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
'''Color by notation.'''
|
|
|
|
__author__ = "Ville Rantanen <ville.q.rantanen@gmail.com>"
|
|
__version__ = "0.2"
|
|
|
|
NEWCHAR="====CHAR"
|
|
NEWCOLOR="====COLOR"
|
|
NEWIMAGE="====EOI"
|
|
ENDOFFILE="====EOF"
|
|
PAUSE="====PAUSE"
|
|
|
|
import sys,os,argparse,re,time
|
|
from argparse import ArgumentParser
|
|
class bc:
|
|
K="\033[1;30m"
|
|
R="\033[1;31m"
|
|
G="\033[1;32m"
|
|
Y="\033[1;33m"
|
|
B="\033[1;34m"
|
|
M="\033[1;35m"
|
|
C="\033[1;36m"
|
|
W="\033[1;37m"
|
|
|
|
k="\033[0;30m"
|
|
r="\033[0;31m"
|
|
g="\033[0;32m"
|
|
y="\033[0;33m"
|
|
b="\033[0;34m"
|
|
m="\033[0;35m"
|
|
c="\033[0;36m"
|
|
w="\033[0;37m"
|
|
|
|
nk="\033[30m"
|
|
nr="\033[31m"
|
|
ng="\033[32m"
|
|
ny="\033[33m"
|
|
nb="\033[34m"
|
|
nm="\033[35m"
|
|
nc="\033[36m"
|
|
nw="\033[37m"
|
|
|
|
bk="\033[40m"
|
|
br="\033[41m"
|
|
bg="\033[42m"
|
|
by="\033[43m"
|
|
bb="\033[44m"
|
|
bm="\033[45m"
|
|
bc="\033[46m"
|
|
bw="\033[47m"
|
|
|
|
S = '\033[1m'
|
|
s = '\033[2m'#strong off
|
|
U = '\033[4m'#underline
|
|
u = '\033[24m'#underline off
|
|
ic = '\033[7m'#inverse colors
|
|
io = '\033[27m'#inverse off
|
|
st = '\033[9m'#strike on
|
|
so = '\033[29m'#strike off
|
|
|
|
Z = '\033[0m'
|
|
CLR = '\033[2J'
|
|
CLREND = '\033[K'
|
|
CLRSCR = CLR+"\033[0;0H"
|
|
|
|
color_keys="pqwertyu01234567;asdfghjPQWERTYUxXczvVbBnN"
|
|
color_list=[K, R, G, Y, B, M, C, W, k, r, g, y, b, m, c, w,
|
|
bk,br,bg,by,bb,bm,bc,bw,nk,nr,ng,ny,nb,nm,nc,nw,
|
|
S,s,CLRSCR,Z,U,u,st,so,ic,io]
|
|
|
|
def pos(self,y,x):
|
|
return "\033["+str(y)+";"+str(x)+"H"
|
|
|
|
def posprint(self, y,x,s):
|
|
sys.stdout.write( (self.pos(y,x) + str(s)).encode('utf-8') )
|
|
|
|
def clear(self):
|
|
sys.stdout.write( (self.CLR+self.pos(0,0)).encode('utf-8') )
|
|
def clear_to_end(self):
|
|
sys.stdout.write( self.CLREND.encode('utf-8') )
|
|
|
|
def color_char(self,s):
|
|
for i,c in enumerate(self.color_keys):
|
|
if c in s:
|
|
s=s.replace(c,self.color_list[i])
|
|
return s
|
|
return ""
|
|
def nocolor_string(self,s):
|
|
return ""
|
|
|
|
def end_drawing(opts):
|
|
if opts.zero_final:
|
|
sys.stdout.write(bc.Z.encode('utf-8'))
|
|
sys.exit(0)
|
|
|
|
def open_data_file(opts):
|
|
if opts.test_image:
|
|
return test_data()
|
|
if opts.filename=='-':
|
|
return sys.stdin
|
|
return open(opts.filename,'r')
|
|
|
|
def setup_options():
|
|
''' Create command line options '''
|
|
usage='''
|
|
Color image renderer in ANSI codes. Example file:
|
|
|
|
first layer of characters
|
|
====CHAR
|
|
more characters
|
|
go here
|
|
====COLOR
|
|
color codes go here (letters 1-7,0 and keys below like a piano)
|
|
12345670 : red green yellow blue magenta cyan white black (darker tone)
|
|
qwertyup : red green yellow blue magenta cyan white black (bright tone)
|
|
QWERTYUP : red green yellow blue magenta cyan white black (without changing brightness)
|
|
asdfghj; : backround colors
|
|
zc: reset colors, clearscreen
|
|
xX: bright color on, bright color off
|
|
vV: underline on, underline off
|
|
bB: strikethrough on, strikethrough off
|
|
nN: inverse on, inverse off
|
|
====EOI
|
|
end of image, start another
|
|
====PAUSE1.5
|
|
another image starts, pause for 1.5 seconds before drawing the next
|
|
====EOF
|
|
end of file
|
|
|
|
'''
|
|
|
|
parser=ArgumentParser(description=usage,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
epilog=__author__)
|
|
|
|
parser.add_argument("-v","--version",action="version",version=__version__)
|
|
parser.add_argument("--no-color","-n",action="store_false",dest="color",default=True,
|
|
help="Disable color.")
|
|
parser.add_argument("--test",action="store_true",dest="test_image",default=False,
|
|
help="Show test image showing features.")
|
|
parser.add_argument("-z",action="store_true",dest="zero",default=False,
|
|
help="Reset color codes at the end of each line.")
|
|
parser.add_argument("-Z",action="store_false",dest="zero_final",default=True,
|
|
help="Disable reset of colors at the end of file.")
|
|
parser.add_argument("-c",action="store_true",dest="clear",default=False,
|
|
help="Clear screen first")
|
|
parser.add_argument("filename",type=str, nargs='?', default=None,
|
|
help="File to show, - for stdin")
|
|
opts=parser.parse_args()
|
|
if not opts.test_image and opts.filename==None:
|
|
parser.error("Need a file or - as argument")
|
|
|
|
return opts
|
|
|
|
def test_data():
|
|
testimage=''' NOT VISIBLE
|
|
_ _ _ ____ ___
|
|
1 red 5 magenta / \ | \ | / ___|_ _|
|
|
2 green 6 cyan / _ \ | \| \___ \| |
|
|
3 yellow 7 white / ___ \| |\ |___) | |
|
|
4 blue 0 black /_/ \_\_| \_|____/___|
|
|
|
|
,ad8888ba, 88 88888888ba
|
|
d8"' `"8b 88 88 "8b
|
|
d8' 88 88 ,8P
|
|
88 ,adPPYba, 88 ,adPPYba, 88aaaaaa8P' ,adPPYba,
|
|
88 a8" "8a 88 a8" "8a 88""""88' I8[ ""
|
|
Y8, 8b d8 88 8b d8 88 `8b `"Y8ba,
|
|
Y8a. .a8P "8a, ,a8" 88 "8a, ,a8" 88 `8b aa ]8I
|
|
`"Y8888Y"' `"YbbdP"' 8888888888 `"YbbdP"' 88 `8b `"YbbdP"'
|
|
====CHAR
|
|
****************************************************
|
|
====COLOR
|
|
1 2 3 4 5 6 7 0 q w e r t y u p Q W EXR TxY U P
|
|
1
|
|
1 5 1
|
|
2 6 1
|
|
3 7 1
|
|
R P Q
|
|
~
|
|
E f z
|
|
e f z
|
|
w f z
|
|
w f z
|
|
w f z
|
|
u f u
|
|
7 f ;
|
|
7 f ;
|
|
====COLOR
|
|
~
|
|
~
|
|
~
|
|
~
|
|
~
|
|
v V b B bB v V
|
|
~
|
|
~ f E
|
|
~ f e
|
|
~ f w
|
|
~ f w
|
|
~ f w
|
|
; f ;
|
|
; f U
|
|
; f U
|
|
====EOI
|
|
Outputs:'''.encode('utf-8')
|
|
return testimage.split('\n')
|
|
|
|
bc=bc()
|
|
opts=setup_options()
|
|
f=open_data_file(opts)
|
|
endoffile=False
|
|
if opts.clear:
|
|
bc.clear()
|
|
if opts.test_image:
|
|
for row in f:
|
|
print(row)
|
|
while True:
|
|
pause=0
|
|
gray=[]
|
|
gray.append([])
|
|
colors=[]
|
|
colorFrame=0
|
|
grayFrame=0
|
|
maxRow=0
|
|
for row in f:
|
|
if not row:
|
|
end_drawing(opts)
|
|
row=row.decode('utf-8').rstrip("\n\r")
|
|
if row.startswith(NEWIMAGE):
|
|
break
|
|
if row.startswith(ENDOFFILE):
|
|
endoffile=True
|
|
if endoffile:
|
|
break
|
|
if row.startswith(PAUSE):
|
|
try:
|
|
new_value=row[len(PAUSE):].strip()
|
|
pause=float(new_value)
|
|
except:
|
|
pause=0
|
|
continue
|
|
if row.startswith(NEWCOLOR):
|
|
colorFrame+=1
|
|
colors.append([])
|
|
continue
|
|
|
|
if colorFrame==0:
|
|
if row.startswith(NEWCHAR):
|
|
grayFrame+=1
|
|
gray.append([])
|
|
continue
|
|
gray[grayFrame].append(row)
|
|
|
|
maxRow=max(maxRow, len(row))
|
|
else:
|
|
colors[colorFrame-1].append(row)
|
|
if len(gray[0])==0:
|
|
end_drawing(opts)
|
|
for i in range(len(gray[0])):
|
|
while maxRow>len(gray[0][i]):
|
|
gray[0][i]=gray[0][i]+" "
|
|
|
|
for i in range(len(gray[0])):
|
|
for frame in colors:
|
|
if len(frame)<i+1:
|
|
frame.append('')
|
|
while len(gray[0][i])>len(frame[i]):
|
|
frame[i]=frame[i]+" "
|
|
if len(gray)>1:
|
|
for layer in gray[1:]:
|
|
if len(layer)<i+1:
|
|
layer.append('')
|
|
while len(gray[0][i])>len(layer[i]):
|
|
layer[i]=layer[i]+" "
|
|
|
|
|
|
for i in range(len(gray[0])):
|
|
for c in range(len(gray[0][i])):
|
|
if opts.color:
|
|
for frame in colors:
|
|
try:
|
|
if frame[i][c]!=" ":
|
|
sys.stdout.write(bc.color_char(frame[i][c]).encode('utf-8'))
|
|
except IndexError:
|
|
pass
|
|
char=" "
|
|
for layer in gray:
|
|
if layer[i][c]==" ": continue
|
|
char=layer[i][c]
|
|
|
|
sys.stdout.write(char.encode('utf-8'))
|
|
if opts.color and opts.zero:
|
|
sys.stdout.write(bc.Z.encode('utf-8'))
|
|
sys.stdout.write("\n")
|
|
if pause>0:
|
|
time.sleep(pause)
|
|
if opts.test_image:
|
|
end_drawing(opts)
|
|
|