Adding missing files

src/Grid/grid_dist_id_iterator_sub.hpp

+/*
+ * grid_dist_id_iterator_sub.hpp
+ *
+ *  Created on: Oct 14, 2015
+ *      Author: i-bird
+ */
+
+#ifndef SRC_GRID_GRID_DIST_ID_ITERATOR_SUB_HPP_
+#define SRC_GRID_GRID_DIST_ID_ITERATOR_SUB_HPP_
+
+
+/*! \brief Distributed grid iterator
+ *
+ * Iterator across the local elements of the distributed grid
+ *
+ * \tparam dim dimensionality of the grid
+ * \tparam device_grid type of basic grid
+ * \tparam impl implementation
+ *
+ */
+template<unsigned int dim, typename device_grid>
+class grid_dist_iterator_sub
+{
+	// sub_set of the grid where to iterate
+	struct sub_set
+	{
+		//! start point where iterate
+		grid_key_dx<dim> start;
+		// ! stop point where iterate
+		grid_key_dx<dim> stop;
+	};
+
+	//! grid list counter
+	size_t g_c;
+
+	//! List of the grids we are going to iterate
+	Vcluster_object_array<device_grid> & gList;
+
+	//! Extension of each grid: domain and ghost + domain
+	openfpm::vector<GBoxes<device_grid::dims>> & gdb_ext;
+
+	//! Actual iterator
+	grid_key_dx_iterator_sub<dim> a_it;
+
+	//! start key
+	grid_key_dx<dim> start;
+
+	//! stop key
+	grid_key_dx<dim> stop;
+
+	//! margin of the grid iterator
+	size_t m;
+
+	/*! \brief compute the subset where it has to iterate
+	 *
+	 * \param g_c Actual grid
+	 * \param start_c adjusted start point for the grid g_c
+	 * \param stop_c adjusted stop point for the grid g_c
+	 *
+	 * \return false if the sub-set does not contain points
+	 *
+	 */
+	bool compute_subset(size_t gc, grid_key_dx<dim> & start_c, grid_key_dx<dim> & stop_c)
+	{
+		// Intersect the grid keys
+
+		for (size_t i = 0 ; i < dim ; i++)
+		{
+			long int start_p = gdb_ext.get(g_c).Dbox.getP1().get(i) + gdb_ext.get(g_c).origin.get(i);
+			long int stop_p = gdb_ext.get(g_c).Dbox.getP2().get(i) + gdb_ext.get(g_c).origin.get(i);
+			if (start.get(i) <= start_p)
+				start_c.set_d(i,gdb_ext.get(g_c).Dbox.getP1().get(i));
+			else if (start.get(i) <= stop_p)
+				start_c.set_d(i,start.get(i) - gdb_ext.get(g_c).origin.get(i));
+			else
+				return false;
+
+			if (stop.get(i) >= stop_p)
+				stop_c.set_d(i,gdb_ext.get(g_c).Dbox.getP2().get(i));
+			else if (stop.get(i) >= start_p)
+				stop_c.set_d(i,stop.get(i) - gdb_ext.get(g_c).origin.get(i));
+			else
+				return false;
+		}
+
+		return true;
+	}
+
+	/*! \brief from g_c increment g_c until you find a valid grid
+	 *
+	 */
+	void selectValidGrid()
+	{
+		// start and stop for the subset grid
+		grid_key_dx<dim> start_c;
+		grid_key_dx<dim> stop_c;
+
+		// When the grid has size 0 potentially all the other informations are garbage
+		while (g_c < gList.size() &&
+			   (gList[g_c].size() == 0 || gdb_ext.get(g_c).Dbox.isValid() == false || compute_subset(g_c,start_c,stop_c) == false ))
+		{g_c++;}
+
+		// get the next grid iterator
+		if (g_c < gList.size())
+		{
+			a_it.reinitialize(gList[g_c].getIterator(start_c,stop_c));
+		}
+	}
+
+	public:
+
+	/*! \brief Copy operator=
+	*
+	* \param tmp iterator to copy
+	*
+	*/
+	grid_dist_iterator_sub<dim,device_grid> & operator=(const grid_dist_iterator_sub<dim,device_grid> & tmp)
+	{
+		g_c = tmp.g_c;
+		gList = tmp.gList;
+		gdb_ext = tmp.gdb_ext;
+		a_it.reinitialize(tmp.a_it);
+		m = tmp.m;
+
+		return *this;
+	}
+
+	/*! \brief Copy constructor
+	*
+	* \param tmp iterator to copy
+	*
+	*/
+	grid_dist_iterator_sub(const grid_dist_iterator_sub<dim,device_grid> & tmp)
+	:g_c(tmp.g_c),gList(tmp.gList),gdb_ext(gdb_ext),m(tmp.m)
+	{
+		a_it.reinitialize(tmp.a_it);
+	}
+
+	/*! \brief Constructor of the distributed grid iterator
+	 *
+	 * \param start position
+	 * \param stop position
+	 * \param gk std::vector of the local grid
+	 * \param gdb_ext information about the local grids
+	 *
+	 */
+	grid_dist_iterator_sub(const grid_key_dx<dim> & start, const grid_key_dx<dim> & stop ,Vcluster_object_array<device_grid> & gk, openfpm::vector<GBoxes<device_grid::dims>> & gdb_ext)
+	:g_c(0),gList(gk),gdb_ext(gdb_ext),start(start),stop(stop),m(0)
+	{
+		// Initialize the current iterator
+		// with the first grid
+		selectValidGrid();
+	}
+
+	// Destructor
+	~grid_dist_iterator_sub()
+	{
+	}
+
+	/*! \brief Get the next element
+	 *
+	 * \return the next grid_key
+	 *
+	 */
+
+	inline grid_dist_iterator_sub<dim,device_grid> operator++()
+	{
+		++a_it;
+
+		// check if a_it is at the end
+
+		if (a_it.isNext() == true)
+			return *this;
+		else
+		{
+			// switch to the new grid
+			g_c++;
+
+			selectValidGrid();
+		}
+
+		return *this;
+	}
+
+	/*! \brief Check if there is the next element
+	 *
+	 * \return true if there is the next, false otherwise
+	 *
+	 */
+	inline bool isNext()
+	{
+		// If there are no other grid stop
+
+		if (g_c >= gList.size())
+			return false;
+
+		return true;
+	}
+
+	/*! \brief Get the actual key
+	 *
+	 * \return the actual key
+	 *
+	 */
+	inline grid_dist_key_dx<dim> get()
+	{
+		return grid_dist_key_dx<dim>(g_c,a_it.get());
+	}
+
+	/*! \brief Convert a g_dist_key_dx into a global key
+	 *
+	 * \see grid_dist_key_dx
+	 * \see grid_dist_iterator
+	 *
+	 * \return the global position in the grid
+	 *
+	 */
+	inline grid_key_dx<dim> getGKey(const grid_dist_key_dx<dim> & k)
+	{
+		// Get the sub-domain id
+		size_t sub_id = k.getSub();
+
+		grid_key_dx<dim> k_glob = k.getKey();
+
+		// shift
+		k_glob = k_glob + gdb_ext.get(sub_id).origin;
+
+		return k_glob;
+	}
+};
+
+
+
+
+
+#endif /* SRC_GRID_GRID_DIST_ID_ITERATOR_SUB_HPP_ */