AutoStacker/Mastercode.py at master · thatoeugine/AutoStacker · 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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import os
import sys
import configparser


#Loading the parameters file:
config = configparser.ConfigParser()
config.read(sys.argv[1])

# working directory
PATH = config.get('pipeline', 'data_path')
os.system('mkdir %s'%PATH)
# source finding container
SOURCE_FINDING_CONTAINER = 'singularity exec /data/exp_soft/containers/sourcefinding.img'
# kern3 container
KERN3_CONTAINER = 'singularity exec /data/exp_soft/containers/kern3.img'
# File names used here
uvstacked_ms = config.get('casa_imager', 'stack_msfile_name')
original_ms = config.get('simulator_params', 'msfile_name')

skymodel= config.get('pipeline', 'project_name')+'_truthcat.txt'
modelimage_ = config.get('field', 'fitsname')
modelimage = modelimage_.replace("-model.fits"," ")
#======================================================================================================
#                                     Simulating with simms+meqtrees+Simuclass
#======================================================================================================
from stack_auto.Stacker import runFoVcut, runImstack, runImageStackData, runStack_clean, runMoveVis, runPlotStackingResults, runPlotStackingImages, runStackingDepth, runStackingDepthProto, runUVStack
from stack_auto.simple_simulate.Simulator import runSimulate, runAddNoise

if config.getboolean('pipeline', 'dosimulate'):
    runSimulate(config, sys.argv[1])
    runAddNoise(config, sys.argv[1])


    if config.getboolean('stacking_params', 'No._of_srcs_of_choice'):
        runStackingDepthProto(config, sys.argv[1])
        os.system(KERN3_CONTAINER+' '+'python simuclass/simuCLASS.py'+' '+sys.argv[1])
        os.system(SOURCE_FINDING_CONTAINER+' '+'python simuclass/Edit_to_Tigger_frendly.py'+' '+PATH+skymodel)

        ####
        #Predicting the visibilities
        ####
        os.system(SOURCE_FINDING_CONTAINER+' '+'wsclean -mem 100 -predict -name'+' '+PATH+modelimage+' '+PATH+original_ms)

    elif config.getboolean('stacking_params', 'stacking_depth'):
        runStackingDepth(config, sys.argv[1])
        os.system(KERN3_CONTAINER+' '+'python simuclass/simuCLASS.py'+' '+sys.argv[1])
        os.system(SOURCE_FINDING_CONTAINER+' '+'python simuclass/Edit_to_Tigger_frendly.py'+' '+PATH+skymodel)

        ####
        #Predicting the visibilities
        ####
        os.system(SOURCE_FINDING_CONTAINER+' '+'wsclean -mem 100 -predict -name'+' '+PATH+modelimage+' '+PATH+original_ms)


    elif config.getboolean('stacking_params', 'full_stack'):
        os.system(KERN3_CONTAINER+' '+'python simuclass/simuCLASS.py'+' '+sys.argv[1])
        os.system(SOURCE_FINDING_CONTAINER+' '+'python simuclass/Edit_to_Tigger_frendly.py'+' '+PATH+skymodel)

        ####
        #Predicting the visibilities
        ####
        os.system(SOURCE_FINDING_CONTAINER+' '+'wsclean -mem 100 -predict -name'+' '+PATH+modelimage+' '+PATH+original_ms)

    elif config.getboolean('stacking_params', 'FOV_size_cut'):
        runFoVcut(config, sys.argv[1])
        os.system(SOURCE_FINDING_CONTAINER+' '+'wsclean -mem 100 -predict -name'+' '+PATH+modelimage+' '+PATH+original_ms)


    runMoveVis(config, sys.argv[1])


#++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
#                              Stacking
#++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
if config.getboolean('pipeline', 'dostacking'):

    if config.getboolean('stacking_params', 'stack_clean'):
        runStack_clean(config, sys.argv[1])

    else:
        pass


            #image stacking:
    runImstack(config, sys.argv[1])

            #vis stacking:
    os.system('CUDA_VISIBLE_DEVICES=2')

    runUVStack(config, sys.argv[1])

    runImageStackData(config, sys.argv[1])

    ####
        #Predicting the visibilities to uv-stacked for comparison with Im-stacked
    ####
    os.system(SOURCE_FINDING_CONTAINER+' '+'wsclean -mem 100 -predict -name'+' '+PATH+'Imstacked'+' '+PATH+uvstacked_ms)

    #Plotting results
    runPlotStackingImages(config, sys.argv[1])
    runPlotStackingResults(config, sys.argv[1])

else:
    'All Tasks Complete'