Skip to content
Snippets Groups Projects
main.cpp 655 B
Newer Older
#include "datastructures/BarnesHutTree.hpp"
#include <iostream>

using namespace nbody;
Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
using namespace std;
int main(int argc, char* argv[]) {
	BarnesHutTree tree;
Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
	if (argc == 2) {
		vector<Body> bodies = Tree::dubinskiParse(argv[1]);

Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
		cout << bodies.size() << endl;
Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
		tree.build(bodies);
		if (!tree.isCorrect()) {
Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
			cout << "Tree not correct" << endl;
Paul Heinzlreiter's avatar
Paul Heinzlreiter committed
		}
		tree.print();
		tree.computeForces();
		tree.moveBodies();
		tree.rebuildTree();
		if (!tree.isCorrect()) {
			cout << "Tree not correct" << endl;
			return -1;
		}
		tree.print();
	} else {
		std::cout << "You need to specify the body input file." << endl;
	}