File: GaussianGridder.h

package info (click to toggle)
esys-particle 2.3.4%2Bdfsg1-4
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 13,036 kB
  • ctags: 10,805
  • sloc: cpp: 80,009; python: 5,872; makefile: 1,243; sh: 313; perl: 225
file content (75 lines) | stat: -rw-r--r-- 2,526 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/////////////////////////////////////////////////////////////
//                                                         //
// Copyright (c) 2003-2014 by The University of Queensland //
// Centre for Geoscience Computing                         //
// http://earth.uq.edu.au/centre-geoscience-computing      //
//                                                         //
// Primary Business: Brisbane, Queensland, Australia       //
// Licensed under the Open Software License version 3.0    //
// http://www.apache.org/licenses/LICENSE-2.0          //
//                                                         //
/////////////////////////////////////////////////////////////

#ifndef ESYS_LSMGAUSSIANGRIDDER_H
#define ESYS_LSMGAUSSIANGRIDDER_H

#include "Foundation/vec3.h"

#include <math.h>

namespace esys
{
  namespace lsm
  {
    class GaussianGridder
    {
    public:
      GaussianGridder(double stdDeviation)
        : m_stdDeviation(stdDeviation)
      {
      }

      static const double SQRT_2PI;
      
      double getWeight(const Vec3 &regPos, const Vec3 &irrPos) const
      {
        return exp( - ((irrPos-regPos).norm2())/(2.0*m_stdDeviation*m_stdDeviation))/(SQRT_2PI*m_stdDeviation);
      }

      template <typename TmplCartesianGrid>
      void transform(const TmplCartesianGrid &irregular, TmplCartesianGrid &regular) const
      {
        typename TmplCartesianGrid::CellIterator regIt = regular.getCellIterator();
        const double radius = 3.99*m_stdDeviation;
        while (regIt.hasNext())
        {
          const typename TmplCartesianGrid::Cell &regCell = regIt.next();
          typename TmplCartesianGrid::CellConstIterator irrIt = irregular.getCellIterator(regCell.getPos(), radius);
          typename TmplCartesianGrid::value_type value;
          value *= 0.0;
          double weightSum = 0.0;
          while (irrIt.hasNext())
          {
            typename TmplCartesianGrid::Cell::ConstIterator pairIt = irrIt.next().getIterator();
            while (pairIt.hasNext())
            {
              const typename TmplCartesianGrid::Cell::PosValuePair &pair = pairIt.next();
              const double weight = getWeight(regCell.getPos(), pair.getPos());
              weightSum += weight;
              value += (pair.getValue()*weight);
            }
          }
          if (weightSum != 0) {
            value /= weightSum;
          }
          regular.insert(regCell.getPos(), value);
        }
      }

    private:
      double m_stdDeviation;
    };
  }
}

#endif