@@ -279,14 +279,15 @@ <h1>Source code for omc3.harpy.handler</h1><div class="highlight"><pre>
279279< div class ="viewcode-block " id ="run_per_bunch ">
280280< a class ="viewcode-back " href ="../../../modules/harpy.html#omc3.harpy.handler.run_per_bunch "> [docs]</ a >
281281< span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> run_per_bunch</ span > < span class ="p "> (</ span >
282- < span class ="n "> tbt_data</ span > < span class ="p "> :</ span > < span class ="n "> TbtData</ span > < span class ="p "> ,</ span > < span class ="n "> harpy_input</ span > < span class ="p "> :</ span > < span class ="n "> DotDict</ span > < span class ="p "> ,</ span > < span class ="n "> file </ span > < span class ="p "> :</ span > < span class ="n " > Path </ span >
282+ < span class ="n "> tbt_data</ span > < span class ="p "> :</ span > < span class ="n "> TbtData</ span > < span class ="p "> ,</ span > < span class ="n "> harpy_input</ span > < span class ="p "> :</ span > < span class ="n "> DotDict</ span > < span class ="p "> ,</ span > < span class ="n "> output_filename </ span > < span class ="p "> :</ span > < span class ="nb " > str </ span >
283283< span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="nb "> dict</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span > < span class ="n "> tfs</ span > < span class ="o "> .</ span > < span class ="n "> TfsDataFrame</ span > < span class ="p "> ]:</ span >
284284< span class ="w "> </ span > < span class ="sd "> """</ span >
285285< span class ="sd "> Cleans data, analyses frequencies and searches for resonances.</ span >
286286
287287< span class ="sd "> Args:</ span >
288288< span class ="sd "> tbt_data: single bunch `TbtData`.</ span >
289289< span class ="sd "> harpy_input: Analysis settings taken from the commandline.</ span >
290+ < span class ="sd "> output_filename: Name of the output files (placed in `harpy_input.outputdir`).</ span >
290291
291292< span class ="sd "> Returns:</ span >
292293< span class ="sd "> Dictionary with a `TfsDataFrame` per plane.</ span >
@@ -296,7 +297,7 @@ <h1>Source code for omc3.harpy.handler</h1><div class="highlight"><pre>
296297 < span class ="n "> model</ span > < span class ="o "> =</ span > < span class ="n "> tfs</ span > < span class ="o "> .</ span > < span class ="n "> read</ span > < span class ="p "> (</ span > < span class ="n "> harpy_input</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="p "> ,</ span > < span class ="n "> index</ span > < span class ="o "> =</ span > < span class ="n "> COL_NAME</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> loc</ span > < span class ="p "> [:,</ span > < span class ="n "> COL_S</ span > < span class ="p "> ]</ span >
297298
298299 < span class ="n "> bpm_datas</ span > < span class ="p "> ,</ span > < span class ="n "> usvs</ span > < span class ="p "> ,</ span > < span class ="n "> lins</ span > < span class ="p "> ,</ span > < span class ="n "> bad_bpms</ span > < span class ="o "> =</ span > < span class ="p "> {},</ span > < span class ="p "> {},</ span > < span class ="p "> {},</ span > < span class ="p "> {}</ span >
299- < span class ="n "> output_file_path</ span > < span class ="o "> =</ span > < span class ="n "> harpy_input</ span > < span class ="o "> .</ span > < span class ="n "> outputdir</ span > < span class ="o "> /</ span > < span class ="n "> file </ span > < span class =" o " > . </ span > < span class =" n " > name </ span >
300+ < span class ="n "> output_file_path</ span > < span class ="o "> =</ span > < span class ="n "> harpy_input</ span > < span class ="o "> .</ span > < span class ="n "> outputdir</ span > < span class ="o "> /</ span > < span class ="n "> output_filename </ span >
300301
301302 < span class ="k "> for</ span > < span class ="n "> plane</ span > < span class ="ow "> in</ span > < span class ="n "> PLANES</ span > < span class ="p "> :</ span >
302303 < span class ="n "> bpm_data</ span > < span class ="o "> =</ span > < span class ="n "> _get_cut_tbt_matrix</ span > < span class ="p "> (</ span > < span class ="n "> tbt_data</ span > < span class ="p "> ,</ span > < span class ="n "> harpy_input</ span > < span class ="o "> .</ span > < span class ="n "> turns</ span > < span class ="p "> ,</ span > < span class ="n "> plane</ span > < span class ="p "> )</ span >
0 commit comments