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);
        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_