stackdisttrans.hh

Go to the documentation of this file.

/* -*- mia-c++  -*-
 *
 * This file is part of MIA - a toolbox for medical image analysis
 * Copyright (c) Leipzig, Madrid 1999-2017 Gert Wollny
 *
 * MIA is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MIA; if not, see <http://www.gnu.org/licenses/>.
 *
 */
 
#ifndef mia_3d_stackdisttrans_hh
#define mia_3d_stackdisttrans_hh
 
#include <vector>
#include <mia/3d/defines3d.hh>
#include <mia/3d/vector.hh>
#include <mia/2d/image.hh>
#include <mia/core/iohandler.hh>
#include <mia/core/ioplugin.hh>
 
NS_MIA_BEGIN

 
class EXPORT_3D C2DStackDistanceTransform: public CIOData
{
public:
       typedef C2DStackDistanceTransform type;

       static const char *const data_descr;

       static const char *const type_descr;
 

       struct DistanceFromPoint {

              DistanceFromPoint(C3DBounds _point, float _distance):
                     point(_point),
                     distance(_distance)
              {};
              C3DBounds point; 
              float distance;  
       };
 
 
       C2DStackDistanceTransform() = default;

       C2DStackDistanceTransform(const C2DStackDistanceTransform& /*other*/) = default;

       C2DStackDistanceTransform *clone() const __attribute__((warn_unused_result));

 
       C2DStackDistanceTransform(const C2DImage& slice, const C3DFVector& voxel_size = C3DFVector::_1);

 
       void read( const C2DImage& slice, int q);
 

       std::vector<DistanceFromPoint> get_slice(size_t s, const C2DImage& image) const;

       bool save(const std::string& filename);
 
private:
 
       float d(float fq, float q, float fv, float v)const;
 
       void dt1d(std::vector<float>& f, float scale)const;
       void dt2d(C2DFImage& image)const;
 
       struct SParabola {
              int k;
              int v;
              float z;
              float fv;
       };
 
       C2DBounds m_size;
       C3DFVector m_voxel_size;
       std::vector<int> m_k;
       std::vector< std::vector<SParabola>> m_zdt;
 
};

 
inline std::ostream& operator << (std::ostream& os, const C2DStackDistanceTransform::DistanceFromPoint& v)
{
       os << "[" << v.point << ":" << v.distance << "]";
       return os;
}
 
inline bool operator == (const C2DStackDistanceTransform::DistanceFromPoint& lhs,
                         const C2DStackDistanceTransform::DistanceFromPoint& rhs)
{
       return (lhs.point == rhs.point) && (lhs.distance == rhs.distance);
}
 

typedef TIOPlugin<C2DStackDistanceTransform> C2DStackDistanceTransformIO;

typedef THandlerSingleton< TIOPluginHandler<C2DStackDistanceTransformIO>> C2DStackDistanceTransformIOPluginHandler;

typedef C2DStackDistanceTransformIOPluginHandler::Instance::DataKey C2DStackDistanceTransformIODataKey;
 
NS_MIA_END
 
#endif