Newer
Older
#include <cmath>
#include <iostream>
#include "Node.hpp"
namespace nbody {
Node::Node(Tree* tree) {
initBox(this->bb);
this->afterSubtree = nullptr;
this->prev = this;
this->next = this;
this->leaf = true;
this->tree = tree;
this->prevSibling = nullptr;
this->nextSibling = nullptr;
this->parent = nullptr;
//check if node needs to be splitted during tree build
bool result = true;
if (this->bodies.size() <= this->tree->maxLeafBodies) {
result = false;
}
//this is to prevent errors with collocated particles
if (volume(this->bb) <= std::numeric_limits<float>::epsilon()) {
result = false;
}
return result;
}
void Node::extendBBforBodies() {
extendForBodies(this->bb, this->bodies);
}
void Node::extendBBtoCube() {
extendToCube(this->bb);
}
std::vector<Body> Node::getBodies() const {
return this->bodies;
}
void Node::insertBefore(Node* node) {
node->next = this;
node->prev = this->prev;
this->prev->next = node;
this->prev = node;
}
void Node::insertAfter(Node* node) {
this->next->insertBefore(node);
}
void Node::unlink() {
this->next->prev = this->prev;
this->prev->next = this->next;
this->next = this;
this->prev = this;
}
return false;
}
if (!isCorrectBox(this->bb)) {
return false;
}
for (int i = 0; i < 3; i++) {
if (this->bb.min[i] > this->bb.max[i]) {
std::cerr << "bb " << i << " min " << this->bb.min[i] << " max " << this->bb.max[i] << '\n';
if (std::any_of(std::begin(bodies), std::end(bodies), [&](const Body& b) {return !isContained(b, bb); })) {
std::cerr << "bb out of bounds\n";
return false;
}
if (!this->leaf) {
Node* current = this->next;
int children = 0;
while (current != nullptr && current != this->afterSubtree) {
current = current->afterSubtree;
children++;
}
return false;
}
if (children != this->tree->numberOfChildren()) {
std::cerr << "wrong number of children " << children << '\n';
return false;
}
current = this->next;
for (int i = 0; i < this->tree->numberOfChildren(); i++) {
current = current->afterSubtree;
}
if (current != this->afterSubtree) {
std::cerr << "last sibling afterSubtree inconsistent\n";
return false;
}
}
if (!this->leaf && this->bodies.size() > 0) {
if (this->leaf && this->nextSibling != nullptr && this->next != this->nextSibling) {
return false;
}
return true;
}
void Node::update() {
double position[3] = {0.0, 0.0, 0.0};
double mass = 0.0;
if (this->leaf) {
for (auto& body : bodies) {
mass += body.mass;
for (int i = 0; i < 3; i++) {
position[i] += body.position[i] * body.mass;
for (Node* node = this->next; node != nullptr; node = node->nextSibling) {
mass += node->representative.mass;
}
}
for (int i = 0; i < 3; i++) {
this->representative.position[i] = position[i] / mass;
}
this->representative.mass = mass;
}
//get criterion to check if node is sufficient for force evaluation
return maxSidelength(this->bb);
}
void Node::print(int parallelId) const {
printBB(parallelId, this->bb);
printBody(parallelId, body);
//check if node is sufficient for force evaluation
bool Node::sufficientForBody(const Body& body) const {
double distance = 0.0;
for (int i = 0; i < 3; i++) {
distance += (this->representative.position[i] - body.position[i]) * (this->representative.position[i] - body.position[i]);
}
return sqrt(distance) > this->getL();
}
//check if node is sufficient for force evaluation for all bodies in box
bool Node::sufficientForBox(const Box& box) const {
return distanceToBox(this->bb, box) > this->getL();
}
void Node::setBodies(const std::vector<Body>& bodies) {
this->bodies = bodies;
}
void Node::extractLocalBodiesTo(std::vector<Body>& result) {
std::copy_if(std::begin(this->bodies), std::end(this->bodies), std::back_inserter(result), [](const Body& b) {return !b.refinement; });
this->bodies.clear();