def plot_grid(fig, params): grid =map(lambda x: x.strip().split(' '), open(au.data_filepath(params['fname'])).readlines()) grid_rs = zeros((len(grid),len(grid[0])),int) + 200 if params['command'] == 'DumpLineageGrid': grid =array(grid, float) grid_colors = zeros(shape(grid),int) grid_colors += grid cols = ['Mut-Averse','Mut-Prone'] plotLinCounts(grid) elif params['command'] == 'DumpTaskGrid': grid_int = array(grid,int) grid_colors = zeros(shape(grid_int),int) -1 while 1: nz =nonzero(greater(grid_int,0)) if len(nz[0]) == 0 : break grid_colors[nz[0],nz[1]]+= 1 grid_int /= 2 grid = grid_colors #(array(grid, float)) cols,data = au.parse_printed('tasks.dat') cols = cols[1:] elif params['command'] == 'PrintResourceData': grid_int = array(grid,int) grid_colors = grid_int cols = [str(i) for i in range(max(grid_int)+1)] else: raise Exception('Grid type not implemented') #grid = params['grid'] x = shape(grid)[0] y = shape(grid)[1] n = x * y if x != y: print "Grid not square, skipping" return fig.clear() ct = mycolors.getct(len(cols)) nc = len(ct) xs,ys,rs,cs = [[] for i in range(4)] c_lambda = lambda x: x>=0 and ct[x] or [0,0,0] arr = zeros((x,y,3)) for i in range(nc): arr[nonzero(equal(grid, i))] = ct[i] #for i in range(x): # for j in range(y): # grid_colors # #for i in range(x): # for j in range(y): # xs.append(i) # ys.append(j) # rs.append(grid_rs[i,j]) # cs.append(c_lambda( grid_colors[i,j])) #ax = fig.add_subplot(111) #ax.scatter(xs,ys,rs,cs) plt.imshow(arr) up.color_legend(fig,ct,cols) plt.draw()
def plot_grid(fig, params, scatter_GEO = True): grid =map(lambda x: x.strip().split(' '), open(au.data_filepath(params['fname'])).readlines()) print(au.data_filepath(params['fname'])) color_grid(grid, params)