marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

Removed hard coded user specific data paths

Browse Source
This commit is contained in:
Kaan Cökerim 2021-08-04 17:05:43 +02:00
parent 4fd27994f6
commit e850e5d4fe
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pylot/core/util/generate_array_maps.py
View File

@ -15,6 +15,7 @@ from pylot.core.util.dataprocessing import Metadata
from pylot.core.util.array_map import Array_map
import matplotlib.pyplot as plt
import argparse
def main(project_file_path, manual=False, auto=True, file_format='png', f_ext='', ncores=None):
@ -75,10 +76,14 @@ def array_map_worker(input_dict):
if __name__ == '__main__':
dataroot = '/home/kaan/master_thesis/waveformData/dmt_dir_proj'
infiles = ['mag_8_sym_proj.plp']
cl = argparse.ArgumentParser()
cl.add_argument('--dataroot', help='Directory containing the PyLoT .plp file', type=str)
cl.add_argument('--infiles', help='.plp files to use', nargs='+')
args = cl.parse_args()
#dataroot = '/home/kaan/master_thesis/waveformData/dmt_dir_proj'
#infiles = ['mag_8_sym_proj.plp']
for infile in infiles:
main(os.path.join(dataroot, infile), f_ext='_correlated_0.03-0.1', ncores=1)
for infile in args.infiles:
main(os.path.join(args.dataroot, infile), f_ext='_correlated_0.03-0.1', ncores=1)
# main('E:\Shared\AlpArray\\test_aa.plp', f_ext='_correlated_0.5Hz', ncores=1)
# main('/home/marcel/alparray_m6.5-6.9_mantle_correlated_v3.plp', f_ext='_correlated_0.5Hz')