DtSpeedTreeForest.h

Go to the documentation of this file.

/****************************************************************************** 
** Copyright (c) 2007 MAK Technologies, Inc. 
** All rights reserved. 
******************************************************************************/ 
/****************************************************************************** 
** $RCSfile: DtSpeedTreeForest.h,v $ $Revision: 1.26 $ $State: Exp $ 
******************************************************************************/ 


#ifndef DtSpeedTreeForest_H_ 
#define DtSpeedTreeForest_H_ 

#include <vrvSt/vrvSt.h> 
#include <vrvOsg/DtOsgForest.h>
#include <osg/ref_ptr>
#include <osg/PositionAttitudeTransform>
#include <osg/Geode>
#include <osg/BoundingBox>

#include <OsgST_Forest.h>

#include <boost/unordered_map.hpp>

namespace makOsgSpeedTree
{
   class TreeInstance;
}

namespace makVrv 
{ 
   class DtAgentUpdateResolverInterface;
   class DtSpeedTreeModelInstance;
   class DtSpeedTreeCore;

   class DT_DLL_VRVST DtSpeedTreeForest : public DtOsgForest
   { 

   public:

      friend class DtSpeedTreeForestClassesCreator;
      friend class DtSpeedTreeForestCreator;

      virtual ~DtSpeedTreeForest();

      virtual void addUniqueTree(const DtTreeRecord& record);
      virtual void removeTree(DtUniqueID record);

      virtual void setTreeClippingPlane(float fFar);
      virtual void setTree3DLodDistances(float fNear, float fFar);
      virtual void setTreeBillboardDistances(float fStart, float fEnd);

      virtual DtSpeedTreeModelInstance* findSpeedTreeModelInstance( const std::string& modelDef);

      void setBoundingBox(const osg::BoundingBox& box);
      const osg::BoundingBox& getBoundingBox(void) const;

      osg::ref_ptr<osg::Node> getRootNode(void);
      
      void programBoundCallback(osg::ref_ptr<makOsgSpeedTree::Forest> pForest, const makOsgSpeedTree::TreeType* pTreeType, uint32 programID);

   protected:

      DtSpeedTreeForest(DtDe& de, DtUniqueID id);

      virtual void slot_treeModelInstanceToBeDestroyed(const std::string& modelDef);

      virtual void slot_treeModelInstanceCreated(const std::string& modelDef);

      virtual DtTreeModel* castUpdaterToTreeModel( DtAgentUpdateResolverInterface* updater );

      virtual bool createTree(const DtTreeRecord& record);

   protected:

      osg::ref_ptr<osg::PositionAttitudeTransform> myPat;

      osg::ref_ptr<makOsgSpeedTree::Forest> myForest;

      osg::BoundingBox myBoundingBox;

      DtSpeedTreeCore& myCore;

   private:
      void connectSignals(void);
      void disconnectSignals(void);

      osg::Vec3f computeWindDirection(float fDirection);

      void setWindStrength(float fStrength);
      void setWindDirection(float fDirection);

      typedef boost::unordered_map<DtUniqueID, makOsgSpeedTree::TreeInstance*> MapIdsToTreeInstances;
      MapIdsToTreeInstances myTreeInstances;

      bool preUpdateCallback(osg::ref_ptr<makOsgSpeedTree::Forest> forest, osg::RenderInfo& renderInfo);
      bool preRenderCallback(osg::ref_ptr<makOsgSpeedTree::Forest> forest, osg::RenderInfo& renderInfo);
      void postUpdateCallback(osg::ref_ptr<makOsgSpeedTree::Forest> forest, osg::RenderInfo& renderInfo);
      void postRenderCallback(osg::ref_ptr<makOsgSpeedTree::Forest> forest, osg::RenderInfo& renderInfo);
      void provideTimeCallback(osg::ref_ptr<makOsgSpeedTree::Forest> forest, double& fTime);

      makOsgSpeedTree::TreeInstanceDescriptor myScratchTreeInstanceDesc;

      typedef std::map<const DtModelDefinition*, std::string> MapModelDefToSTFileName;
      MapModelDefToSTFileName myMapModelDefToSTFileName;

      static const float myCellSize;

      int myPreviousTU;
      void pushGraphicsState(osg::RenderInfo& renderInfo);
      void popGraphicsState(osg::RenderInfo& renderInfo);

      //keeps track of the current state of vrvVisualOverride so the programBoundCallback can apply it to the uniform
      osg::Camera* myCamera;

      std::map<int, bool> myShaderBoundMap;

      
   }; 

   class DT_DLL_VRVST DtSpeedTreeForestSignaler : public DtVirtualBaseClass
   {
   public:
       static DtSpeedTreeForestSignaler& instance(DtDe&);

       ~DtSpeedTreeForestSignaler();

       boost::signal<void (DtSpeedTreeForest&)> signal_forestCreated;

       boost::signal<void (DtSpeedTreeForest&)> signal_forestToBeDestroyed;

   protected:
       DtSpeedTreeForestSignaler();

   private: 
       DtSpeedTreeForestSignaler(const DtSpeedTreeForestSignaler&);
       DtSpeedTreeForestSignaler& operator=(const DtSpeedTreeForestSignaler&);
   };

   class DT_DLL_VRVST DtSpeedTreeForestCreator : public DtOsgForestCreator
   {
   public:
      DtSpeedTreeForestCreator(){}
      virtual DtForestObject* create(DtDe& de, DtUniqueID id);

   private:

   };
} 

#endif