hermiteexpansioncoefficient.h

Go to the documentation of this file.

#ifndef HERMITEEXPANSIONCOEFFICIENT_H
#define HERMITEEXPANSIONCOEFFICIENT_H

#include <math/vector3.h>
#include <armadillo>

using namespace arma;
using namespace std;

class HermiteExpansionCoefficient
{
public:
    explicit HermiteExpansionCoefficient(int dimension);

    void reset(int dimension);
    void set(double a, double b, const Vector3 &A, const Vector3 &B, int iA, int iB, int jA, int jB, int kA, int kB);

    const cube &operator [](const uword row) const;
    cube &operator[](const uword row);
    void setupE(int iA, int iB, int jA, int jB, int kA, int kB);
    bool checkIndexCombinationForE(int iA, int iB, int t);
    double operator ()(int iA, int jA, int kA, int iB, int jB, int kB, const uword t, const uword u, const uword v) const;
protected:
    double m_a;
    double m_b;
    Vector3 m_A;
    Vector3 m_B;
    int m_dimensionMax;
    cube m_E[3]; // t, i, j
    Vector3 m_AB;
    Vector3 m_P;
    Vector3 m_PA;
    Vector3 m_PB;
};

inline const cube &HermiteExpansionCoefficient::operator[](const uword dimension) const {
    return m_E[dimension];
}

inline cube &HermiteExpansionCoefficient::operator[](const uword row)
{
    return const_cast<cube&>(static_cast<const HermiteExpansionCoefficient&>(*this)[row]);
}

inline double HermiteExpansionCoefficient::operator()(int iA, int jA, int kA, int iB, int jB, int kB, const uword t, const uword u, const uword v) const {
    return m_E[0](iA, iB, t) * m_E[1](jA, jB, u) * m_E[2](kA, kB, v);
}

#endif // HERMITEEXPANSIONCOEFFICIENT_H