from algebraic_hbm import softening_Duffing, Algebraic_HBM

# HBM ansatz order.
n = 1

# Initialize HBM for a softening Duffing oscillator and ansatz order 'n=1'.
HBM = Algebraic_HBM(ODE=softening_Duffing, order=n)

# Generate multivariate polynomials.
HBM.generate_multivariate_polynomials()

# Get monomial coefficient matrix for Macauly matrix framework 
# evaluated at a given excitation frequency 'a'
a = 1
A = HBM.get_monomial_coefficient_matrix(a)
print(f"{A=}")