def chromatic_beating(input_files, measure_input, tune_dict):
"""
Main function to compute chromatic optics beating.
Args:
tune_dict:
input_files: `InputFiles` object containing frequency spectra files (linx/y).
measure_input:` OpticsInput` object containing analysis settings.
Returns:
"""
dpps = np.array([dpp_val for dpp_val in set(input_files.dpps("X"))])
if np.max(dpps) - np.min(dpps) == 0.0:
return
for plane in PLANES:
betas = []
for dpp_val in dpps:
dpp_meas_input = deepcopy(measure_input)
dpp_meas_input["dpp"] = dpp_val
phase_dict, out_dfs = phase.calculate(dpp_meas_input, input_files,
tune_dict, plane)
beta_df, _ = beta_from_phase.calculate(dpp_meas_input,
tune_dict, phase_dict,
OrderedDict(), plane)
betas.append(beta_df)
output_df = chromatic.calculate_w_and_phi(betas, dpps, input_files,
measure_input, plane)
tfs.write(os.path.join(measure_input.outputdir,
f"{CHROM_BETA_NAME}{plane.lower()}{EXT}"),
output_df, {},
save_index="NAME")
def measure_optics(input_files, measure_input):
"""
Main function to compute various lattice optics parameters from frequency spectra.
Args:
input_files: `InputFiles` object containing frequency spectra files (linx/y).
measure_input: `OpticsInput` object containing analysis settings.
Returns:
"""
LOGGER.info(f"Calculating optics parameters - code version {VERSION}")
iotools.create_dirs(measure_input.outputdir)
logging_tools.add_module_handler(
logging_tools.file_handler(
os.path.join(measure_input.outputdir, LOG_FILE)))
tune_dict = tune.calculate(measure_input, input_files)
common_header = _get_header(measure_input, tune_dict)
invariants = {}
for plane in PLANES:
phase_dict, out_dfs = phase.calculate(measure_input, input_files,
tune_dict, plane)
phase.write(out_dfs, [common_header, common_header],
measure_input.outputdir, plane)
phase.write_special(measure_input, phase_dict, tune_dict[plane]["QF"],
plane)
if measure_input.only_coupling:
continue
beta_df, beta_header = beta_from_phase.calculate(
measure_input, tune_dict, phase_dict, common_header, plane)
beta_from_phase.write(beta_df, beta_header, measure_input.outputdir,
plane)
ratio = beta_from_amplitude.calculate(measure_input, input_files,
tune_dict, beta_df,
common_header, plane)
invariants[plane] = kick.calculate(measure_input, input_files, ratio,
common_header, plane)
ip_df = interaction_point.betastar_from_phase(measure_input,
phase_dict)
interaction_point.write(ip_df, common_header, measure_input.outputdir,
plane)
dispersion.calculate_orbit(measure_input, input_files, common_header,
plane)
dispersion.calculate_dispersion(measure_input, input_files,
common_header, plane)
if plane == "X":
dispersion.calculate_normalised_dispersion(measure_input,
input_files, beta_df,
common_header)
# coupling.calculate_coupling(measure_input, input_files, phase_dict, tune_dict, common_header)
if 'rdt' in measure_input.nonlinear:
iotools.create_dirs(os.path.join(measure_input.outputdir, "rdt"))
rdt.calculate(measure_input, input_files, tune_dict, invariants,
common_header)
if 'crdt' in measure_input.nonlinear:
iotools.create_dirs(os.path.join(measure_input.outputdir, "crdt"))
crdt.calculate(measure_input, input_files, invariants, common_header)
if measure_input.chromatic_beating:
chromatic_beating(input_files, measure_input, tune_dict)