MpiSimulation.hpp

#ifndef MPI_SIMULATION_HPP
#define MPI_SIMULATION_HPP


#include <mpi.h>
#include <Simulation.hpp>
#include <BarnesHutTree.hpp>
#include <vector>

namespace nbody {
	using namespace std;

	typedef struct _SendStore {
		Body* bodies;
		MPI_Request request;
		int size;
	} SendStore;

	class MpiSimulation : public Simulation {
	protected:
		MPI_Datatype bodyType;
		MPI_Datatype boxType;
		Box* domains;
		Box overallDomain;
		vector<SendStore> sendStores;

		virtual SendStore* availableSendStore(int numElems);
		virtual void send(vector<Body> bodies, int target);
		virtual void recv(vector<Body>& bodies);
	public:
		MpiSimulation(int& argc, char**& argv);
		virtual ~MpiSimulation();
		virtual void cleanup();
		virtual int getNumberOfProcesses();
		virtual int getProcessId();
		virtual bool stateCorrect();
		virtual void distributeBodies();
		virtual void distributeDomains(vector<Body> localBodies);
		virtual void distributeDomains(Box localDomain);
		virtual void distributeDomains();
		virtual void distributeLETs();
		virtual void rebuildTree();
		virtual void buildTree();
		virtual void runStep();
		virtual MPI_Datatype* getDatatype();
	};
}

#endif