BasisFunctionCache.h

Go to the documentation of this file.

// $Id$
//==============================================================================
//==============================================================================
 
#ifndef _BASIS_FUNCTION_CACHE_H
#define _BASIS_FUNCTION_CACHE_H
 
#include "BasisFunctionVals.h"
#include "ASMenums.h"
 
#include <array>
#include <memory>
 
class Integrand;
 
 
template<size_t Dim> class BasisFunctionCache
{
public:
  BasisFunctionCache();
 
  BasisFunctionCache(const BasisFunctionCache& rhs);
 
  virtual ~BasisFunctionCache() = default;
 
  void clear();
 
  bool init(int nd);
 
  const BasisFunctionVals& getVals(size_t el, size_t gp, bool reduced = false);
 
  void setIntegrand(const Integrand* itg) { integrand = itg; }
 
  const std::array<int,Dim>& nGauss(bool reduced = false)
  { return reduced ? reducedQ->ng : mainQ->ng; }
 
  const std::array<const double*,Dim>& weight(bool reduced = false) const
  { return reduced ? reducedQ->wg : mainQ->wg; }
 
  const std::array<const double*,Dim>& coord(bool reduced = false) const
  { return reduced ? reducedQ->xg : mainQ->xg; }
 
  bool hasReduced() const { return !reducedQ->gpar[0].empty(); }
 
  virtual double getParam(int dir, size_t el, size_t gp, bool reduced = false) const;
 
  int basis = 1; 
 
  void resizeThreadBuffers();
 
protected:
  struct Quadrature {
    std::array<int,Dim> ng; 
    std::array<const double*,Dim> xg; 
    std::array<const double*,Dim> wg; 
    std::array<RealArray,Dim> gpar; 
 
    void reset()
    {
      ng.fill(0);
      xg.fill(nullptr);
      wg.fill(nullptr);
      gpar.fill({});
    }
  };
 
  virtual bool internalInit() = 0;
 
  virtual BasisFunctionVals calculatePt(size_t el, size_t gp,
                                        bool reduced = false) const = 0;
 
  virtual void calculateAll() = 0;
 
  virtual size_t index (size_t el, size_t gp, bool reduced) const;
 
  std::array<size_t,Dim> gpIndex(size_t gp, bool reduced) const;
 
  std::vector<BasisFunctionVals> values; 
  std::vector<BasisFunctionVals> valuesRed; 
  const Integrand* integrand = nullptr; 
  int nderiv = 0; 
 
  std::shared_ptr<Quadrature> mainQ; 
  std::shared_ptr<Quadrature> reducedQ; 
 
  size_t nTotal = 0; 
  size_t nTotalRed = 0; 
};
 
#endif