def _prepare_and_get_arguments_list(self, path_to_out_run_dir_root,
dir_name):
path_to_run_input = os.path.join(TestOutput.path_to_input, dir_name)
path_to_run_output = os.path.join(path_to_out_run_dir_root, dir_name)
Utilities.iotools.create_dirs(path_to_run_output)
args_dict = {}
args_dict["--mad"] = madxrunner.get_sys_dependent_path_to_mad_x()
args_dict[
"--bbsource"] = Utilities.iotools.get_absolute_path_to_betabeat_root(
)
args_dict["--path"] = path_to_run_input
args_dict["--save"] = path_to_run_output
args_dict["--twiss"] = os.path.join(path_to_run_input, "model",
"twiss_elements.dat")
# model is presumed to be in subdir model of the run directory
args_list = []
for key in args_dict:
args_list.append(key + " " + args_dict[key])
args_list.append(
self._get_args_from_file_in_run_dir(path_to_run_input))
return args_list
def _run_Segment_by_Segment(self, output_path, sequence):
print "Running Segment by Segment..."
madx_bin_path = madxrunner.get_sys_dependent_path_to_mad_x()
bb_source_path = iotools.get_absolute_path_to_betabeat_root()
sbs_output_path = os.path.join(output_path, "sbs")
twiss_model_path = os.path.join(output_path, "twiss_elements.dat")
ip_segments = ""
for ip in IP_SEGMENTS_TO_RUN:
beam = sequence.replace("LHC", "")
ip_segments += IP_SEGMENT_START + str(
ip) + "." + beam + "," + IP_SEGMENT_END + str(
ip) + "." + beam + ",IP" + str(ip) + ","
ip_segments = ip_segments[:-1] # Remove last comma
self.assertTrue(len(ip_segments) > 0, "Cannot create segment list.")
call_command = sys.executable + " " + os.path.abspath(self._path_to_sbs) + \
" --path=" + output_path + \
" --save=" + sbs_output_path + \
" --twiss=" + twiss_model_path + \
" --mad=" + madx_bin_path + \
" --bbsource=" + bb_source_path + \
" --accel=" + sequence + " --start=" + ip_segments
self._run_outer_process(call_command, "Segment by Segment")
def _run_Segment_by_Segment(self, output_path, sequence):
print "Running Segment by Segment..."
madx_bin_path = madxrunner.get_sys_dependent_path_to_mad_x()
bb_source_path = iotools.get_absolute_path_to_betabeat_root()
sbs_output_path = os.path.join(output_path, "sbs")
twiss_model_path = os.path.join(output_path, "twiss_elements.dat")
ip_segments = ""
for ip in IP_SEGMENTS_TO_RUN:
beam = sequence.replace("LHC", "")
ip_segments += (
IP_SEGMENT_START
+ str(ip)
+ "."
+ beam
+ ","
+ IP_SEGMENT_END
+ str(ip)
+ "."
+ beam
+ ",IP"
+ str(ip)
+ ","
)
ip_segments = ip_segments[:-1] # Remove last comma
self.assertTrue(len(ip_segments) > 0, "Cannot create segment list.")
call_command = (
sys.executable
+ " "
+ os.path.abspath(self._path_to_sbs)
+ " --path="
+ output_path
+ " --save="
+ sbs_output_path
+ " --twiss="
+ twiss_model_path
+ " --mad="
+ madx_bin_path
+ " --bbsource="
+ bb_source_path
+ " --accel="
+ sequence
+ " --start="
+ ip_segments
)
self._run_outer_process(call_command, "Segment by Segment")
def _prepare_and_get_arguments_list(self, path_to_out_run_dir_root, dir_name):
path_to_run_input = os.path.join(TestOutput.path_to_input, dir_name)
path_to_run_output = os.path.join(path_to_out_run_dir_root, dir_name)
Utilities.iotools.create_dirs(path_to_run_output)
args_dict = {}
args_dict["--mad"] = madxrunner.get_sys_dependent_path_to_mad_x()
args_dict["--bbsource"] = Utilities.iotools.get_absolute_path_to_betabeat_root()
args_dict["--path"] = path_to_run_input
args_dict["--save"] = path_to_run_output
args_dict["--twiss"] = os.path.join(path_to_run_input, "model", "twiss_elements.dat")
# model is presumed to be in subdir model of the run directory
args_list = []
for key in args_dict:
args_list.append(key + " " + args_dict[key])
args_list.append(self._get_args_from_file_in_run_dir(path_to_run_input))
return args_list
