def construct_patches(self):
"""
"""
try:
visual = self.settings['DEBUG']['visual']['patch_map']
except KeyError:
visual = False
try:
verbose = self.settings['DEBUG']['verbose']['patch_generation']
except KeyError:
verbose = False
try:
magx_tilt_1 = self.settings['grid_settings']['patch_generation'][
'magx_tilt_1']
except KeyError:
magx_tilt_1 = 0.0
try:
magx_tilt_2 = self.settings['grid_settings']['patch_generation'][
'magx_tilt_2']
except KeyError:
magx_tilt_2 = 0.0
xpt1 = self.LineTracer.NSEW_lookup['xpt1']['coor']
xpt2 = self.LineTracer.NSEW_lookup['xpt2']['coor']
magx = np.array([
self.settings['grid_settings']['rmagx'] +
self.settings['grid_settings']['patch_generation']['rmagx_shift'],
self.settings['grid_settings']['zmagx'] +
self.settings['grid_settings']['patch_generation']['zmagx_shift']
])
psi_1 = self.settings['grid_settings']['psi_1']
psi_2 = self.settings['grid_settings']['psi_2']
psi_core = self.settings['grid_settings']['psi_core']
psi_pf_1 = self.settings['grid_settings']['psi_pf_1']
psi_pf_2 = self.settings['grid_settings']['psi_pf_2']
if self.settings['grid_settings']['patch_generation'][
'strike_pt_loc'] == 'limiter':
WestPlate1 = self.parent.LimiterData.copy()
WestPlate2 = self.parent.LimiterData.copy()
EastPlate1 = self.parent.LimiterData.copy()
EastPlate2 = self.parent.LimiterData.copy()
else:
WestPlate1 = self.PlateData['plate_W1']
WestPlate2 = self.PlateData['plate_W2']
EastPlate1 = self.PlateData['plate_E1']
EastPlate2 = self.PlateData['plate_E2']
# Generate Horizontal Mid-Plane lines
LHS_Point = Point(magx[0] - 1e6 * np.cos(magx_tilt_1),
magx[1] - 1e6 * np.sin(magx_tilt_1))
RHS_Point = Point(magx[0] + 1e6 * np.cos(magx_tilt_1),
magx[1] + 1e6 * np.sin(magx_tilt_1))
midline_1 = Line([LHS_Point, RHS_Point])
LHS_Point = Point(magx[0] - 1e6 * np.cos(magx_tilt_2),
magx[1] - 1e6 * np.sin(magx_tilt_2))
RHS_Point = Point(magx[0] + 1e6 * np.cos(magx_tilt_2),
magx[1] + 1e6 * np.sin(magx_tilt_2))
midline_2 = Line([LHS_Point, RHS_Point])
# Generate Vertical Mid-Plane line
Lower_Point = Point(magx[0], magx[1] - 1e6)
Upper_Point = Point(magx[0], magx[1] + 1e6)
topLine = Line([Lower_Point, Upper_Point])
# Tracing primary-separatrix: core-boundary
# H1_E / B1_W
xpt1N__psiMinCore = self.LineTracer.draw_line(xpt1['N'],
{'psi': psi_core},
option='rho',
direction='cw',
show_plot=visual,
text=verbose)
H1_E = xpt1N__psiMinCore
B1_W = H1_E.reverse_copy()
# B1_N / B2_S
xpt1NW__midline_1 = self.LineTracer.draw_line(xpt1['NW'],
{'line': midline_1},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
B1_N = xpt1NW__midline_1
B2_S = B1_N.reverse_copy()
# H2_S / H1_N
xpt1NE__midline_2 = self.LineTracer.draw_line(xpt1['NE'],
{'line': midline_2},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
# C1_N / C2_S
midline_1__topLine = self.LineTracer.draw_line(xpt1NW__midline_1.p[-1],
{'line': topLine},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
C1_N = midline_1__topLine
C2_S = C1_N.reverse_copy()
# D1_N / D2_S
midline_2__topLine = self.LineTracer.draw_line(xpt1NE__midline_2.p[-1],
{'line': topLine},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
D1_N = midline_2__topLine.reverse_copy()
D2_S = D1_N.reverse_copy()
# Tracing core: psi-min-core region (rho = 1)
# / B1_S
psiMinCore__midline_1_core = self.LineTracer.draw_line(
xpt1N__psiMinCore.p[-1], {'line': midline_1},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
B1_S = psiMinCore__midline_1_core.reverse_copy()
# H1_E1
psiMinCore__midline_2_core = self.LineTracer.draw_line(
xpt1N__psiMinCore.p[-1], {'line': midline_2},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
# / C1_S
midline_1_core__topLine = self.LineTracer.draw_line(
psiMinCore__midline_1_core.p[-1], {'line': topLine},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
C1_S = midline_1_core__topLine.reverse_copy()
# D1_S
midline_2_core__topLine = self.LineTracer.draw_line(
psiMinCore__midline_2_core.p[-1], {'line': topLine},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
D1_S = midline_2_core__topLine
# A1_E / I1_W
xpt1__psiMinPF1 = self.LineTracer.draw_line(xpt1['S'],
{'psi': psi_pf_1},
option='rho',
direction='cw',
show_plot=visual,
text=verbose)
A1_E = xpt1__psiMinPF1
I1_W = A1_E.reverse_copy()
# A1_S
psiMinPF1__WestPlate1 = self.LineTracer.draw_line(
xpt1__psiMinPF1.p[-1], {'line': WestPlate1},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
A1_S = psiMinPF1__WestPlate1
# / I1_S
psiMinPF1__EastPlate1 = self.LineTracer.draw_line(
xpt1__psiMinPF1.p[-1], {'line': EastPlate1},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
I1_S = psiMinPF1__EastPlate1.reverse_copy()
# A2_S / A1_N
xpt1__WestPlate1 = self.LineTracer.draw_line(xpt1['SW'],
{'line': WestPlate1},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
A2_S = xpt1__WestPlate1
A1_N = A2_S.reverse_copy()
# I1_N / I2_S
xpt1__EastPlate1 = self.LineTracer.draw_line(xpt1['SE'],
{'line': EastPlate1},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
I1_N = xpt1__EastPlate1
I2_S = I1_N.reverse_copy()
# Drawing the portion of the separatrix found in the double-null configuration
# E2_E / H2_W
xpt2N__outer_core = self.LineTracer.draw_line(
xpt2['N'], {'line': xpt1NE__midline_2},
option='rho',
direction='cw',
show_plot=visual,
text=verbose)
E2_E = xpt2N__outer_core
H2_W = E2_E.reverse_copy()
# E1_E / H1_W
xpt2N__outer_core__inner_core = self.LineTracer.draw_line(
xpt2N__outer_core.p[-1], {'line': psiMinCore__midline_2_core},
option='rho',
direction='cw',
show_plot=visual,
text=verbose)
E1_E = xpt2N__outer_core__inner_core
H1_W = E1_E.reverse_copy()
E1_N, H1_N = xpt1NE__midline_2.reverse_copy().split(
xpt2N__outer_core.p[-1], add_split_point=True)
E2_S = E1_N.reverse_copy()
H2_S = H1_N.reverse_copy()
H1_S, E1_S = psiMinCore__midline_2_core.split(
xpt2N__outer_core__inner_core.p[-1], add_split_point=True)
# H2_N__I2_N
xpt2NW__EastPlate1 = self.LineTracer.draw_line(xpt2['NW'],
{'line': EastPlate1},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
# E3_S / E2_N
xpt2NE__midline_2 = self.LineTracer.draw_line(xpt2['NE'],
{'line': midline_2},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
E3_S = xpt2NE__midline_2
E2_N = E3_S.reverse_copy()
# D3_S / D2_N
xpt2__midline_2__topLine = self.LineTracer.draw_line(
xpt2NE__midline_2.p[-1], {'line': topLine},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
D3_S = xpt2__midline_2__topLine
D2_N = D3_S.reverse_copy()
# C3_S / C2_N
xpt2__topLine__midline_1 = self.LineTracer.draw_line(
xpt2__midline_2__topLine.p[-1], {'line': midline_1},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
C3_S = xpt2__topLine__midline_1
C2_N = C3_S.reverse_copy()
xpt2__midline_1__WestPlate1 = self.LineTracer.draw_line(
xpt2__topLine__midline_1.p[-1], {'line': WestPlate1},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
if self.settings['grid_settings']['patch_generation']['use_xpt1_W']:
tilt = self.settings['grid_settings']['patch_generation'][
'xpt1_W_tilt']
B2_W = self.LineTracer.draw_line(
xpt1['W'], {'line': (xpt2__midline_1__WestPlate1, tilt)},
option='z_const',
direction='ccw',
show_plot=visual,
text=verbose)
else:
B2_W = self.LineTracer.draw_line(
xpt1['W'], {'line': xpt2__midline_1__WestPlate1},
option='rho',
direction='ccw',
show_plot=visual,
text=verbose)
A2_E = B2_W.reverse_copy()
if self.settings['grid_settings']['patch_generation']['use_xpt1_E']:
tilt = self.settings['grid_settings']['patch_generation'][
'xpt1_E_tilt']
I2_W = self.LineTracer.draw_line(
xpt1['E'], {'line': (xpt2NW__EastPlate1, tilt)},
option='z_const',
direction='cw',
show_plot=visual,
text=verbose)
else:
I2_W = self.LineTracer.draw_line(xpt1['E'],
{'line': xpt2NW__EastPlate1},
option='rho',
direction='ccw',
show_plot=visual,
text=verbose)
H2_E = I2_W.reverse_copy()
# / A3_S, B3_S
A2_N, B2_N = xpt2__midline_1__WestPlate1.reverse_copy().split(
B2_W.p[-1], add_split_point=True)
A3_S = A2_N.reverse_copy()
B3_S = B2_N.reverse_copy()
# / H3_S, I3_S
H2_N, I2_N = xpt2NW__EastPlate1.split(I2_W.p[-1], add_split_point=True)
H3_S = H2_N.reverse_copy()
I3_S = I2_N.reverse_copy()
xpt2__psiMinPF2 = self.LineTracer.draw_line(xpt2['S'],
{'psi': psi_pf_2},
option='rho',
direction='cw',
show_plot=visual,
text=verbose)
F1_W, F2_W = xpt2__psiMinPF2.reverse_copy().split(
xpt2__psiMinPF2.p[len(xpt2__psiMinPF2.p) // 2],
add_split_point=True)
G1_E = F1_W.reverse_copy()
G2_E = F2_W.reverse_copy()
F1_S = self.LineTracer.draw_line(F1_W.p[0], {
'line': EastPlate2
},
option='theta',
direction='cw',
show_plot=visual,
text=verbose).reverse_copy()
G1_S = self.LineTracer.draw_line(F1_W.p[0], {'line': WestPlate2},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
F1_N = self.LineTracer.draw_line(F2_W.p[0], {'line': EastPlate2},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
F2_S = F1_N.reverse_copy()
G1_N = self.LineTracer.draw_line(F2_W.p[0], {
'line': WestPlate2
},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose).reverse_copy()
G2_S = G1_N.reverse_copy()
F2_N = self.LineTracer.draw_line(xpt2['SE'], {'line': EastPlate2},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
F3_S = F2_N.reverse_copy()
G2_N = self.LineTracer.draw_line(xpt2['SW'], {
'line': WestPlate2
},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose).reverse_copy()
G3_S = G2_N.reverse_copy()
if self.settings['grid_settings']['patch_generation']['use_xpt2_W']:
tilt = self.settings['grid_settings']['patch_generation'][
'xpt2_W_tilt']
H3_W = self.LineTracer.draw_line(xpt2['W'],
{'psi_horizontal': (psi_2, tilt)},
option='z_const',
direction='ccw',
show_plot=visual,
text=verbose)
else:
H3_W = self.LineTracer.draw_line(xpt2['W'], {'psi': psi_2},
option='rho',
direction='ccw',
show_plot=visual,
text=verbose)
G3_E = H3_W.reverse_copy()
H3_W__EastPlate1 = self.LineTracer.draw_line(H3_W.p[-1],
{'line': EastPlate1},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
I3_W = self.LineTracer.draw_line(I2_W.p[-1],
{'line': H3_W__EastPlate1},
option='rho',
direction='ccw',
show_plot=visual,
text=verbose)
H3_E = I3_W.reverse_copy()
H3_N, I3_N = H3_W__EastPlate1.split(I3_W.p[-1], add_split_point=True)
G3_N = self.LineTracer.draw_line(H3_W.p[-1], {
'line': WestPlate2
},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose).reverse_copy()
if self.settings['grid_settings']['patch_generation']['use_xpt2_E']:
tilt = self.settings['grid_settings']['patch_generation'][
'xpt2_E_tilt']
F3_W = self.LineTracer.draw_line(xpt2['E'],
{'psi_horizontal': (psi_1, tilt)},
option='z_const',
direction='cw',
show_plot=visual,
text=verbose)
else:
F3_W = self.LineTracer.draw_line(xpt2['E'], {'psi': psi_1},
option='rho',
direction='ccw',
show_plot=visual,
text=verbose)
F3_W = self.LineTracer.draw_line(xpt2['E'], {'psi': psi_1},
option='rho',
direction='ccw',
show_plot=visual,
text=verbose)
E3_E = F3_W.reverse_copy()
F3_N = self.LineTracer.draw_line(F3_W.p[-1], {'line': EastPlate2},
option='theta',
direction='cw',
show_plot=visual,
text=verbose)
E3_N = self.LineTracer.draw_line(F3_W.p[-1], {
'line': midline_2
},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose).reverse_copy()
D3_N = self.LineTracer.draw_line(E3_N.p[0], {
'line': topLine
},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose).reverse_copy()
C3_N = self.LineTracer.draw_line(D3_N.p[0], {
'line': midline_1
},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose).reverse_copy()
midline_1__WestPlate1 = self.LineTracer.draw_line(C3_N.p[0],
{'line': WestPlate1},
option='theta',
direction='ccw',
show_plot=visual,
text=verbose)
B3_W = self.LineTracer.draw_line(B2_W.p[-1],
{'line': midline_1__WestPlate1},
option='rho',
direction='ccw',
show_plot=visual,
text=verbose)
A3_E = B3_W.reverse_copy()
A3_N, B3_N = midline_1__WestPlate1.reverse_copy().split(
B3_W.p[-1], add_split_point=True)
B1_E = Line([B1_N.p[-1], B1_S.p[0]])
C1_W = B1_E.reverse_copy()
B2_E = Line([B2_N.p[-1], B2_S.p[0]])
C2_W = B2_E.reverse_copy()
B3_E = Line([B3_N.p[-1], B3_S.p[0]])
C3_W = B3_E.reverse_copy()
C1_E = Line([C1_N.p[-1], C1_S.p[0]])
D1_W = C1_E.reverse_copy()
C2_E = Line([C2_N.p[-1], C2_S.p[0]])
D2_W = C2_E.reverse_copy()
C3_E = Line([C3_N.p[-1], C3_S.p[0]])
D3_W = C3_E.reverse_copy()
D1_E = Line([D1_N.p[-1], D1_S.p[0]])
E1_W = D1_E.reverse_copy()
D2_E = Line([D2_N.p[-1], D2_S.p[0]])
E2_W = D2_E.reverse_copy()
D3_E = Line([D3_N.p[-1], D3_S.p[0]])
E3_W = D3_E.reverse_copy()
A1_W = trim_geometry(WestPlate1, A1_S.p[-1], A1_N.p[0])
A2_W = trim_geometry(WestPlate1, A2_S.p[-1], A2_N.p[0])
A3_W = trim_geometry(WestPlate1, A3_S.p[-1], A3_N.p[0])
G1_W = trim_geometry(WestPlate2, G1_S.p[-1], G1_N.p[0])
G2_W = trim_geometry(WestPlate2, G2_S.p[-1], G2_N.p[0])
G3_W = trim_geometry(WestPlate2, G3_S.p[-1], G3_N.p[0])
I1_E = trim_geometry(EastPlate1, I1_N.p[-1], I1_S.p[0])
I2_E = trim_geometry(EastPlate1, I2_N.p[-1], I2_S.p[0])
I3_E = trim_geometry(EastPlate1, I3_N.p[-1], I3_S.p[0])
F1_E = trim_geometry(EastPlate2, F1_N.p[-1], F1_S.p[0])
F2_E = trim_geometry(EastPlate2, F2_N.p[-1], F2_S.p[0])
F3_E = trim_geometry(EastPlate2, F3_N.p[-1], F3_S.p[0])
# ============== Patch A1 ==============
A1 = Patch([A1_N, A1_E, A1_S, A1_W],
patch_name='A1',
plate_patch=True,
plate_location='W')
# ============== Patch A2 ==============
A2 = Patch([A2_N, A2_E, A2_S, A2_W],
patch_name='A2',
plate_patch=True,
plate_location='W')
# ============== Patch A3 ==============
A3 = Patch([A3_N, A3_E, A3_S, A3_W],
patch_name='A3',
plate_patch=True,
plate_location='W')
# ============== Patch B1 ==============
B1 = Patch([B1_N, B1_E, B1_S, B1_W], patch_name='B1')
# ============== Patch B2 ==============
B2 = Patch([B2_N, B2_E, B2_S, B2_W], patch_name='B2')
# ============== Patch B3 ==============
B3 = Patch([B3_N, B3_E, B3_S, B3_W], patch_name='B3')
# ============== Patch C1 ==============
C1 = Patch([C1_N, C1_E, C1_S, C1_W], patch_name='C1')
# ============== Patch C2 ==============
C2 = Patch([C2_N, C2_E, C2_S, C2_W], patch_name='C2')
# ============== Patch C3 ==============
C3 = Patch([C3_N, C3_E, C3_S, C3_W], patch_name='C3')
# ============== Patch D1 ==============
D1 = Patch([D1_N, D1_E, D1_S, D1_W], patch_name='D1')
# ============== Patch D2 ==============
D2 = Patch([D2_N, D2_E, D2_S, D2_W], patch_name='D2')
# ============== Patch D3 ==============
D3 = Patch([D3_N, D3_E, D3_S, D3_W], patch_name='D3')
# ============== Patch E1 ==============
E1 = Patch([E1_N, E1_E, E1_S, E1_W], patch_name='E1')
# ============== Patch E2 ==============
E2 = Patch([E2_N, E2_E, E2_S, E2_W], patch_name='E2')
# ============== Patch E3 ==============
E3 = Patch([E3_N, E3_E, E3_S, E3_W], patch_name='E3')
# ============== Patch F1 ==============
F1 = Patch([F1_N, F1_E, F1_S, F1_W],
patch_name='F1',
plate_patch=True,
plate_location='E')
# ============== Patch F2 ==============
F2 = Patch([F2_N, F2_E, F2_S, F2_W],
patch_name='F2',
plate_patch=True,
plate_location='E')
# ============== Patch F3 ==============
F3 = Patch([F3_N, F3_E, F3_S, F3_W],
patch_name='F3',
plate_patch=True,
plate_location='E')
# ============== Patch G1 ==============
G1 = Patch([G1_N, G1_E, G1_S, G1_W],
patch_name='G1',
plate_patch=True,
plate_location='W')
# ============== Patch G2 ==============
G2 = Patch([G2_N, G2_E, G2_S, G2_W],
patch_name='G2',
plate_patch=True,
plate_location='W')
# ============== Patch G3 ==============
G3 = Patch([G3_N, G3_E, G3_S, G3_W],
patch_name='G3',
plate_patch=True,
plate_location='W')
# ============== Patch H1 ==============
H1 = Patch([H1_N, H1_E, H1_S, H1_W], patch_name='H1')
# ============== Patch H2 ==============
H2 = Patch([H2_N, H2_E, H2_S, H2_W], patch_name='H2')
# ============== Patch H3 ==============
H3 = Patch([H3_N, H3_E, H3_S, H3_W], patch_name='H3')
# ============== Patch I1 ==============
I1 = Patch([I1_N, I1_E, I1_S, I1_W],
patch_name='I1',
plate_patch=True,
plate_location='E')
# ============== Patch I2 ==============
I2 = Patch([I2_N, I2_E, I2_S, I2_W],
patch_name='I2',
plate_patch=True,
plate_location='E')
# ============== Patch I3 ==============
I3 = Patch([I3_N, I3_E, I3_S, I3_W],
patch_name='I3',
plate_patch=True,
plate_location='E')
patches = [
A3, A2, A1, B3, B2, B1, C3, C2, C1, D3, D2, D1, E3, E2, E1, F3, F2,
F1, G3, G2, G1, H3, H2, H1, I3, I2, I1
]
self.patches = {}
for patch in patches:
patch.parent = self
patch.PatchTagMap = self.PatchTagMap
self.patches[patch.patch_name] = patch
self.OrderPatches()
def construct_patches(self):
"""
Draws lines and creates patches for both USN and LSN configurations.
Patch Labeling Key:
I: Inner,
O: Outer,
DL: Divertor Leg,
PF: Private Flux,
T: Top,
B: Bottom,
S: Scrape Off Layer,
C: Core.
"""
try:
visual = self.settings['DEBUG']['visual']['patch_map']
except KeyError:
visual = False
try:
verbose = self.settings['DEBUG']['verbose']['patch_generation']
except KeyError:
verbose = False
try:
magx_tilt_1 = self.settings['grid_settings']['patch_generation']['magx_tilt_1']
except KeyError:
magx_tilt_1 = 0.0
try:
magx_tilt_2 = self.settings['grid_settings']['patch_generation']['magx_tilt_2']
except KeyError:
magx_tilt_2 = 0.0
xpt1 = self.LineTracer.NSEW_lookup['xpt1']['coor']
xpt2 = self.LineTracer.NSEW_lookup['xpt2']['coor']
magx = np.array([self.settings['grid_settings']['rmagx'] + self.settings['grid_settings']['patch_generation']['rmagx_shift'],
self.settings['grid_settings']['zmagx'] + self.settings['grid_settings']['patch_generation']['zmagx_shift']])
psi_1 = self.settings['grid_settings']['psi_1']
psi_2 = self.settings['grid_settings']['psi_2']
psi_core = self.settings['grid_settings']['psi_core']
psi_pf_1 = self.settings['grid_settings']['psi_pf_1']
psi_pf_2 = self.settings['grid_settings']['psi_pf_2']
psi_separatrix_2 = Point(xpt2['center']).psi(self)
if self.settings['grid_settings']['patch_generation']['strike_pt_loc'] == 'limiter':
WestPlate1 = self.parent.LimiterData.copy()
WestPlate2 = self.parent.LimiterData.copy()
EastPlate1 = self.parent.LimiterData.copy()
EastPlate2 = self.parent.LimiterData.copy()
else:
WestPlate1 = self.PlateData['plate_W1']
WestPlate2 = self.PlateData['plate_W2']
EastPlate1 = self.PlateData['plate_E1']
EastPlate2 = self.PlateData['plate_E2']
# Generate Horizontal Mid-Plane lines
LHS_Point = Point(magx[0] - 1e6 * np.cos(magx_tilt_1), magx[1] - 1e6 * np.sin(magx_tilt_1))
RHS_Point = Point(magx[0] + 1e6 * np.cos(magx_tilt_1), magx[1] + 1e6 * np.sin(magx_tilt_1))
midline_1 = Line([LHS_Point, RHS_Point])
LHS_Point = Point(magx[0] - 1e6 * np.cos(magx_tilt_2), magx[1] - 1e6 * np.sin(magx_tilt_2))
RHS_Point = Point(magx[0] + 1e6 * np.cos(magx_tilt_2), magx[1] + 1e6 * np.sin(magx_tilt_2))
midline_2 = Line([LHS_Point, RHS_Point])
# Generate Vertical Mid-Plane line
Lower_Point = Point(magx[0], magx[1] - 1e6)
Upper_Point = Point(magx[0], magx[1] + 1e6)
topLine = Line([Lower_Point, Upper_Point])
E1_E = self.LineTracer.draw_line(xpt1['N'], {'psi': psi_core}, option='rho', direction='cw', show_plot=visual, text=verbose)
H1_W = E1_E.reverse_copy()
E1_S = self.LineTracer.draw_line(E1_E.p[-1], {'line': midline_2}, option='theta', direction='ccw', show_plot=visual, text=verbose)
D1_S = self.LineTracer.draw_line(E1_S.p[-1], {'line': topLine}, option='theta', direction='ccw', show_plot=visual, text=verbose)
B1_S__H1_S = self.LineTracer.draw_line(E1_E.p[-1], {'line': midline_1}, option='theta', direction='cw', show_plot=visual, text=verbose).reverse_copy()
C1_S = self.LineTracer.draw_line(B1_S__H1_S.p[0], {'line': topLine}, option='theta', direction='cw', show_plot=visual, text=verbose).reverse_copy()
H1_N__B1_N = self.LineTracer.draw_line(xpt1['NW'], {'line': midline_1}, option='theta', direction='cw', show_plot=visual, text=verbose)
C1_N = self.LineTracer.draw_line(H1_N__B1_N.p[-1], {'line': topLine}, option='theta', direction='cw', show_plot=visual, text=verbose)
C2_S = C1_N.reverse_copy()
E2_S = self.LineTracer.draw_line(xpt1['NE'], {'line': midline_2}, option='theta', direction='ccw', show_plot=visual, text=verbose)
E1_N = E2_S.reverse_copy()
D2_S = self.LineTracer.draw_line(E2_S.p[-1], {'line': topLine}, option='theta', direction='ccw', show_plot=visual, text=verbose)
D1_N = D2_S.reverse_copy()
H2_E = self.LineTracer.draw_line(xpt2['N'], {'line': H1_N__B1_N}, option='rho', direction='cw', show_plot=visual, text=verbose)
B2_W = H2_E.reverse_copy()
H1_N, B1_N = H1_N__B1_N.split(H2_E.p[-1], add_split_point=True)
H2_S, B2_S = H1_N.reverse_copy(), B1_N.reverse_copy()
H1_E = self.LineTracer.draw_line(H2_E.p[-1], {'line': B1_S__H1_S}, option='rho', direction='cw', show_plot=visual, text=verbose)
B1_W = H1_E.reverse_copy()
B1_S, H1_S = B1_S__H1_S.split(H1_E.p[-1], add_split_point=True)
if self.settings['grid_settings']['patch_generation']['use_xpt1_E']:
tilt = self.settings['grid_settings']['patch_generation']['xpt1_E_tilt']
F2_W__F3_W = self.LineTracer.draw_line(xpt1['E'], {'psi_horizontal': (psi_1, tilt)}, option='z_const', direction='cw', show_plot=visual, text=verbose)
else:
F2_W__F3_W = self.LineTracer.draw_line(xpt1['E'], {'psi': psi_1}, option='rho', direction='ccw', show_plot=visual, text=verbose)
if self.settings['grid_settings']['patch_generation']['use_xpt2_W']:
tilt = self.settings['grid_settings']['patch_generation']['xpt2_W_tilt']
B3_W = self.LineTracer.draw_line(xpt2['W'], {'psi_horizontal': (psi_1, tilt)}, option='z_const', direction='ccw', show_plot=visual, text=verbose)
else:
B3_W = self.LineTracer.draw_line(xpt2['W'], {'psi': psi_1}, option='rho', direction='ccw', show_plot=visual, text=verbose)
A3_E = B3_W.reverse_copy()
A3_N = self.LineTracer.draw_line(B3_W.p[-1], {'line': WestPlate2}, option='theta', direction='ccw', show_plot=visual, text=verbose).reverse_copy()
B3_N = self.LineTracer.draw_line(A3_N.p[-1], {'line': midline_1}, option='theta', direction='cw', show_plot=visual, text=verbose)
C3_N = self.LineTracer.draw_line(B3_N.p[-1], {'line': topLine}, option='theta', direction='cw', show_plot=visual, text=verbose)
F3_N = self.LineTracer.draw_line(F2_W__F3_W.p[-1], {'line': EastPlate1}, option='theta', direction='cw', show_plot=visual, text=verbose)
E3_N = self.LineTracer.draw_line(F3_N.p[0], {'line': midline_2}, option='theta', direction='ccw', show_plot=visual, text=verbose).reverse_copy()
D3_N = self.LineTracer.draw_line(E3_N.p[0], {'line': topLine}, option='theta', direction='ccw', show_plot=visual, text=verbose).reverse_copy()
B2_N = self.LineTracer.draw_line(xpt2['NW'], {'line': midline_1}, option='theta', direction='cw', show_plot=visual, text=verbose)
B3_S = B2_N.reverse_copy()
C2_N = self.LineTracer.draw_line(B2_N.p[-1], {'line': topLine}, option='theta', direction='cw', show_plot=visual, text=verbose)
C3_S = C2_N.reverse_copy()
D2_N = self.LineTracer.draw_line(C2_N.p[-1], {'line': midline_2}, option='theta', direction='cw', show_plot=visual, text=verbose)
D3_S = D2_N.reverse_copy()
E2_N = self.LineTracer.draw_line(D2_N.p[-1], {'line': F2_W__F3_W}, option='theta', direction='cw', show_plot=visual, text=verbose)
E3_S = E2_N.reverse_copy()
F2_W, F3_W = F2_W__F3_W.split(E2_N.p[-1], add_split_point=True)
E2_E, E3_E = F2_W.reverse_copy(), F3_W.reverse_copy()
F2_N = self.LineTracer.draw_line(E2_N.p[-1], {'line': EastPlate1}, option='theta', direction='cw', show_plot=visual, text=verbose)
F3_S = F2_N.reverse_copy()
F1_N = self.LineTracer.draw_line(xpt1['SE'], {'line': EastPlate1}, option='theta', direction='cw', show_plot=visual, text=verbose)
F2_S = F1_N.reverse_copy()
G1_E = self.LineTracer.draw_line(xpt1['S'], {'psi': psi_pf_1}, option='rho', direction='cw', show_plot=visual, text=verbose)
F1_W = G1_E.reverse_copy()
G1_S = self.LineTracer.draw_line(G1_E.p[-1], {'line': WestPlate1}, option='theta', direction='ccw', show_plot=visual, text=verbose)
F1_S = self.LineTracer.draw_line(G1_E.p[-1], {'line': EastPlate1}, option='theta', direction='cw', show_plot=visual, text=verbose).reverse_copy()
G2_S = self.LineTracer.draw_line(xpt1['SW'], {'line': WestPlate1}, option='theta', direction='ccw', show_plot=visual, text=verbose)
G1_N = G2_S.reverse_copy()
H3_S__G3_S = self.LineTracer.draw_line(xpt2['NE'], {'line': WestPlate1}, option='theta', direction='ccw', show_plot=visual, text=verbose)
if self.settings['grid_settings']['patch_generation']['use_xpt1_W']:
tilt = self.settings['grid_settings']['patch_generation']['xpt1_W_tilt']
H2_W = self.LineTracer.draw_line(xpt1['W'], {'line': (H3_S__G3_S, tilt)}, option='z_const', direction='ccw', show_plot=visual, text=verbose)
else:
H2_W = self.LineTracer.draw_line(xpt1['W'], {'line': H3_S__G3_S}, option='rho', direction='ccw', show_plot=visual, text=verbose)
G2_E = H2_W.reverse_copy()
H3_S, G3_S = H3_S__G3_S.split(H2_W.p[-1], add_split_point=True)
H2_N, G2_N = H3_S.reverse_copy(), G3_S.reverse_copy()
if self.settings['grid_settings']['patch_generation']['use_xpt2_E']:
tilt = self.settings['grid_settings']['patch_generation']['xpt2_E_tilt']
I3_W = self.LineTracer.draw_line(xpt2['E'], {'psi_horizontal': (psi_2, tilt)}, option='z_const', direction='cw', show_plot=visual, text=verbose)
else:
I3_W = self.LineTracer.draw_line(xpt2['E'], {'psi': psi_2}, option='rho', direction='ccw', show_plot=visual, text=verbose)
H3_E = I3_W.reverse_copy()
I3_N = self.LineTracer.draw_line(I3_W.p[-1], {'line': EastPlate2}, option='theta', direction='cw', show_plot=visual, text=verbose)
G3_N__H3_N = self.LineTracer.draw_line(I3_W.p[-1], {'line': WestPlate1}, option='theta', direction='ccw', show_plot=visual, text=verbose).reverse_copy()
if self.settings['grid_settings']['patch_generation']['use_xpt1_W']:
tilt = self.settings['grid_settings']['patch_generation']['xpt1_W_tilt']
H3_W = self.LineTracer.draw_line(G2_N.p[-1], {'line': (G3_N__H3_N, tilt)}, option='z_const', direction='ccw', show_plot=visual, text=verbose)
else:
H3_W = self.LineTracer.draw_line(G2_N.p[-1], {'line': G3_N__H3_N}, option='rho', direction='ccw', show_plot=visual, text=verbose)
G3_E = H3_W.reverse_copy()
G3_N, H3_N = G3_N__H3_N.split(H3_W.p[-1], add_split_point=True)
A2_E__A1_E = self.LineTracer.draw_line(xpt2['S'], {'psi': psi_pf_2}, option='rho', direction='cw', show_plot=visual, text=verbose)
A2_E, A1_E = A2_E__A1_E.split(A2_E__A1_E.p[len(A2_E__A1_E.p) // 2], add_split_point=True)
I2_W, I1_W = A2_E.reverse_copy(), A1_E.reverse_copy()
A1_S = self.LineTracer.draw_line(A1_E.p[-1], {'line': WestPlate2}, option='theta', direction='ccw', show_plot=visual, text=verbose)
I1_S = self.LineTracer.draw_line(A1_E.p[-1], {'line': EastPlate2}, option='theta', direction='cw', show_plot=visual, text=verbose).reverse_copy()
A2_S = self.LineTracer.draw_line(A2_E.p[-1], {'line': WestPlate2}, option='theta', direction='ccw', show_plot=visual, text=verbose)
A1_N = A2_S.reverse_copy()
I2_S = self.LineTracer.draw_line(A2_E.p[-1], {'line': EastPlate2}, option='theta', direction='cw', show_plot=visual, text=verbose).reverse_copy()
I1_N = I2_S.reverse_copy()
A3_S = self.LineTracer.draw_line(xpt2['SW'], {'line': WestPlate2}, option='theta', direction='ccw', show_plot=visual, text=verbose)
A2_N = A3_S.reverse_copy()
I3_S = self.LineTracer.draw_line(xpt2['SE'], {'line': EastPlate2}, option='theta', direction='cw', show_plot=visual, text=verbose).reverse_copy()
I2_N = I3_S.reverse_copy()
B3_E = Line([B3_N.p[-1], B3_S.p[0]])
C3_W = B3_E.reverse_copy()
B2_E = Line([B2_N.p[-1], B2_S.p[0]])
C2_W = B2_E.reverse_copy()
B1_E = Line([B1_N.p[-1], B1_S.p[0]])
C1_W = B1_E.reverse_copy()
C3_E = Line([C3_N.p[-1], C3_S.p[0]])
D3_W = C3_E.reverse_copy()
C2_E = Line([C2_N.p[-1], C2_S.p[0]])
D2_W = C2_E.reverse_copy()
C1_E = Line([C1_N.p[-1], C1_S.p[0]])
D1_W = C1_E.reverse_copy()
D3_E = Line([D3_N.p[-1], D3_S.p[0]])
E3_W = D3_E.reverse_copy()
D2_E = Line([D2_N.p[-1], D2_S.p[0]])
E2_W = D2_E.reverse_copy()
D1_E = Line([D1_N.p[-1], D1_S.p[0]])
E1_W = D1_E.reverse_copy()
A1_W = trim_geometry(WestPlate2, A1_S.p[-1], A1_N.p[0])
A2_W = trim_geometry(WestPlate2, A2_S.p[-1], A2_N.p[0])
A3_W = trim_geometry(WestPlate2, A3_S.p[-1], A3_N.p[0])
F1_E = trim_geometry(EastPlate1, F1_N.p[-1], F1_S.p[0])
F2_E = trim_geometry(EastPlate1, F2_N.p[-1], F2_S.p[0])
F3_E = trim_geometry(EastPlate1, F3_N.p[-1], F3_S.p[0])
I1_E = trim_geometry(EastPlate2, I1_N.p[-1], I1_S.p[0])
I2_E = trim_geometry(EastPlate2, I2_N.p[-1], I2_S.p[0])
I3_E = trim_geometry(EastPlate2, I3_N.p[-1], I3_S.p[0])
G1_W = trim_geometry(WestPlate1, G1_S.p[-1], G1_N.p[0])
G2_W = trim_geometry(WestPlate1, G2_S.p[-1], G2_N.p[0])
G3_W = trim_geometry(WestPlate1, G3_S.p[-1], G3_N.p[0])
# ============== Patch A1 ==============
A1 = Patch([A1_N, A1_E, A1_S, A1_W], patch_name='A1', plate_patch=True, plate_location='W')
# ============== Patch A2 ==============
A2 = Patch([A2_N, A2_E, A2_S, A2_W], patch_name='A2', plate_patch=True, plate_location='W')
# ============== Patch A3 ==============
A3 = Patch([A3_N, A3_E, A3_S, A3_W], patch_name='A3', plate_patch=True, plate_location='W')
# ============== Patch B1 ==============
B1 = Patch([B1_N, B1_E, B1_S, B1_W], patch_name='B1')
# ============== Patch B2 ==============
B2 = Patch([B2_N, B2_E, B2_S, B2_W], patch_name='B2')
# ============== Patch B3 ==============
B3 = Patch([B3_N, B3_E, B3_S, B3_W], patch_name='B3')
# ============== Patch C1 ==============
C1 = Patch([C1_N, C1_E, C1_S, C1_W], patch_name='C1')
# ============== Patch C2 ==============
C2 = Patch([C2_N, C2_E, C2_S, C2_W], patch_name='C2')
# ============== Patch C3 ==============
C3 = Patch([C3_N, C3_E, C3_S, C3_W], patch_name='C3')
# ============== Patch D1 ==============
D1 = Patch([D1_N, D1_E, D1_S, D1_W], patch_name='D1')
# ============== Patch D2 ==============
D2 = Patch([D2_N, D2_E, D2_S, D2_W], patch_name='D2')
# ============== Patch D3 ==============
D3 = Patch([D3_N, D3_E, D3_S, D3_W], patch_name='D3')
# ============== Patch E1 ==============
E1 = Patch([E1_N, E1_E, E1_S, E1_W], patch_name='E1')
# ============== Patch E2 ==============
E2 = Patch([E2_N, E2_E, E2_S, E2_W], patch_name='E2')
# ============== Patch E3 ==============
E3 = Patch([E3_N, E3_E, E3_S, E3_W], patch_name='E3')
# ============== Patch F1 ==============
F1 = Patch([F1_N, F1_E, F1_S, F1_W], patch_name='F1', plate_patch=True, plate_location='E')
# ============== Patch F2 ==============
F2 = Patch([F2_N, F2_E, F2_S, F2_W], patch_name='F2', plate_patch=True, plate_location='E')
# ============== Patch F3 ==============
F3 = Patch([F3_N, F3_E, F3_S, F3_W], patch_name='F3', plate_patch=True, plate_location='E')
# ============== Patch G1 ==============
G1 = Patch([G1_N, G1_E, G1_S, G1_W], patch_name='G1', plate_patch=True, plate_location='W')
# ============== Patch G2 ==============
G2 = Patch([G2_N, G2_E, G2_S, G2_W], patch_name='G2', plate_patch=True, plate_location='W')
# ============== Patch G3 ==============
G3 = Patch([G3_N, G3_E, G3_S, G3_W], patch_name='G3', plate_patch=True, plate_location='W')
# ============== Patch H1 ==============
H1 = Patch([H1_N, H1_E, H1_S, H1_W], patch_name='H1')
# ============== Patch H2 ==============
H2 = Patch([H2_N, H2_E, H2_S, H2_W], patch_name='H2')
# ============== Patch H3 ==============
H3 = Patch([H3_N, H3_E, H3_S, H3_W], patch_name='H3')
# ============== Patch I1 ==============
I1 = Patch([I1_N, I1_E, I1_S, I1_W], patch_name='I1', plate_patch=True, plate_location='E')
# ============== Patch I2 ==============
I2 = Patch([I2_N, I2_E, I2_S, I2_W], patch_name='I2', plate_patch=True, plate_location='E')
# ============== Patch I3 ==============
I3 = Patch([I3_N, I3_E, I3_S, I3_W], patch_name='I3', plate_patch=True, plate_location='E')
patches = [A3, A2, A1, B3, B2, B1, C3, C2, C1, D3, D2, D1, E3, E2, E1,
F3, F2, F1, G3, G2, G1, H3, H2, H1, I3, I2, I1]
self.patches = {}
for patch in patches:
patch.parent = self
patch.PatchTagMap = self.PatchTagMap
self.patches[patch.patch_name] = patch
self.OrderPatches()