conncomponents.h

Go to the documentation of this file.

#pragma once
 
/*+
________________________________________________________________________
 
 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Bo Zhang/Yuancheng Liu
 Date:          July 2012
 RCS:           $Id$
________________________________________________________________________
 
 
-*/
 
 
#include "algomod.h"
#include "factory.h"
 
 
template <class T> class Array2D;
template <class T> class Array2DSlice;
template <class T> class Array3D;
 
class TaskRunner;
 
mExpClass(Algo) ConnComponents
{
public:    
 
                                ConnComponents(const Array2D<bool>&);
                                ~ConnComponents();
 
    void                        compute(TaskRunner* tr=0); 
 
    int                         nrComponents() const; 
    const TypeSet<int>*         getComponent(int compidx);
    const Array2D<int>*         getLabel() const        { return label_; }
    void                        trimCompBranches(TypeSet<int>& comp);
    static void                 trimCompBranches(TypeSet<int>& comp,int sz1);
    static void                 getCompSticks(TypeSet<int>& comp,int sz1,
                                        int allowgapsz, int minsticksz,
                                        TypeSet<TypeSet<int> >& sticks);
                                /*Will change comp, make copy before call this*/
 
    float                       overLapRate(int componentidx);
                                /*Minimum rate of all dimensions.*/
 
protected:
 
    void                        classifyMarks(Array2D<int>& mark);
    void                        setMark(Array2D<int>& r,int source,int newval);
 
    const Array2D<bool>&        input_;
    Array2D<int>*               label_;
    TypeSet< TypeSet<int> >     components_;
    TypeSet<int>                sortedindex_;
};
 
 
mExpClass(Algo) ConnComponents3D 
{
public:    
 
                                ConnComponents3D(const Array3D<bool>&);
                                ~ConnComponents3D();
 
    void                        compute(TaskRunner* tr=0); 
 
    struct VPos {
                                VPos() : i(-1),j(-1),k(-1) {}
                                int i;int j;int k;
                };
    int                         nrComponents() const; 
    const ObjectSet<VPos>*      getComponent(int compidx);
                                /*<Comp size is sorted descending. */
 
protected:
 
    void                        addToComponent(
                                const TypeSet<TypeSet<TypeSet<int> > >& comps, 
                                int sliceidx,int compidx,
                                TypeSet<TypeSet<unsigned char> >& usedcomps,
                                ObjectSet<VPos>& rescomp);
 
    const Array3D<bool>&        input_;
    ObjectSet< ObjectSet<VPos> > components_;
    TypeSet<int>                sortedindex_;
};