import string from pprint import pprint def between(s,first,second): if s.find(first) > -1: return (s.split(first)[1].split(second)[0]) def do(cmd='ls'): print 'dodot.do::', cmd from os import popen3 pIn,pOut,pErr=0,1,2 popenResults=popen3(cmd) #showTheImage: I was running this twice pOut=popenResults[pOut].read() pErr=popenResults[pErr].read() print pErr,pOut return pOut def makeTestfile(fname='t.dot',content=''): tfile=open(fname,'w') if content: tfile.write(content) else: tfile.write('digraph g{a->b}') tfile.close() def showTestfile(fname): print open(fname).read() def doDot(fname): do('/sw/bin/dot ' + fname) def testdot(): makeTestfile('t.dot','digraph g{a->b->c->d}') doDot('t.dot') def BracketPhrases(remainder='[one phrase] [middle] [and yet another'): if remainder and remainder.strip()[-1]<>']': remainder=remainder.strip()+']' attrs=[] while between(remainder,'[',']'): attrs.append( between(remainder,'[',']' ) ) choppoint=remainder.find(']') + 1 remainder=remainder[choppoint : ] return attrs # with end-brackets removed #print BracketPhrases() # ['one phrase', 'another', 'and yet another'] def components(s): if s.count(']')<>s.count('['): return 'ERROR: unmatched brackets:\n' + s if ( ('[' in s) or (']' in s) ): if (s[0]<>'[' and s[-1]<> ']'): return 'ERROR: brackets go at beginning or end of line:\n' + s BPs=BracketPhrases(s) N=len(BPs) if N==0: return [ '' , s , '' ] if N==2: return [ BPs[0].strip(), between(s, ']', '[').strip(), BPs[1].strip() ] if N==1: if s.find('[')==0: return [ BPs[0].strip(), s.split(']')[1].strip(), '' ] else: return ['', s.split('[')[0].strip(), BPs[0].strip() ] return 'ERROR: too many bracket phrases:\n' + s def tstComponents(): s1='no brackets here' s2='brackets at right [right here]' s3='[at left]brackets at left ' s4= '[left] brackets both [right]' s5='[left] too many should make error [uh oh] more [right]' s6='unmatched] trouble' s7='brackets in middle [here] not ends' s8='[its all brackets]' assert components(s1) == ['', 'no brackets here', ''] assert components(s2) == ['', 'brackets at right', 'right here'], components(s2) assert components(s3) ==['at left', 'brackets at left', ''] assert components(s4) ==['left', 'brackets both', 'right'] assert components(s5) =='ERROR: too many bracket phrases:\n' + s5 assert components(s6) =='ERROR: unmatched brackets:\n' + s6 assert components(s7) =='ERROR: brackets go at beginning or end of line:\n' + s7 if __debug__: tstComponents() def wwrap(text='this is pretty very long', width=10): """ A word-wrap function that preserves existing line breaks and most spaces in the text. Expects that existing line breaks are posix newlines (\n). """ return reduce(lambda line, word, width=width: '%s%s%s' % (line, ' \n'[(len(line[line.rfind('\n')+1:]) + len(word.split('\n',1)[0]) >= width)], word), text.split(' ') ) #print wwrap() def wordwrap(text='this is pretty very long', width=10): if width<1: return text return wwrap(text=text, width=width).replace('\n','\\n') def Shorthands(chunk): colors = """aliceblue antiquewhite antiquewhite1 antiquewhite2 antiquewhite3 antiquewhite4 aquamarine aquamarine1 aquamarine2 aquamarine3 aquamarine4 azure azure1 azure2 azure3 azure4 beige bisque bisque1 bisque2 bisque3 bisque4 black blanchedalmond blue blue1 blue2 blue3 blue4 blueviolet brown brown1 brown2 brown3 brown4 burlywood burlywood1 burlywood2 burlywood3 burlywood4 cadetblue cadetblue1 cadetblue2 cadetblue3 cadetblue4 chartreuse chartreuse1 chartreuse2 chartreuse3 chartreuse4 chocolate chocolate1 chocolate2 chocolate3 chocolate4 coral coral1 coral2 coral3 coral4 cornflowerblue cornsilk cornsilk1 cornsilk2 cornsilk3 cornsilk4 crimson cyan cyan1 cyan2 cyan3 cyan4 darkgoldenrod darkgoldenrod1 darkgoldenrod2 darkgoldenrod3 darkgoldenrod4 darkgreen darkkhaki darkolivegreen darkolivegreen1 darkolivegreen2 darkolivegreen3 darkolivegreen4 darkorange darkorange1 darkorange2 darkorange3 darkorange4 darkorchid darkorchid1 darkorchid2 darkorchid3 darkorchid4 darksalmon darkseagreen darkseagreen1 darkseagreen2 darkseagreen3 darkseagreen4 darkslateblue darkslategray darkslategray1 darkslategray2 darkslategray3 darkslategray4 darkslategrey darkturquoise darkviolet deeppink deeppink1 deeppink2 deeppink3 deeppink4 deepskyblue deepskyblue1 deepskyblue2 deepskyblue3 deepskyblue4 dimgray dimgrey dodgerblue dodgerblue1 dodgerblue2 dodgerblue3 dodgerblue4 firebrick firebrick1 firebrick2 firebrick3 firebrick4 floralwhite forestgreen gainsboro ghostwhite gold gold1 gold2 gold3 gold4 goldenrod goldenrod1 goldenrod2 goldenrod3 goldenrod4 gray gray0 gray1 gray10 gray100 gray11 gray12 gray13 gray14 gray15 gray16 gray17 gray18 gray19 gray2 gray20 gray21 gray22 gray23 gray24 gray25 gray26 gray27 gray28 gray29 gray3 gray30 gray31 gray32 gray33 gray34 gray35 gray36 gray37 gray38 gray39 gray4 gray40 gray41 gray42 gray43 gray44 gray45 gray46 gray47 gray48 gray49 gray5 gray50 gray51 gray52 gray53 gray54 gray55 gray56 gray57 gray58 gray59 gray6 gray60 gray61 gray62 gray63 gray64 gray65 gray66 gray67 gray68 gray69 gray7 gray70 gray71 gray72 gray73 gray74 gray75 gray76 gray77 gray78 gray79 gray8 gray80 gray81 gray82 gray83 gray84 gray85 gray86 gray87 gray88 gray89 gray9 gray90 gray91 gray92 gray93 gray94 gray95 gray96 gray97 gray98 gray99 green green1 green2 green3 green4 greenyellow grey grey0 grey1 grey10 grey100 grey11 grey12 grey13 grey14 grey15 grey16 grey17 grey18 grey19 grey2 grey20 grey21 grey22 grey23 grey24 grey25 grey26 grey27 grey28 grey29 grey3 grey30 grey31 grey32 grey33 grey34 grey35 grey36 grey37 grey38 grey39 grey4 grey40 grey41 grey42 grey43 grey44 grey45 grey46 grey47 grey48 grey49 grey5 grey50 grey51 grey52 grey53 grey54 grey55 grey56 grey57 grey58 grey59 grey6 grey60 grey61 grey62 grey63 grey64 grey65 grey66 grey67 grey68 grey69 grey7 grey70 grey71 grey72 grey73 grey74 grey75 grey76 grey77 grey78 grey79 grey8 grey80 grey81 grey82 grey83 grey84 grey85 grey86 grey87 grey88 grey89 grey9 grey90 grey91 grey92 grey93 grey94 grey95 grey96 grey97 grey98 grey99 honeydew honeydew1 honeydew2 honeydew3 honeydew4 hotpink hotpink1 hotpink2 hotpink3 hotpink4 indianred indianred1 indianred2 indianred3 indianred4 indigo ivory ivory1 ivory2 ivory3 ivory4 khaki khaki1 khaki2 khaki3 khaki4 lavender lavenderblush lavenderblush1 lavenderblush2 lavenderblush3 lavenderblush4 lawngreen lemonchiffon lemonchiffon1 lemonchiffon2 lemonchiffon3 lemonchiffon4 lightblue lightblue1 lightblue2 lightblue3 lightblue4 lightcoral lightcyan lightcyan1 lightcyan2 lightcyan3 lightcyan4 lightgoldenrod lightgoldenrod1 lightgoldenrod2 lightgoldenrod3 lightgoldenrod4 lightgoldenrodyellow lightgray lightgrey lightpink lightpink1 lightpink2 lightpink3 lightpink4 lightsalmon lightsalmon1 lightsalmon2 lightsalmon3 lightsalmon4 lightseagreen lightskyblue lightskyblue1 lightskyblue2 lightskyblue3 lightskyblue4 lightslateblue lightslategray lightslategrey lightsteelblue lightsteelblue1 lightsteelblue2 lightsteelblue3 lightsteelblue4 lightyellow lightyellow1 lightyellow2 lightyellow3 lightyellow4 limegreen linen magenta magenta1 magenta2 magenta3 magenta4 maroon maroon1 maroon2 maroon3 maroon4 mediumaquamarine mediumblue mediumorchid mediumorchid1 mediumorchid2 mediumorchid3 mediumorchid4 mediumpurple mediumpurple1 mediumpurple2 mediumpurple3 mediumpurple4 mediumseagreen mediumslateblue mediumspringgreen mediumturquoise mediumvioletred midnightblue mintcream mistyrose mistyrose1 mistyrose2 mistyrose3 mistyrose4 moccasin navajowhite navajowhite1 navajowhite2 navajowhite3 navajowhite4 navy navyblue oldlace olivedrab olivedrab1 olivedrab2 olivedrab3 olivedrab4 orange orange1 orange2 orange3 orange4 orangered orangered1 orangered2 orangered3 orangered4 orchid orchid1 orchid2 orchid3 orchid4 palegoldenrod palegreen palegreen1 palegreen2 palegreen3 palegreen4 paleturquoise paleturquoise1 paleturquoise2 paleturquoise3 paleturquoise4 palevioletred palevioletred1 palevioletred2 palevioletred3 palevioletred4 papayawhip peachpuff peachpuff1 peachpuff2 peachpuff3 peachpuff4 peru pink pink1 pink2 pink3 pink4 plum plum1 plum2 plum3 plum4 powderblue purple purple1 purple2 purple3 purple4 red red1 red2 red3 red4 rosybrown rosybrown1 rosybrown2 rosybrown3 rosybrown4 royalblue royalblue1 royalblue2 royalblue3 royalblue4 saddlebrown salmon salmon1 salmon2 salmon3 salmon4 sandybrown seagreen seagreen1 seagreen2 seagreen3 seagreen4 seashell seashell1 seashell2 seashell3 seashell4 sienna sienna1 sienna2 sienna3 sienna4 skyblue skyblue1 skyblue2 skyblue3 skyblue4 slateblue slateblue1 slateblue2 slateblue3 slateblue4 slategray slategray1 slategray2 slategray3 slategray4 slategrey snow snow1 snow2 snow3 snow4 springgreen springgreen1 springgreen2 springgreen3 springgreen4 steelblue steelblue1 steelblue2 steelblue3 steelblue4 tan tan1 tan2 tan3 tan4 thistle thistle1 thistle2 thistle3 thistle4 tomato tomato1 tomato2 tomato3 tomato4 transparent turquoise turquoise1 turquoise2 turquoise3 turquoise4 violet violetred violetred1 violetred2 violetred3 violetred4 wheat wheat1 wheat2 wheat3 wheat4 white whitesmoke yellow yellow1 yellow2 yellow3 yellow4 yellowgreen""" shapes="""box polygon ellipse circle point egg triangle plaintext diamond trapezium parallelogram house hexagon octagon doublecircle doubleoctagon tripleoctagon invtriangle invtrapezium invhouse Mdiamond Msquare Mcircle""" styles="""filled solid dashed dotted bold invis""" arrowends="""none normal inv dot odot invdot invodot tee empty invempty open halfopen diamond odiamond box obox crow""" colors=colors.split() shapes=shapes.split() styles=styles.split() arrowends=arrowends.split() chunk=chunk.strip() if chunk in colors: chunk='color='+chunk elif chunk in shapes: chunk='shape='+chunk elif chunk in styles: chunk='style='+chunk elif chunk in arrowends: chunk='arrowhead='+chunk if chunk =='colorize': chunk='colorize=1' elif chunk in ['LR','TB']: chunk= 'rankdir='+chunk elif chunk=='debug': chunk='debug=YES' elif chunk=='tight': chunk='ranksep="0.1" nodesep="0.1"' elif chunk=='endfile': chunk='endfile=HERE' elif chunk=='URL' or chunk=='jump': chunk='fontcolor=blue URL="' + foundURL(line) + '"' return chunk def Unpack(bracketPhrase='yellow box tight dotted, inv'): bp=bracketPhrase attrs=[] for bp in bp.replace(',','').split(): #remove commas, split at whitespace attrs.append(Shorthands(bp) ) return ' '.join(attrs) def tstUnpack(): print(Unpack('yellow box tight dotted, inv') )=='color=yellow shape=box ranksep="0.1" nodesep="0.1" style=dotted arrowhead=inv' def paddedbracket(s): return '[' +Unpack(s) + ']' def o2dls(outline="""one\n two""",wrap=10): #o2dConditions lines=outline.split('\n') LinesWithFacets=[] #create this list of dotline components... for line in lines: if line.strip(): #empty lines are discarded deep=0 while line[deep] in string.whitespace: #count leading white space chars deep+=1 comps=components(line.strip()) #should return ltag,meat, rtag or else... for tag in (0,2): #unpack tags if comps[tag]: try: comps[tag]=paddedbracket(comps[tag]) except: print 'TROUBLE WITH', line if not comps[1][0] == '#': #leave comments alone if comps[1] not in ['node', 'edge', 'graph']: if 'digraph g{' in comps[1] or '}' in comps[1] or '->' in comps[1] : pass #allow dot language if begins with 'digraph' and ends with '}' else: if not comps[1][0] in '"': # doublequotes comps[1]=wordwrap(comps[1], wrap) comps[1]='"'+comps[1]+'"' #double-quote multiword lines assert type(comps)<>type('string'), comps #...an Error messages as a string LinesWithFacets.append( [deep] + comps) dotlines=[] ancestry=[] depth,ltags,meat,rtags=0,1,2,3 #useful names for i in range(len(LinesWithFacets)): #compute parentage count, ltags,line,rtags =LinesWithFacets[i] if line.startswith('#') or line.split()[0] in ['graph', 'node', 'edge']: dotlines.append(line + rtags) else: if i == 0: lastline=line ancestry=[] else: j=0 while i-j>0: j+=1 lastcount,lastline=LinesWithFacets[i-1][depth], LinesWithFacets[i-j][meat] if not (lastline.startswith('#') or lastline.split()[0] in ['graph', 'node', 'edge']): break #reach back for the last relevant line if count > lastcount: #we've indented assert count-lastcount==1, 'EXTRA INDENT at line ' + line ancestry.append(lastline) #who was my parent? print 'appended', lastline, 'to ancestry' elif count<lastcount: #we've outdented for j in range(lastcount-count): #my ancestor (if any) is farther back ancestry.pop() tabs=count*'\t' dotlines.append(tabs + line + rtags) if count >0: dotlines.append(tabs + ancestry[-1] + '->' + line + ltags) return dotlines def dotText(dls): #wrap dotlines in dgraph g{} and convert to a string, unless the firstword is graph or digraph (allows pure dot to go this way too) firstword=dls[0].split()[0] if firstword in ['graph', 'digraph'] : pass else: dls.insert(0,'digraph h{') dls.append('}') return '\n'.join(dls) Gdot='/Users/jis/python/G/MacOS/dot' Gneato='/Users/jis/python/G/MacOS/neato' Gcirco='/Users/jis/python/G/MacOS/circo' Gtwopi='/Users/jis/python/G/MacOS/neato' def doGraph(prog=Gdot, fname='t.dot', output='pdf', content='digraph g{x->y-z}' ): makeTestfile(fname,content) fnamepdf=fname.split('.')[0]+'.' + output if output == 'pdf' : output='epdf' do(prog + ' -T' + output + ' -o' +fnamepdf + ' ' + fname) do('open ' + fnamepdf ) def o2g(prog=Gdot, o='Your ad here', fname='o2g.dot', wrap=10, verbose=1, output='pdf'): # an Outline defines links by indents. # Attributes at the left modify the link. Attributes at the right modify the node. o=o.strip() print 'o.split()[o]', o.split()[0], 'startswith digraph:', o.startswith('digraph') if o.startswith('graph g{') or o.startswith('digraph g{'): dtext=o else: dotlines=o2dls(o,wrap) # dotlines is a list of strings suitable for graphviz dtext= dotText(dotlines) # dtext is one big string wrapped in 'digraph h{' if verbose: print 'DotText:\n', dtext doGraph(prog=prog, content=dtext, fname=fname, output=output) return dtext if __name__=='__main__': o2g(verbose=1, prog=Gdot, fname='o2gtest.dot',wrap=15, output='pdf',o=""" o2g Outline to Graphviz A line of words makes a node. No quoting is necessary wordwrapping is automatic indents denote children ad infinitum ad infinitum :-) [wheat filled doublecircle] attributes can be written the usual way [shape=box, color=pink, style=filled] but we also understand shorthands like 'pink filled box' [pink filled box] [red bold] attributes to the left flavor the edge, attributes to the right 'flavor' the node [lightblue filled diamond] you can also assign default attributes to graph, node, and edges the graph attributes work retroactively, but (as with graphviz), the node and edge attributes only work proactively # here come default settings for graph, node, and edge # keep them aligned with the previous line in order to not upset indenting graph [LR] node [red] edge [orange] thus, new defaults henceforth #its ok to indent comments for clarity and by the way, if you begin an outline with DIGRAPH G{ or GRAPH G{ (but all lower case) it will be passed, untouched to Graphviz [box] end of tour """) # o2g.bak o2d.Bak6 tstO2G o2gtest subGraphs o2r