1 files changed, 85 insertions, 0 deletions
diff --git a/src/communication/heuristicLoadBalancer.h b/src/communication/heuristicLoadBalancer.h
new file mode 100644
index 0000000..f2e1fd9
--- /dev/null
+++ b/src/communication/heuristicLoadBalancer.h
@@ -0,0 +1,85 @@
+/*  This file is part of the OpenLB library
+ *
+ *  Copyright (C) 2012 Jonas Fietz, Mathias J. Krause
+ *  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 HEURISTIC_LOAD_BALANCER_H
+#define HEURISTIC_LOAD_BALANCER_H
+
+
+#include "communication/mpiManager.h"
+#include "geometry/cuboidGeometry2D.h"
+#include "geometry/cuboid3D.h"
+#include "geometry/cuboid3D.hh"
+#include "geometry/cuboidGeometry3D.h"
+#include "geometry/cuboidGeometry3D.hh"
+#include "communication/loadBalancer.h"
+
+
+
+
+namespace olb {
+
+/** Constructs a load balancer from a given cuboid geometry using a heurist.
+ * \param cGeometry     cuboid geometry to base the load balance on
+ * \param blockGeometry used to determine number of full and empty cells if given
+ * \param ratioFullEmpty time it takes for full cells in relation to empty cells
+ *
+ * This class has a virtual method call in its destructor and should therefore
+ * not be used as a base class.
+ */
+template<typename T>
+class HeuristicLoadBalancer final : public LoadBalancer<T> {
+private:
+  // Handles the MPI communication
+#ifdef PARALLEL_MODE_MPI
+  singleton::MpiNonBlockingHelper _mpiNbHelper;
+#endif
+  CuboidGeometry3D<T>* _cGeometry3d;
+  CuboidGeometry2D<T>* _cGeometry2d;
+
+  double _ratioFullEmpty;
+
+public:
+  HeuristicLoadBalancer() {};
+
+  ~HeuristicLoadBalancer() override;
+
+  HeuristicLoadBalancer(CuboidGeometry3D<T>& cGeometry3d, const double ratioFullEmpty=1., const double weightEmpty=.0);
+
+  HeuristicLoadBalancer(CuboidGeometry2D<T>& cGeometry2d, const double ratioFullEmpty=1., const double weightEmpty=.0);
+
+  void reInit(CuboidGeometry3D<T>& cGeometry3d, const double ratioFullEmpty=1., const double weightEmpty=.0) override;
+
+  void reInit(CuboidGeometry2D<T>& cGeometry2d, const double ratioFullEmpty=1., const double weightEmpty=.0) override;
+
+  /// Swap method
+  void swap(HeuristicLoadBalancer<T>& loadBalancer);
+
+  /// Write itself into Stringstream
+  void writeToStream(std::ostream& stream) override;
+  /// Write itself into File
+  void writeToFile(std::string fileName) override;
+};
+}  // namespace olb
+
+#endif