Example #1
0
            continue
        correlations = {}
        num_channels = 4
        cross_combinations = list(
            itertools.combinations(range(num_channels), 2))
        for comb in cross_combinations:
            filename = "{a}x{b}.npy".format(a=comb[0], b=comb[1])
            with open("{d}/{t}/{f}".format(d=directory,
                                           t=timestamp,
                                           f=filename)) as f:
                signal = np.load(f)
                correlations[comb] = signal
        correlator = FakeCorrelator(signals=correlations)
        correlator.add_cable_length_calibrations(
            '/home/jgowans/workspace/directionFinder_backend/config/cable_length_calibration_actual_array.json'
        )
        correlator.add_frequency_bin_calibrations(
            '/home/jgowans/workspace/directionFinder_backend/config/frequency_domain_calibration_through_chain.json'
        )
        correlator.apply_frequency_domain_calibrations()

        array = AntennaArray.mk_from_config(args.array_geometry_file)

        df = DirectionFinder(correlator, array, args.f_start,
                             logger.getChild('df'))

        df.df_strongest_signal(args.f_start,
                               args.f_stop,
                               directory,
                               t=timestamp)
        os.mkdir(df_raw_dir)

    array = AntennaArray.mk_from_config(args.array_geometry_file)
    correlator = Correlator(logger = logger.getChild('correlator'))
    correlator.set_accumulation_len(args.acc_len)
    correlator.add_cable_length_calibrations('/home/jgowans/workspace/directionFinder_backend/config/cable_length_calibration_actual_array.json')
    correlator.add_frequency_bin_calibrations('/home/jgowans/workspace/directionFinder_backend/config/frequency_domain_calibration_through_chain.json')
    df = DirectionFinder(correlator, array, args.f_start, logger.getChild('df'))

    if args.impulse == True:
        df.set_time()  # go into time mode
        # 100 impulse filter len = 0.5 us
        correlator.set_impulse_filter_len(100)
        correlator.set_impulse_setpoint(args.impulse_setpoint)
        correlator.re_sync()
        time.sleep(0.1)
        correlator.impulse_arm()

    while True:
        if args.impulse == True:
            if df.fetch_impulse() == True:
                correlator.save_time_domain_snapshots(df_raw_dir)
                # not necessary to apply cal as it's done in the correlation routine
                df.df_impulse(df_raw_dir)
        else:
            df.fetch_frequency_crosses()
            correlator.save_frequency_correlations(df_raw_dir)
            correlator.apply_frequency_domain_calibrations()
            df.df_strongest_signal(args.f_start, args.f_stop, df_raw_dir)

    correlator.set_accumulation_len(args.acc_len)
    correlator.add_cable_length_calibrations(
        '/home/jgowans/workspace/directionFinder_backend/config/cable_length_calibration_actual_array.json'
    )
    correlator.add_frequency_bin_calibrations(
        '/home/jgowans/workspace/directionFinder_backend/config/frequency_domain_calibration_through_chain.json'
    )
    df = DirectionFinder(correlator, array, args.f_start,
                         logger.getChild('df'))

    if args.impulse == True:
        df.set_time()  # go into time mode
        # 100 impulse filter len = 0.5 us
        correlator.set_impulse_filter_len(100)
        correlator.set_impulse_setpoint(args.impulse_setpoint)
        correlator.re_sync()
        time.sleep(0.1)
        correlator.impulse_arm()

    while True:
        if args.impulse == True:
            if df.fetch_impulse() == True:
                correlator.save_time_domain_snapshots(df_raw_dir)
                # not necessary to apply cal as it's done in the correlation routine
                df.df_impulse(df_raw_dir)
        else:
            df.fetch_frequency_crosses()
            correlator.save_frequency_correlations(df_raw_dir)
            correlator.apply_frequency_domain_calibrations()
            df.df_strongest_signal(args.f_start, args.f_stop, df_raw_dir)
    parser.add_argument('--array_geometry_file', default=None)
    args = parser.parse_args()

    directory = args.d

    contents = os.listdir(directory)
    contents.sort()
    for timestamp_str in contents:
        try:
            timestamp = float(timestamp_str)
        except ValueError:
            continue
        correlations = {}
        num_channels = 4
        cross_combinations = list(itertools.combinations(range(num_channels), 2))
        for comb in cross_combinations:
            filename = "{a}x{b}.npy".format(a = comb[0], b = comb[1])
            with open("{d}/{t}/{f}".format(d = directory, t = timestamp, f = filename)) as f:
                signal = np.load(f)
                correlations[comb] = signal
        correlator = FakeCorrelator(signals = correlations)
        correlator.add_cable_length_calibrations('/home/jgowans/workspace/directionFinder_backend/config/cable_length_calibration_actual_array.json')
        correlator.add_frequency_bin_calibrations('/home/jgowans/workspace/directionFinder_backend/config/frequency_domain_calibration_through_chain.json')
        correlator.apply_frequency_domain_calibrations()

        array = AntennaArray.mk_from_config(args.array_geometry_file)

        df = DirectionFinder(correlator, array, args.f_start, logger.getChild('df'))

        df.df_strongest_signal(args.f_start, args.f_stop, directory, t = timestamp)