def parse_args(args=None):
parser = _get_optics_parser()
options, acc_args = parser.parse_known_args(args)
accelerator = manager.get_accel_instance(acc_args)
options.accelerator = accelerator
optics_input = OpticsInput.init_from_options(options)
return optics_input
def _do_analysis(directory, options):
LOGGER.info("Performing analysis...")
if options.analyse == 'sussix':
LOGGER.info(" -> Running drive...")
with silence():
_run_drive(directory, options)
else:
LOGGER.info(" -> Running hole_in_one...")
with silence():
_run_harpy(directory, options)
tbt_path = _get_tbt_path(directory)
twiss_path = _get_twiss_path(directory)
err_def_path = _copy_error_def_file(directory, options)
#shutil.copy(err_def_path)
# with silence():
if options.new:
LOGGER.info(" -> Running measure_optics...")
inputs = measure_optics.InputFiles(tbt_path)
meas_input = optics_input.OpticsInput()
meas_input.outputdir = directory
meas_input.accelerator = manager.get_accel_instance(accel="lhc", lhc_mode="lhc_runII_2016", beam=BEAM, model_dir=directory)
measure_optics.measure_optics(inputs, meas_input)
else:
LOGGER.info(" -> Running GetLLM...")
GetLLM.main(directory, tbt_path, twiss_path, accel="LHCB" + str(BEAM), errordefspath = err_def_path, bpmu="mm")
LOGGER.info("")
def parse_args(args=None):
parser = _get_optics_parser()
options, acc_args = parser.parse_known_args(args)
accelerator = manager.get_accel_instance(acc_args)
options.accelerator = accelerator
optics_input = OpticsInput.init_from_options(options)
return optics_input
def create_instance_and_model(opt, accel_opt):
numeric_level = 0
if (sys.flags.debug):
numeric_level = getattr(logging, "DEBUG", None)
#print "DEBUG Level %d" % numeric_level
ch = logging.StreamHandler(sys.stdout)
formatter = logging.Formatter(
' %(asctime)s %(levelname)s | %(name)s : %(message)s')
ch.setFormatter(formatter)
logging.getLogger().addHandler(ch)
logging.getLogger().setLevel(numeric_level)
else:
numeric_level = getattr(logging, "WARNING", None)
#print "WARNING Level %d" % numeric_level
logging.basicConfig(level=numeric_level) # warning level to stderr
create_dirs(opt.output)
accel_inst = manager.get_accel_instance(accel_opt)
create_model(
accel_inst,
opt.type,
opt.output,
writeto=opt.writeto,
logfile=opt.logfile,
)
def _create_input():
sourcespath = abspath(join(dirname(__file__), "..", "measurements"))
iotools.create_dirs(sourcespath)
accelerator = {"model_dir": join(dirname(__file__), "..", "inputs", "models", "25cm_beam1"),
"accel": "lhc", "lhc_mode": "lhc_runII_2018", "beam": 1}
files_to_load = optics_measurement_test_files(accelerator["model_dir"], sourcespath)
measure_input = optics_input.OpticsInput()
measure_input.outputdir = abspath(join(dirname(__file__), "..", "results"))
measure_input.accelerator = manager.get_accel_instance(accelerator)
input_files = measure_optics.InputFiles(files_to_load)
return input_files, measure_input
def _create_input():
sourcespath = abspath(join(dirname(__file__), "..", "measurements"))
iotools.create_dirs(sourcespath)
accelerator = {
"model_dir":
join(dirname(__file__), "..", "inputs", "models", "25cm_beam1"),
"accel":
"lhc",
"lhc_mode":
"lhc_runII_2018",
"beam":
1
}
files_to_load = optics_measurement_test_files(accelerator["model_dir"],
sourcespath)
measure_input = optics_input.OpticsInput()
measure_input.outputdir = abspath(join(dirname(__file__), "..", "results"))
measure_input.accelerator = manager.get_accel_instance(accelerator)
input_files = measure_optics.InputFiles(files_to_load)
return input_files, measure_input
def parse_args(args=None):
main_parser = _get_main_parser()
main_options, rest = main_parser.parse_known_args(args)
if "optics" in rest:
rest.append('--files={}'.format(main_options.file))
rest.append('--outputdir={}'.format(main_options.outputdir))
rest.append('--model_dir={}'.format(
dirname(abspath(main_options.model))))
subparsers = {
"clean": _get_clean_parser,
"harpy": _get_harpy_parser,
"optics": _get_optics_parser,
}
suboptions = {}
for subparser_name in subparsers.keys():
if subparser_name in rest:
rest.remove(subparser_name)
suboption, rest = subparsers[subparser_name]().parse_known_args(
rest)
suboptions[subparser_name] = suboption
main_input = MainInput.init_from_options(main_options)
clean_input = None
harpy_input = None
optics_input = None
if "clean" in suboptions:
clean_input = CleanInput.init_from_options(suboptions["clean"])
if "harpy" in suboptions:
harpy_input = HarpyInput.init_from_options(suboptions["harpy"])
if "optics" in suboptions:
accelerator = manager.get_accel_instance(rest)
suboptions["optics"].accelerator = accelerator
optics_input = OpticsInput.init_from_options(suboptions["optics"])
python_path = "/afs/cern.ch/work/o/omc/anaconda/bin/python "
return main_input, clean_input, harpy_input, optics_input, (
python_path + " ".join(sys.argv))
def parse_args(args=None):
main_parser = _get_main_parser()
main_options, rest = main_parser.parse_known_args(args)
if "optics" in rest:
rest.append('--files={}'.format(main_options.file))
rest.append('--outputdir={}'.format(main_options.outputdir))
rest.append('--model_dir={}'.format(dirname(abspath(main_options.model))))
subparsers = {
"clean": _get_clean_parser,
"harpy": _get_harpy_parser,
"optics": _get_optics_parser,
}
suboptions = {}
for subparser_name in subparsers.keys():
if subparser_name in rest:
rest.remove(subparser_name)
suboption, rest = subparsers[subparser_name]().parse_known_args(rest)
suboptions[subparser_name] = suboption
main_input = MainInput.init_from_options(main_options)
clean_input = None
harpy_input = None
optics_input = None
if "clean" in suboptions:
clean_input = CleanInput.init_from_options(suboptions["clean"])
if "harpy" in suboptions:
harpy_input = HarpyInput.init_from_options(suboptions["harpy"])
if "optics" in suboptions:
accelerator = manager.get_accel_instance(rest)
suboptions["optics"].accelerator = accelerator
optics_input = OpticsInput.init_from_options(suboptions["optics"])
python_path = "/afs/cern.ch/work/o/omc/anaconda/bin/python "
return main_input, clean_input, harpy_input, optics_input, (python_path + " ".join(sys.argv))
def _do_analysis(directory, options):
LOGGER.info("Performing analysis...")
if options.analyse == 'sussix':
LOGGER.info(" -> Running drive...")
with silence():
_run_drive(directory, options)
else:
LOGGER.info(" -> Running hole_in_one...")
with silence():
_run_harpy(directory, options)
tbt_path = _get_tbt_path(directory)
err_def_path = _copy_error_def_file(directory, options)
#shutil.copy(err_def_path)
LOGGER.info(" -> Running GetLLM...")
# with silence():
accel = manager.get_accel_instance(accel="lhc",
lhc_mode="lhc_runII_2016",
beam=BEAM,
model_dir=directory)
GetLLM.main(accel, directory, directory, tbt_path, bpmu="mm")
LOGGER.info("")