LHC example with errors

examples/run3_injection_no_bb_with_apertures_and_errors/000_build_collider_from_mad_model.py

@@ -0,0 +1,48 @@
+from cpymad.madx import Madx
+import xtrack as xt
+
+import xmask as xm
+import xmask.lhc as xmlhc
+
+# Import user-defined optics-specific tools
+import optics_specific_tools_run3 as ost
+
+# Read config file
+with open('config.yaml','r') as fid:
+    config = xm.yaml.load(fid)
+config_mad_model = config['config_mad']
+
+# Make mad environment
+xm.make_mad_environment(links=config_mad_model['links'])
+
+# Start mad
+mad_b1b2 = Madx(command_log="mad_collider.log")
+mad_b4 = Madx(command_log="mad_b4.log")
+
+# Build sequences
+ost.build_sequence(mad_b1b2, mylhcbeam=1)
+ost.build_sequence(mad_b4, mylhcbeam=4)
+
+# Apply optics (only for b1b2, b4 will be generated from b1b2)
+ost.apply_optics(mad_b1b2, optics_file=config_mad_model['optics_file'])
+
+# Build xsuite collider
+collider = xmlhc.build_xsuite_collider(
+    sequence_b1=mad_b1b2.sequence.lhcb1,
+    sequence_b2=mad_b1b2.sequence.lhcb2,
+    sequence_b4=mad_b4.sequence.lhcb2,
+    beam_config=config_mad_model['beam_config'],
+    install_apertures=config_mad_model['install_apertures'],
+    enable_imperfections=config_mad_model['enable_imperfections'],
+    enable_knob_synthesis=config_mad_model['enable_knob_synthesis'],
+    pars_for_imperfections={'par_myseed': 6, 'par_on_errors_LHC': 1},
+    ver_lhc_run=1.,
+    ver_hllhc_optics=None,
+    call_after_last_use=ost.call_after_last_use)
+
+# Save to file
+collider.to_json('collider_00_from_mad.json')
+
+
+
+

examples/run3_injection_no_bb_with_apertures_and_errors/002_knobs_and_tuning.py

@@ -0,0 +1,41 @@
+import xtrack as xt
+import xmask as xm
+
+# Load collider anf build trackers
+collider = xt.Environment.from_json('collider_00_from_mad.json')
+collider.build_trackers()
+
+# Read knobs and tuning settings from config file
+with open('config.yaml','r') as fid:
+    config = xm.yaml.load(fid)
+conf_knobs_and_tuning = config['config_knobs_and_tuning']
+
+# Set all knobs (crossing angles, dispersion correction, rf, crab cavities,
+# experimental magnets, etc.)
+for kk, vv in conf_knobs_and_tuning['knob_settings'].items():
+    collider.vars[kk] = vv
+
+# Tunings
+for line_name in ['lhcb1', 'lhcb2']:
+
+    knob_names = {kk: vv.lower() for kk, vv in conf_knobs_and_tuning['knob_names'][line_name].items()}
+
+    targets = {
+        'qx': conf_knobs_and_tuning['qx'][line_name],
+        'qy': conf_knobs_and_tuning['qy'][line_name],
+        'dqx': conf_knobs_and_tuning['dqx'][line_name],
+        'dqy': conf_knobs_and_tuning['dqy'][line_name],
+    }
+
+    xm.machine_tuning(line=collider[line_name],
+        enable_closed_orbit_correction=True,
+        enable_linear_coupling_correction=False,
+        enable_tune_correction=True,
+        enable_chromaticity_correction=True,
+        knob_names=knob_names,
+        targets=targets,
+        line_co_ref=collider[line_name+'_co_ref'],
+        co_corr_config=conf_knobs_and_tuning['closed_orbit_correction'][line_name],
+        verbose=True)
+
+collider.to_json('collider_02_tuned_bb_off.json')