NodalBasis.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//               Rapid Optimization Library (ROL) Package
//
// Copyright 2014 NTESS and the ROL contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#include<iostream>
#include"ROL_StdVector.hpp"
#include"OrthogonalPolynomials.hpp"
#include"Lagrange.hpp"
#include"LinearAlgebra.hpp"
 
#ifndef __NODAL_BASIS__
#define __NODAL_BASIS__
 
 
template<class Real>
struct NodalBasis{

        ROL::Ptr<Teuchos::LAPACK<int,Real> > lapack_;
 
        const int ni_;

        const int nq_;
 
        NodalBasis(ROL::Ptr<Teuchos::LAPACK<int,Real> > lapack, const int ni, const int nq);    
        ~NodalBasis();

        std::vector<Real> xi_;

        std::vector<Real> xq_;

        std::vector<Real> wq_;

        std::vector<Real> L_;
        std::vector<Real> Lp_;
        
        ROL::Ptr<Lagrange<Real> > lagrange_; 
 
         
};

template<class Real>
NodalBasis<Real>::NodalBasis(ROL::Ptr<Teuchos::LAPACK<int,Real> > const lapack, const int ni, const int nq): 
    lapack_(lapack), ni_(ni), nq_(nq), xi_(ni_,0), xq_(nq_,0), wq_(nq_,0), L_(ni_*nq_,0), Lp_(ni_*nq_,0)
    {
 
    // Generate Legendre-Gauss-Lobatto nodes and weights (unused)
    std::vector<Real> ai(ni_,0);
    std::vector<Real> bi(ni_,0);
    std::vector<Real> wi(ni_,0);
    rec_jacobi(this->lapack_,0,0,ai,bi); 
    rec_lobatto(this->lapack_,-1.0,1.0,ai,bi);
    gauss(this->lapack_,ai,bi,xi_,wi);
 
    // Generate Legendre-Gauss nodes and weights
    std::vector<Real> aq(nq_,0);
    std::vector<Real> bq(nq_,0);
    rec_jacobi(this->lapack_,0,0,aq,bq); 
    gauss(this->lapack_,aq,bq,xq_,wq_);
 
    std::vector<Real> e(ni_,0);
    std::vector<Real> ell(nq,0);
 
    lagrange_ = ROL::makePtr<Lagrange<Real>>(xi_,xq_);
 
    // Loop over canonical vectors
    for(int i=0;i<ni_;++i) {
 
        lagrange_->interpolant(i,ell);
 
        std::copy(ell.begin(),ell.end(),L_.begin()+i*nq_);
      
 
        lagrange_->derivative(i,ell);
        std::copy(ell.begin(),ell.end(),Lp_.begin()+i*nq_);
 
        // Rezero the vector 
        std::fill(e.begin(),e.end(),0);
    }     
 
}
 
template<class Real>
NodalBasis<Real>::~NodalBasis(){
}
 
#endif