Initialize at openlb-1-3

1 files changed, 50 insertions, 0 deletions

diff --git a/src/particles/forces/weightForce3D.hh b/src/particles/forces/weightForce3D.hh
new file mode 100644
index 0000000..a5f098e
--- /dev/null
+++ b/src/particles/forces/weightForce3D.hh
@@ -0,0 +1,50 @@
+/*  This file is part of the OpenLB library
+ *
+ *  Copyright (C) 2016 Thomas Henn
+ *  E-mail contact: info@openlb.net
+ *  The most recent release of OpenLB can be downloaded at
+ *  <http://www.openlb.net/>
+ *
+ *  This program is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU General Public License
+ *  as published by the Free Software Foundation; either version 2
+ *  of the License, or (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public
+ *  License along with this program; if not, write to the Free
+ *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ *  Boston, MA  02110-1301, USA.
+*/
+
+#ifndef WEIGHTFORCE_3D_HH
+#define WEIGHTFORCE_3D_HH
+
+#include <cmath>
+#include "weightForce3D.h"
+
+namespace olb {
+
+template<typename T, template<typename U> class PARTICLETYPE>
+WeightForce3D<T, PARTICLETYPE>::WeightForce3D(std::vector<T> direction, T g) :  Force3D<T, PARTICLETYPE>(), _direction(direction), _g(g)
+{
+  T directionNorm = sqrt(pow(_direction[0], 2.)+pow(_direction[1], 2.)+pow(_direction[2], 2.));
+  for (int i=0; i<3; ++i) {
+    _direction[i]/=directionNorm;
+  }
+}
+
+template<typename T, template<typename U> class PARTICLETYPE>
+void WeightForce3D<T, PARTICLETYPE>::applyForce(typename std::deque<PARTICLETYPE<T> >::iterator p, int pInt, ParticleSystem3D<T, PARTICLETYPE>& psSys)
+{
+  for (int j=0; j<3; ++j) {
+    p->getForce()[j] += p->getMass() * _g * _direction[j];
+  }
+}
+
+}
+#endif /* WEIGHTFORCE_3D_HH */