Plots a plane and generates a Cubit file for an OpenMC general plane

plane_test.py

import openmc
from openmc_cad_adapter import to_cubit_journal

import matplotlib.pyplot as plt
import numpy as np

# Define the coefficients of the plane
a, b, c, d = 1, 2, -3, 1

plane = openmc.Plane(a, b, c, d)

c1 = openmc.Cell(region=-plane)
c2 = openmc.Cell(region=+plane)

u = openmc.Universe(cells=[c1, c2])

to_cubit_journal(openmc.Geometry(u), world=[100, 100, 100])

# Create the x, y, and z arrays
xx, yy = np.meshgrid(range(10), range(10))
z = (d - a*xx - b*yy) * 1.0 / c

# Create a 3D plot
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(xx, yy, z)

# Set the axis labels
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# Show the plot