add_noise.py

from modules.pymcfost import pymcfost
#from multiprocessing import Pool
import os, shutil

# Define the directory
directory = '../Output/S2_Final/Moment/'
filename = 'pseudo_casa.fits'

# Define the parameters of the simulated data model
Delta_v = 0.1
iTrans = 0
bmaj = 0.15
bmin = 0.15
bpa = -38
convulations = 2

# Create a list model using the Moment data
model = pymcfost.Line('../Output/S2_Final/Moment/')
model.P.mol.nv = model.nv/convulations

# Define CPUS
#num_cpu = 4 # int(os.environ['SLURM_NTASKS'])*int(os.environ['SLURM_CPUS_PER_TASK'])
#pool = Pool(num_cpu)


# These arguments can be added online with a forked branch from pymcfost
pymcfost.pseudo_CASA_simdata(model, Delta_v=Delta_v, iTrans=iTrans, bmaj=bmaj, bmin=bmin, bpa=bpa, subtract_cont=True, SNR=5)

# Move the file from the directory
try: os.remove(directory + filename)
except: pass
shutil.move("CASA/" + filename, directory)
os.rmdir("CASA")

# Print the complete information
print("Noise Data Complete on Line Data. Moved to %s%s." % (directory, filename))