GalaxySim/gen_gal_input.py at master · aritraghsh09/GalaxySim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
############################################
#  gen_gal_input.py
#
#  A python script to generate GALFIT Template files
#############################################

from multiprocessing import Pool
import numpy as np
import time

NUM_ITER = 100000    #number of galaxies.
NUM_THREADS = 15    #number of threads available
FILE_PATH = "./"
IMG_PATH = "./"
#FILE_PATH = IMG_PATH = "./"

def file_write(i):

	template_file = open(FILE_PATH + 'galfit_temp_'+str(i),'w')

	template_file.write("===============================================================================\n")

	#File parameters
	template_file.write("A) gal.fits\n")  # Input data image (FITS file)
	template_file.write("B) "+IMG_PATH+"output_img_"+str(i)+".fits"+"\n") # Output data image block
	template_file.write("C) none\n") #Sigma Image
	template_file.write("D) none\n") #PSF
	template_file.write("E) 1\n") #PSF fine sampling factor relative to data
	template_file.write("F) none\n")
	template_file.write("G) none\n")
	template_file.write("H) 1 83 1 83\n") #Image region to fit (xmin xmax ymin ymax)
	template_file.write("I) 83 83\n") #Size of the convolution box (x y)
	template_file.write("J) 25.95\n") # Magnitude photometric zeropoint
	template_file.write("K) 0.06 0.06\n")# Plate scale (dx dy)    [arcsec per pixel]
	template_file.write("O) regular\n") # Display type (regular, curses, both)
	template_file.write("P) 1\n\n") # Choose: 0=optimize, 1=model, 2=imgblock, 3=subcomps

	#Object number 1   #the flag after the variable value refer to the fact whether the value is free to fit -- 1 or not 0
	template_file.write(" 0) sersic\n") #object type
	template_file.write(" 1) "+str(int(x_pos[i]))+" "+str(int(y_pos[i]))+" 0 0\n")#position x y
	#template_file.write(" 1) 83.0 83.0 0 0\n")#position x y
	template_file.write(" 3) "+str(inte_mag[i])+" 0\n")#Integrated Magnitude
	template_file.write(" 4) "+str(half_light_radius[i])+" 0\n")#R_e (half-light radius)   [pix]
	template_file.write(" 5) "+str(sersic_idx[i])+" 0\n")#Sersic index n (de Vaucouleurs n=4)
	template_file.write(" 6) 0.0000 0\n")
	template_file.write(" 7) 0.0000 0\n")
	template_file.write(" 8) 0.0000 0\n")
	template_file.write(" 9) "+str(axis_ratio[i])+" 0\n")#axis ratio (b/a)
	template_file.write("10) "+str(position_angle[i])+" 0\n")#position angle (PA) [deg: Up=0, Left=90]
	template_file.write(" Z) 0\n\n")#output option (0 = resid., 1 = Don't subtract)

	#Object number 2
	template_file.write(" 0) sky\n")
	template_file.write(" 1) "+ str(sky_back[i]) +" 0\n") # sky background at center of fitting region [ADUs]
	#template_file.write(" 1) 1.3920 0\n") # sky background at center of fitting region [ADUs]
	template_file.write(" 2) 0.0000 0\n") # dsky/dx (sky gradient in x)
	template_file.write(" 3) 0.0000 0\n")  #  dsky/dy (sky gradient in y)
	template_file.write(" Z) 0\n\n") #  output option (0 = resid., 1 = Don't subtract)

	template_file.write("===============================================================================")
	template_file.close()

if __name__ == '__main__':

	start_timestamp = time.time()

	#Draw the parameters for the sersic object from appropriate distributions
	sersic_idx = np.random.uniform(0.0, 8.0, NUM_ITER)
	half_light_radius = np.random.uniform(1.0,22.0,NUM_ITER)
	axis_ratio = np.random.uniform(0.1,1.0,NUM_ITER)
	position_angle = np.random.uniform(-90.0,90.0,NUM_ITER)
	inte_mag = np.random.uniform(17.0,27.8,NUM_ITER)
	x_pos = np.random.uniform(39.43,43.57,NUM_ITER)
	y_pos = np.random.uniform(39.43,43.57,NUM_ITER)
	#sky_back = np.random.choice(np.genfromtxt(FILE_PATH+"norm_skyval.txt"),NUM_ITER)
	sky_back = np.random.uniform(0.001,0.005,NUM_ITER)

	#store all generated parameter values corresponding to the images in a file
	para_file = open(FILE_PATH+"sim_para.txt","w") ###MAKE THIS .GZ later if size is a problem###
	stacked_para = np.column_stack((sersic_idx,half_light_radius,axis_ratio,position_angle,inte_mag,x_pos,y_pos,sky_back))
	np.savetxt(para_file,stacked_para,delimiter=" ",header="sersic_idx R_e axis_ratio PA Inte_Mag x_pos y_pos sky_back",fmt="%.4f")
	para_file.close()

	print( "Parameters generated. Real time taken:- %s seconds\nCreating Files....." %(time.time() - start_timestamp) )

	pl = Pool(NUM_THREADS)
	pl.map(file_write,range(0,NUM_ITER))  #This just call the function file_write with the input variables being range(0,NUM_ITER)

	print("Finished.Real Time of Execution:- %s seconds" % (time.time() - start_timestamp) )