vtkCompositeMapperHelper2.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
 
#ifndef vtkCompositeMapperHelper2_h
#define vtkCompositeMapperHelper2_h
 
#ifndef __VTK_WRAP__
 
#include "vtkOpenGLPolyDataMapper.h"
#include "vtkRenderingOpenGL2Module.h" // for export macro
 
class vtkPolyData;
class vtkCompositePolyDataMapper2;
 
// this class encapsulates values tied to a
// polydata
class vtkCompositeMapperHelperData
{
public:
  vtkPolyData* Data;
  unsigned int FlatIndex;
  double Opacity;
  bool IsOpaque;
  bool Visibility;
  bool Pickability;
  bool OverridesColor;
  vtkColor3d AmbientColor;
  vtkColor3d DiffuseColor;
  vtkColor3d SelectionColor;
  double SelectionOpacity;
 
  bool Marked;
 
  unsigned int StartVertex;
  unsigned int NextVertex;
 
  // point line poly strip edge stripedge
  unsigned int StartIndex[vtkOpenGLPolyDataMapper::PrimitiveEnd];
  unsigned int NextIndex[vtkOpenGLPolyDataMapper::PrimitiveEnd];
 
  // stores the mapping from vtk cells to gl_PrimitiveId
  vtkNew<vtkOpenGLCellToVTKCellMap> CellCellMap;
};
 
//===================================================================
class VTKRENDERINGOPENGL2_EXPORT vtkCompositeMapperHelper2 : public vtkOpenGLPolyDataMapper
{
public:
  static vtkCompositeMapperHelper2* New();
  vtkTypeMacro(vtkCompositeMapperHelper2, vtkOpenGLPolyDataMapper);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  void SetParent(vtkCompositePolyDataMapper2* p) { this->Parent = p; }
 
  vtkCompositeMapperHelperData* AddData(vtkPolyData* pd, unsigned int flatIndex);
 
  void RenderPiece(vtkRenderer* ren, vtkActor* act) override;
 
  void ClearMark();
  void RemoveUnused();
  bool GetMarked() { return this->Marked; }
  void SetMarked(bool v) { this->Marked = v; }
 
  std::vector<vtkPolyData*> GetRenderedList() const;
 
  void ProcessSelectorPixelBuffers(
    vtkHardwareSelector* sel, std::vector<unsigned int>& pixeloffsets, vtkProp* prop) override;
 
  virtual void ProcessCompositePixelBuffers(vtkHardwareSelector* sel, vtkProp* prop,
    vtkCompositeMapperHelperData* hdata, std::vector<unsigned int>& mypixels);
 
protected:
  vtkCompositePolyDataMapper2* Parent;
  std::map<vtkPolyData*, vtkCompositeMapperHelperData*> Data;
 
  bool Marked;
 
  void UpdateCameraShiftScale(vtkRenderer* ren, vtkActor* actor) override;
 
  vtkCompositeMapperHelper2() { this->Parent = nullptr; };
  ~vtkCompositeMapperHelper2() override;
 
  void DrawIBO(vtkRenderer* ren, vtkActor* actor, int primType, vtkOpenGLHelper& CellBO,
    GLenum mode, int pointSize);
 
  virtual void SetShaderValues(
    vtkShaderProgram* prog, vtkCompositeMapperHelperData* hdata, size_t primOffset);
 
  void UpdateShaders(vtkOpenGLHelper& cellBO, vtkRenderer* ren, vtkActor* act) override;
 
  void ReplaceShaderColor(
    std::map<vtkShader::Type, vtkShader*> shaders, vtkRenderer* ren, vtkActor* act) override;
 
  bool GetNeedToRebuildBufferObjects(vtkRenderer* ren, vtkActor* act) override;
 
  void BuildBufferObjects(vtkRenderer* ren, vtkActor* act) override;
  virtual void AppendOneBufferObject(vtkRenderer* ren, vtkActor* act,
    vtkCompositeMapperHelperData* hdata, vtkIdType& flat_index, std::vector<unsigned char>& colors,
    std::vector<float>& norms);
 
  void BuildSelectionIBO(
    vtkPolyData* poly, std::vector<unsigned int> (&indices)[4], vtkIdType offset) override;
 
  int CanUseTextureMapForColoring(vtkDataObject*) override;
 
  std::vector<unsigned int> VertexOffsets;
 
  // vert line poly strip edge stripedge
  std::vector<unsigned int> IndexArray[PrimitiveEnd];
 
  void RenderPieceDraw(vtkRenderer* ren, vtkActor* act) override;
 
  bool PrimIDUsed;
  bool OverideColorUsed;
 
  vtkHardwareSelector* CurrentSelector;
 
  std::vector<std::vector<unsigned int>> PickPixels;
 
  std::map<vtkAbstractArray*, vtkDataArray*> ColorArrayMap;
 
private:
  vtkCompositeMapperHelper2(const vtkCompositeMapperHelper2&) = delete;
  void operator=(const vtkCompositeMapperHelper2&) = delete;
};
 
#endif
 
#endif