Beispiel #1
0
def draw(asw,xc,yc):
    print asw
    N,R = 100,50
    if sim[asw][0] == 'quasar':
        flag,R = 'Q',20
    if sim[asw][0] == 'galaxy':
        flag,R = 'G',20
    if sim[asw][0] == 'cluster':
        flag,R = 'C',42
    x = linspace(-R,R,N)
    y = 1*x
    kappa,arriv = many.grids(asw,x,y)
    fig = figure()
    panel = fig.add_subplot(1,1,1)
    panel.set_aspect('equal')
    swsquare = imread('asw/'+asw+'.png')
    panel.imshow(swsquare[yc-R:yc+R+1,xc-R:xc+R+1])
    lo,hi = amin(arriv), amax(arriv)
    lev = linspace(lo,lo+.2*(hi-lo),100)
#    x = 1.2*x
#    y = 1.2*y
    panel.contour(x+R,-y+R,arriv,lev)
    show()
Beispiel #2
0
def draw_sim(asw, sim):
  
    path = os.path.join(simdir, asw)
#    try:
#      path = os.path.join(simdir, asw)
#      #os.makedirs(path)
#    except OSError as e:
#      print 'error creating dir', e
#      #return

    filenames = ['arriv', 'kappa', 'm_encl']
    
    print '> drawing sim %s'%asw,    
    
    #prevent submodules prints..
    old_stdout = sys.stdout
    old_stderr = sys.stderr
    devnull = open(os.devnull, 'w')
    sys.stdout = devnull
    #sys.stderr = devnull
      
    print asw
    N,R = 100,50
    if sim[asw][0] == 'quasar':
        flag,R = 'Q',20
    if sim[asw][0] == 'galaxy':
        flag,R = 'G',20
    if sim[asw][0] == 'cluster':
        flag,R = 'C',50
    x = np.linspace(-R,R,N)
    y = 1*x
    kappa,arriv = many.grids(asw,x,y)


    fig = pl.figure()
    panel = fig.add_subplot(1,1,1)
    panel.set_aspect('equal')
    lev = np.linspace(0,10,41)
    pc = panel.contour(x,y,kappa,lev)
    panel.clabel(pc, inline=1, fontsize=10)
    pl.savefig(os.path.join(path, '%s.%s'%(filenames[1],ext)))


    fig = pl.figure()
    panel = fig.add_subplot(1,1,1)
    rad = np.linspace(0,R,20)[1:]
    radq = rad*rad
    sum = 0*rad
    for i in range(len(x)):
        for j in range(len(y)):
            rsq = x[i]**2 + y[j]**2
            for k in range(len(rad)):
                if rsq < radq[k]:
                    sum[k] += kappa[j,i]
    dx = x[1]-x[0]
    for k in range(len(rad)):
        sum[k] *= dx*dx/(pi*radq[k])
    fil = open(os.path.join(path, asw+'.txt'),'w')
    for k in range(len(rad)):
        fil.write('%9.2e %9.2e\n' % (rad[k],sum[k]))
    fil.close()
    fil = open(os.path.join(simdir, asw+'.txt'),'w')
    for k in range(len(rad)):
        fil.write('%9.2e %9.2e\n' % (rad[k],sum[k]))
    fil.close()
    panel.scatter(rad,sum)
    panel.set_xlabel('radius [pixels]')
    panel.set_ylabel('average interior \textkappa [1]')
    pl.savefig(os.path.join(path, '%s.%s'%(filenames[2],ext)))
    
    
    fig = pl.figure()
    panel = fig.add_subplot(1,1,1)
    panel.set_aspect('equal')
    lo,hi = np.amin(arriv), np.amax(arriv)
    lev = np.linspace(lo,lo+.2*(hi-lo),100)
    panel.contour(x,y,arriv,lev)
    pl.savefig(os.path.join(path, '%s.%s'%(filenames[0],ext)))

    #restore stdout
    sys.stdout = old_stdout
    sys.stderr = old_stderr
    
    print '...DONE.'
    for f in filenames: print '  - %s'%(f+'.'+ext)