cxVtkHelperClasses.h

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.
                 
Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.
                 
CustusX is released under a BSD 3-Clause license.
                 
See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/


#ifndef CXVTKHELPERCLASSES_H_
#define CXVTKHELPERCLASSES_H_

#include "cxResourceVisualizationExport.h"

#include "cxTransform3D.h"
#include "cxGraphicalPrimitives.h"

#include "vtkForwardDeclarations.h"

#include <QColor>

// --------------------------------------------------------
namespace cx
{
// --------------------------------------------------------

template <typename vtkPropertyOrProperty2DPtr>
void setColorAndOpacity(vtkPropertyOrProperty2DPtr property, QColor color)
{
        property->SetColor(color.redF(), color.greenF(), color.blueF());
        property->SetOpacity(color.alphaF());
}

Vector3D getColorAsVector3D(QColor color) ;

class cxResourceVisualization_EXPORT RGBColor : public boost::array<double,3>
{
public:
        RGBColor();
        RGBColor(double r, double g, double b);
        explicit RGBColor(const double* rgb);
        RGBColor(const RGBColor& t);
        RGBColor& operator=(const RGBColor& t);
        RGBColor(QColor c);
        RGBColor& operator=(const QColor& c);
        ~RGBColor() {}
};

class cxResourceVisualization_EXPORT OffsetPoint
{
        public:
                OffsetPoint( vtkRendererPtr renderer );
                ~OffsetPoint();
                void setRadius ( int radius );
                void setValue(Vector3D point, RGBColor color);
                void setValue(Vector3D point, Vector3D color);
                void update( const Vector3D& point );
                vtkActor2DPtr getActor();
        private:
                vtkSphereSourcePtr source;
                vtkPolyDataMapper2DPtr mapper;
                vtkActor2DPtr actor;
                vtkRendererPtr mRenderer;
};
typedef boost::shared_ptr<OffsetPoint> OffsetPointPtr;

class cxResourceVisualization_EXPORT LineSegment
{
        public:
                LineSegment(vtkRendererPtr renderer);
                ~LineSegment();
                void setPoints(Vector3D point1, Vector3D point2, RGBColor color, int stipplePattern = 0xFFFF);
                void setResolution(int res);
                void updatePosition(Vector3D point1, Vector3D point2);
                vtkActor2DPtr getActor();
                void setColor(RGBColor color);
                void setPattern(int stipplePattern);
                void setWidth(float width);

        private:
                vtkPolyDataMapper2DPtr mapper2d;
                vtkActor2DPtr actor2d;
                vtkRendererPtr mRenderer;
                vtkLineSourcePtr source;
};

class cxResourceVisualization_EXPORT TextDisplay
{
        public:
                TextDisplay(const QString& text, const QColor& color, int fontsize);
                ~TextDisplay();
                void setRenderer( vtkRendererPtr renderer );
                void setPosition( float x, float y);
                void setPosition( const Vector3D& pos );
                void setCentered();
                void updateText(const QString& text );
                void setColor(QColor color);
                vtkTextProperty* textProperty();
                vtkActor2DPtr getActor();
                void setMaxWidth( int width, vtkViewport *vp );
                int getMaxWidth();
                int getWidth( vtkViewport *vp );
                vtkTextMapperPtr getMapper() { return mapper; }

                static void forceUseVtkTextMapper();

        private:
                vtkTextMapperPtr mapper;
                vtkActor2DPtr actor;
                vtkRendererPtr mRenderer;
                int maxWidth;
                QString text;

                static void verifyVtkTextMapper();
};
typedef boost::shared_ptr<class TextDisplay> TextDisplayPtr;


class cxResourceVisualization_EXPORT CrossHair2D
{
        public:
                CrossHair2D( vtkRendererPtr renderer);
                ~CrossHair2D();
                void setValue(const Vector3D& focalPoint, int width, int height, double bordarOffset, const RGBColor& color);
                //void update( const Vector3D& crossPos );
                void update( const Vector3D& crossPos, const DoubleBoundingBox3D& vp);
                void updateRegCross( const Vector3D& crossPos );
                vtkActor2DPtr getActor();
        private:
                vtkCursor2DPtr mCursor2D;
                vtkPolyDataMapper2DPtr mapper;
                vtkActor2DPtr actor;
                vtkRendererPtr mRenderer;
};
typedef boost::shared_ptr<class CrossHair2D> CrossHair2DPtr;

class cxResourceVisualization_EXPORT Axes3D
{
public:
        Axes3D(vtkRendererPtr renderer = vtkRendererPtr());
        ~Axes3D();
        void setPosition(const Transform3D& pos);
        vtkProp3DPtr getProp();
private:
        void setCaption(vtkCaptionActor2D* captionActor, const QString& caption, RGBColor color);

        vtkAxesActorPtr mAxes;
        vtkRendererPtr mRenderer;
};
typedef boost::shared_ptr<Axes3D> Axes3DPtr;

// --------------------------------------------------------
}//     namespace vm
// --------------------------------------------------------


#endif // CXVTKHELPERCLASSES_H_