# %% 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')
# %%