import matplotlib.pyplot as plt from math import * import sys if len(sys.argv) < 4: print 'rtp: not enough parameter!' print '\tUsage: rt.py dir_name snap_num radius' print '\tExample: rtp.py PKlh-1 0010 20.0' sys.exit(-1) dir_name = sys.argv[1] snap_num = sys.argv[2] rsize = float(sys.argv[3]) ntot = 0 ntot, arr = mp.read_dump(mp.prefix, dir_name, snap_num) r = arr['rad'][:] T = arr['T'][:] idw = np.where(arr['type'] == mp.TYPE_WARM) idc = np.where(arr['type'] == mp.TYPE_COLD) rc('font', size=17) plt.xlabel('Radius [kpc]') plt.ylabel('Temperature [K]') plt.subplots_adjust(left=0.15, right=0.95, top=0.93, bottom=0.12) plt.yscale('log') print 'len of r:', len(r)
#base_name = '1E3' #snap_num = '0001' #nproc = 1 if len(sys.argv) < 4: print 'Dsort: not enough parameter!' print '\tUsage: Dsort.py dir_name snap_num range' print '\tExample: Dsort.py PKc-1 0001 50' sys.exit(1) dir_name = sys.argv[1] snap_num = sys.argv[2] size = float(sys.argv[3]) ntot = 0 ntot, rarr = mp.read_dump(mp.prefix, dir_name, snap_num) print 'Sorting...' arr = np.sort(rarr, order = 'den') print 'done!' x = arr['pos'][:, 0] y = arr['pos'][:, 1] z = arr['pos'][:, 2] den = arr['den'][:] idw = np.where(arr['type'] == mp.TYPE_WARM) idc = np.where(arr['type'] == mp.TYPE_COLD) ids = np.where(arr['type'] == mp.TYPE_STAR_PARAL)