Some customization for HIER

Author: lkotipal

dccrg.hpp

@@ -7489,6 +7489,7 @@ template <
 	unsigned int neighborhood_length = 1;
 
 	std::string load_balancing_method{"RCB"};
+	int hierarchical_level_zero {16};
 
 	// maximum value an MPI tag can have
 	unsigned int max_tag;
@@ -7963,12 +7964,12 @@ template <
 
 		// Hardcoded for now
 		if (this->load_balancing_method == "HIER") {
-			add_partitioning_level(4);	// Level 0 - Nodes
+			add_partitioning_level(this->hierarchical_level_zero);	// Level 0 - Nodes
 			add_partitioning_option(0, "LB_METHOD", "HYPERGRAPH");
 
 			add_partitioning_level(1);	// Level 1 - Processes
-			add_partitioning_option(1, "LB_METHOD", "RCB");
-			//add_partitioning_option(1, "LB_METHOD", "RIB");
+			add_partitioning_option(1, "LB_METHOD", "RIB");
+			//add_partitioning_option(1, "LB_METHOD", "RCB");
 		}
 
 		// reserved options that the user cannot change
@@ -8498,6 +8499,23 @@ template <
 	const std::string& get_load_balancing_method() const {
 		return this->load_balancing_method;
 	}
+	
+	/*!
+	Sets amount of tasks for node-level of HIER partition for Zoltan
+
+	Must be called before initialize().
+
+	\see balance_load()
+	*/
+	Dccrg<
+		Cell_Data,
+		Geometry,
+		std::tuple<Additional_Cell_Items...>,
+		std::tuple<Additional_Neighbor_Items...>
+	>& set_load_balancing_hier_level_zero(const int n) {
+		this->hierarchical_level_zero = n;
+		return *this;
+	}
 
 
 private: