statdirdata.h

Go to the documentation of this file.

#ifndef statdirdata_h
#define statdirdata_h
/*+
________________________________________________________________________

 (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&     get( int isect, int ipart )
                        { return (*((*this)[isect]))[ipart]; }
    const SectorPartData& get( 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

#endif