cxSyntheticVolume.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 SYNTHETIC_VOLUME_HPP
#define SYNTHETIC_VOLUME_HPP

#include "cxtestutilities_export.h"

#include "cxUSFrameData.h"
#include <vector>
#include <cstdlib>
#include <time.h>
#include "cxMathBase.h"


namespace cx {

class CXTESTUTILITIES_EXPORT cxSyntheticVolume {
public:

        cxSyntheticVolume(Vector3D bounds)
        {
                // Seed the random number generator
                srand(time(NULL));
                mBounds = bounds;
        }
        virtual ~cxSyntheticVolume() { };

        virtual Vector3D
        getBounds()
        {
                return mBounds;
        }

        virtual ProcessedUSInputDataPtr
        sampleUsData(const std::vector<Transform3D>& planes_rMf,
                     const Eigen::Array2f& pixelSpacing,
                     const Eigen::Array2i& sliceDimension,
                                 const Transform3D& output_dMr,
                     const double noiseSigma,
                     const unsigned char noiseMean) const;
        virtual ProcessedUSInputDataPtr
        sampleUsData(const std::vector<Transform3D>& planes_rMt,
                                 const ProbeDefinition& probe,
                                 const Transform3D& output_dMr,
                                 const double noiseSigma=0,
                                 const unsigned char noiseMean=0) const;
        vtkImageDataPtr
        sampleUsData(const Transform3D& plane_rMt,
                                 const ProbeDefinition& probe,
                                 const double noiseSigma=0,
                                 const unsigned char noiseMean=0) const;
        vtkImageDataPtr
        sampleUsData(const Transform3D& plane_rMf,
                                 const Eigen::Array2f& pixelSpacing,
                                 const Eigen::Array2i& sliceDimension,
                                 const double noiseSigma,
                                 const unsigned char noiseMean) const;

        virtual unsigned char evaluate(const Vector3D& p) const = 0;
        virtual void printInfo() const {}

        void fillVolume(cx::ImagePtr vol);
        virtual float computeRMSError(cx::ImagePtr vol);

protected:
        unsigned char constrainToUnsignedChar(const int val) const;
        vtkImageDataPtr createEmptyMask(const Eigen::Array2i& sliceDimension) const;

        Vector3D mBounds;

};

typedef boost::shared_ptr<cxSyntheticVolume> cxSyntheticVolumePtr;

CXTESTUTILITIES_EXPORT double calculateRMSError(vtkImageDataPtr a, vtkImageDataPtr b);
CXTESTUTILITIES_EXPORT cx::Vector3D calculateCentroid(cx::ImagePtr image);
CXTESTUTILITIES_EXPORT double calculateMass(cx::ImagePtr image);

}
#endif