PCBasis.h

Go to the documentation of this file.

/* =====================================================================================
 
                      The UQ Toolkit (UQTk) version 3.1.5
                          Copyright (2024) NTESS
                        https://www.sandia.gov/UQToolkit/
                        https://github.com/sandialabs/UQTk
 
     Copyright 2024 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
     Under the terms of Contract DE-NA0003525 with NTESS, the U.S. Government
     retains certain rights in this software.
 
     This file is part of The UQ Toolkit (UQTk)
 
     UQTk is open source software: you can redistribute it and/or modify
     it under the terms of BSD 3-Clause License
 
     UQTk is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     BSD 3 Clause License for more details.
 
     You should have received a copy of the BSD 3 Clause License
     along with UQTk. If not, see https://choosealicense.com/licenses/bsd-3-clause/.
 
     Questions? Contact the UQTk Developers at https://github.com/sandialabs/UQTk/discussions
     Sandia National Laboratories, Livermore, CA, USA
===================================================================================== */
 
#ifndef PCBASIS_H_SEEN
#define PCBASIS_H_SEEN
 
#include <iostream>
#include <string.h>
#include "Array1D.h"
#include "Array2D.h"
// #include "Array3D.h"
#include "ftndefs.h"
#include "dsfmt_add.h"
 
 
class PCBasis {
public:
  PCBasis(const string type="LU", const double alpha=0.0, const double betta=1.0, const int maxord=10);
 
 
  ~PCBasis() {};
 
  void Init1dQuadPoints(int qdpts);
 
  void Eval1dBasisAtQuadPoints();
 
  void Eval1dBasisAtCustPoints(Array2D<double>& psi,int kord, const Array1D<double>& custPoints);
 
  double EvalBasis(const double &xi, Array1D<double> &basisEvals) const;
  double EvalBasis(const double &xi, const int kord, double *basisEvals) const;
 
  void Eval1dNormSq_Exact(int kord);
 
 
  /***************************************************
  New derivative functionality
  ***************************************************/
  void EvalDerivBasis(const double& xi, Array1D<double>& basisDEvals);
  void Eval1dDerivBasisAtCustPoints(Array2D<double>& dpsi,int kord, const Array1D<double>& custPoints);
 
  void Eval2ndDerivBasis(const double& xi,Array1D<double>& ddP);
  void Eval2ndDerivCustPoints(Array2D<double>& psi, int kord, Array1D<double>& custPoints);
  /***************************************************
  ***************************************************/
 
  void Get1dNormsSq(Array1D<double>& psi1dSq) const {psi1dSq=psi1dSq_; return;}
 
  void Get1dNormsSqExact(Array1D<double>& psi1dSqExact) const {psi1dSqExact=psi1dSqExact_; return;}
 
  void GetRandSample(Array1D<double>& randSamples);
 
  void GetRandSample(double* randSamples, const int& nSamp);
 
  int GetSeed() const {return rSeed_;}
 
  void SeedRandNumGen(const int& seed);
 
  void GetQuadRule(Array2D<double>& qPoints, Array1D<double>& qWeights, Array2D<int>& qIndices);
 
  void GetQuadPoints(Array2D<double>& quadPoints) const { quadPoints=quadPoints_; return;}
 
  void GetQuadWeights(Array1D<double>& quadWeights) const { quadWeights=quadWeights_; return;}
 
  void GetQuadIndices(Array2D<int>& quadIndices) const { quadIndices=quadIndices_; return;}
 
  void GetBasisAtQuadPoints(Array2D<double>& psi1d) const { psi1d=psi1d_; return;}
 
  string GetPCType() const {return type_;}
 
  double GetAlpha() const {return alpha_;}
 
  double GetBeta() const {return beta_;}
 
private:
  // PCBasis(): type_("NA") {};
 
  PCBasis(const PCBasis &obj):type_(obj.type_) {};
 
 
  void Eval1dNormSq(int kord);
 
 
  double NormSq_Exact(int kord);
 
  string type_;
 
  Array2D<double> quadPoints_;
 
  Array1D<double> quadWeights_;
 
  Array2D<int> quadIndices_;
 
 
  Array2D<double> psi1d_;
 
  Array1D<double> psi1dSq_;
 
  Array1D<double> psi1dSqExact_;
 
  int maxord_;
 
  int narg_;
 
  double alpha_;
 
  double beta_;
 
  dsfmt_t  rnstate_ ;
 
  int rSeed_;
 
};
 
#endif /* PCBASIS_H_SEEN */