osgEarthTerrainSourceTracker.h

Go to the documentation of this file.

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


#pragma once

#include <vantageTerrainImplementation/terrainSourceTracker.h>
#include <vantageTerrainImplementation/pageInVisitor.h>
#include <vantageTerrainImplementation/bulletUtil.h>
#include <vantageTerrainImplementation/vantageTerrainImplementationDefines.h>
#include <vantageTerrainImplementation/vantageTerrainImplementationUtil.h>

#include <vantageTerrainImplementation/sharedMemoryBulletDebugDrawer.h>

#include <terrainInterface/terrainProcessPolygonsBox.h>

#include <vrfutil/vertexList.h>

//Mak Terrain Toolkit includes
#include <geometry/sphere.h>

//#include <vlutil/vlSmartPointers.h>

#include <osgTerrain/TerrainTile>
#include <osgEarth/Profile>
#include <osgEarth/TileKey>
#include <osgEarth/MapNode>
#include <osgEarth/TerrainTileModel>

#include <string>
#include <list>
#include <vector>

#include <boost/ptr_container/ptr_vector.hpp>

#include <tbb/spin_rw_mutex.h>

class DtPageInTilesCommand;
class DtPageOutTilesCommand;
class DtVantageTerrainEngine;
class DtPoint;
class DtGeodeticCoord;
class DtTerrainPagingChord;
class DtTerrainPagingSphere;
class DtTerrainPagingShape;
class DtOsgEarthTerrainSourceTracker;

typedef std::list<osgEarth::TileKey> DtTileKeyList;

struct DT_DLL_vantageTerrainImplementation DtTileData
{
   enum TileType
   {
      Normal,
      Empty
   };

   struct DT_DLL_vantageTerrainImplementation Extent
   {
      typedef std::vector<Extent> Vector;

      static Vector Simplify(const Vector& extents);

      explicit Extent(const osgEarth::TileKey&);
      explicit Extent(const DtVector& min, const DtVector& max);

      DtVector min, max;
   };

   struct DT_DLL_vantageTerrainImplementation IndexExtent
   {
      typedef std::vector<IndexExtent> Vector;

      explicit IndexExtent(const DtTileData& tile, const Extent& extent);

      unsigned minx, miny, maxx, maxy;
   };

   explicit DtTileData(
      const DtOsgEarthTerrainSourceTracker& tracker,
      const osgEarth::TileKey& key);

   virtual ~DtTileData();

   const DtOsgEarthTerrainSourceTracker& tracker;

   const osgEarth::TileKey tileKey;

   osg::ref_ptr<osgEarth::TerrainTileModel> tileModel;

   Extent extent;

   virtual void addMaskExtents(const Extent::Vector& extents);

   virtual bool insideValidArea(const DtPoint& p) const;

   virtual bool extentInsideValidArea(unsigned x1, unsigned y1, unsigned x2, unsigned y2) const;

   virtual bool extentOutsideValidArea(unsigned x1, unsigned y1, unsigned x2, unsigned y2) const;

   virtual bool addActiveTileKey(const osgEarth::TileKey& key) = 0;

   virtual void recomputeActiveTileKeyAreas() = 0;

   TileType type;

   std::vector< osg::ref_ptr<osg::Node> > geometries;

   virtual std::string description() const = 0;

   virtual unsigned numberOfVertices() const = 0;

   virtual unsigned numberOfTrianglesForVertices() const;

   virtual void getTriangleVertices(unsigned xindex, unsigned yindex, double& x, double& y) const = 0;

   virtual void getTriangleIndices(double x, double y, unsigned& xindex, unsigned& yindex) const = 0;

   virtual void heightfieldIndexForTriangleIndex(
      unsigned triangleXIndex, unsigned triangleYIndex,
      unsigned& heightFieldXUsed, unsigned& heightFieldYUsed) const = 0;

   virtual double interpolateHeight(
      unsigned triangleXIndex, unsigned triangleYIndex,
      unsigned* heightFieldXUsed = 0, unsigned* heightFieldYUsed = 0) const = 0;

   tbb::atomic<unsigned int> numTriangles;

   bool pendingRemoval;

   virtual DtTerrainIntersectIDType getLastUsed() const = 0;

   virtual bool updateLastUsed(DtTerrainIntersectIDType value) = 0;

   virtual bool findIntersections(
      const DtChord& chord,
      unsigned x1, unsigned y1, unsigned x2, unsigned y2,
      DtVrfChordIntersectRecordList& intList,
      bool singleCell = false,
      bool insideValidAreaChecked = false) const;

   virtual bool findIntersections(
      const DtChord& chord,
      DtVrfChordIntersectRecordList& intList) const = 0;

   virtual void findTriangles(
      const DtTerrainProcessPolygonsBox& box,
      unsigned x1, unsigned y1, unsigned x2, unsigned y2,
      DtTriangleOutputInterface&,
      bool singleCell = false,
      bool insideValidAreaChecked = false) const;

   virtual void findTriangles(
      const DtTerrainProcessPolygonsBox& box,
      DtTriangleOutputInterface&) const = 0;

protected:
   mutable tbb::spin_rw_mutex maskExtentsMutex;

   Extent::Vector maskExtents;

   IndexExtent::Vector maskIndexExtents;
};

class DT_DLL_vantageTerrainImplementation DtOsgEarthTerrainSourceTracker : public DtTerrainSourceTracker
{
public:
   DtOsgEarthTerrainSourceTracker(
      DtVantageTerrainEngine*, DtPagerCommandQueue* commandQueue, 
      osgEarth::MapNode* mapNode);

   virtual ~DtOsgEarthTerrainSourceTracker();

   virtual bool useRawElevation() const;

   virtual unsigned getMaxLOD() const;
   virtual void setMaxLOD(unsigned);

   virtual double triangleOutputMergeTolerance() const;
   virtual void setTriangleOutputMergeTolerance(double tol);
   virtual double intersetionTriangleMergeTolerance() const;
   virtual void setIntersectionTriangleMergeTolerance(double tol);

   virtual unsigned tileCount() const;

   virtual unsigned cellCount() const;

   virtual void addPagedInTile(
      const osgEarth::TileKey& tilekey,
      const boost::shared_ptr<DtTileData>& tileData);

   virtual void removePagedInTile(const osgEarth::TileKey tileKey);

   virtual bool dataAvailable(DtPagingMetaDataPtr metaData, 
      DtTerrainPagingGeometryPtr geometry);

   virtual bool findHeightFieldIntersections(
      const DtChord& chord,
      DtVrfChordIntersectRecordList& intList,
      bool* dataAvailable = 0) const;

   virtual void processTriangles(
      DtTriangleOutputInterface& output,
      const DtTerrainProcessPolygonsBox& box) const;

   virtual void pageInData(DtPagingMetaDataPtr metaData,
      DtTerrainPagingGeometryPtr geometry);

   virtual DtTerrainIntersectIDType findOldestPagedData(unsigned int& numTrianglesOut);

   virtual void pageOutOldestData(const int priority);

   virtual void drawDebugInfo(btIDebugDraw* debugDrawer);

   virtual void drawElevation();

   virtual bool canPageOutData(DtTerrainIntersectIDType intersectId);

   virtual osgEarth::TileKey getBestTileKey(const osgEarth::TileKey&) const;

   virtual void findMissingTiles(const DtTerrainPagingGeometryPtr& geometry, DtTileKeyList& tiles) const;

protected:

   virtual DtPageInTilesCommand* createPageInTilesCommand(
      DtPagingMetaDataPtr metaData);

   virtual DtPageOutTilesCommand* createPageOutTilesCommand(
      DtPagingMetaDataPtr metaData);

   virtual bool setDrawMode(DtDrawMode);

protected:
    
   class TileSpatialIndex
   {
   public:
      explicit TileSpatialIndex(const DtOsgEarthTerrainSourceTracker&, const osgEarth::TileKey& tilekey);

      bool intersects(const DtTerrainPagingGeometry& geometry) const;
      bool intersectsSubTile(const DtTerrainPagingGeometry& geometry, int i) const;

      bool chordIntersects(const DtChord& chord) const;
      bool extentIntersects(const DtExtent& extent) const;
      bool sphereIntersects(const DtSphere& sphere) const;
      bool shapeIntersects(const DtLocalVertexList& shape) const;
      bool chordIntersectsSubTile(const DtChord& chord, int i) const;
      bool extentIntersectsSubTile(const DtExtent& shape, int i) const;
      bool sphereIntersectsSubTile(const DtSphere& sphere, int i) const;
      bool shapeIntersectsSubTile(const DtLocalVertexList& shape, int i) const;

      void findMissing(
         const DtTerrainPagingGeometry& geometry,
         DtTileKeyList& missing) const;

      bool findData(
         const DtTerrainPagingGeometry* geometry,
         std::vector< boost::shared_ptr<DtTileData> >& data) const;

      boost::shared_ptr<DtTileData> findOldest() const;

      bool addTile(const osgEarth::TileKey& key, const boost::shared_ptr<DtTileData>& data);
      void removeTile(const osgEarth::TileKey& tilekey);
      unsigned tileCount() const;
      unsigned cellCount() const;

      const osgEarth::TileKey tileKey() const { return myTileKey;  }

   protected:
      void getExtent(DtVector& lower, DtVector& upper) const;
      void getSubTileExtent(unsigned i, DtVector& lower, DtVector& upper, bool useElevation = true) const;

      bool isTileEmpty() const;

      virtual void recomputePagedInData();

      const DtOsgEarthTerrainSourceTracker& myTracker;

      const osgEarth::TileKey myTileKey;
      const osgEarth::GeoExtent myGeoExtent;

      struct SubTileSlot
      {
         osgEarth::GeoExtent extent;
         boost::scoped_ptr<TileSpatialIndex> subtile;
      };

      SubTileSlot mySubTiles[4];

      boost::shared_ptr<DtTileData> myTileData;

      bool myAllPagedInFlag;

      bool myMaxDataLevelFlag;
   };

   virtual void convertGeometry(
      const DtTerrainPagingGeometryPtr& geometry,
      std::vector<DtTerrainPagingGeometryPtr>& output) const;

   virtual bool convertGeometry(
      const DtTerrainPagingChord& geometry,
      std::vector<DtTerrainPagingGeometryPtr>& output) const;

   virtual bool convertGeometry(
      const DtTerrainPagingSphere& geometry,
      std::vector<DtTerrainPagingGeometryPtr>& output) const;

   virtual bool convertGeometry(
      const DtTerrainPagingShape& geometry,
      std::vector<DtTerrainPagingGeometryPtr>& output) const;

   virtual bool nearCenterOfEarth(const DtChord& chord) const;

   virtual bool findData(
      const DtTerrainPagingGeometry* geometry,
      std::vector< boost::shared_ptr<DtTileData> >& data,
      const DtPagingMetaDataPtr& metaData = DtPagingMetaDataPtr()) const;

   virtual void findMissing(
      const DtTerrainPagingGeometry& geometry,
      DtTileKeyList& missing) const;

   virtual boost::shared_ptr<DtTileData> findOldest() const;
   virtual boost::shared_ptr<DtTileData> findOldestUnlocked() const;

   typedef boost::ptr_vector<TileSpatialIndex> TileRoots;

   const osg::ref_ptr<osgEarth::MapNode> myMapNode;
   std::vector< osg::ref_ptr<osgEarth::ElevationLayer> > myElevationLayers;

   bool myUseRawElevation;
   unsigned int myMaxLOD;
   double myTriangleOutputMergeTolerance;
   double myIntersetionTriangleMergeTolerance;

   TileRoots myTileSpatialIndexRoots;
   double myWorldWidth;

   boost::weak_ptr<DtTileData> myOldestData;

   typedef tbb::spin_mutex DrawerMutex;
   mutable DrawerMutex myDrawerMutex;
   std::auto_ptr<DtSharedMemoryBulletDebugDrawer> myExtentDrawer;
   std::auto_ptr<DtSharedMemoryBulletDebugDrawer> myElevationDrawer;
};