File: StressTensorCalculator.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 (107 lines) | stat: -rw-r--r-- 3,063 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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/////////////////////////////////////////////////////////////
//                                                         //
// 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_LSMSTRESSTENSORCACLULATOR_H
#define ESYS_LSMSTRESSTENSORCACLULATOR_H

#include "Foundation/vec3.h"
#include "Foundation/Matrix3.h"
#include "Tools/StressCalculator/StressTensor.h"

#include <vector>
#include <map>

namespace esys
{
  namespace lsm
  {
    class IntercentreStressTensorCalculator
    {
    public:

      IntercentreStressTensorCalculator()
      {
      }

      template <typename TmplContactReference>
      void updateTensor(Matrix3 &tensor, const Vec3 &tensorPos, TmplContactReference &contact)
      {
          const Vec3 r = contact.getCentrePos2() - tensorPos;
          for (int i = 0; i < 3; i++) {
            for (int j = 0; j < 3; j++) {
              tensor(i,j) += contact.getForce()[i] * r[j];
            }
          }
      }
      
      template <typename TmplContactIterator>
      StressTensor calculate(TmplContactIterator &it)
      {
        Matrix3 tensor;
        ParticleData tensorData;
        if (it.hasNext())
        {
          tensorData = it.next().getParticle1();
          updateTensor(tensor, tensorData.getPos(), it.current());
        }
        
        while (it.hasNext())
        {
          updateTensor(tensor, tensorData.getPos(), it.next());
        }
        return StressTensor(tensorData, tensor);
      }
    };

    class ContactPtTensorCalculator
    {
    public:

      ContactPtTensorCalculator()
      {
      }

      template <typename TmplContactReference>
      void updateTensor(Matrix3 &tensor, const Vec3 &tensorPos, TmplContactReference &contact)
      {
        const Vec3 r = contact.getForcePos() - tensorPos;
        for (int i = 0; i < 3; i++) {
          for (int j = 0; j < 3; j++) {
            tensor(i,j) += r[i] * contact.getForce()[j];
          }
        }
      }

      template <typename TmplContactIterator>
      StressTensor calculate(TmplContactIterator &it)
      {
        Matrix3 tensor;
        ParticleData tensorData;
        double volume = 1.0;
        if (it.hasNext())
        {
          tensorData = it.next().getParticle1();
          volume     = it.current().getVolume1();
          updateTensor(tensor, tensorData.getPos(), it.current());
        }
        while (it.hasNext())
        {
          updateTensor(tensor, tensorData.getPos(), it.next());
        }
        return StressTensor(tensorData, tensor/volume);
      }
    };    
  }
}

#endif