vtkProteinRibbonFilter.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 vtkProteinRibbonFilter_h
#define vtkProteinRibbonFilter_h
 
#include "vtkDomainsChemistryModule.h" // for export macro
#include "vtkPolyDataAlgorithm.h"
 
#include "vtkColor.h" // For vtkColor3ub.
#include <map>        // For element to color map.
 
VTK_ABI_NAMESPACE_BEGIN
class vtkVector3f;
class vtkStringArray;
 
class VTKDOMAINSCHEMISTRY_EXPORT vtkProteinRibbonFilter : public vtkPolyDataAlgorithm
{
public:
  vtkTypeMacro(vtkProteinRibbonFilter, vtkPolyDataAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  static vtkProteinRibbonFilter* New();
 
 
  vtkGetMacro(CoilWidth, float);
  vtkSetMacro(CoilWidth, float);
 
 
  vtkGetMacro(HelixWidth, float);
  vtkSetMacro(HelixWidth, float);
 
 
  vtkGetMacro(SubdivideFactor, int);
  vtkSetMacro(SubdivideFactor, int);
 
 
  vtkGetMacro(DrawSmallMoleculesAsSpheres, bool);
  vtkSetMacro(DrawSmallMoleculesAsSpheres, bool);
 
 
  vtkGetMacro(SphereResolution, int);
  vtkSetMacro(SphereResolution, int);
 
protected:
  vtkProteinRibbonFilter();
  ~vtkProteinRibbonFilter() override;
 
  int FillInputPortInformation(int, vtkInformation*) override;
 
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  void CreateThinStrip(vtkPolyData* poly, vtkUnsignedCharArray* pointsColors, vtkPoints* p,
    std::vector<std::pair<vtkVector3f, bool>>& p1, std::vector<std::pair<vtkVector3f, bool>>& p2,
    std::vector<vtkColor3ub>& colors);
 
  void CreateAtomAsSphere(vtkPolyData* poly, vtkUnsignedCharArray* pointsColors, double* pos,
    const vtkColor3ub& color, float radius, float scale);
 
  static std::vector<vtkVector3f>* Subdivide(std::vector<std::pair<vtkVector3f, bool>>& p, int div);
 
  void SetColorByAtom(std::vector<vtkColor3ub>& colors, vtkStringArray* atomTypes);
 
  void SetColorByStructure(std::vector<vtkColor3ub>& colors, vtkStringArray* atomTypes,
    vtkUnsignedCharArray* ss, const vtkColor3ub& helixColor, const vtkColor3ub& sheetColor);
 
  std::map<std::string, vtkColor3ub> ElementColors;
 
  float CoilWidth;
  float HelixWidth;
  int SphereResolution;
  int SubdivideFactor;
  bool DrawSmallMoleculesAsSpheres;
 
private:
  vtkProteinRibbonFilter(const vtkProteinRibbonFilter&) = delete;
  void operator=(const vtkProteinRibbonFilter&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif // vtkProteinRibbonFilter_h