d_distribution = Maxwellian(d_density, temperature, bulk_velocity, elements.deuterium.atomic_weight * atomic_mass)
he2_distribution = Maxwellian(he2_density, temperature, bulk_velocity, elements.helium.atomic_weight * atomic_mass)
c6_distribution = Maxwellian(c6_density, temperature, bulk_velocity, elements.carbon.atomic_weight * atomic_mass)
ne10_distribution = Maxwellian(ne10_density, temperature, bulk_velocity, elements.neon.atomic_weight * atomic_mass)
e_distribution = Maxwellian(e_density, temperature, bulk_velocity, electron_mass)
d_species = Species(elements.deuterium, 1, d_distribution)
he2_species = Species(elements.helium, 2, he2_distribution)
c6_species = Species(elements.carbon, 6, c6_distribution)
ne10_species = Species(elements.neon, 10, ne10_distribution)
# define species
plasma.b_field = ConstantVector3D(Vector3D(1.0, 1.0, 1.0))
plasma.electron_distribution = e_distribution
plasma.composition = [d_species, he2_species, c6_species, ne10_species]
# BEAM ------------------------------------------------------------------------
beam = Beam(parent=world, transform=translate(1.0, 0.0, 0) * rotate(90, 0, 0))
beam.plasma = plasma
beam.atomic_data = adas
beam.energy = 60000
beam.power = 3e6
beam.element = elements.deuterium
beam.sigma = 0.025
beam.divergence_x = 0.5
beam.divergence_y = 0.5
beam.length = 3.0
beam.attenuator = SingleRayAttenuator(clamp_to_zero=True)
beam.models = [
BeamCXLine(Line(elements.helium, 1, (4, 3))),
pedestal_top=pedestal_top)
imp_distribution = Maxwellian(imp_density, imp_temperature,
velocity_profile,
impurity.atomic_weight * atomic_mass)
species.append(Species(impurity, ionisation, imp_distribution))
# define the electron distribution
e_density = IonFunction(peak_density, 0, pedestal_top=pedestal_top)
e_temperature = IonFunction(peak_temperature, 0, pedestal_top=pedestal_top)
e_distribution = Maxwellian(e_density, e_temperature, velocity_profile,
electron_mass)
# define species
plasma.b_field = ConstantVector3D(Vector3D(0, 0, 0))
plasma.electron_distribution = e_distribution
plasma.composition = species
####################
# Visualise Plasma #
h0 = plasma.composition.get(hydrogen, 0)
h1 = plasma.composition.get(hydrogen, 1)
c6 = plasma.composition.get(carbon, 6)
# Run some plots to check the distribution functions and emission profile are as expected
h1_temp = h1.distribution.effective_temperature
r, _, z, t_samples = sample3d(h1_temp, (-1, 2, 200), (0, 0, 1), (-1, 1, 200))
plt.imshow(np.transpose(np.squeeze(t_samples)), extent=[-1, 2, -1, 1])
plt.colorbar()
plt.axis('equal')
plt.xlabel('x axis')
d_distribution = Maxwellian(d_density, temperature, bulk_velocity,
deuterium_mass)
nitrogen_mass = nitrogen.atomic_weight * atomic_mass
n_distribution = Maxwellian(n_density, temperature, bulk_velocity,
nitrogen_mass)
e_distribution = Maxwellian(e_density, temperature, bulk_velocity,
electron_mass)
d0_species = Species(deuterium, 0, d_distribution)
d1_species = Species(deuterium, 1, d_distribution)
n1_species = Species(nitrogen, 1, n_distribution)
# define species
plasma.b_field = ConstantVector3D(Vector3D(1.0, 1.0, 1.0))
plasma.electron_distribution = e_distribution
plasma.composition = [d0_species, d1_species, n1_species]
# setup the Balmer lines
hydrogen_I_410 = Line(deuterium, 0, (6, 2)) # n = 6->2: 410.12nm
hydrogen_I_396 = Line(deuterium, 0, (7, 2)) # n = 7->2: 396.95nm
# setup the Nitrgon II line with multiplet splitting instructions
nitrogen_II_404 = Line(nitrogen, 1,
("2s2 2p1 4f1 3G13.0", "2s2 2p1 3d1 3F10.0"))
multiplet = [[403.509, 404.132, 404.354, 404.479, 405.692],
[0.205, 0.562, 0.175, 0.029, 0.029]]
# add all lines to the plasma
plasma.models = [
ExcitationLine(hydrogen_I_410, lineshape=StarkBroadenedLine),
RecombinationLine(hydrogen_I_410, lineshape=StarkBroadenedLine),
Blend2D(Constant2D(0.0), IsoMapper2D(psin_2d, density_c6_psi),
inside_lcfs))
density_d = lambda x, y, z: electron_density(x, y, z) - 6 * density_c6(x, y, z)
d_distribution = Maxwellian(density_d, ion_temperature, flow_velocity,
deuterium.atomic_weight * atomic_mass)
c6_distribution = Maxwellian(density_c6, ion_temperature, flow_velocity,
carbon.atomic_weight * atomic_mass)
e_distribution = Maxwellian(electron_density, ion_temperature, flow_velocity,
electron_mass)
d_species = Species(deuterium, 1, d_distribution)
c6_species = Species(carbon, 6, c6_distribution)
plasma.electron_distribution = e_distribution
plasma.composition = [d_species, c6_species]
# ########################### NBI CONFIGURATION ############################# #
print('Loading JET PINI configuration...')
attenuation_instructions = (SingleRayAttenuator, {'clamp_to_zero': True})
beam_emission_instructions = [(BeamCXLine, {'line': Line(carbon, 5, (8, 7))})]
pini_8_1 = load_pini_from_ppf(PULSE, '8.1', plasma, adas,
attenuation_instructions,
beam_emission_instructions, world)
pini_8_2 = load_pini_from_ppf(PULSE, '8.2', plasma, adas,
attenuation_instructions,
beam_emission_instructions, world)
