TreeInformation.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef TreeInformation_h
#define TreeInformation_h
 
#include "vtkCesium3DTilesWriter.h"
#include <vtkSmartPointer.h>
 
#include <vtk_nlohmannjson.h>
#include VTK_NLOHMANN_JSON(json.hpp)
 
#include <array>
#include <vector>
 
VTK_ABI_NAMESPACE_BEGIN
class vtkActor;
class vtkCompositeDataSet;
class vtkDataArray;
class vtkIdList;
class vtkImageData;
class vtkIntArray;
class vtkPolyData;
class vtkPointSet;
class vtkRenderWindow;
class vtkRenderWindowInteractor;
class vtkIncrementalOctreeNode;
 
class TreeInformation
{
public:
 
  // buildings
  TreeInformation(vtkIncrementalOctreeNode* root, int numberOfNodes,
    const std::vector<vtkSmartPointer<vtkCompositeDataSet>>* buildings,
    const std::string& textureBaseDirectory, const std::string& propertyTextureFile,
    bool saveTextures, bool contentGLTF, bool contentGLTFSaveGLB, const char* crs,
    const std::string& outputDir);
  // points
  TreeInformation(vtkIncrementalOctreeNode* root, int numberOfNodes, vtkPointSet* points,
    bool contentGLTF, bool contentGLTFSaveGLB, const char* crs, const std::string& output);
  // mesh
  TreeInformation(vtkIncrementalOctreeNode* root, int numberOfNodes, vtkPolyData* mesh,
    const std::string& textureBaseDirectory, const std::string& propertyTextureFile,
    bool saveTextures, bool contentGLTF, bool contentGLTFSaveGLB, const char* crs,
    const std::string& output);
 
  void PrintNode(vtkIncrementalOctreeNode* node);
 
 
  std::array<double, 6> GetNodeTightBounds(int i) { return NodeTightBounds[i]; }
  bool GetNodeTightBounds(int i, double* bounds);
  static bool GetNodeTightBounds(void* data, vtkIncrementalOctreeNode* node, double* bounds);
 
  void AddGeometricError(vtkPolyData* representation);
  void Compute();
  void SaveTilesBuildings(bool mergeTilePolyData, size_t mergedTextureWidth);
  void SaveTilesMesh();
  void SaveTilesPoints();
  void SaveTileset(const std::string& output);
  static void PrintBounds(const char* name, const double* bounds);
  static void PrintBounds(const std::string& name, const double* bounds)
  {
    PrintBounds(name.c_str(), bounds);
  }
  static std::array<double, 6> ExpandBounds(double* first, double* second);
 
protected:
  void PostOrderTraversal(void (TreeInformation::*Visit)(vtkIncrementalOctreeNode* node, void* aux),
    vtkIncrementalOctreeNode* node, void* aux);
  void PreOrderTraversal(void (TreeInformation::*Visit)(vtkIncrementalOctreeNode* node, void* aux),
    vtkIncrementalOctreeNode* node, void* aux);
  void SaveTileset(vtkIncrementalOctreeNode* root, const std::string& output);
  nlohmann::json GenerateTileJson(vtkIncrementalOctreeNode* node);
  bool ConvertTileCartesianBuildings(vtkIncrementalOctreeNode* node);
  bool ConvertDataSetCartesian(vtkPointSet* points);
 
 
  void VisitCompute(vtkIncrementalOctreeNode* node, void* aux);
  void VisitComputeGeometricError(vtkIncrementalOctreeNode* node, void* aux);
  void SaveTileBuildings(vtkIncrementalOctreeNode* node, void* auxData);
  void SaveTileMesh(vtkIncrementalOctreeNode* node, void* auxData);
  void WriteTileTexture(
    vtkIncrementalOctreeNode* node, const std::string& fileName, vtkImageData* tileImage);
  vtkSmartPointer<vtkImageData> SplitTileTexture(
    vtkPolyData* tileMesh, vtkImageData* textureImage, vtkDataArray* tcoordsTile);
  void SaveTilePoints(vtkIncrementalOctreeNode* node, void* auxData);
 
 
  double ComputeGeometricErrorTilesetBuildings();
  double ComputeGeometricErrorTilesetPoints();
  double ComputeGeometricErrorTilesetMesh();
  double ComputeGeometricErrorTileset();
  double ComputeGeometricErrorNodeBuildings(vtkIncrementalOctreeNode* node, void* aux);
  double ComputeGeometricErrorNodePoints(vtkIncrementalOctreeNode* node, void* aux);
  double ComputeGeometricErrorNodeMesh(vtkIncrementalOctreeNode* node, void* aux);
  double ComputeGeometricErrorNode(vtkIncrementalOctreeNode* node, void* aux);
  std::array<double, 6> ComputeTightBB(vtkIdList* tileBuildings);
  std::string ContentTypeExtension() const;
  void Initialize();
  double GetRootLength2();
  bool ForEachBuilding(vtkIncrementalOctreeNode* node, std::function<bool(vtkPolyData*)> Execute);
 
private:
  enum vtkCesium3DTilesWriter::InputType InputType;
  vtkIncrementalOctreeNode* Root;
 
  const std::vector<vtkSmartPointer<vtkCompositeDataSet>>* Buildings;
  vtkPointSet* Points;
  vtkPolyData* Mesh;
 
  std::string OutputDir;
  std::string TextureBaseDirectory;
  std::string PropertyTextureFile;
  bool SaveTextures;
  bool ContentGLTF;
  bool ContentGLTFSaveGLB;
 
  const char* CRS;
  std::vector<std::array<double, 6>> NodeTightBounds;
  std::vector<bool> EmptyNode;
  std::vector<double> GeometricError;
  nlohmann::json RootJson;
};
 
VTK_ABI_NAMESPACE_END
#endif
// VTK-HeaderTest-Exclude: TreeInformation.h