hermiteintegral.h

Go to the documentation of this file.

#ifndef HERMITEINTEGRAL_H
#define HERMITEINTEGRAL_H

#include "math/vector3.h"
#include "math/boysfunction.h"

#include <armadillo>

using namespace arma;
using namespace std;

class HermiteIntegral
{
public:
    HermiteIntegral(int dimensionMax);

    void reset(int dimension);
    void set(double alpha, const Vector3 &A, int t, int u, int v);
    void setupR(int t, int u, int v);

    const cube &operator [](const uword row) const;
    cube &operator[](const uword row);
    const cube &operator()(const uword row) const;
    cube &operator()(const uword row);
    double operator ()(const uword n, const uword t, const uword u, const uword v) const;
protected:
    double m_alpha;
    Vector3 m_A;
    field<cube> m_R;
    int m_dimensionMax;
    BoysFunction boysFunction;
};

inline const cube &HermiteIntegral::operator()(const uword row) const {
    return m_R(row);
}

inline double HermiteIntegral::operator()(const uword n, const uword t, const uword u, const uword v) const {
    return m_R(n)(t,u,v);
}

inline const cube &HermiteIntegral::operator[](const uword row) const {
    return m_R[row];
}

// Non-const version based on Scott Meyer's Item 3 in Effective C++ Third Edition

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

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

#endif // HERMITEINTEGRAL_H