cxVideoGraphics.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 CXVIDEOGRAPHICS_H_
#define CXVIDEOGRAPHICS_H_

#include "cxResourceVisualizationExport.h"

#include "vtkForwardDeclarations.h"
#include "cxForwardDeclarations.h"

typedef vtkSmartPointer<class vtkTransformTextureCoords> vtkTransformTextureCoordsPtr;
typedef vtkSmartPointer<class vtkDataSetMapper> vtkDataSetMapperPtr;
typedef vtkSmartPointer<class vtkImageMask> vtkImageMaskPtr;
typedef vtkSmartPointer<class UltrasoundSectorSource> UltrasoundSectorSourcePtr;

namespace cx
{
typedef boost::shared_ptr<class VideoSourceGraphics> VideoSourceGraphicsPtr;

class cxResourceVisualization_EXPORT VideoGraphics
{
public:
        VideoGraphics();
        virtual ~VideoGraphics();

        vtkActorPtr getActor();
        void setActorUserMatrix(vtkMatrix4x4Ptr rMu);
        void setVisibility(bool visible);
        void setFlipVertical(bool on);
        bool getFlipVertical();

        void setMask(vtkImageDataPtr mask);

        void setClip(vtkPolyDataPtr sector);

        void setInputVideo(vtkImageDataPtr video);

        void update();

private:
        void setLookupTable();

        void setupPipeline();
        void connectVideoImageToPipeline();
        void updatePlaneSourceBounds();
        void updateLUT();
        bool inputImageIsEmpty();

        vtkImageDataPtr mInputMask;
        vtkPolyDataPtr mInputSector;
        vtkImageDataPtr mInputVideo;

        vtkLookupTablePtr mLUT;
        vtkImageChangeInformationPtr mDataRedirecter;
        vtkActorPtr mPlaneActor;
        vtkPlaneSourcePtr mPlaneSource;
        vtkTexturePtr mTexture;
        UltrasoundSectorSourcePtr mUSSource;
        vtkDataSetMapperPtr mDataSetMapper;
        vtkTransformTextureCoordsPtr mTransformTextureCoords;
        vtkTextureMapToPlanePtr mTextureMapToPlane;

        vtkImageThresholdPtr mMapZeroToOne;
        vtkImageMaskPtr mMaskFilter;
};
typedef boost::shared_ptr<VideoGraphics> VideoGraphicsPtr;

} // namespace cx

#endif // CXVIDEOGRAPHICS_H_