numba_example.py

import numpy as np
from numba import njit

MACH_CRITIC = 0.78
C_X_MIN = 0.095


@njit(nogil=True, fastmath=True)
def compute_cx(alpha, mach):
    """
    Compute the drag coefficient at M = 0 depending on alpha
    (the higher alpha the higher the drag)
    """
    cx = np.power((np.degrees(alpha) + 5) * 0.03, 2) + C_X_MIN
    if mach < MACH_CRITIC:
        return cx / np.sqrt(1 - (mach ** 2))
    if mach < 1:
        return cx * 15 * (mach - MACH_CRITIC) + (cx / np.sqrt(1 - (mach ** 2)))