Пример #1
0
def calc_run_stats(run_id):
    """Calculates all the run statistics for a run for the database."""
    srk_settings, run_settings = srkdata.get_settings_from_database(run_id)

    # Calculate all the statistics related to each orientation
    p_stats = calc_orientation_stats(run_id, True)
    a_stats = calc_orientation_stats(run_id, False)

    # Check if data for both orientations exists
    both_orientations = False
    if len(p_stats) > 0 or len(a_stats) > 0:
        both_orientations = True

    p_stats['DipolePositionBelowChamber'] = get_dist_bottom_from_pos(srk_settings['DipolePosition'],
                                                                     srk_settings['ChamberHeight'])
    pos = [float(x) for x in srk_settings['DipolePosition'].split(' ')]  # Splits up space separates position list
    p_stats['DipolePositionX'] = pos[0]

    # Return here if both orientations
    if not both_orientations:
        return srkdata.merge_dicts(p_stats, a_stats)

    # Calculate stats related to frequency difference between orientaitons
    run_stats = srkdata.default_delta_omega_stats()

    if run_settings['RunType'] == 'deltaOmega':
        # ufloat - floats with uncertainty
        p_phase = ufloat(p_stats['Par_PhiMean'], p_stats['Par_PhiError'])
        a_phase = ufloat(a_stats['Anti_PhiMean'], a_stats['Anti_PhiError'])

        delta_phase = p_phase - a_phase
    elif run_settings['RunType'] == 'deltaOmegaSame':
        dp = calc_delta_stats_same_tracks(run_id)
        delta_phase = ufloat(dp[0], dp[1] / run_settings['NumTracksPer'])
    else:
        return srkdata.merge_dicts(p_stats, a_stats)

    print str(run_id) + ":"
    print "Delta Phase: " + str(delta_phase)

    delta_omega = delta_phase / srk_settings['TimeLimit']
    print "Delta Omega: " + str(delta_omega)

    if srk_settings['E0FieldStrength'] == 0.:
        false_edm = ufloat(0., 0.)
    else:
        false_edm = calc_false_edm(delta_omega, srk_settings['E0FieldStrength'])
    print "False EDM: " + str(false_edm)

    run_stats['DeltaPhase'] = delta_phase.nominal_value
    run_stats['DeltaPhaseError'] = delta_phase.std_dev
    run_stats['DeltaOmega'] = delta_omega.nominal_value
    run_stats['DeltaOmegaError'] = delta_omega.std_dev
    run_stats['FalseEDM'] = false_edm.nominal_value
    run_stats['FalseEDMError'] = false_edm.std_dev

    pr_stats = calc_dipole_predictions_pignol_and_rocia(srk_settings)

    return srkdata.merge_dicts(run_stats, p_stats, a_stats, pr_stats)
Пример #2
0
def calc_orientation_stats(run_id, is_parallel):
    """Calc stats related to an orientation of the E0 and B0 fields (parallel and anti-parallel)"""

    if is_parallel:
        letter = 'P'
        run_type = 'Par'
    else:
        letter = 'A'
        run_type = 'Anti'
    file_path = srkdata.srkglobal.results_dir + "Results_RID" + str(run_id) + "_" + letter + ".root"

    stats = calc_stats_for_results_file(file_path)

    if len(stats) == 0:
        return {}

    stats = srkdata.prefix_dict_keys(stats, run_type + '_')

    srk_settings, run_settings = srkdata.get_settings_from_database(run_id)
    pr_stats = calc_dipole_predictions_pignol_and_rocia(srk_settings)

    return srkdata.merge_dicts(stats, pr_stats)
Пример #3
0
def calc_orientation_stats(run_id, is_parallel):
    """Calc stats related to an orientation of the E0 and B0 fields (parallel and anti-parallel)"""

    if is_parallel:
        letter = 'P'
        run_type = 'Par'
    else:
        letter = 'A'
        run_type = 'Anti'
    file_path = srkdata.srkglobal.results_dir + "Results_RID" + str(
        run_id) + "_" + letter + ".root"

    stats = calc_stats_for_results_file(file_path)

    if len(stats) == 0:
        return {}

    stats = srkdata.prefix_dict_keys(stats, run_type + '_')

    srk_settings, run_settings = srkdata.get_settings_from_database(run_id)
    pr_stats = calc_dipole_predictions_pignol_and_rocia(srk_settings)

    return srkdata.merge_dicts(stats, pr_stats)
Пример #4
0
def calc_run_stats(run_id):
    """Calculates all the run statistics for a nedm-type run for the database."""
    srk_settings, run_settings = srkdata.get_settings_from_database(run_id)

    # Calculate all the statistics related to each orientation
    p_stats = calc_orientation_stats(run_id, True)
    a_stats = calc_orientation_stats(run_id, False)

    # Check if data for both orientations exists
    both_orientations = False
    if len(p_stats) > 0 or len(a_stats) > 0:
        both_orientations = True

    p_stats['DipolePositionBelowChamber'] = get_dist_bottom_from_pos(
        srk_settings['DipolePosition'], srk_settings['ChamberHeight'])
    pos = [float(x) for x in srk_settings['DipolePosition'].split(' ')
           ]  # Splits up space separates position list
    p_stats['DipolePositionX'] = pos[0]

    # Return here if both orientations
    if not both_orientations:
        return srkdata.merge_dicts(p_stats, a_stats)

    # Calculate stats related to frequency difference between orientaitons
    run_stats = srkdata.default_delta_omega_stats()

    if run_settings['RunType'] == 'deltaOmega':
        # ufloat - floats with uncertainty
        p_phase = ufloat(p_stats['Par_PhiMean'], p_stats['Par_PhiError'])
        a_phase = ufloat(a_stats['Anti_PhiMean'], a_stats['Anti_PhiError'])

        delta_phase = p_phase - a_phase
    elif run_settings['RunType'] == 'deltaOmegaSame':
        dp = calc_delta_stats_same_tracks(run_id)
        delta_phase = ufloat(dp[0], dp[1] / run_settings['NumTracksPer'])
    else:
        return srkdata.merge_dicts(p_stats, a_stats)

    print str(run_id) + ":"
    print "Delta Phase: " + str(delta_phase)

    delta_omega = delta_phase / srk_settings['TimeLimit']
    print "Delta Omega: " + str(delta_omega)

    if srk_settings['E0FieldStrength'] == 0.:
        false_edm = ufloat(0., 0.)
    else:
        false_edm = calc_false_edm(delta_omega,
                                   srk_settings['E0FieldStrength'])
    print "False EDM: " + str(false_edm)

    run_stats['DeltaPhase'] = delta_phase.nominal_value
    run_stats['DeltaPhaseError'] = delta_phase.std_dev
    run_stats['DeltaOmega'] = delta_omega.nominal_value
    run_stats['DeltaOmegaError'] = delta_omega.std_dev
    run_stats['FalseEDM'] = false_edm.nominal_value
    run_stats['FalseEDMError'] = false_edm.std_dev

    pr_stats = calc_dipole_predictions_pignol_and_rocia(srk_settings)

    return srkdata.merge_dicts(run_stats, p_stats, a_stats, pr_stats)