# %% Make markdown import madxp try: madxp.madx2md('macros_bb.madx', 'macros_bb.md') except FileNotFoundError: import os os.chdir('/afs/cern.ch/work/s/sterbini/madxp/examples/macros_bb') madxp.madx2md('macros_bb.madx', 'macros_bb.md') # %%
madxp.madxp('input_thin.madx', pythonData=pythonData) except FileNotFoundError: import os os.chdir('/afs/cern.ch/work/s/sterbini/madxp/examples/run3') madxp.madxp('input_thin.madx', pythonData=pythonData) myList.append(copy.deepcopy(pythonData)) # to append the pythonData dictionary gc.collect() # to free memory # %% myDF=pd.DataFrame(myList) myDF['q1 of B1']=myDF['B1summ'].apply(lambda x: x['q1'].values[0]) myDF['q2 of B1']=myDF['B1summ'].apply(lambda x: x['q2'].values[0]) myDF['q1 of B2']=myDF['B2summ'].apply(lambda x: x['q1'].values[0]) myDF['q2 of B2']=myDF['B2summ'].apply(lambda x: x['q2'].values[0]) # %% aux=myDF[myDF['makedipedgevalue']==0] plt.plot(aux['slicefactor'], aux['q1 of B1']-62, 'o-b', label='Q1, makedipedge=False') plt.plot(aux['slicefactor'], aux['q2 of B1']-60, 's-r', label='Q2, makedipedge=False') aux=myDF[myDF['makedipedgevalue']==1] plt.plot(aux['slicefactor'], aux['q1 of B1']-62, 'o:b', label='Q1, makedipedge=True') plt.plot(aux['slicefactor'], aux['q2 of B1']-60, 's:r', label='Q2, makedipedge=True') plt.plot(aux['slicefactor'],.31+0*aux['slicefactor'],'b-.', label='Q1 reference') plt.plot(aux['slicefactor'],.32+0*aux['slicefactor'],'r-.', label='Q2 reference') plt.xlabel('slicefactor') plt.ylabel('fractional tunes') plt.grid(True) plt.legend(loc='best') # %% madxp.madx2md('input_thin.madx','input_thin.md') # %%
import madxp try: madxp.madxp('mask.madx', verbose=False) except: import os os.chdir('/afs/cern.ch/work/s/sterbini/madxp/examples/maskExample' ) # %% Read the output.pkl madxp.madxp('mask.madx', verbose=False) # %% import pandas as pd myDF = pd.read_pickle( '/afs/cern.ch/work/s/sterbini/madxp/examples/maskExample/output.pkl') myDF # %% Make markdown madxp.madx2md('mask.madx', 'mask.md') # %% plt.figure(figsize=[20, 10]) plt.plot(myDF['on_x1']) plt.xticks(rotation='vertical') plt.ylabel('on_x1') # %% plt.figure(figsize=[20, 10]) plt.semilogy(myDF['execution time [s]']) plt.xticks(rotation='vertical') plt.ylabel('execution time [s]') # %%
# %% Run it basically pythonData = {} pythonData = {'quadrupole_length': 6} import madxp try: madxp.madxp('input.madx', pythonData=pythonData) except FileNotFoundError: import os os.chdir('/afs/cern.ch/work/s/sterbini/madxp/examples/simpleFodo') madxp.madxp('input.madx') # %% Read the output.pkl import pandas as pd myDF = pd.read_pickle('output.pkl') myDF # %% Make markdown madxp.madx2md('input.madx', 'input.md') # %%