* node representation

0f4bd471 · Paul Heinzlreiter · 1fb38c2f · 0f4bd471 · 0f4bd471 · 0f4bd471
Commit 0f4bd471 authored 8 years ago by Paul Heinzlreiter
--- a/bh_tree_mpi/datastructures/Body.cpp
+++ b/bh_tree_mpi/datastructures/Body.cpp
@@ -116,6 +116,19 @@ namespace nbody {
 		return this->position[index];
 	}

+	double Body::getMass() {
+		return this->weight;
+	}
+
+	void Body::setPosition(int index, double value) {
+		this->position[index] = value;
+	}
+
+	void Body::setMass(double value) {
+		this->weight = value;
+	}
+
+
 	void Body::print() {
 		cout << this->position[0] << " " << this->position[1] << " " << this->position[2] << endl;
 	}

--- a/bh_tree_mpi/datastructures/Body.hpp
+++ b/bh_tree_mpi/datastructures/Body.hpp
 #ifndef BODY_HPP
 #define BODY_HPP

+
 namespace nbody {
 	static const double timestep = 1.0;
 	class Derivative;
@@ -21,6 +22,9 @@ namespace nbody {
 		Body(double positionX, double positionY, double positionZ, double velocityX, double velocityY, double velocityZ, double weight);
 		virtual ~Body();
 		virtual double getPosition(int index);
+		virtual double getMass();
+		virtual void setPosition(int index, double value);
+		virtual void setMass(double value);
 		virtual void integrate();
 		virtual void print();
 	};

--- a/bh_tree_mpi/datastructures/TreeNode.cpp
+++ b/bh_tree_mpi/datastructures/TreeNode.cpp
@@ -133,6 +133,12 @@ namespace nbody {
 		return true;
 	}

+	void TreeNode::update() {
+		if (this->leaf) {
+
+		}
+	}
+
 	void TreeNode::print() {
 		this->bb.print();
 		for (vector<Body>::iterator it = this->bodies.begin(); it != this->bodies.end(); it++) {

--- a/bh_tree_mpi/datastructures/TreeNode.hpp
+++ b/bh_tree_mpi/datastructures/TreeNode.hpp
@@ -24,6 +24,7 @@ namespace nbody {
 		TreeNode* afterSubtree;
 		bool leaf;
 		Tree* tree;
+		Body representative;
 	public:
 		TreeNode(Tree* tree);
 		virtual ~TreeNode();
@@ -35,6 +36,7 @@ namespace nbody {
 		virtual void insertBefore(TreeNode* node);
 		virtual void insertAfter(TreeNode* node);
 		virtual void unlink();
+		virtual void update();
 		virtual bool isCorrect();
 		virtual void print();
 	};