houghtransform.h

Go to the documentation of this file.

#ifndef houghtransform_h
#define houghtransform_h
/*
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        N. Fredman
 Date:          18-12-2002
 RCS:           $Id$
________________________________________________________________________

*/


#include "algomod.h"
#include "threadlock.h"
#include "arrayndimpl.h"
#include "ranges.h"

class Array3DInfo;
template <class T> class Array3D;
class SequentialTask;
template <class T, class B> class TopList;
class PlaneFrom3DSpaceHoughTransformTask;
class Plane3;

mExpClass(Algo) PlaneFrom3DSpaceHoughTransform
{
public:
                                PlaneFrom3DSpaceHoughTransform();
    virtual                     ~PlaneFrom3DSpaceHoughTransform();

    void                        setResolution( double dangle,
                                               int distsize );
    int                         getParamSpaceSize() const;
    int                         getNrDistVals() const;

    void                        setClipRate( float );
    float                       clipRate() const;

    void                        setData( const Array3D<float>* );
    ObjectSet<SequentialTask>*  createCalculators();

    TopList<unsigned int, unsigned int>*
                                sortParamSpace(int) const;
    Plane3                      getPlane( int plane ) const;
    int                         getNrPointsAfterClip() const; 
protected:
    void                        incParamPos( int normal, double dist );
    float                       cliprate_;
    TypeSet<unsigned int>       calcpositions_;
    Array3DInfo*                datainfo_;

    Array2D<unsigned int>*      paramspace_;
    double                      deltadist_;
    TypeSet<Coord3>*            normals_;
    friend                      class ::PlaneFrom3DSpaceHoughTransformTask;

    Threads::Lock               paramspacelock_;
};


mExpClass(Algo) LineFrom2DSpaceHoughTransform
{
public:
                                LineFrom2DSpaceHoughTransform(
                                        const Array2D<float>&);
                                ~LineFrom2DSpaceHoughTransform();

    bool                        compute();

    void                        setThreshold(float val,bool above_val);
                                /*<Make hough array based on above/below val.*/
    void                        setLineAngleRange(Interval<float>rg);
                                /*<rg is between 0-PI in radian.*/
    void                        setTopList(int nr)      { toplistnr_ = nr; }
                                /*<nr is the number of top confident lines.*/
    Array2D<unsigned char>*     getResult() const       { return result_; }
                                /*<0/1 array, 1 means on a line. */

//protected:

    bool                        setLineFlag(float radius,float theta);

    const Array2D<float>&       input_;
    Array2D<int>*               origcnt_;       
    Array2D<int>*               hougharr_;
    Array2D<unsigned char>*     result_;

    float                       threshold_;
    bool                        abovethreshold_;
    int                         toplistnr_; 
    Interval<float>             anglerg_;
};


#endif