spatialVrfObjectManager.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2019 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/

#pragma once


#include "vrfobjcore/vrfobjcoreDefines.h"
#include <vlutil/vlConfig.h>
#include "geometry/spatialSubdivision.h"
#include <vrfobjcore/managedObjectList.h>
#include <list>
#include <vector>
#include <boost/unordered_set.hpp>
#include "vrfobjcore/octree.h"
#include "vrfobjcore/simObject.h"


class DT_DLL_vrfobjcore DtSpatialSelectionFunctor : public DtSpatialSelectionFunctorTemplate<const DtSimObject>
{
public:
   virtual ~DtSpatialSelectionFunctor() {};

   virtual bool operator()(const DtSimObject *object) = 0;
};


class DT_DLL_vrfobjcore DtSpatialVrfObjectManagerInterface
{
public:
   virtual ~DtSpatialVrfObjectManagerInterface() {};

   virtual void addVrfObjects(const DtManagedObjectListConstPtr& vrfObjects) = 0;

   virtual void removeVrfObject(const DtSimObject * vrfObject) = 0;

   virtual bool setResolutions(unsigned int xRes, unsigned int yRes, 
      unsigned int zRes) = 0;

   virtual void clearVrfObjects() = 0;
   
   virtual void reset(const DtManagedObjectListConstPtr& vrfObjects) = 0;

    


   typedef std::vector<const DtSimObject*>       DtVrfObjectUniqueContainer;
   typedef std::vector<const DtSimObject*> DtVrfConstObjectUniqueContainer;

   typedef std::vector<const DtSimObject*>            DtVrfObjectContainer;

   virtual bool getVrfObjects(const DtExtent& extent,
      DtVrfConstObjectUniqueContainer& objects) const = 0;

   virtual bool getVrfObjects(const DtVrfObjectPredicate& predicate, const DtExtent& extent,
      DtVrfConstObjectUniqueContainer& objects) const = 0;

   virtual bool getVrfObjects(const DtChord& chord, std::list<const DtSimObject*>& objects) const = 0;
   
   virtual bool getVrfObjects(const DtVrfObjectPredicate& predicate, const DtChord& chord,
      std::list<const DtSimObject*>& objects) const = 0;


   virtual void visitVrfObjects(const DtExtent& extent, DtSpatialSelectionFunctor &callback) const = 0;
   virtual void visitVrfObjects(const DtChord& chord, DtSpatialSelectionFunctor& callback) const = 0; 

   virtual unsigned int numObjects() const = 0;
};


class DT_DLL_vrfobjcore DtSpatialVrfObjectManager : public DtSpatialVrfObjectManagerInterface
{
public:
   typedef DtSpatialSubdivision<const DtSimObject*> DtSpatialContainer;
   typedef std::vector<const DtSimObject*> DtNonSpatialContainer;
   typedef DtNonSpatialContainer::iterator DtNonSpatialVrfObjectIter;
   typedef DtNonSpatialContainer::const_iterator DtNonSpatialVrfObjectConstIter;


   DtSpatialVrfObjectManager(unsigned int xRes = 1, unsigned int yRes = 1, 
      unsigned int zRes = 1);

   virtual ~DtSpatialVrfObjectManager();

private:
   DtSpatialVrfObjectManager(const DtSpatialVrfObjectManager& original);
   DtSpatialVrfObjectManager& operator=(const DtSpatialVrfObjectManager& original);

public:

   virtual bool initialize(const DtManagedObjectListConstPtr& vrfObjects,
      unsigned int cellCountX, unsigned int cellCountY, unsigned int cellCountZ);

   virtual void addVrfObjects(const DtManagedObjectListConstPtr& vrfObjects);

   virtual void removeVrfObject(const DtSimObject* vrfObject);

   virtual void clearVrfObjects();
   
   virtual bool setResolutions(unsigned int xRes, unsigned int yRes, 
      unsigned int zRes);

   virtual unsigned int xResolution() const;
   virtual unsigned int yResolution() const;
   virtual unsigned int zResolution() const;

   virtual void resolutions(unsigned int& xRes, unsigned int& yRes, 
      unsigned int& zRes) const;

   virtual void reset(const DtManagedObjectListConstPtr& vrfObjects);

    


   //bool getVrfObjects(const DtExtent& extent, DtVrfObjectUniqueContainer& objects);
   bool getVrfObjects(const DtExtent& extent, DtVrfConstObjectUniqueContainer& objects) const override;

   //bool getVrfObjects(const DtVrfObjectPredicate& predicate, const DtExtent& extent,
   bool getVrfObjects(const DtVrfObjectPredicate& predicate, const DtExtent& extent,
      DtVrfConstObjectUniqueContainer& objects) const override;

   bool getVrfObjects(const DtChord& chord, std::list<const DtSimObject*>& objects) const override;
   
   bool getVrfObjects(const DtVrfObjectPredicate& predicate, const DtChord& chord,
      std::list<const DtSimObject*>& objects) const override;


   void visitVrfObjects(const DtExtent& extent, DtSpatialSelectionFunctor &callback) const override;
   void visitVrfObjects(const DtChord& chord, DtSpatialSelectionFunctor& callback) const override;
   
   virtual const DtSpatialContainer* spatialContainer() const;
   virtual DtSpatialContainer* spatialContainer();

   virtual const DtNonSpatialContainer* nonSpatiallyOrganizedObjects() const;
   virtual DtNonSpatialContainer* nonSpatiallyOrganizedObjects();

protected:

   virtual bool createSpatialContainer(const DtExtent& extent, unsigned int cellCountX, 
      unsigned int cellCountY, unsigned int cellCountZ);

   virtual void setExtent(const DtManagedObjectListConstPtr& vrfObjects);

   virtual void expandExtent(const DtManagedObjectListConstPtr& vrfObjects, 
      DtExtent& expandedExtent) const;

   bool testInvariant() const;
   

protected:

   DtSpatialContainer* myVrfObjectSpatialContainer;

   DtNonSpatialContainer myNonSpatiallyOrganizedObjects;
};

static DtOctree<const DtSimObject>* createOctree(const DtExtent& extent)
{
   return new DtOctree<const DtSimObject>(extent);
}