import logging
from colorlog import ColoredFormatter

if __name__ == '__main__':
    logger = logging.getLogger('main')
    handler = logging.StreamHandler()
    colored_formatter = ColoredFormatter("%(log_color)s%(asctime)s%(levelname)s:%(name)s:%(message)s")
    handler.setFormatter(colored_formatter)
    logger.addHandler(handler)
    logger.setLevel(logging.DEBUG)

    correlator = Correlator(logger = logger.getChild('correlator'))
    time.sleep(1)
    correlator.set_impulse_filter_len(50)
    correlator.set_impulse_setpoint(3772)
    correlator.add_time_domain_calibration('/home/jgowans/workspace/directionFinder_backend/config/time_domain_calibration_long_sma_cables_only.json')
    correlator.add_cable_length_calibrations('/home/jgowans/workspace/directionFinder_backend/config/cable_length_calibration.json')
    correlator.re_sync()
    correlator.impulse_arm()

    fig = plt.figure()
    axes = [[], []]

    axes[0].append(fig.add_subplot(2, 4, 1))
    axes[0].append(fig.add_subplot(2, 4, 2, sharex=axes[0][0], sharey=axes[0][0]))
    axes[0].append(fig.add_subplot(2, 4, 3, sharex=axes[0][0], sharey=axes[0][0]))
    axes[0].append(fig.add_subplot(2, 4, 4, sharex=axes[0][0], sharey=axes[0][0]))
    axes[0][0].set_ylim([-130, 130])

    axes[1].append(fig.add_subplot(2, 1, 2))
    handler.setFormatter(colored_formatter)
    handler.setLevel(logging.DEBUG)

    root = logging.getLogger()
    root.addHandler(handler)
    root.setLevel(logging.INFO)

    logger = logging.getLogger('main')
    logger.propagate = False
    logger.addHandler(handler)
    logger.setLevel(logging.INFO)

    correlator = Correlator(logger = logger.getChild('correlator'))
    time.sleep(1)
    correlator.set_impulse_filter_len(100)
    correlator.set_impulse_setpoint(100)
    correlator.add_time_domain_calibration('/home/jgowans/workspace/directionFinder_backend/bin/time_domain_calibration.json')
    correlator.re_sync()
    correlator.impulse_arm()

    impulse_levels = []

    for _ in range(1000):
        impulse_levels.append(
            correlator.get_current_impulse_level())
        #time.sleep(0.01)

    logger.info("Mean: {}".format(np.mean(impulse_levels)))
    logger.info("Std dev: {}".format(np.std(impulse_levels)))
    logger.info("Max: {}".format(np.max(impulse_levels)))
if __name__ == '__main__':
    logger = logging.getLogger('main')
    handler = logging.StreamHandler()
    colored_formatter = ColoredFormatter(
        "%(log_color)s%(asctime)s%(levelname)s:%(name)s:%(message)s")
    handler.setFormatter(colored_formatter)
    logger.addHandler(handler)
    logger.setLevel(logging.DEBUG)

    correlator = Correlator(logger=logger.getChild('correlator'))
    time.sleep(1)
    correlator.set_impulse_filter_len(50)
    correlator.set_impulse_setpoint(3772)
    correlator.add_time_domain_calibration(
        '/home/jgowans/workspace/directionFinder_backend/config/time_domain_calibration_long_sma_cables_only.json'
    )
    correlator.add_cable_length_calibrations(
        '/home/jgowans/workspace/directionFinder_backend/config/cable_length_calibration.json'
    )
    correlator.re_sync()
    correlator.impulse_arm()

    fig = plt.figure()
    axes = [[], []]

    axes[0].append(fig.add_subplot(2, 4, 1))
    axes[0].append(
        fig.add_subplot(2, 4, 2, sharex=axes[0][0], sharey=axes[0][0]))
    axes[0].append(
        fig.add_subplot(2, 4, 3, sharex=axes[0][0], sharey=axes[0][0]))
    handler.setLevel(logging.DEBUG)

    root = logging.getLogger()
    root.addHandler(handler)
    root.setLevel(logging.INFO)

    logger = logging.getLogger('main')
    logger.propagate = False
    logger.addHandler(handler)
    logger.setLevel(logging.INFO)

    correlator = Correlator(logger=logger.getChild('correlator'))
    time.sleep(1)
    correlator.set_impulse_filter_len(100)
    correlator.set_impulse_setpoint(100)
    correlator.add_time_domain_calibration(
        '/home/jgowans/workspace/directionFinder_backend/bin/time_domain_calibration.json'
    )
    correlator.re_sync()
    correlator.impulse_arm()

    impulse_levels = []

    for _ in range(1000):
        impulse_levels.append(correlator.get_current_impulse_level())
        #time.sleep(0.01)

    logger.info("Mean: {}".format(np.mean(impulse_levels)))
    logger.info("Std dev: {}".format(np.std(impulse_levels)))
    logger.info("Max: {}".format(np.max(impulse_levels)))