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