00001 /* @HEADER@ */ 00002 // ************************************************************************ 00003 // 00004 // Sundance 00005 // Copyright (2005) Sandia Corporation 00006 // 00007 // Copyright (year first published) Sandia Corporation. Under the terms 00008 // of Contract DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government 00009 // retains certain rights in this software. 00010 // 00011 // This library is free software; you can redistribute it and/or modify 00012 // it under the terms of the GNU Lesser General Public License as 00013 // published by the Free Software Foundation; either version 2.1 of the 00014 // License, or (at your option) any later version. 00015 // 00016 // This library is distributed in the hope that it will be useful, but 00017 // WITHOUT ANY WARRANTY; without even the implied warranty of 00018 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 00019 // Lesser General Public License for more details. 00020 // 00021 // You should have received a copy of the GNU Lesser General Public 00022 // License along with this library; if not, write to the Free Software 00023 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 00024 // USA 00025 // Questions? Contact Kevin Long (krlong@sandia.gov), 00026 // Sandia National Laboratories, Livermore, California, USA 00027 // 00028 // ************************************************************************ 00029 /* @HEADER@ */ 00030 00031 #ifndef SUNDANCE_ATOCDENSITYSAMPLER_H 00032 #define SUNDANCE_ATOCDENSITYSAMPLER_H 00033 00034 #include "SundanceDefs.hpp" 00035 #include "SundanceMesh.hpp" 00036 #include "SundanceExpr.hpp" 00037 #include "SundanceDiscreteFunction.hpp" 00038 #include "SundanceDiscreteSpace.hpp" 00039 #include "SundanceAToCPointLocator.hpp" 00040 00041 namespace Sundance 00042 { 00043 using namespace Teuchos; 00044 using namespace Playa; 00045 00046 /** 00047 * AToCDensitySampler samples a distribution of particles to compute a 00048 * density function on a discrete space. 00049 * 00050 * Note: not tested in parallel. 00051 */ 00052 class AToCDensitySampler 00053 { 00054 public: 00055 /** */ 00056 AToCDensitySampler(const AToCPointLocator& locator, 00057 const VectorType<double>& vecType); 00058 /** */ 00059 AToCDensitySampler(const AToCPointLocator& locator, 00060 const std::vector<double>& origin, 00061 const std::vector<double>& rotationalAxis, 00062 const VectorType<double>& vecType); 00063 00064 /** */ 00065 Expr sample(const std::vector<double>& positions, 00066 const double& particleWeight) const ; 00067 00068 /** */ 00069 Expr resetCounts() const ; 00070 00071 /** */ 00072 void addToCounts(const std::vector<double>& positions, 00073 const double& particleWeight, 00074 Expr density) const ; 00075 00076 00077 private: 00078 void init(); 00079 Point vec2point(const std::vector<double>& x) const ; 00080 Point normPoint(const Point& x) const ; 00081 00082 DiscreteSpace discSpace_; 00083 int dim_; 00084 Mesh mesh_; 00085 RCP<Array<int> > elemToVecIndexMap_; 00086 Expr elemWeights_; 00087 Vector<double> elemWeightVec_; 00088 AToCPointLocator locator_; 00089 bool isAxisymmetric_; 00090 Point origin_; 00091 Point axis_; 00092 }; 00093 } 00094 00095 00096 #endif