def main():
    print TITLE
    print
    for cam in CAMERAS:
        methods = deepcopy(METHODS)
        if cam == 'nf' and 'gaussian' in METHODS:
            methods.remove('gaussian')
        elif cam == 'ff' and 'tophat' in METHODS:
            methods.remove('tophat')

        base_i = None
        for i, test in enumerate(TESTS):
            if 'baseline' in test:
                base_i = i
                continue       

            print cam, test
            new_object = NEW_OBJECTS or cam + '_obj.pkl' not in os.listdir(FOLDER + test + '/')
            if new_object:
                ambient = 'ambient/'
                dark = 'dark/'
                save_new_object(FOLDER, test, cam, ambient, dark)

            if NEW_DATA or new_object:
                set_new_data(FOLDER + test, cam, methods,
                             fiber_method=FIBER_METHOD,
                             kernel_size=kernel)

        if base_i is not None:
            new_baseline = NEW_BASELINE or cam + '_obj.pkl' not in os.listdir(FOLDER + TESTS[base_i] + '/')
            if new_baseline:
                save_baseline_object(FOLDER + TESTS[base_i], cam,
                                     TESTS[base_i-1],
                                     fiber_method=FIBER_METHOD,
                                     kernel=kernel)

            if NEW_DATA or new_baseline:
                set_new_data(FOLDER + TESTS[base_i], cam, methods,
                             fiber_method=FIBER_METHOD,
                             kernel_size=kernel)

        if 'fft' in methods:
            methods.remove('fft')
            save_fft_plot(FOLDER, TESTS, cam, LABELS, TITLE)

        save_modal_noise_data(FOLDER, TESTS, cam, LABELS, methods, TITLE)
def main():
    print TITLE
    print
    for cam in CAMERAS:
        methods = deepcopy(METHODS)
        if cam == 'nf' and 'gaussian' in METHODS:
            methods.remove('gaussian')
        elif cam == 'ff' and 'tophat' in METHODS:
            methods.remove('tophat')

        kernel = KERNEL
        if cam == 'ff':
            kernel = None

        for i, test in enumerate(TESTS):
            folder = FOLDER + test

            print cam, test
            new_object = NEW_OBJECTS or cam + '_obj.pkl' not in os.listdir(folder)
            if new_object:
                dark_folder = '../dark/'
                ambient_folder = '../ambient_1s/'
                if '1s_new' in folder:
                    ambient_folder = '../ambient_1s_new/'
                if '8s_new' in test:
                    ambient_folder = '../ambient_8s_new/'
                elif '8s' in test:
                    ambient_folder = '../ambient_8s/'
                if '10s' in test:
                    ambient_folder = '../ambient_10s/'
                if '80s' in test:
                    ambient_folder = '../ambient_80s/'
                save_new_object(folder, cam, ambient_folder, dark_folder)

            if NEW_DATA or new_object:
                set_new_data(folder, cam, methods,
                             fiber_method=FIBER_METHOD,
                             kernel_size=kernel)
        if 'fft' in methods:
            methods.remove('fft')
            save_fft_plot(FOLDER, TESTS, cam, LABELS, TITLE)

        save_modal_noise_bar_plot(FOLDER, TESTS, cam, LABELS, methods[0], TITLE)
            if 'baseline' in test:
                base_i = i
                continue

            print cam, test
            new_object = NEW_OBJECTS or cam + '_obj.pkl' not in os.listdir(FOLDER + test + '/')
            if new_object:
                dark_folder = 'dark/'
                ambient_folder = 'ambient_1s/'
                if '8s' in test:
                    ambient_folder = 'ambient_8s/'
                if '10s' in test:
                    ambient_folder = 'ambient_10s/'
                if '80s' in test:
                    ambient_folder = 'ambient_80s/'
                save_new_object(FOLDER, test, cam, ambient_folder)

            if NEW_DATA or new_object:
                set_new_data(FOLDER + test, cam, methods,
                             fiber_method=FIBER_METHOD,
                             kernel_size=KERNEL)

        if base_i is not None:
            new_baseline = NEW_BASELINE or cam + '_obj.pkl' not in os.listdir(FOLDER + TESTS[base_i] + '/')
            if new_baseline:
                save_baseline_object(FOLDER + TESTS[base_i], cam,
                                     TESTS[base_i-1],
                                     fiber_method=FIBER_METHOD,
                                     kernel=KERNEL)

            if NEW_DATA or new_baseline:
        for i, test in enumerate(TESTS):
            if 'baseline_filter' in test:
                base_i = i
                continue  

            print cam, test
            new_object = NEW_OBJECTS or cam + '_obj.pkl' not in os.listdir(FOLDER + test + '/')
            if new_object:
                ambient_folder = 'ambient_1s/'
                if '8s' in test:
                    ambient_folder = 'ambient_8s/'
                if '10s' in test:
                    ambient_folder = 'ambient_10s/'
                if '80s' in test:
                    ambient_folder = 'ambient_80s/'
                save_new_object(FOLDER, test, cam, ambient_folder)

            if NEW_DATA or new_object:
                set_new_data(FOLDER, test, cam, methods, FIBER_METHOD, KERNEL)

        if base_i is not None:
            new_baseline = NEW_BASELINE or cam + '_obj.pkl' not in os.listdir(FOLDER + TESTS[base_i] + '/')
            if new_baseline:
                save_baseline_object(FOLDER, TESTS[base_i], cam, TESTS[base_i-1], FIBER_METHOD)

            if NEW_DATA or new_baseline:
                set_new_data(FOLDER, TESTS[base_i], cam, FIBER_METHOD, KERNEL)
        if 'fft' in methods:
            methods.remove('fft')
            save_fft_plot(FOLDER, TESTS, cam, LABELS, TITLE)
Esempio n. 5
0
        if cam == 'nf' and 'gaussian' in METHODS:
            methods.remove('gaussian')
        elif cam == 'ff' and 'tophat' in METHODS:
            methods.remove('tophat')

        base_i = None
        for i, test in enumerate(TESTS):
            if 'baseline' in test:
                base_i = i
                continue

            print cam, test
            new_object = NEW_OBJECTS or cam + '_obj.pkl' not in os.listdir(
                FOLDER + test + '/')
            if new_object:
                save_new_object(FOLDER, test, cam)

            if NEW_DATA or new_object:
                set_new_data(FOLDER, test, cam, methods, FIBER_METHOD, KERNEL)

        if base_i is not None:
            new_baseline = NEW_BASELINE or cam + '_obj.pkl' not in os.listdir(
                FOLDER + TESTS[base_i] + '/')
            if new_baseline:
                save_baseline_object(FOLDER, TESTS[base_i], cam,
                                     TESTS[base_i - 1], FIBER_METHOD, KERNEL)

            if NEW_DATA or new_baseline:
                set_new_data(FOLDER, TESTS[base_i], cam, methods, FIBER_METHOD,
                             KERNEL)
        elif cam == 'ff' and 'tophat' in METHODS:
            methods.remove('tophat')

        base_i = None
        for i, test in enumerate(TESTS):
            if 'baseline_filter' in test:
                base_i = i
                continue

            print cam, test
            new_object = NEW_OBJECTS or cam + '_obj.pkl' not in os.listdir(
                FOLDER + test + '/')
            if new_object:
                save_new_object(FOLDER,
                                test,
                                cam,
                                ambient_folder=CAL[i] + 'ambient/',
                                dark_folder=CAL[i] + 'dark/')

            if NEW_DATA or new_object:
                set_new_data(FOLDER, test, cam, methods, FIBER_METHOD)

        if base_i is not None:
            new_baseline = NEW_BASELINE or cam + '_obj.pkl' not in os.listdir(
                FOLDER + TESTS[base_i] + '/')
            if new_baseline:
                save_baseline_object(FOLDER, TESTS[base_i], cam,
                                     TESTS[base_i - 1], FIBER_METHOD)

            if NEW_DATA or new_baseline:
                set_new_data(FOLDER, TESTS[base_i], cam, methods, FIBER_METHOD)