You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

104 lines
2.8 KiB
C++

/*=========================================================================
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.
=========================================================================*/
/**
* @class vtkOpenGLStickMapper
* @brief use imposters to draw cylinders
*
* PolyDataMapper that uses imposters to draw cylinders/sticks
* for ball/stick style molecular rendering. Supports picking.
*/
#ifndef vtkOpenGLStickMapper_h
#define vtkOpenGLStickMapper_h
#include "vtkOpenGLPolyDataMapper.h"
#include "vtkRenderingOpenGL2Module.h" // For export macro
class VTKRENDERINGOPENGL2_EXPORT vtkOpenGLStickMapper : public vtkOpenGLPolyDataMapper
{
public:
static vtkOpenGLStickMapper* New();
vtkTypeMacro(vtkOpenGLStickMapper, vtkOpenGLPolyDataMapper);
void PrintSelf(ostream& os, vtkIndent indent) override;
//@{
/**
* Convenience method to set the array to scale with.
*/
vtkSetStringMacro(ScaleArray);
//@}
//@{
/**
* Convenience method to set the array to orient with
*/
vtkSetStringMacro(OrientationArray);
//@}
//@{
/**
* Convenience method to set the array to select with
*/
vtkSetStringMacro(SelectionIdArray);
//@}
protected:
vtkOpenGLStickMapper();
~vtkOpenGLStickMapper() override;
/**
* Create the basic shaders before replacement
*/
void GetShaderTemplate(
std::map<vtkShader::Type, vtkShader*> shaders, vtkRenderer* ren, vtkActor* act) override;
/**
* Perform string replacements on the shader templates
*/
void ReplaceShaderValues(
std::map<vtkShader::Type, vtkShader*> shaders, vtkRenderer* ren, vtkActor* act) override;
/**
* Set the shader parameters related to the Camera
*/
void SetCameraShaderParameters(vtkOpenGLHelper& cellBO, vtkRenderer* ren, vtkActor* act) override;
/**
* Set the shader parameters related to the actor/mapper
*/
void SetMapperShaderParameters(vtkOpenGLHelper& cellBO, vtkRenderer* ren, vtkActor* act) override;
const char* ScaleArray;
const char* OrientationArray;
const char* SelectionIdArray;
/**
* Does the VBO/IBO need to be rebuilt
*/
bool GetNeedToRebuildBufferObjects(vtkRenderer* ren, vtkActor* act) override;
/**
* Update the VBO to contain point based values
*/
void BuildBufferObjects(vtkRenderer* ren, vtkActor* act) override;
void RenderPieceDraw(vtkRenderer* ren, vtkActor* act) override;
private:
vtkOpenGLStickMapper(const vtkOpenGLStickMapper&) = delete;
void operator=(const vtkOpenGLStickMapper&) = delete;
};
#endif