Skip to content
Snippets Groups Projects
BarnesHutTree.cpp 956 B
Newer Older
#include "BarnesHutTree.hpp"

namespace nbody {
	BarnesHutTree::BarnesHutTree() {
	}

	BarnesHutTree::~BarnesHutTree() {
	}

	vector<Box> BarnesHutTree::splitBB(TreeNode* node) {
		vector<Box> result;

		for (int i = 0; i < 8; i++) {
			Box current = node->getBB();

			for (int j = 0; j < 3; j++) {
				double middle = node->getBB().getMin(j) + (node->getBB().getMax(j) - node->getBB().getMin(j)) / 2.0;

				if (i & (1 >> j)) {
					current.setMin(j, middle);
				} else {
					current.setMax(j, middle);
				}
			}
			result.push_back(current);
		}
		return result;
	}

	void BarnesHutTree::build(vector<Body> bodies) {
Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
		vector<TreeNode*>::iterator it;

		this->clean();
		this->nodes.push_back(new TreeNode());
		std::copy(bodies.begin(), bodies.end(), this->nodes.front()->bodies.begin());
		this->nodes.front()->extendBBforBodies();
		this->nodes.front()->extendBBtoCube();
Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
		it = this->nodes.begin();
		while (it != this->nodes.end()) {

		}