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)
                    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:
                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, test, cam, LABELS, TITLE)

        save_modal_noise_data(FOLDER, TESTS, cam, LABELS, methods, TITLE)
                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)

        save_modal_noise_data(FOLDER, TESTS, cam, methods, TITLE)
Example #4
0
                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)

        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)

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

        save_modal_noise_data(FOLDER, test, cam, methods, TITLE)