"""
1999 Manuel Gutierrez Algaba
You are free to use , modify, distribute and copy this piece of
python code, if you keep this free copyright notice in it.

grapher.py Version 1.0

This small programm produces state diagrams, just specifying the
relationship among the different states

"""

class graphstate:
   def __init__(self, label1, label2 = None):
       self.label1= label1
       self.label2 = label2
       self.next = None
       self.goto=[]
       self.label_next = None
       self.make_width()
       self.size = ""
       self.numberlabel = None
       self.figure = "ellipse"
       self.already_drawn = 0
       self.already_drawn_arrows = 0
       self.index_of_goto_arrows = 0

   def do_size(self, s):
       """
       valid sizes are: 0.9 and 0.8, or greater
       """
       self.size = s

   def rel(self,n=None, l= None, f = None, g=[] ):
       """
       This makes a relationship among this state and those stated
       in n and g.
       n = the next one state,
       g = a list of states where this state goes
       l = label of the transition of this state to the next one
       f = forced_next , this forces f to be the next node to be visited.
       """
       self.next = n
       self.label_next = l
       self.goto = g
       self.forced_next = f

   def make_width(self):
       if self.label2 is None:
           t = len(self.label1)
       else:
           t = max(len(self.label1), len(self.label2))

       self.width = t

   def generate_gpic_code(self, file ):
       f = open( file, "w")
       self.write_headings(f)
       self.write_body(f)
       self.write_footnote(f)
       f.close()

   def make_size(self, size ):
       self.size = size

   def write_headings(self, f):
       if self.size=="":
           f.write(".PS \ndown\narrowhead=7\narrow\n")
       else:
           f.write(".PS "+`self.size`+"\ndown\narrowhead=7\narrow\n")

   def write_footnote(self, f):
       f.write(".PE\n")

   def write_body(self , f, numberlabel = None):
       self.assign_numberlabel()
       self.start_drawing(f)
       self.arrows_and_such(f)

   def start_drawing(self,f):
       print self.numberlabel, self.already_drawn
       if self.already_drawn == 1:
           return
       self.already_drawn = 1
       self.write_label(f)
       self.write_enclosing_figure(f)
       self.write_movement(f)
       self.write_rest_of_it(f)

   def write_rest_of_it(self,f):
       if not self.forced_next is None:
           self.forced_next.start_drawing(f)

       if not self.next is None:
           self.next.start_drawing(f)

       for i in self.goto:
           i[0].start_drawing(f)

   def write_rest_of_it_arrows(self,f):
       if not self.next is None:
           self.next.arrows_and_such(f)

       for i in self.goto:
           i[0].arrows_and_such(f)

   def write_movement(self,f):
       if self.label_next is None:
           f.write("move 0.2\n")
       else:
           f.write("move 0.5\n")

   def write_enclosing_figure(self, f ):
       if self.label2 is None:
           part1 = self.figure+  ' "'+self.label1  + '" '
       else:
           part1 = self.figure+ ' "'+ self.label1 + '" "' + self.label2 + '" '

           #print "line 200",self.width
       part2 = " ht 0.5 wid " + str( 0.12 *  self.width )
       f.write(part1 + part2 + "\n")

   def convert_numberlabel(self):
       return self.numberlabel[0]+str(self.numberlabel[1])

   def write_label(self,f):
       f.write(self.convert_numberlabel() +":  ")

   def what_numberlabel(self):
       return self.numberlabel

   def assign_numberlabel(self, numberlabel= None):
       """
       We go through all the states in the drawing. And we label them.
       """
       if not self.numberlabel is None:
           return
       # first we assign our own numberlabel
       if numberlabel is None:
           self.numberlabel = ('L', 0)
       else:
           #print "line 224",self.numberlabel
           self.numberlabel = ('L', numberlabel + 1)
       # then we propagate the numberlabels
       propagated_numberlabel = self.numberlabel[1]

       if not self.forced_next is None:
           propagated_numberlabel = self.forced_next.assign_numberlabel( propagated_numberlabel)

       if not self.next is None:
           propagated_numberlabel = self.next.assign_numberlabel( propagated_numberlabel)

       for i in self.goto:
           if i[0].what_numberlabel() is None:
               propagated_numberlabel = i[0].assign_numberlabel(propagated_numberlabel)

       return propagated_numberlabel

   def arrows_and_such(self, f):
       if self.already_drawn_arrows == 1:
           return
       self.already_drawn_arrows = 1
       self.write_arrows(f)
       self.write_rest_of_it_arrows(f)

   def write_arrows(self, file):
       '''
       Cases:
       arrow next:
         arrow   "health" rjust from Gairst.s to Econd.n
         arrow  from Econd.s to Erd.n
       arrow self:
          PAPER : spline    from Erd.e then up 0.2 right  0.3 then down 0.4 then up 0.2 left 0.3 ->
          box invis  "apteros"  at PAPER + ( 0.6, 0.0)
       arrow goto:
          D: arc ->  from Erd.e to Gairst.e
          sprintf("zoom") ljust at D.e + (0.2, 0.0)

       '''
       self.write_arrow_next(file)
       self.write_arrow_goto(file)

   def write_arrow_next(self, f):
       if not self.forced_next is None:
           self.arrow_forced_west((self.next, self.label_next),f )
           return

       if not self.next is None:
           if not self.label_next is None:
               f.write('arrow "'+self.label_next+'" rjust from '+self.convert_numberlabel()+'.s to '+ self.next.convert_numberlabel()+'.n\n')
           else:
               f.write('arrow from '+self.convert_numberlabel()+'.s to '+ self.next.convert_numberlabel()+'.n\n')

   def write_arrow_goto(self,f):
       '''
       arrow goto down to up:
          D: arc ->  from Erd.e to Gairst.e
          sprintf("zoom") ljust at D.e + (0.2, 0.0)
       arrow goto up to down :
          D: arc ->  from Erd.w to Gairst.w
          sprintf("zoom") rjust at D.e + (-0.2, 0.0)

       '''
       for i in self.goto:
           if i[0] != self:
               self.make_arrow(i,f)
           else:
               self.write_arrow_self(i,f)

   def make_arrow(self,arrow,f):
       if  self.is_lower(arrow[0]):
           self.arrow_down_up(arrow,f)
       else:
           self.arrow_up_down(arrow,f)

   def arrow_down_up(self, arrow, f):
       intlabel = self.generate_goto_arrow_internal_label(f)
       self.generate_arc(arrow,f,"DOWN-UP")
       self.generate_goto_arrow_label(arrow,f,intlabel,"DOWN-UP")

   def arrow_up_down(self, arrow, f):
       intlabel = self.generate_goto_arrow_internal_label(f)
       self.generate_arc(arrow,f,"UP-DOWN")
       self.generate_goto_arrow_label(arrow,f,intlabel,"UP-DOWN")
   def arrow_forced_west(self, arrow, f):
       intlabel = self.generate_goto_arrow_internal_label(f)
       self.generate_arc(arrow,f,"FORCED-WEST")
       self.generate_goto_arrow_label(arrow,f,intlabel,"FORCED-WEST")

   def generate_goto_arrow_internal_label(self, f):
       self.index_of_goto_arrows = self.index_of_goto_arrows + 1
       to_return = self.convert_numberlabel()+`self.index_of_goto_arrows`+" :"
       f.write(to_return)
       return to_return

   def generate_arc(self, arrow, f, location):
       if location =="DOWN-UP":
           f.write('arc -> from '+self.convert_numberlabel()+'.w to '+ arrow[0].convert_numberlabel()+'.w\n')
       elif location =="UP-DOWN":
           f.write('arc -> from '+self.convert_numberlabel()+'.e to '+ arrow[0].convert_numberlabel()+'.e\n')
       elif location=="FORCED-WEST":
           f.write('arc -> from '+self.convert_numberlabel()+'.w to '+ arrow[0].convert_numberlabel()+'.w\n')

   def generate_goto_arrow_label(self, arrow,f,intlabel,location):
       if arrow[1] is None:
           return
       if location =="DOWN-UP":
           #print "line 163", `arrow[1]`, intlabel
           f.write('sprintf("'+arrow[1]+'") ljust at '+intlabel[:-2]+'.w +(0.2, 0.0)\n')
       elif location =="UP-DOWN":
           f.write('sprintf("'+arrow[1]+'") rjust at '+intlabel[:-2]+'.e +(-0.2, 0.0)\n')
       elif location=="FORCED-WEST":
           f.write('sprintf("'+arrow[1]+'") ljust at '+intlabel[:-2]+'.w +(0.2, 0.0)\n')



   def is_lower(self, another_state):
       return self.what_numberlabel()[1] < another_state.what_numberlabel()[1]

   def write_arrow_self(self, arrow, f):
       '''
       PAPER : spline    from Erd.e then up 0.2 right  0.3 then down 0.4 then up 0.2 left 0.3 ->
       box invis  "apteros"  at PAPER + ( 0.6, 0.0)
       '''
       intlabel = self.generate_goto_arrow_internal_label(f)
       if arrow[1] is None:
           label= ""
       else:
           label = arrow[1]

       f.write(' spline from '+ self.convert_numberlabel()+'.e then up 0.2 right  0.3 then down 0.4 then up 0.2 left 0.3 -> \n \
box invis  "'+label +'" at '+ intlabel[:-2] + " +( 0.6, 0.0)\n")


def gs(label1 = None, label2 = None):
   return graphstate(label1,label2)


def __test24__():

   st1 = graphstate("wake up")
   st2 = graphstate("breakfast")
   st3 = graphstate("homework", "done?")
   st4 = graphstate("nice ", "class day")
   st5 = graphstate("bad", "class day")
   st6 = graphstate("have lunch")
   st7 = graphstate("go to","party")
   st8 = graphstate("have fun")
   st9 = graphstate("next day")
   #st1.do_size(2)
   st1.rel(n=st2)
   st2.rel(n=st3)
   st3.rel(n=st4, l='Yes', g=[(st5,'No')])
   st4.rel(n=st6,f=st5)
   st5.rel(n=st6)
   st6.rel(n=st7)
   st7.rel(n=st8)
   st8.rel(n=st9,g=[(st8, "have fun")])
   st9.rel(g=[(st1,"Sleep")])

   st1.generate_gpic_code("test1.gpic")

def __test25__():

   st1 = graphstate("consideraciones metafisicas up")
   st2 = graphstate("kuntakinte en norte america")
   st3 = graphstate("burros y otros", "el burro espa'nol")

   st1.rel(n=st2)
   st2.rel(n=st3)
   st3.rel(g=[(st1,"hola"),(st2,"adios")])

   st1.generate_gpic_code("test1.gpic")

if __name__=='__main__':
   __test24__()

You are viewing proxied material from ftp.icm.edu.pl. The copyright of proxied material belongs to its original authors. Any comments or complaints in relation to proxied material should be directed to the original authors of the content concerned. Please see the disclaimer for more details.