Geometry.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2024 Univ. Grenoble Alpes, CNRS, Grenoble INP - UGA, TIMC, 38000 Grenoble, France
 *
 * Visit http://camitk.imag.fr for more information
 *
 * This file is part of CamiTK.
 *
 * CamiTK is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * CamiTK is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License version 3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with CamiTK.  If not, see <http://www.gnu.org/licenses/>.
 *
 * $CAMITK_LICENCE_END$
 ****************************************************************************/
 
#ifndef GEOMETRY_H
#define GEOMETRY_H
 
// -- Core stuff
#include "CamiTKAPI.h"
#include "InterfaceGeometry.h"
 
#include <QMap>
 
//-- VTK Classes
class vtkPointSet;
class vtkAlgorithmOutput;
class vtkDataSetMapper;
class vtkActor;
class vtkProp;
class vtkTexture;
class vtkGlyph3D;
class vtkTransform;
class vtkTubeFilter;
class vtkCastToConcrete;
class vtkTextMapper;
class vtkSphereSource;
class vtkTransformFilter;
 
namespace camitk {
class CAMITK_API Geometry : public InterfaceGeometry {
 
public:
    Geometry(QString label, vtkSmartPointer<vtkPointSet> pointSet, const InterfaceGeometry::RenderingModes mode = InterfaceGeometry::Surface);
 
    ~Geometry() override;
 
    const QString getLabel() const override {
        return label;
    }
 
    void setLabel(QString newName) override {
        label = newName;
        updateLabel();
    }
 
    vtkSmartPointer<vtkPointSet> getPointSet() override {
        return pointSet;
    }
 
    void setPointSet(vtkSmartPointer<vtkPointSet> ds) override;
 
    void setMeshWorldTransform(vtkSmartPointer<vtkTransform>) override;
 
    vtkSmartPointer<vtkAlgorithmOutput> getDataPort() const override {
        return dataOutput;
    }
 
    void setDataConnection(vtkSmartPointer<vtkAlgorithmOutput>) override;
 
    void setPointData(vtkSmartPointer<vtkDataArray>) override;
 
    vtkSmartPointer<vtkActor> getActor(const RenderingModes) override;
 
    void setTexture(vtkSmartPointer<vtkTexture> texture) override;
 
    void pointPicked(vtkIdType, bool) override {};
 
    void cellPicked(vtkIdType, bool) override {};
 
 
 
    vtkSmartPointer<vtkProp> getProp(const QString&) override;
 
    unsigned int getNumberOfProp() const override;
 
    vtkSmartPointer<vtkProp> getProp(unsigned int) override;
 
    bool addProp(const QString&,  vtkSmartPointer<vtkProp>) override final;
 
    bool removeProp(const QString&) override;
 
 
    void getBounds(double* bounds) override final;
 
    double getBoundingRadius() override final;
 
    void setPointPosition(const unsigned int orderNumber, const double x, const double y, const double z) override;
 
 
 
 
    void setRenderingModes(const RenderingModes rMode) override {
        renderingModes = rMode;
    }
 
    const RenderingModes getRenderingModes() const override {
        return renderingModes;
    }
 
    void setEnhancedModes(const EnhancedModes) override;
 
    const EnhancedModes getEnhancedModes() const override {
        return enhancedModes;
    }
 
    void setActorColor(const RenderingModes, double*) override;
 
    void setActorColor(const RenderingModes, const double, const double, const double) override;
 
    void getActorColor(const RenderingModes, double*) const override final;
 
    void setColor(const double, const double, const double) override;
 
    void setColor(const double, const double, const double, const double) override;
 
    void setActorOpacity(const RenderingModes, const double) override final;
 
    double getActorOpacity(const RenderingModes) const override;
 
    void setOpacity(const double) override;
 
    void setMapperScalarRange(double min, double max) override;
 
    void setGlyphType(const GlyphTypes type, const double size = 0.0) override final;
 
    void setLinesAsTubes(bool isTubes = true, bool radiusFromLength = true, double radiusFactor = 1.0 / 40.0, int numberOfSides = 5) override;
 
    void setColorMode(int vtkColorMode = VTK_COLOR_MODE_DEFAULT) override;
 
 
 
private:
 
    vtkSmartPointer<vtkPointSet> pointSet;
 
    vtkSmartPointer<vtkAlgorithmOutput> dataOutput;
 
    vtkSmartPointer<vtkAlgorithmOutput> customPipelineOutput;
 
    vtkSmartPointer<vtkCastToConcrete> concreteData;
 
    vtkSmartPointer<vtkDataSetMapper> mapper;
 
    vtkSmartPointer<vtkTextMapper> labelActorMapper;
 
    vtkSmartPointer<vtkActor> surfaceActor;
 
    vtkSmartPointer<vtkActor> wireframeActor;
 
    vtkSmartPointer<vtkActor> pointsActor;
 
    vtkSmartPointer<vtkTexture> texture;
 
    vtkSmartPointer<vtkTubeFilter> tube;
 
    vtkSmartPointer<vtkTransformFilter> worldTransformFilter;
 
    vtkSmartPointer<vtkSphereSource> sphereGeom;
 
    InterfaceGeometry::RenderingModes renderingModes;
 
    InterfaceGeometry::EnhancedModes enhancedModes;
 
    double alphaShaded;
 
    QString label;
 
    double glyphSize;
 
    void buildLabel();
 
    void updateLabel();
 
    void buildGlyph(const GlyphTypes type);
 
    void createPointCloudVisualization();
 
    double surfaceColor[4];
    double wireframeColor[4];
    double pointsColor[4];
 
    double oldAlphaSurface;
    double oldAlphaWireframe;
    double oldAlphaPoints;
 
 
    QMap<QString, vtkSmartPointer<vtkProp> > extraProp;
 
};
 
 
 
}
 
#endif