statdirdata.h

Go to the documentation of this file.

#pragma once
/*+
________________________________________________________________________
 
 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Bert Bril
 Date:          Mar 2009
 RCS:           $Id$
________________________________________________________________________
 
-*/
 
#include "algomod.h"
#include "manobjectset.h"
#include "angles.h"
#include "ranges.h"
 
namespace Stats
{
 
mClass(Algo) SectorPartData
{
public:
 
                        SectorPartData( float v=0, float p=0.5, int cnt=0 )
                            : val_(v), pos_(p), count_(cnt)  {}
    bool                operator ==( const SectorPartData& spd ) const
                        { return pos_ == spd.pos_; }
 
    float               pos_;   
    float               val_;   
    int                 count_; 
 
};
 
 
typedef TypeSet<SectorPartData> SectorData;
 
 
mClass(Algo) DirectionalData : public ManagedObjectSet<SectorData>
{
public:
 
    mClass(Algo) Setup
    {
    public:
                        Setup()
                            : usrposrg_(0,1)
                            , angle0_(0)
                            , angletype_(Angle::UsrDeg) {}
 
        Interval<float> usrposrg_;
        float           angle0_;
        Angle::Type     angletype_;
    };
 
                        DirectionalData(int nrsectors=0,int nrparts=0);
 
    void                init(int nrsectors,int nrparts);
 
    SectorPartData&     getPartData( int isect, int ipart )
                        { return (*((*this)[isect]))[ipart]; }
    const SectorPartData& getPartData( int isect, int ipart ) const
                        { return (*((*this)[isect]))[ipart]; }
    inline int          nrSectors() const
                        { return size(); }
    inline int          nrParts( int isect ) const
                        { return ((*this)[isect])->size(); }
    inline float        angle(int isect,int bound=0) const;
    inline float        angle(int isect,Angle::Type,int bound=0) const;
    int                 sector(float ang) const;
    int                 sector(float ang,Angle::Type) const;
    int                 part(int isect,float pos) const;
 
    Setup               setup_;
 
};
 
 
inline float DirectionalData::angle( int isect, int bound ) const
{
    if ( isEmpty() )
        return mUdf(float);
 
    float fullc; Angle::getFullCircle( setup_.angletype_, fullc );
    const float angstep = fullc / size();
    const float centerang = setup_.angle0_ + angstep * isect;
    return centerang + bound * angstep * .5f;
}
 
 
inline float DirectionalData::angle( int isect, Angle::Type t, int bound ) const
{
    const float ang = angle( isect, bound );
    return Angle::convert( setup_.angletype_, ang, t );
}
 
 
inline int DirectionalData::sector( float ang, Angle::Type t ) const
{
    return sector( Angle::convert(t,ang,setup_.angletype_) );
}
 
 
inline int DirectionalData::sector( float ang ) const
{
    if ( mIsUdf(ang) )
        return 0;
 
    ang -= setup_.angle0_;
    const float usrang = Angle::convert(setup_.angletype_,ang,Angle::UsrDeg);
    const float fsect = size() * (usrang / 360);
    int sect = mNINT32(fsect);
    if ( sect >= size() ) sect = 0;
    return sect;
}
 
 
inline int DirectionalData::part( int isect, float pos ) const
{
    const int nrparts = nrParts( isect );
    const float fpart = nrparts * pos / setup_.usrposrg_.width();
    int prt = (int)fpart;
    if ( prt<0 ) prt = 0;
    if ( prt>=nrparts ) prt = nrparts-1;
    return prt;
}
 
 
inline DirectionalData::DirectionalData( int nrsect, int nrparts )
{
    init( nrsect, nrparts );
}
 
 
inline void DirectionalData::init( int nrsect, int nrparts )
{
    erase();
 
    for ( int isect=0; isect<nrsect; isect++ )
    {
        SectorData* sd = new SectorData;
        *this += sd;
        for ( int ipart=0; ipart<nrparts; ipart++ )
            *sd += SectorPartData( 0, (ipart + .5f) / nrparts, 0 );
    }
}
 
} // namespace Stats