fwdsolver_mw.h

// -*-C++-*-
// ==========================================================================
// Forward model: finite element method (multi-wavelength version)
// ==========================================================================

#ifndef __FWDSOLVER_MW_H
#define __FWDSOLVER_MW_H

#include "fwdsolver.h"
#ifdef TOAST_MPI
#include "toast_mpi.h"
#endif

// =========================================================================

template<class T> class TFwdSolverMW: public TFwdSolver<T> {
public:
    TFwdSolverMW (const QMMesh *mesh, LSOLVER linsolver, double tol = 1e-10);

    TFwdSolverMW (const QMMesh *mesh, char *solver, double tol = 1e-10);

    TFwdSolverMW (const QMMesh *mesh, ParamParser &pp);

    ~TFwdSolverMW ();

    TVector<T> ProjectAll_wavel (const TCompRowMatrix<T> &qvec,
        const TCompRowMatrix<T> &mvec, const MWsolution &sol, double omega,
        DataScale scl = DATA_DEFAULT);

    RVector ProjectAll_wavel_real (const TCompRowMatrix<T> &qvec,
        const TCompRowMatrix<T> &mvec, const MWsolution &sol, double omega,
        DataScale scl = DATA_DEFAULT);

protected:
    void Setup();

    void Cleanup();

private:
#ifdef TOAST_MPI
    MPI_Datatype mpitp;
    int sze, rnk;
    int *projall_count;
    int *projall_ofs;
    int *qidx;
#endif
};

// ==========================================================================
// template typedefs

typedef TFwdSolverMW<double>         RFwdSolverMW;
typedef TFwdSolverMW<std::complex<double> > CFwdSolverMW;

// ==========================================================================
// extern declarations of FwdSolverMW (only required for VS)

#ifndef __FWDSOLVER_MW_CC
extern template class STOASTLIB TFwdSolverMW<double>;
extern template class STOASTLIB TFwdSolverMW<std::complex<double> >;
#endif // !__FWDSOLVER_MW_CC

#endif // __FWDSOLVER_MW_H