flake8ed

Author: doutriaux1

Doc/xmgrace_tutorial_Numeric.py

@@ -9,118 +9,121 @@
 TEMPDIR = './'
 
 # First let's create some data
-npoints=50
-X=numpy.arrayrange(npoints,dtype='d')
-X=X/float(npoints)*2.*numpy.pi
+npoints = 50
+X = numpy.arrayrange(npoints, dtype='d')
+X = X / float(npoints) * 2. * numpy.pi
 
-Y1=numpy.ma.sin(X)
-Y2=numpy.ma.cos(X)
-Y3=numpy.ma.tan(X)
+Y1 = numpy.ma.sin(X)
+Y2 = numpy.ma.cos(X)
+Y3 = numpy.ma.tan(X)
 
-# Now the real grace thing, plot the 2 on one graph and the diff on another graph
+# Now the real grace thing, plot the 2 on one graph and the diff on
+# another graph
 
-from genutil import xmgrace # first we need to import xmgrace module
+from genutil import xmgrace  # first we need to import xmgrace module
 
-x=xmgrace.init() # create our xmgrace object
+x = xmgrace.init()  # create our xmgrace object
 
 # First let's set our graphs area
 # graph 0, exist by default, but we need to add 1 graph, therefore:
-x.add_graph() # adds one graph to our graph list (x.Graph)
+x.add_graph()  # adds one graph to our graph list (x.Graph)
 
 # Let's change the orientation of the page to portrait
 x.portrait()
 
 # Now let's set the area of graph 0
-x.Graph[0].vymin=.55  # starts at 55% of the page
-x.Graph[0].vymax=.9   # stops at 90% of the page
-x.Graph[0].vxmin=.1  # starts at 10% of the page
-x.Graph[0].vxmax=.75   # stops at 75% of the page
+x.Graph[0].vymin = .55  # starts at 55% of the page
+x.Graph[0].vymax = .9   # stops at 90% of the page
+x.Graph[0].vxmin = .1  # starts at 10% of the page
+x.Graph[0].vxmax = .75   # stops at 75% of the page
 # and the area of graph 1
-x.Graph[1].vymin=.1 # starts at 10% of the page
-x.Graph[1].vymax=.45 # 45 % of page
+x.Graph[1].vymin = .1  # starts at 10% of the page
+x.Graph[1].vymax = .45  # 45 % of page
 # Let's offset the 2 graphs just for fun
-x.Graph[1].vxmin=.25  # starts at 55% of the page
-x.Graph[1].vxmax=.9   # stops at 90% of the page
+x.Graph[1].vxmin = .25  # starts at 55% of the page
+x.Graph[1].vxmax = .9   # stops at 90% of the page
 
 # let's set the titles and subtitles
-x.Graph[0].title='TRIGO'
-x.Graph[0].stitle='SIN and COS'
-x.Graph[1].stitle='TAN'
+x.Graph[0].title = 'TRIGO'
+x.Graph[0].stitle = 'SIN and COS'
+x.Graph[1].stitle = 'TAN'
 
 
 # ok now let's set the min ans max for the Yaxis
-x.Graph[0].yaxis.min=-1. # ymin for graph 0
-x.Graph[0].yaxis.max=1. # ymax for graph 0
-x.Graph[1].yaxis.min=-1.5 # ymin for graph 1
-x.Graph[1].yaxis.max=1.5  # ymax for graph 1
+x.Graph[0].yaxis.min = -1.  # ymin for graph 0
+x.Graph[0].yaxis.max = 1.  # ymax for graph 0
+x.Graph[1].yaxis.min = -1.5  # ymin for graph 1
+x.Graph[1].yaxis.max = 1.5  # ymax for graph 1
 
 # Now let's set the tick marks for Graph 0
-x.Graph[0].yaxis.tick.inc=1 # Main tick every unit
-x.Graph[0].yaxis.tick.minor_ticks=4 # 4 sub in between , 1 every .25 units
+x.Graph[0].yaxis.tick.inc = 1  # Main tick every unit
+x.Graph[0].yaxis.tick.minor_ticks = 4  # 4 sub in between , 1 every .25 units
 # Now let's set the tick marks for Graph 1
-x.Graph[1].yaxis.tick.inc=10 # Main tick every 10
-x.Graph[1].yaxis.tick.minor_ticks=4 # 4 sub in between , 1 every 2.5 units
+x.Graph[1].yaxis.tick.inc = 10  # Main tick every 10
+x.Graph[1].yaxis.tick.minor_ticks = 4  # 4 sub in between , 1 every 2.5 units
 
 # X values are between 0 and 2pi
 # therefore let's set the axis  from -1,1
-x.Graph[0].xaxis.min=0.
-x.Graph[0].xaxis.max=2.*numpy.pi
-x.Graph[1].xaxis.min=0.
-x.Graph[1].xaxis.max=2.*numpy.pi
-
-# Set the xaxis ticks 
-dic={0.:'0',numpy.pi/2:'PI/2',numpy.pi:'PI',3*numpy.pi/2:'3PI/2',2*numpy.pi:'2PI'}
-x.Graph[0].xaxis.tick.spec.loc=dic
+x.Graph[0].xaxis.min = 0.
+x.Graph[0].xaxis.max = 2. * numpy.pi
+x.Graph[1].xaxis.min = 0.
+x.Graph[1].xaxis.max = 2. * numpy.pi
+
+# Set the xaxis ticks
+dic = {0.: '0', numpy.pi / 2: 'PI/2', numpy.pi: 'PI',
+       3 * numpy.pi / 2: '3PI/2', 2 * numpy.pi: '2PI'}
+x.Graph[0].xaxis.tick.spec.loc = dic
 # way 2 the "less convential" way
-x.Graph[1].xaxis.tick.spec.loc=dic
+x.Graph[1].xaxis.tick.spec.loc = dic
 # Finally let's reduce the size a bit
-x.Graph[0].xaxis.tick.label.char_size=.8
-x.Graph[1].xaxis.tick.label.char_size=.8
+x.Graph[0].xaxis.tick.label.char_size = .8
+x.Graph[1].xaxis.tick.label.char_size = .8
 
 # Ok now the legend
-x.Graph[0].legend.char_size=.8
-x.Graph[0].legend.x=.8  # Legend at 80% in x
-x.Graph[0].legend.x=.8  # Legend at 80% in y
-#or
-x.Graph[1].legend.char_size=.8
-x.Graph[1].legend.x=.05
-x.Graph[1].legend.y=.35
+x.Graph[0].legend.char_size = .8
+x.Graph[0].legend.x = .8  # Legend at 80% in x
+x.Graph[0].legend.x = .8  # Legend at 80% in y
+# or
+x.Graph[1].legend.char_size = .8
+x.Graph[1].legend.x = .05
+x.Graph[1].legend.y = .35
 
 # Ok now let's play with the appearence of the sets themself
 # Ok by default we only have 2 sets (1 per graph)
 # So we need to add 2 sets
-x.add_set(0) # adding to graph 0 (since it's graph 0, the 0 is optional)
+x.add_set(0)  # adding to graph 0 (since it's graph 0, the 0 is optional)
 
 # First let's change the line width
-x.Graph[0].Set[0].line.linewidth=2
-x.Graph[0].Set[1].line.linewidth=2
-x.Graph[1].Set[0].line.linewidth=2
+x.Graph[0].Set[0].line.linewidth = 2
+x.Graph[0].Set[1].line.linewidth = 2
+x.Graph[1].Set[0].line.linewidth = 2
 # Let's set the colors
-x.Graph[0].Set[0].line.color='red' # xmgrace default colors are coded
-x.Graph[0].Set[1].line.color=4 # color 4 is blue
+x.Graph[0].Set[0].line.color = 'red'  # xmgrace default colors are coded
+x.Graph[0].Set[1].line.color = 4  # color 4 is blue
 # Let's define a new color for the tangeante
 x.add_color('purple')
-x.Graph[1].Set[0].line.color='purple' # use the color we just defined
+x.Graph[1].Set[0].line.color = 'purple'  # use the color we just defined
 # Let's set the second's set style to dash
-x.Graph[1].Set[0].line.linestyle='solid'
-x.Graph[0].Set[1].line.linestyle='dash'
-x.Graph[0].Set[0].line.linestyle=2 # 2 is dot
+x.Graph[1].Set[0].line.linestyle = 'solid'
+x.Graph[0].Set[1].line.linestyle = 'dash'
+x.Graph[0].Set[0].line.linestyle = 2  # 2 is dot
 
 # Now let's set the name for each set, to be used in the legend
-x.Graph[0].Set[0].legend='SIN'
-x.Graph[0].Set[1].legend='COS'
-x.Graph[1].Set[0].legend='TAN'
+x.Graph[0].Set[0].legend = 'SIN'
+x.Graph[0].Set[1].legend = 'COS'
+x.Graph[1].Set[0].legend = 'TAN'
 # ok one last trick, in order to have the 0 line plotted
 # let's create a dummy array with zeros in it:
-zero=[0,0]
+zero = [0, 0]
 # add a dataset and assign it graph 1 black color
-x.add_set(1,'black')
+x.add_set(1, 'black')
 
 # Finally just for fun let's place a big red "Sample" accross it
-x.add_string(0.5,0.5,'Sample',color='red',char_size=9,rot=55,just=14)
+x.add_string(0.5, 0.5, 'Sample', color='red', char_size=9, rot=55, just=14)
 
 # And plot these babies:
-x.plot([Y1,Y2,Y3,zero],xs=[X,X,X,[0.,2.*numpy.pi]]) # You MUST pass a list of slab, even if only one slab
+# You MUST pass a list of slab, even if only one slab
+x.plot([Y1, Y2, Y3, zero], xs=[X, X, X, [0., 2. * numpy.pi]])
 # or you can plot then 1 by 1
 # first let's clear
 x('kill G0.S0')
@@ -130,48 +133,51 @@
 
 # Now the tan is pretty ugly because of extreme let's mask everything
 # that is greater than 1.5 and redraw that
-Y3=numpy.ma.masked_greater(Y3,1.5)
+Y3 = numpy.ma.masked_greater(Y3, 1.5)
 
 # Also we're going to add error bars 10% of the value
 # dx for the sin
-YY1=numpy.ma.zeros((2,npoints),dtype='d')
-YY1[0]=Y1
-YY1[1]=Y1*.1
-x.Graph[0].Set[0].type='xydx'
-x.Graph[0].Set[0].errorbar.status='on'
-x.Graph[0].Set[0].errorbar.color='red'
+YY1 = numpy.ma.zeros((2, npoints), dtype='d')
+YY1[0] = Y1
+YY1[1] = Y1 * .1
+x.Graph[0].Set[0].type = 'xydx'
+x.Graph[0].Set[0].errorbar.status = 'on'
+x.Graph[0].Set[0].errorbar.color = 'red'
 # dy for the cos
-YY2=numpy.ma.zeros((2,npoints),dtype='d')
-YY2[0]=Y2
-YY2[1]=Y2*.1
-x.Graph[0].Set[1].type='xydy'
-x.Graph[0].Set[1].errorbar.status='on'
-x.Graph[0].Set[1].errorbar.color=x.Graph[0].Set[1].line.color
+YY2 = numpy.ma.zeros((2, npoints), dtype='d')
+YY2[0] = Y2
+YY2[1] = Y2 * .1
+x.Graph[0].Set[1].type = 'xydy'
+x.Graph[0].Set[1].errorbar.status = 'on'
+x.Graph[0].Set[1].errorbar.color = x.Graph[0].Set[1].line.color
 # dy and dx for the tan
-YY3=numpy.ma.zeros((3,npoints),dtype='d')
-YY3[0]=Y3
-YY3[1]=Y3*.1
-YY3[2]=Y3*.1
-x.Graph[1].Set[0].type='xydxdy'
-x.Graph[1].Set[0].errorbar.status='on'
-x.Graph[1].Set[0].errorbar.color='purple'
-
-x.plot(YY1,xs=X,G=0,S=0)
-x.plot(YY2,xs=X,G=0,S=1)
-x.plot(YY3,xs=X,G=1,S=0)
-# Now since we are passing lists and not numpy.mas we need to wrap them into a list
-x.plot([zero],xs=[[0.,2.*numpy.pi]],G=1,S=1) 
+YY3 = numpy.ma.zeros((3, npoints), dtype='d')
+YY3[0] = Y3
+YY3[1] = Y3 * .1
+YY3[2] = Y3 * .1
+x.Graph[1].Set[0].type = 'xydxdy'
+x.Graph[1].Set[0].errorbar.status = 'on'
+x.Graph[1].Set[0].errorbar.color = 'purple'
+
+x.plot(YY1, xs=X, G=0, S=0)
+x.plot(YY2, xs=X, G=0, S=1)
+x.plot(YY3, xs=X, G=1, S=0)
+# Now since we are passing lists and not numpy.mas we need to wrap them
+# into a list
+x.plot([zero], xs=[[0., 2. * numpy.pi]], G=1, S=1)
 # Finally let's save the result
-x.ps(TEMPDIR+'xmgrace_demo_Numeric') # postscript
-x.ps(TEMPDIR+'xmgrace_demo_Numeric_gray',color='grayscale') # grayscale postscript
-x.jpeg(TEMPDIR+'xmgrace_demo_Numeric',dpi=300,quality=80) # jpeg 300dpi, 80% quality compression
-x.pdf(TEMPDIR+'xmgrace_demo_Numeric')
-x.eps(TEMPDIR+'xmgrace_demo_Numeric')
-x.svg(TEMPDIR+'xmgrace_demo_Numeric')
-x.mif(TEMPDIR+'xmgrace_demo_Numeric')
-x.pnm(TEMPDIR+'xmgrace_demo_Numeric')
-x.png(TEMPDIR+'xmgrace_demo_Numeric')
-x.metafile(TEMPDIR+'xmgrace_demo_Numeric')
+x.ps(TEMPDIR + 'xmgrace_demo_Numeric')  # postscript
+x.ps(TEMPDIR + 'xmgrace_demo_Numeric_gray',
+     color='grayscale')  # grayscale postscript
+# jpeg 300dpi, 80% quality compression
+x.jpeg(TEMPDIR + 'xmgrace_demo_Numeric', dpi=300, quality=80)
+x.pdf(TEMPDIR + 'xmgrace_demo_Numeric')
+x.eps(TEMPDIR + 'xmgrace_demo_Numeric')
+x.svg(TEMPDIR + 'xmgrace_demo_Numeric')
+x.mif(TEMPDIR + 'xmgrace_demo_Numeric')
+x.pnm(TEMPDIR + 'xmgrace_demo_Numeric')
+x.png(TEMPDIR + 'xmgrace_demo_Numeric')
+x.metafile(TEMPDIR + 'xmgrace_demo_Numeric')
 
 # Finish
 x.close()