def test_transverse_output(self):
print('\nTesting transverse plot, scale log, symlog')
read_cfg = pytlwall.CfgIo('input/one_layer.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZTrans()
savedir = 'output/one_layer/img/'
savename = 'ZTransReal.png'
imped_type = "T"
f = mywall.f
list_Z = [
mywall.ZDipX.real, mywall.ZDipY.real, mywall.ZQuadX.real,
mywall.ZQuadY.real
]
title = 'Transverse impedance Real'
list_label = [
'Dipolar X', 'Dipolar Y', 'Quadrupolar X', 'Quadrupolar Y'
]
plot.plot_list_Z_vs_f(f, list_Z, list_label, 'T', title, savedir,
savename, 'log', 'symlog')
savename = 'ZTransImag.png'
list_Z = [
mywall.ZDipX.imag, mywall.ZDipY.imag, mywall.ZQuadX.imag,
mywall.ZQuadY.imag
]
list_label = [
'Dipolar X', 'Dipolar Y', 'Quadrupolar X', 'Quadrupolar Y'
]
title = 'Transverse impedance Imaginary Part'
plot.plot_list_Z_vs_f(f, list_Z, list_label, 'T', title, savedir,
savename, 'log', 'symlog')
def test_onelayer(self):
print('\nTesting one layer cfg')
read_cfg = pytlwall.CfgIo()
chamber = read_cfg.read_chamber('input/one_layer.cfg')
self.assertEqual('prova', chamber.component_name)
self.assertEqual(2e-2, chamber.pipe_hor_m)
self.assertEqual('CW', chamber.layers[0].layer_type)
self.assertEqual('PEC', chamber.layers[1].layer_type)
def test_longitudinal_output(self):
print('\nTesting longitudinal output, text')
read_cfg = pytlwall.CfgIo('input/one_layer.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
filedir = 'output/'
filename = 'one_layerZLong.txt'
label = "longitudinal impedance"
txt.save_ZLong(filedir, filename, mywall.f, mywall.ZLong, label)
def test_transverse_output(self):
print('\nTesting transverse output, text')
read_cfg = pytlwall.CfgIo('input/one_layer.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZTrans()
filedir = 'output/'
filename = 'one_layerZTrans.txt'
label = "transverse impedance"
txt.save_ZTrans(filedir, filename, mywall.f, mywall.ZTrans, label)
def test_all_transverse_output(self):
print('\nTesting all the transverse output, text')
read_cfg = pytlwall.CfgIo('input/test001.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZTrans()
filedir = 'output/'
filename = 'test001ZAllTrans.txt'
label = "transverse impedance"
txt.save_ZAllTrans(filedir, filename, mywall.f, mywall.ZDipX,
mywall.ZDipY, mywall.ZQuadX, mywall.ZQuadY, label)
def test_onelayer_circ(self):
print('\nTesting one layer circular relativ')
read_cfg = pytlwall.CfgIo('input/test001.cfg')
# plot all save all
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
mywall.calc_ZTrans()
savedir = 'output/test001/'
savename = 'ZLong.txt'
label = 'Longitudinal'
txt.save_ZLong(savedir, savename, mywall.f, mywall.ZLong, label)
savename = 'ZTrans.txt'
label = 'Transversal'
txt.save_ZTrans(savedir, savename, mywall.f, mywall.ZTrans, label)
savedir = 'output/test001/img/'
savename = 'ZLong.png'
imped_type = "L"
title = 'Longitudinal impedance'
plot.plot_Z_vs_f_simple(mywall.f,
mywall.ZLong,
imped_type,
title,
savedir,
savename,
xscale='log',
yscale='log')
savename = 'ZTransReal.png'
imped_type = "T"
list_Z = [
mywall.ZDipX.real, mywall.ZDipY.real, mywall.ZQuadX.real,
mywall.ZQuadY.real
]
title = 'Transverse impedance Real'
list_label = [
'Dipolar X', 'Dipolar Y', 'Quadrupolar X', 'Quadrupolar Y'
]
plot.plot_list_Z_vs_f(mywall.f, list_Z, list_label, 'T', title,
savedir, savename, 'log', 'symlog')
savename = 'ZTransImag.png'
list_Z = [
mywall.ZDipX.imag, mywall.ZDipY.imag, mywall.ZQuadX.imag,
mywall.ZQuadY.imag
]
list_label = [
'Dipolar X', 'Dipolar Y', 'Quadrupolar X', 'Quadrupolar Y'
]
title = 'Transverse impedance Imaginary Part'
plot.plot_list_Z_vs_f(mywall.f, list_Z, list_label, 'T', title,
savedir, savename, 'log', 'symlog')
def test_longitudinal_output(self):
print('\nTesting longitudinal simple plot, scale log')
read_cfg = pytlwall.CfgIo('input/one_layer.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
savedir = 'output/one_layer/img/'
savename = 'ZLong.png'
imped_type = "L"
title = 'Longitudinal impedance'
plot.plot_Z_vs_f_simple(mywall.f,
mywall.ZLong,
imped_type,
title,
savedir,
savename,
xscale='log',
yscale='log')
def test_onelayer_circSurf(self):
print('\nTesting one layer circular relativ surface impedance')
read_cfg = pytlwall.CfgIo('input/test001.cfg')
# plot all save all
mywall = read_cfg.read_pytlwall()
ZLongSurf = mywall.ZLongSurf
ZTransSurf = mywall.ZTransSurf
savedir = 'output/test001/'
savename = 'ZLongSurf.txt'
label = 'Longitudinal'
txt.save_ZLong(savedir, savename, mywall.f, ZLongSurf, label)
savename = 'ZTransSurf.txt'
label = 'Transversal'
txt.save_ZTrans(savedir, savename, mywall.f, ZTransSurf, label)
savedir = 'output/test001/img/'
savename = 'ZLongSurf.png'
imped_type = "L"
title = 'Longitudinal impedance'
plot.plot_Z_vs_f_simple(mywall.f,
ZLongSurf,
imped_type,
title,
savedir,
savename,
xscale='log',
yscale='log')
savename = 'ZTransSurf.png'
imped_type = "T"
plot.plot_Z_vs_f_simple(mywall.f,
ZTransSurf,
imped_type,
title,
savedir,
savename,
xscale='log',
yscale='log')
import pandas as pd
import pytlwall
import pytlwall.plot_util as plot
read_cfg = pytlwall.CfgIo('ex_lowbeta/ex_lowbeta.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
mywall.calc_ZTrans()
# save all the data in a dataframe
savedir = 'ex_lowbeta/output/'
data = {
'f': mywall.f,
'ZLong real': mywall.ZLong.real,
'ZLong imag': mywall.ZLong.imag,
'ZTrans real': mywall.ZTrans.real,
'ZTrans imag': mywall.ZTrans.imag
}
df = pd.DataFrame(data)
df.to_excel(savedir + 'output.xlsx')
# plot the impedances
savedir = 'ex_lowbeta/img/'
savename = 'ZLong.png'
title = 'Longitudinal impedance'
ZLong = mywall.ZLong + mywall.ZLongISC
plot.plot_Z_vs_f_simple(mywall.f,
ZLong,
'L',
title,
savedir,
def test_pytlwall(self):
print('\nTesting pytlwall')
read_cfg = pytlwall.CfgIo('input/one_layer.cfg')
mywall = read_cfg.read_pytlwall()
self.assertEqual(0.01, mywall.beam.test_beam_shift)
self.assertEqual(1.e5, round(mywall.beam.gammarel, 3))
def test_beam(self):
print('\nTesting beam')
read_cfg = pytlwall.CfgIo()
beam = read_cfg.read_beam('input/one_layer.cfg')
self.assertEqual(0.01, beam.test_beam_shift)
self.assertEqual(1e5, round(beam.gammarel, 3))
def test_frequency(self):
print('\nTesting frequency')
read_cfg = pytlwall.CfgIo()
freq = read_cfg.read_freq('input/one_layer.cfg')
self.assertEqual(2, freq.fstep)
import pandas as pd
import pytlwall
import pytlwall.plot_util as plot
read_cfg = pytlwall.CfgIo('ex_PEC/ex_PEC.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
mywall.calc_ZTrans()
# save all the data in a dataframe
savedir = 'ex_PEC/output/'
data = {
'f': mywall.f,
'ZLong real': mywall.ZLong.real,
'ZLong imag': mywall.ZLong.imag,
'ZTrans real': mywall.ZTrans.real,
'ZTrans imag': mywall.ZTrans.imag
}
df = pd.DataFrame(data)
df.to_excel(savedir + 'output.xlsx')
# plot the impedances
savedir = 'ex_PEC/img/'
savename = 'ZLong.png'
title = 'Longitudinal impedance'
plot.plot_Z_vs_f_simple(mywall.f,
mywall.ZLong,
'L',
title,
savedir,
savename,
import pandas as pd
import pytlwall
import pytlwall.plot_util as plot
read_cfg = pytlwall.CfgIo('ex_Vac/ex_Vac.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
mywall.calc_ZTrans()
# save all the data in a dataframe
savedir = 'ex_Vac/output/'
data = {
'f': mywall.f,
'ZLong real': mywall.ZLong.real,
'ZLong imag': mywall.ZLong.imag,
'ZTrans real': mywall.ZTrans.real,
'ZTrans imag': mywall.ZTrans.imag
}
df = pd.DataFrame(data)
df.to_excel(savedir + 'output.xlsx')
# plot the impedances
savedir = 'ex_Vac/img/'
savename = 'ZLong.png'
title = 'Longitudinal impedance'
plot.plot_Z_vs_f_simple(mywall.f,
mywall.ZLong,
'L',
title,
savedir,
savename,
configurator file or interactively taking the information from the user
'''
import sys
import pytlwall
import pytlwall.shell_interface as shell
if len(sys.argv) == 1:
shell.welcome_message()
shell.help_pytlwall()
else:
param = sys.argv[1]
if param == '-a':
filename = sys.argv[2]
read_cfg = pytlwall.CfgIo(filename)
read_cfg.read_output()
read_cfg.calc_wall()
read_cfg.print_wall()
read_cfg.plot_wall()
elif param == '-i':
choice = ''
shell.welcome_message()
myconfig = pytlwall.CfgIo()
while choice.upper() != 'X' and choice.lower() != 'exit':
if ('chamber' in locals() and 'beam' in locals()
and 'freq' in locals() and 'wall' not in locals()):
if len(chamber.layers) > 0:
defined_wall = True
wall = pytlwall.TlWall(chamber, beam, freq)
shell.menu0_pytlwall()
import pandas as pd
import pytlwall
import pytlwall.plot_util as plot
read_cfg = pytlwall.CfgIo('ex_surface_impedance/ex_surface_impedance.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
mywall.calc_ZTrans()
# save all the data in a dataframe
savedir = 'ex_surface_impedance/output/'
data = {
'f': mywall.f,
'ZLong real': mywall.ZLong.real,
'ZLong imag': mywall.ZLong.imag,
'ZTrans real': mywall.ZTrans.real,
'ZTrans imag': mywall.ZTrans.imag,
'ZLong Equivalent Surface real': mywall.ZLongSurf.real,
'ZLong Equivalent Surface imag': mywall.ZLongSurf.imag,
'ZTrans Equivalent Surface real': mywall.ZTransSurf.real,
'ZTrans Equivalent Surface imag': mywall.ZTransSurf.imag
}
df = pd.DataFrame(data)
df.to_excel(savedir + 'output.xlsx')
# plot the impedances
savedir = 'ex_surface_impedance/img/'
savename = 'ZLongSurf.png'
title = 'Eq. Long. Surface imp.'
plot.plot_Z_vs_f_simple(mywall.f,
mywall.ZLongSurf,
import pandas as pd
import pytlwall
import pytlwall.plot_util as plot
import pytlwall.txt_util as txt
read_cfg = pytlwall.CfgIo(
'ex_surface_impedance_input/ex_surface_impedance_input.cfg')
mywall = read_cfg.read_pytlwall()
filename = 'ex_surface_impedance_input/surface_impedance_input.txt'
new_f, new_KZ = txt.read_surface_impedance_txt(filename, skipped_rows=1)
mywall.chamber.layers[0].set_surf_imped(new_f, new_KZ)
mywall.calc_ZLong()
mywall.calc_ZTrans()
# save all the data in a dataframe
savedir = 'ex_surface_impedance_input/output/'
data = {
'f': mywall.f,
'ZLong real': mywall.ZLong.real,
'ZLong imag': mywall.ZLong.imag,
'ZTrans real': mywall.ZTrans.real,
'ZTrans imag': mywall.ZTrans.imag,
}
df = pd.DataFrame(data)
df.to_excel(savedir + 'output.xlsx')
# plot the impedances
savedir = 'ex_surface_impedance_input/img/'
savename = 'ZLong.png'
title = 'Long. impedance'
import pandas as pd
import pytlwall
import pytlwall.plot_util as plot
read_cfg = pytlwall.CfgIo('ex_CW/ex_CW.cfg')
mywall = read_cfg.read_pytlwall()
mywall.calc_ZLong()
mywall.calc_ZTrans()
# save all the data in a dataframe
savedir = 'ex_CW/output/'
data = {'f': mywall.f,
'ZLong real': mywall.ZLong.real, 'ZLong imag': mywall.ZLong.imag,
'ZTrans real': mywall.ZTrans.real, 'ZTrans imag': mywall.ZTrans.imag}
df = pd.DataFrame(data)
df.to_excel(savedir + 'output.xlsx')
# plot the impedances
savedir = 'ex_CW/img/'
savename = 'ZLong.png'
title = 'Longitudinal impedance'
plot.plot_Z_vs_f_simple(mywall.f, mywall.ZLong, 'L', title,
savedir, savename,
xscale='log', yscale='log')
savename = 'ZTrans.png'
title = 'Transversal impedance'
plot.plot_Z_vs_f_simple(mywall.f, mywall.ZTrans, 'T', title,
savedir, savename,
xscale='log', yscale='log')
