casa_inference_batch.py

import sys
import shutil
import os
import subprocess

if (len(sys.argv) != 2):
        print("e.g. python auto_batch.shift.py test_data/data.csv")
        sys.exit(1)

#mData = os.path.basename(os.path.normpath(sys.argv[1]))
mData = sys.argv[1]

models_array = [(0,'llda_models/model_planetary_tr_500it/','llda_train_input/planetary_tr_labelmap.sub'),(2,'llda_models/model_planetary_2_500it/','llda_train_input/planetary_2_labelmap.sub'),(5,'llda_models/model_planetary_full_500it/','llda_train_input/planetary_full_labelmap.sub'), (5,'llda_models/model_planetary_full_expanded_3-0.001_500it','llda_train_input/planetary_full_labelmap.sub')]
#models_array = [(0,'llda_models/model_dark_clouds_tr_500it/','llda_train_input/dark_clouds_tr_labelmap.sub'),(2,'llda_models/model_dark_clouds_2_500it/','llda_train_input/dark_clouds_2_labelmap.sub'),(5,'llda_models/model_dark_clouds_full_500it/','llda_train_input/dark_clouds_full_labelmap.sub'),(5,'llda_models/model_dark_clouds_full_expanded_3-0.001_500it','llda_train_input/dark_clouds_full_labelmap.sub')]
#models_array = [(0,'llda_models/model_diffuse_clouds_tr_500it/','llda_train_input/diffuse_clouds_tr_labelmap.sub'),(2,'llda_models/model_diffuse_clouds_2_500it/','llda_train_input/diffuse_clouds_2_labelmap.sub'),(5,'llda_models/model_diffuse_clouds_full_500it/','llda_train_input/diffuse_clouds_full_labelmap.sub')]
#models_array = [(0,'llda_models/model_comets_tr_500it/','llda_train_input/comets_tr_labelmap.sub'),(2,'llda_models/model_comets_2_500it/','llda_train_input/comets_2_labelmap.sub'),(5,'llda_models/model_comets_full_500it/','llda_train_input/comets_full_labelmap.sub')]
#models_array = [(0,'llda_models/model_extragalactic_tr_500it/','llda_train_input/extragalactic_tr_labelmap.sub'),(2,'llda_models/model_extragalactic_2_500it/','llda_train_input/extragalactic_2_labelmap.sub'),(5,'llda_models/model_extragalactic_full_500it/','llda_train_input/extragalactic_full_labelmap.sub')]
#models_array = [(5,'llda_models/model_dark_clouds_full_expanded_3-0.001_500it/','llda_train_input/dark_clouds_full_labelmap.sub'),(5,'llda_models/model_diffuse_clouds_full_expanded_3-0.001_500it/','llda_train_input/diffuse_clouds_full_labelmap.sub'),(5,'llda_models/model_comets_full_expanded_3-0.001_500it/','llda_train_input/comets_full_labelmap.sub'),(5,'llda_models/model_extragalactic_full_expanded_3-0.001_500it/','llda_train_input/extragalactic_full_labelmap.sub'),]
for channeling, model, labelmap in models_array:
        with open(mData) as f:
                for line in f:
                        tokens = line.strip().split(';')
                        if (len(tokens) < 3):
                                continue
                        #print(tokens)
                        os.system("python auto.shift.py "+model+" "+str(channeling)+" "+tokens[2]+" casa_spectrums/"+tokens[1]+" "+labelmap)

                        data_name = tokens[1].split('.')[0]
                        model_name = model.split('/')[1]
                        os.system("mv output.dat "+data_name+"."+model_name+".output")