#setup = Setup('SN')
#sf = SF(setup.filename)
#eq = EQ(sf,pf,sigma=0.2,boundary=rb.get_fw(expand=0.25),n=7.5e4)
#eq.plotj(trim=True)
#pl.plot(sf.rbdry,sf.zbdry,color=0.75*np.ones(3),lw=1.5)
for conf in ['SN', 'X', 'SFm', 'SX', 'SXex']: #
print(conf)
setup = Setup(conf)
sf = SF(setup.filename)
pf = PF(sf.eqdsk)
rb = RB(setup, sf)
rb.firstwall(calc=False, plot=True, debug=False)
rb.vessel()
pf.plot(coils=pf.coil, label=True, plasma=False, current=False)
tf = TF(nTF=18,
shape={
'vessel': rb.loop,
'pf': pf,
'fit': False,
'setup': setup,
'plot': True,
'config': conf,
'coil_type': 'A'
})
tf.fill()
pl.plot(tf.Rcl, tf.Zcl)
coil = {'Rcl': tf.Rcl, 'Zcl': tf.Zcl, 'nTF': tf.nTF, 'Iturn': 1}
} # inner radius
shp.loop.xo['upper'] = {'value': 0.33, 'lb': 0.5, 'ub': 1}
shp.loop.xo['lower'] = {'value': 0.33, 'lb': 0.5, 'ub': 1}
shp.add_bound({'r': rvv, 'z': zvv}, 'internal') # vessel
shp.plot_bounds()
shp.minimise()
#shp.loop.plot()
tf = TF(profile, sf=sf)
tf.fill()
demo = DEMO()
#demo.fill_part('Blanket')
#demo.fill_part('Vessel')
#demo.plot_limiter()
'''
rb.vessel()
rb.trim_sol()
profile = Profile(config['TF'],family='S',part='TF')
shp = Shape(profile,obj='L',nTF=18,eqconf=config['eq'])
#shp.cage.pattern(plot=True)
rvv,zvv = geom.rzSLine(rb.loop.R,rb.loop.Z,20)
rvv,zvv = geom.offset(rvv,zvv,0.2)
rmin = np.min(rvv)
rvv[rvv<=rmin+0.12] = rmin+0.12
shp.add_bound({'r':rvv,'z':zvv},'internal') # vessel
shp.plot_bounds()
'''