Geometry.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2018 Univ. Grenoble Alpes, CNRS, TIMC-IMAG UMR 5525 (GMCAO)
 *
 * 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 {
        return label;
    }

    void setLabel(QString newName) {
        label = newName;
        updateLabel();
    }

    vtkSmartPointer<vtkPointSet> getPointSet() {
        return pointSet;
    }

    void setPointSet(vtkSmartPointer<vtkPointSet> ds) override;

    void setMeshWorldTransform(vtkSmartPointer<vtkTransform>) override;

    vtkSmartPointer<vtkAlgorithmOutput> getDataPort() const {
        return dataOutput;
    }

    void setDataConnection(vtkSmartPointer<vtkAlgorithmOutput>) override;

    void setPointData(vtkSmartPointer<vtkDataArray>) override;

    vtkSmartPointer<vtkActor> getActor(const RenderingModes) 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;

    bool removeProp(const QString&) override;

    void setTexture(vtkSmartPointer<vtkTexture> texture) override;

    void pointPicked(vtkIdType, bool) {};

    void cellPicked(vtkIdType, bool) {};



    void getBounds(double* bounds) override;

    double getBoundingRadius() override;

    void setPointPosition(const unsigned int orderNumber, const double x, const double y, const double z) override;




    void setRenderingModes(const RenderingModes rMode) {
        renderingModes = rMode;
    }

    const RenderingModes getRenderingModes() const {
        return renderingModes;
    }

    void setEnhancedModes(const EnhancedModes) override;

    const EnhancedModes getEnhancedModes() const {
        return enhancedModes;
    }

    void setActorColor(const RenderingModes, double*) override;

    void setActorColor(const RenderingModes, const double, const double, const double) override;

    void getActorColor(const RenderingModes, double*) override;

    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;

    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;

    void setLinesAsTubes(bool tubes = false) override;



private:

    vtkSmartPointer<vtkPointSet> pointSet;

    vtkSmartPointer<vtkAlgorithmOutput> dataOutput;

    vtkSmartPointer<vtkAlgorithmOutput> customPipelineOutput;

    vtkSmartPointer<vtkCastToConcrete> concreteData;

    vtkSmartPointer<vtkDataSetMapper> mapper;

    QMap<QString, vtkSmartPointer<vtkProp> > extraProp;

    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;
    double oldPointsColor[4];
protected:
    vtkSmartPointer< vtkPointSet > New();
};



}

#endif