pce_example.cpp

Go to the documentation of this file.
00001 // $Id$ 
00002 // $Source$ 
00003 // @HEADER
00004 // ***********************************************************************
00005 // 
00006 //                           Sacado Package
00007 //                 Copyright (2006) Sandia Corporation
00008 // 
00009 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
00010 // the U.S. Government retains certain rights in this software.
00011 // 
00012 // This library is free software; you can redistribute it and/or modify
00013 // it under the terms of the GNU Lesser General Public License as
00014 // published by the Free Software Foundation; either version 2.1 of the
00015 // License, or (at your option) any later version.
00016 //  
00017 // This library is distributed in the hope that it will be useful, but
00018 // WITHOUT ANY WARRANTY; without even the implied warranty of
00019 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00020 // Lesser General Public License for more details.
00021 //  
00022 // You should have received a copy of the GNU Lesser General Public
00023 // License along with this library; if not, write to the Free Software
00024 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
00025 // USA
00026 // Questions? Contact David M. Gay (dmgay@sandia.gov) or Eric T. Phipps
00027 // (etphipp@sandia.gov).
00028 // 
00029 // ***********************************************************************
00030 // @HEADER
00031 
00032 // pce_example
00033 //
00034 //  usage: 
00035 //     pce_example
00036 //
00037 //  output:  
00038 //     prints the Hermite Polynomial Chaos Expansion of a simple function
00039 //     using the class Sacado::PCE::Hermite
00040 
00041 #include <iostream>
00042 
00043 #include "Sacado.hpp"
00044 #include "Sacado_PCE_OrthogPoly.hpp"
00045 #include "Stokhos.hpp"
00046 
00047 int main(int argc, char **argv)
00048 {
00049   try {
00050     const unsigned int d = 7;
00051     typedef Stokhos::HermiteBasis<int,double> basis_type;
00052     Teuchos::Array< Teuchos::RCP<const Stokhos::OneDOrthogPolyBasis<int,double> > > bases(1); 
00053     bases[0] = Teuchos::rcp(new basis_type(d));
00054     Teuchos::RCP<const Stokhos::CompletePolynomialBasis<int,double> > basis = 
00055       Teuchos::rcp(new Stokhos::CompletePolynomialBasis<int,double>(bases));
00056     Teuchos::RCP<Stokhos::OrthogPolyExpansion<int,double> > expansion = 
00057       Teuchos::rcp(new Stokhos::DerivOrthogPolyExpansion<int,double>(basis));
00058     Sacado::PCE::OrthogPoly<double> u(expansion);
00059     u.fastAccessCoeff(0) = 1.0;
00060     u.fastAccessCoeff(1) = 0.4;
00061     u.fastAccessCoeff(2) = 0.06;
00062     u.fastAccessCoeff(3) = 0.002;
00063 
00064     Sacado::PCE::OrthogPoly<double> w = std::log(u);
00065     Sacado::PCE::OrthogPoly<double> v = std::sinh(1.0/(w*w + 1.0));
00066 
00067     std::cout << "u (hermite basis) = " << u;
00068     std::cout << "v (hermite basis) = " << v;
00069 
00070     std::cout << "\nExample passed!" << std::endl;
00071   }
00072   catch (std::exception& e) {
00073     std::cout << e.what() << std::endl;
00074   }
00075 }

Generated on Wed May 12 21:39:33 2010 for Sacado Package Browser (Single Doxygen Collection) by  doxygen 1.4.7