cxUSFrameData.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 CXUSFRAMEDATA_H_
#define CXUSFRAMEDATA_H_
 
#include "cxResourceExport.h"
 
#include "cxProbeSector.h"
#include "cxUSReconstructInputData.h"
typedef vtkSmartPointer<class vtkImageImport> vtkImageImportPtr;
 
namespace cx
{
typedef boost::shared_ptr<class ImageDataContainer> ImageDataContainerPtr;
typedef boost::shared_ptr<class CachedImageDataContainer> CachedImageDataContainerPtr;
}
 
namespace cx
{
 
 
typedef boost::shared_ptr<class USFrameData> USFrameDataPtr;
//typedef boost::shared_ptr<class TimedPosition> TimedPositionPtr;
 
class cxResource_EXPORT ProcessedUSInputData
{
public:
        ProcessedUSInputData(std::vector<vtkImageDataPtr> frames, std::vector<TimedPosition> pos, vtkImageDataPtr mask, QString path, QString uid);
 
        unsigned char* getFrame(unsigned int index) const;
        Eigen::Array3i getDimensions() const;
        Vector3D getSpacing() const;
        std::vector<TimedPosition> getFrames() const;
        vtkImageDataPtr getMask();
 
        QString getFilePath();
        QString getUid();
        bool validate() const;
 
private:
        std::vector<vtkImageDataPtr> mProcessedImage;
        std::vector<TimedPosition> mFrames;
        vtkImageDataPtr mMask;
        QString mPath;
        QString mUid;
};
typedef boost::shared_ptr<ProcessedUSInputData> ProcessedUSInputDataPtr;
 
 
class cxResource_EXPORT USFrameData
{
public:
        static USFrameDataPtr create(ImagePtr inputFrameData);
        static USFrameDataPtr create(QString inputFilename, FileManagerServicePtr fileManager);
        static USFrameDataPtr create(QString name, cx::ImageDataContainerPtr images);
        static USFrameDataPtr create(QString name, std::vector<vtkImageDataPtr> frames);
        ~USFrameData();
 
        Eigen::Array3i getDimensions() const;
        Vector3D getSpacing() const;
        QString getName() const;
        cx::ImageDataContainerPtr getImageContainer() { return mImageContainer; }
        unsigned getNumImages();
 
        void resetRemovedFrames();
        void removeFrame(unsigned int index);
        void setCropBox(IntBoundingBox3D mCropbox);
        void fillImageImport(vtkImageImportPtr import, int index); 
        void setPurgeInputDataAfterInitialize(bool value);
 
        std::vector<std::vector<vtkImageDataPtr> > initializeFrames(std::vector<bool> angio);
 
        virtual USFrameDataPtr copy();
        void purgeAll();
 
        bool is4D();//Do this object contain 4D data? (Opposed to the usual 3D data)
        bool is8bit() const;
 
protected:
        USFrameData();
        vtkImageDataPtr useAngio(vtkImageDataPtr inData, vtkImageDataPtr grayFrame, int frameNum) const;
 
        vtkImageDataPtr cropImageExtent(vtkImageDataPtr input, IntBoundingBox3D cropbox) const;
        vtkImageDataPtr to8bitGrayscaleAndEffectuateCropping(vtkImageDataPtr input) const;
 
        std::vector<int> mReducedToFull; 
        IntBoundingBox3D mCropbox;
 
        QString mName;
        cx::ImageDataContainerPtr mImageContainer;
        bool mPurgeInput;
private:
        vtkImageDataPtr convertTo8bit(vtkImageDataPtr input) const;
};
 
}//namespace cx
 
#endif // CXUSFRAMEDATA_H_