Examples of Device Coilsets

Notebook displaying example coilset positions and sizes for different devices. To be used in examples and tests.

import json
from pathlib import Path

import matplotlib.pyplot as plt

from bluemira.base.file import get_bluemira_root
from bluemira.equilibria.coils import Coil, CoilSet, symmetrise_coilset

def read_coil_json(name):
    """Read coil info and return data."""
    root_path = get_bluemira_root()

    file_path = Path(root_path, "tests/equilibria/test_data/coilsets/", name)
    with open(file_path) as f:
        return json.load(f)

def make_coilset(data):
    """Make a coilset from position info. Currents not set."""
    coils = []
    for xi, zi, dxi, dzi, name, ctype in zip(
        data["xc"],
        data["zc"],
        data["dxc"],
        data["dzc"],
        data["coil_names"],
        data["coil_types"],
        strict=False,
    ):
        coil = Coil(x=xi, z=zi, dx=dxi, dz=dzi, name=name, ctype=ctype)
        coils.append(coil)
    return CoilSet(*coils)

# DEMO Single Null
data = read_coil_json("DEMO-SN_coilset.json")
demo_sn_coils = make_coilset(data)
demo_sn_coils.plot()
plt.show()

# DEMO Double Null
data = read_coil_json("DEMO-DN_coilset.json")
demo_dn_coils = make_coilset(data)
demo_dn_coils.plot()
plt.show()

# MAST-U
data = read_coil_json("MAST-U_coilset.json")
mastu_coils = make_coilset(data)
mastu_coils.plot()
plt.show()

# ITER
data = read_coil_json("ITER_coilset.json")
iter_coils = make_coilset(data)
iter_coils.plot()
plt.show()

A quick look at applying symmetrise_coilset:

print("MAST")
print("------")
print(mastu_coils)
print("DEMO DN")
print("------")
print(demo_dn_coils)

new_mastu_coils = symmetrise_coilset(mastu_coils)
new_demo_dn_coils = symmetrise_coilset(demo_dn_coils)
print("MAST")
print("------")
print(new_mastu_coils)
print("DEMO DN")
print("------")
print(new_demo_dn_coils)