nurbspy
is a Python package for Non-Uniform Rational Basis Spline (NURBS) curves and surfaces.
The classes and methods were inspired by the algorithms presented in The NURBS Book and the code was implemented using vectorized Numpy functions and Numba's just-in-time compilation decorators to achieve C-like speed.
nurbspy
aims to be a simple NURBS library, not a fully fledged CAD kernel. If you need a powerful, open source CAD kernel we recommend you to check out the C++ OpenCascade library. If you feel that OpenCascade is too complex or you are not sure how to start using it, this repository might be useful for you!
nurbspy
has the following features for NURBS curves:
- Constructors for rational and non-rational Bézier and B-Spline curves
- Methods to evaluate curve coordinates
- Methods to evaluate arbitrary-order derivatives analytically
- Methods to evaluate the tangent, normal, and binormal unitary vectors (Frenet-Serret frame of reference)
- Methods to compute the curvature and torsion
- Methods to compute the arc-length of the curve by numerical quadrature
- Methods to for point projection / point inversion
- Methods to visualize the curve using the Matplotlib library
In addition, nurbspy
provides the following capabilities for NURBS surfaces:
- Constructors for rational and non-rational Bézier and B-Spline surfaces
- Additional constructors for some common special surfaces:
- Bilinear surfaces
- Ruled surfaces
- Extruded surfaces
- Revolution surfaces
- Coons surfaces
- Methods to evaluate surface coordinates
- Methods to evaluate arbitrary-order derivatives analytically
- Methods to evaluate the unitary normal vectors
- Methods to evaluate the mean and Gaussian curvatures
- Methods to compute u- and v-isoparametic curves
- Methods to for point projection / point inversion
- Methods to visualize the surface using the Matplotlib library
In addition, nurbspy
can work with real and complex data types natively. This allows to compute accurate (down to machine epsilon!) shape derivatives using the complex step method and avoid the numerical error incurred by finite-difference derivative approximations. This shape sensitivity information is necessary to solve shape optimization problems with many design variables using gradient based-optimization algorithms. To our knowledge, nurbspy
is the only Python package that provides the flexibility to work with complex numbers right away.
nurbspy
has the following mandatory runtime dependencies:
- numpy (multidimensional array library)
- scipy (scientific computing library)
- numba (just-in-time Python compiler)
- pygmo (optimization library)
- matplotlib (visualization library)
In addition, nurbspy
uses pytest for local tests.
nurbspy
is available on Linux via the pip package manager. The installation with pip is straightfoward:
pip install nurbspy
nurbspy
is also available on Linux, Windows, and macOS via the conda installer. In order to install nurbspy
via conda you need to add conda-forge
and roberagro
to the list of available channels:
conda install nurbspy --channel conda-forge --channel roberagro
You can verify that nurbspy
was installed successfully with this minimal Python script:
# Nurbspy minimal working example
import nurbspy
nurbspy.minimal_example.run()
or by typing this one-liner on your terminal:
python3 -c "import nurbspy; nurbspy.minimal_example.run()"
nurbspy
can be used to create Bézier, B-Spline and NURBS curves. The type of curve depends on the arguments used to initialize the curve class. As an example, the following piece of code can be used to generate a degree four Bézier curve in two dimensions:
# Import packages
import numpy as np
import nurbspy as nrb
import matplotlib.pyplot as plt
# Define the array of control points
P = np.zeros((2,5))
P[:, 0] = [0.20, 0.50]
P[:, 1] = [0.40, 0.70]
P[:, 2] = [0.80, 0.60]
P[:, 3] = [0.80, 0.40]
P[:, 4] = [0.40, 0.20]
# Create and plot the Bezier curve
bezierCurve = nrb.NurbsCurve(control_points=P)
bezierCurve.plot()
plt.show()
If the installation was succesful, you should be able to see the Bézier curve when you execute the previous code snippet.
Check out the curve demos directory to see more examples showing the capabilities of the library and how to use them.
Similarly, nurbspy
can be used to create Bézier, B-Spline and NURBS surfaces. The type of surface depends on the arguments used to initialize the surface class. As an example, the following code snippet can be used to generate a simple Bézier surface of degree 3 in the u-direction and degree 2 in the v-direction:
# Import packages
import numpy as np
import nurbspy as nrb
import matplotlib.pyplot as plt
# Define the array of control points
n_dim, n, m = 3, 4, 3
P = np.zeros((n_dim, n, m))
# First row
P[:, 0, 0] = [0.00, 0.00, 0.00]
P[:, 1, 0] = [1.00, 0.00, 1.00]
P[:, 2, 0] = [2.00, 0.00, 1.00]
P[:, 3, 0] = [3.00, 0.00, 0.00]
# Second row
P[:, 0, 1] = [0.00, 1.00, 1.00]
P[:, 1, 1] = [1.00, 1.00, 2.00]
P[:, 2, 1] = [2.00, 1.00, 2.00]
P[:, 3, 1] = [3.00, 1.00, 1.00]
# Third row
P[:, 0, 2] = [0.00, 2.00, 0.00]
P[:, 1, 2] = [1.00, 2.00, 1.00]
P[:, 2, 2] = [2.00, 2.00, 1.00]
P[:, 3, 2] = [3.00, 2.00, 0.00]
# Create and plot the Bezier surface
bezierSurface = nrb.NurbsSurface(control_points=P)
bezierSurface.plot(control_points=True, isocurves_u=6, isocurves_v=6)
plt.show()
If the installation was successful, you should be able to see the Bézier surface when you execute the previous script.
Check out the surface demos directory to see more examples showing the capabilities of the library and how to use them.
Check out the Bézier, B-Spline, and NURBS notes if you want to learn more about the definition and mathematical properties of these curves and surfaces
nurbspy
was developed by Roberto Agromayor under the supervision of Associate Professor Lars O. Nord at the Norwegian University of Science and Technology (NTNU) as part of his PhD on turbomachinery shape optimization. Please, drop us an email to roberto.agromayor@ntnu.no if you have questions about the code or you have a bug to report!