'savefig.jpeg_quality': 100,
'savefig.pad_inches': 0.1,
'lines.linewidth': 0.75
}
sns.set(context='paper',
style='white',
font='sans-serif',
palette='Set2',
font_scale=7 / 8,
rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
conf = Config('SN')
sf = SF(conf)
conf.TF(sf)
rb = RB(conf, sf, Np=500)
a = (sf.LFPr - sf.HFPr) / 2
R = (sf.LFPr + sf.HFPr) / 2
AR = R / a
r95, z95 = sf.get_boundary(alpha=0.95)
ru = r95[np.argmax(z95)] # triangularity
rl = r95[np.argmin(z95)]
del_u = (R - ru) / a
del_l = (R - rl) / a
kappa = (np.max(z95) - np.min(z95)) / (2 * a)
}
sns.set(context='paper',
style='white',
font='sans-serif',
palette='Set2',
font_scale=7 / 8,
rc=rc)
color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
eqdsk = 'vde'
#eqdsk = 'SN'
sf = SF(Config(eqdsk))
sf.eq['ncoil'] = 0
if eqdsk == 'vde':
eq = EQ(sf, dCoil=1, limit=[4.25, 8, -4.5, 2], n=8e3)
#eq = EQ(sf,dCoil=1,limit=[3,9,-6,6],n=5e3)
else:
eq = EQ(sf, dCoil=1, limit=[5, 13, -5.5, 5], n=1e3)
eq.set_eq_psi()
levels = eq.sf.contour()
psi_o = eq.psi.copy()
psi_flux = np.linspace(0, 1, 500)
from eqConfig import Config
from itertools import cycle
import seaborn as sns
rc = {'figure.figsize':[3.14*12/16,3.14],'savefig.dpi':350, #*12/16
'savefig.jpeg_quality':100,'savefig.pad_inches':0.1,
'lines.linewidth':0.75}
sns.set(context='paper',style='white',font='sans-serif',palette='Set2',
font_scale=7/8,rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
conf = Config('SX')
sf = SF(conf)
'''
zcs_space = np.linspace(6,-8,13)
rcs_space = 2.9*np.ones(len(zcs_space))
pl.plot(rcs_space,zcs_space,'-o',markersize=1.5)
for i in range(5):
del sf.coil['Coil{:1.0f}'.format(i)]
for i in range(len(zcs_space)):
name = 'Coilcs{:1.0f}'.format(i)
sf.coil[name] = {'r':rcs_space[i],'z':zcs_space[i],'dr':1,
'dz':abs(np.diff(zcs_space))[0],'I':1}
'''
fig = pl.figure(figsize=(15, 5))
gd = gridspec.GridSpec(1, 4, wspace=0.0, hspace=0.0)
for i, config in enumerate(['SN', 'X', 'SF', 'SX']):
sns.set_palette('Set2')
ax = pl.Subplot(fig, gd[i])
fig.add_subplot(ax)
fig.sca(ax)
pl.axis('equal')
pl.axis('off')
pl.tight_layout()
#pl.xlim([4,14]),pl.ylim([-13,8])
pl.xlim([3, 17]), pl.ylim([-12, 10])
conf = Config(config, inside=False)
sf = SF(conf)
sf.contour()
sf.plot_coils(Color)
conf.TF(sf)
rb = RB(conf, sf, Np=200)
rb.divertor_outline(True)
rb.fill(dt=conf.tfw, alpha=0.7, color=next(Color))
rb.fill(dt=conf.BB, alpha=0.7, color=next(Color))
rb.fill(dt=conf.tBBsupport, alpha=0.7, color=next(Color))
rb.BBsheild_fill(dt=conf.sheild,
ref_o=2 / 8 * np.pi,
alpha=0.7,
color=next(Color))
rb.fill(dt=conf.VV, alpha=0.7, color=next(Color), loop=True)
rb.set_TFbound() # TF boundary conditions
palette='Set2',
font_scale=7 / 8,
rc=rc)
Color = cycle(sns.color_palette('Set2', 3))
figheight = 8.0
mpl.rcParams['figure.figsize'] = [figheight, figheight * 5 / 9]
xlim, ylim = [8, 12], [-8, -7]
pl.figure()
pl.axis('equal')
pl.axis('off')
pl.xlim(xlim)
pl.ylim(ylim)
conf = Config('SX7')
sf = SF(conf, sample=1)
conf.TF(sf)
rb = RB(conf, sf, Np=1e3)
rb.divertor_outline(True, color=next(Color))
Rt, Zt = rb.targets['outer']['R'], rb.targets['outer']['Z']
Ro, Zo = rb.targets['outer']['Rsol'][-1], rb.targets['outer']['Zsol'][-1]
index = np.argmin((Rt - Ro)**2 + (Zt - Zo)**2)
R, Z = Rt[:index + 1], Zt[:index + 1]
graze = sf.get_graze([Ro, Zo], [R[-2] - R[-1], Z[-2] - Z[-1]])
print(graze * 180 / np.pi, rb.targets['outer']['theta'] * 180 / np.pi)
conf = Config('SX8')
sf = SF(conf, sample=1)
conf.TF(sf)
rb = RB(conf, sf, Np=1e3)
font_scale=1.5,
rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
pkl = PKL('moveSX')
sf, eq, inv = pkl.fetch(['sf', 'eq', 'inv'])
sf.config = 'SXex'
inv.swing_fix(np.mean(inv.Swing))
inv.solve_slsqp()
eq.run(update=False)
conf = Config('SXex')
conf.TF(sf)
sf.conf = conf
rb = RB(conf, sf, Np=500)
rb.divertor_outline(False, plot=True, debug=False)
eq.grid(boundary={'R': rb.Rb, 'Z': rb.Zb}, n=5e3)
eq.set_sf_psi() # set psi
eq.run()
sf.set_plasma({'r': eq.r, 'z': eq.z, 'psi': eq.psi}, contour=True)
inv.plot_fix()
sf.eqwrite()
sf = SF(conf, sample=1, Nova=True)
sf.contour(lw=0.5)
'savefig.pad_inches': 0.1,
'lines.linewidth': 2
}
sns.set(context='talk',
style='white',
font='sans-serif',
palette='Set2',
font_scale=7 / 8,
rc=rc)
pl.figure()
text = linelabel(Ndiv=42, value=' 1.0f', postfix='m', loc='end')
for config in ['SN', 'SFp', 'SFm']: #,'SFm','SFp','X','SX8','SXex']: #
color = next(Color)
conf = Config(config)
sf = SF(conf, sample=1)
rb = RB(conf, sf, Np=250)
rb.divertor_outline(False, plot=False, debug=False)
'''
sf.eq['ncoil'] = 0
eq = EQ(sf,boundary={'R':rb.Rb,'Z':rb.Zb,'expand':0.5},n=1e4) #'expand':0.5,'zmax':sf.Xpoint[1]+sf.rcirc+sf.drcirc}
if 'ex' not in config:
eq.run()
eq.set_eq_psi()
'''
sf.sol(Nsol=21, update=True, plot=False)
rb.trim_sol(plot=False) # trim legs to targets
sf.add_core() # referance low field midplane
pkl = PKL('moveSX_Dev2')
import seaborn as sns
rc = {'figure.figsize':[7*12/16,7],'savefig.dpi':150, #*12/16
'savefig.jpeg_quality':100,'savefig.pad_inches':0.1,
'lines.linewidth':0.75}
sns.set(context='paper',style='white',font='sans-serif',palette='Set2',
font_scale=7/8,rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
conf = Config('SN')
sf = SF(conf)
conf = Config('X')
sf.conf = conf
conf.TF(sf)
#rb = RB(conf,sf,Np=200)
#rb.divertor_outline(False,plot=False,debug=False)
eq = EQ(sf,dCoil=0.5,limit=[5,12,-8,5.5],n=1e4)
eq.set_sf_psi() # set psi
eq.gen()
inv = INV(sf,eq,configTF='X',config='X')
Lpf = inv.grid_PF(nPF=5)
Lcs = inv.grid_CS(nCS=5)
Lo = np.append(Lpf,Lcs)
'savefig.jpeg_quality': 100,
'savefig.pad_inches': 0.1,
'lines.linewidth': 0.75
}
sns.set(context='paper',
style='white',
font='sans-serif',
palette='Set2',
font_scale=7 / 8,
rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
conf = Config('SN')
sf = SF(conf)
pl.plot(sf.rbdry, sf.zbdry)
eq = EQ(sf, dCoil=1, limit=[4, 15.5, -13, 5.5], n=1e4)
eq.get_plasma_coil()
#eq.set_coil_psi()
#eq.get_Xpsi()
#eq.get_Mpsi()
inv = INV(sf, eq, configTF='SN', config='SN')
inv.fix_boundary_psi(N=25, alpha=0.99, factor=1) # add boundary points
inv.add_null(factor=1, point=sf.Xpoint)
inv.add_alpha(1, factor=1, point=sf.Xpoint)
F = np.zeros((inv.nC,nSwing,2))
I = np.zeros((inv.nC,nSwing))
Fcs = np.zeros((nSwing,2))
for j,swing in enumerate(Swing):
inv.fix_flux(swing)
inv.solve_slsqp()
Fcoil = inv.ff.get_force()
F[:,j] = Fcoil['F']
Fcs[j,0] = Fcoil['CS']['sep']
Fcs[j,1] = Fcoil['CS']['zsum']
from eqConfig import Config
conf = Config('SXex')
#conf.TF(sf)
#rb = RB(conf,sf,Np=150)
# animation function
def animate(swing): #,data,ax
Color = cycle(sns.color_palette('Set2'))
pl.sca(ax1)
ax1.cla()
#ax1.set_axis_off()
ax.set_ylim([3,14])
#pl.axis('equal')
'savefig.dpi': 200, #*12/16
'savefig.jpeg_quality': 100,
'savefig.pad_inches': 0.1,
'lines.linewidth': 0.75
}
sns.set(context='paper',
style='white',
font='sans-serif',
palette='Set2',
font_scale=7 / 8,
rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
#pl.figure(figsize=(12,8))
pl.axis('equal')
pl.axis('off')
#pl.xlim([4,14]),pl.ylim([-13,8])
pl.xlim([6, 11]), pl.ylim([-6.5, -4.5])
config = 'SFp' # SN,X,SX,SF,SX2
conf = Config(config, inside=False)
sf = SF(conf, sample=1)
rb = RB(conf, sf, Np=360)
rb.divertor_outline(True, color=next(Color))
config = 'SFm' # SN,X,SX,SF,SX2
conf = Config(config, inside=False)
sf = SF(conf, sample=1)
rb = RB(conf, sf, Np=360)
rb.divertor_outline(True, color=next(Color))
'savefig.pad_inches': 0.1,
'lines.linewidth': 0.75
}
sns.set(context='paper',
style='white',
font='sans-serif',
palette='Set2',
font_scale=7 / 8,
rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
conf = Config('SFm')
sf = SF(conf)
sf.contour(Nstd=1.5, color=next(Color))
eq = EQ([4, 13], [-6, 6], 1e4, sf)
eq.Vtarget = sf.Mpoint[1]
rbdry, zbdry = sf.get_boundary() # update boundary
pl.plot(rbdry, zbdry)
pl.plot(sf.Mpoint[0], sf.Mpoint[1], 'o', markersize=1)
eq.gen(Verr=1e-4)
sf.contour(levels=sf.cs.levels, color=next(Color))
rbdry, zbdry = sf.get_boundary() # update boundary
pl.plot(rbdry, zbdry)
from eqConfig import Config
from itertools import cycle
import seaborn as sns
rc = {'figure.figsize':[3.14*12/16,3.14],'savefig.dpi':400, #*12/16
'savefig.jpeg_quality':100,'savefig.pad_inches':0.1,
'lines.linewidth':0.75}
sns.set(context='paper',style='white',font='sans-serif',palette='Set2',
font_scale=7/8,rc=rc)
Color = cycle(sns.color_palette('Set2'))
pl.figure()
pl.axis('equal')
pl.axis('off')
conf = Config('X')
sf = SF(conf)
rb = RB(conf,sf,Np=150)
eq = EQ([4,12.5],[-9.5,6.5],5e4,sf)
sf.plasma_coils(N=11,dL=0.25)
eq.coils(delta=0.25) # multi-filiment coils
r,z = sf.get_boundary(alpha=0.95)
L = sf.length(r,z)
Lc = np.linspace(0,1,21)[:-1]
fix = {}
fix['r'],fix['z'] = interp1(L,r)(Lc),interp1(L,z)(Lc)
fix['value'] = eq.Pcoil(fix['r'],fix['z'])
fix['BC'] = np.array(['psi']*len(fix['r']))
import numpy as np
from radial_build import RB
from shelf import PKL
from eqConfig import Config
import seaborn as sns
rc = {
'figure.figsize': [6, 6 * 12 / 16],
'savefig.dpi': 125, #*12/16
'savefig.jpeg_quality': 100,
'savefig.pad_inches': 0.1,
'lines.linewidth': 0.75
}
sns.set(context='paper',
style='white',
font='sans-serif',
palette='Set2',
font_scale=1.5,
rc=rc)
Color = cycle(sns.color_palette('Set2'))
conf = Config('SN')
sf = SF(conf, sample=1, Nova=False)
#sf.contour(lw=0.5)
pl.figure()
pl.plot(np.linspace(0, 1, len(sf.eq['pressure'])), sf.eq['pressure'])
pl.figure()
pl.plot(np.linspace(0, 1, len(sf.eq['pressure'])), sf.eq['fpol'])
#sf.eqwrite()
