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)
