User:User A1/svgTinker.py
- !/usr/bin/python
from BeautifulSoup import BeautifulStoneSoup
import string
import sys
import re
def epsilon():
eps=float(1.0);
while eps + 1.0 > 1.0 : eps = eps//2;
return eps;
- Takes a stone-soup tag and applies various
- workaround fixes of dubious effectiveness
def fontFix(tag):
print type(tag); bold=False; italic=False; dejavu=False;
dejaVuRe=re.compile("'?(?i)dejavusans.*") boldRe=re.compile("(?i).*-bold.*"); italicRe=re.compile("(?i).*-italic.*"); fontRe=re.compile("(?i)-.*");
for i in tag.attrs :
print i if i[0] == "font-family" : #Check the font types and perform font substitution bold=boldRe.match(i[1]); italic=italicRe.match(i[1]); dejavu=dejaVuRe.match(i[1]); #Strip font bold/italic embed tmp = fontRe.split(i[1]); fontAttr=tmp[0]; i = (i[0],fontAttr) break;
#if none of the above apply we can skip if(not bold and not italic and not dejavu): return;
str=""
if (bold) : str=str+"bad bolding method "; if(italic): str=str+"bad italicising method "; if(dejavu): str=str+"wrong font name";
print "Fixing tag : " + str print tag #Otherwise we have work to do!
haveWeight=False; haveStyle=False;
for i in tag.attrs: #find any bold font-weight tag if i[0] == "font-weight": haveWeight=True; continue; if i[0] == "font-style": haveStyle=True; continue;
#Check for bold
if(bold):
if(haveWeight):
if not re.match(i[1],".*(?i)bold.*"):
tag["font-weight"]+=";Bold";
else:
tag["font-weight"]="Bold";
tag["font-family"]=re.sub("(?i)-Bold","",tag["font-family"])
#Check for italics
if(italic and haveStyle ):
if not re.match(i[1],".*(?i)italic.*"):
tag["font-style"]+=";Italic";
else:
if italic and not haveStyle:
tag["font-style"]="Italic";
#Fix dejavu vs Deja Vu if (dejavu): tag["font-family"]="DejaVu Sans";
- Check to see if a small font is being used in conjunction with
def fontSizeFix(tag):
#without a transformatio there is nothing we can do if(not tag["transform"] ) : return
#grab the matrix
trans=re.sub(".*(?i)matrix\(","",tag["transform"]);
trans=re.sub("\)","",trans);
#split the transformation matirx m = re.split("(\ |,)",trans);
m=filter(lambda x: not (x=="" or x==" " or x==",") ,m);
#Transform should be of the form y=Mx+b print m assert(len(m) == 6);
mF=[]; for i in m: mF.append(float(i));
m=mF;
print m
EPSILON=0.001;
if(abs(m[1]) < EPSILON and abs(m[2]) < EPSILON ):
#OK, so M is a diagonal matrix
print "so far so good"
if abs(m[0]) > abs(m[3]) :
factor=m[0];
else:
factor=m[3];
if (factor > 1 ):
#Pump up the font size by factor, then reduce the matrix
fontSize =float(tag["font-size"]);
tag["font-size"] = fontSize*factor;
m[0] = m[0]/factor;
m[3] = m[3]/factor;
tag["transform"] = "matrix(" + str(m[0]) + " " + str(m[1]) + " " + str(m[2]) + " " + str(m[3]) + " "+ str(m[4]) + " " + str(m[5]) + ")";
return;
- Crappy font substitution routine
def fontSub(tag):
preferredFont = [];
preferredFont.append((re.compile("(?i)Arial.*"),"DejaVu Sans"));
preferredFont.append((re.compile("(?i)Times new roman.*"),"Times"));
for i in tag.attrs : if i[0] == "font-family" : #Substitute fonts from our preferred font table for j in preferredFont: if j[0].match(i[1]): tag["font-family"]=j[1]; break; return;
def main():
if(not len(sys.argv) == 3): print "Usage: svgTinker.py inputFile outputFile" quit(1);
f = open(sys.argv[1]);
if not f : print "File does not exist or could not be read" quit(1)
xmlText = f.read();
soup=BeautifulStoneSoup(xmlText);
tags=soup.findAll("text");
#Correct all font tags for i in tags: fontTag=False; if(len(i.attrs)): for j in i.attrs : if re.match("(?i)font-family",j[0]): fontTag=True;
break;
if fontTag : fontFix(i); fontSizeFix(i); fontSub(i); continue;
tags=soup.findAll("tspan");
#Nuke the tspans, preserving children for i in tags: i.extract();
try: f=open(sys.argv[2],'w'); except: print('Unable to open file for writing. aborting'); quit(1);
#save modified svg data
f.write(str(soup));
print("Wrote file : " + sys.argv[2]);
if __name__ == "__main__": main()