OctreeStaticNodeSelector.h

Go to the documentation of this file.

 
#pragma once
 
#include <memory>
 
#include "autopas/containers/octree/OctreeInnerNode.h"
#include "autopas/containers/octree/OctreeLeafNode.h"
#include "autopas/containers/octree/OctreeNodeWrapper.h"
 
namespace autopas {
template <typename Particle_T>
class OctreeInnerNode;
 
template <typename Particle_T, typename FunctionType>
decltype(auto) withStaticNodeType(const std::unique_ptr<OctreeNodeInterface<Particle_T>> &root,
                                  FunctionType &&function) {
  OctreeNodeInterface<Particle_T> *nodePtr = root.get();
  // TODO: These should be static casts because we already do the runtime check via hasChildren()
  if (nodePtr->hasChildren()) {
    return function(dynamic_cast<OctreeInnerNode<Particle_T> *>(nodePtr));
  } else {
    return function(dynamic_cast<OctreeLeafNode<Particle_T> *>(nodePtr));
  }
}
}  // namespace autopas