Stokhos_OrthogPolyApprox.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Stokhos Package
//
// Copyright 2009 NTESS and the Stokhos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef STOKHOS_ORTHOGPOLYAPPROX_HPP
#define STOKHOS_ORTHOGPOLYAPPROX_HPP

#include "Teuchos_RCP.hpp"             // class data member
#include "Stokhos_OrthogPolyBasis.hpp" // class data member
#include <ostream>                 // for std::ostream

#include "Stokhos_StandardStorage.hpp"
#include "Stokhos_ProductBasisUtils.hpp"

namespace Stokhos {

  template <typename ordinal_type, typename value_type, 
      typename storage_type = Stokhos::StandardStorage<ordinal_type, 
                   value_type> >
  class OrthogPolyApprox {
  public:

    typedef typename storage_type::reference reference;
    typedef typename storage_type::const_reference const_reference;
    typedef typename storage_type::pointer pointer;
    typedef typename storage_type::const_pointer const_pointer;


    OrthogPolyApprox(
      const Teuchos::RCP<const Stokhos::OrthogPolyBasis<ordinal_type, value_type> >& basis = Teuchos::null,
      ordinal_type sz = 0,
      const value_type* vals = NULL);
    
    OrthogPolyApprox(const OrthogPolyApprox& x);
    
    ~OrthogPolyApprox();
    
    OrthogPolyApprox& operator=(const OrthogPolyApprox& x);

    OrthogPolyApprox& operator=(const value_type& v);

    void init(const value_type& v);

    void init(const value_type* v);

    template <typename S>
    void init(const OrthogPolyApprox<ordinal_type, value_type, S>& v) {
      coeff_.init(v.coeff());
    }

    void load(value_type* v);

    template <typename S>
    void load(OrthogPolyApprox<ordinal_type, value_type, S>& v) {
      coeff_.load(v.coeff());
    }

    Teuchos::RCP<const Stokhos::OrthogPolyBasis<ordinal_type, value_type> > 
    basis() const;


    void reset(const Teuchos::RCP<const Stokhos::OrthogPolyBasis<ordinal_type, value_type> >& new_basis, ordinal_type sz = 0);

    void resize(ordinal_type sz);

    ordinal_type size() const;

    pointer coeff();

    const_pointer coeff() const;

    reference operator[](ordinal_type i);

    const_reference operator[](ordinal_type i) const;

    reference term(ordinal_type dimension, ordinal_type order);

    const_reference term(ordinal_type dimension, ordinal_type order) const;

    const MultiIndex<ordinal_type>& order(ordinal_type term) const;

    value_type evaluate(const Teuchos::Array<value_type>& point) const;

    value_type evaluate(const Teuchos::Array<value_type>& point,
      const Teuchos::Array<value_type>& basis_vals) const;

    value_type mean() const;

    value_type standard_deviation() const;

    value_type two_norm() const;

    value_type two_norm_squared() const;

    value_type inner_product(const OrthogPolyApprox& b) const;

    std::ostream& print(std::ostream& os) const;

  protected:

    Teuchos::RCP<const Stokhos::OrthogPolyBasis<ordinal_type, value_type> > basis_;

    storage_type coeff_;
    
  }; // class OrthogPolyApprox

  template <typename ordinal_type, typename value_type, typename node_type> 
  std::ostream& 
  operator << (std::ostream& os, 
         const OrthogPolyApprox<ordinal_type,value_type,node_type>& a);

} // namespace Stokhos

#include "Stokhos_OrthogPolyApproxImp.hpp"

#endif //  STOKHOS_ORTHOGPOLYAPPROX_HPP