linear.h

Go to the documentation of this file.

#pragma once

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        A.H.Bril
 Date:          Jan 2005
________________________________________________________________________

-*/


#include "algomod.h"
#include "coord.h"
#include "mathfunc.h"

class Plane3;

template <class T>
mClass(Algo) LineParameters : public MathFunction<T,T>
{
public:
                LineParameters( T i0=0, T i1=0 )
                : a0(i0), ax(i1)                {}

    inline T    getValue( T x ) const
                        { return a0 + ax * x; }
    inline T    getXValue( T y ) const
                        { return ax ? (y - a0) / ax : 0; }
    inline T    getProjectedX( T x, T y ) const
                        { return (x + ax * (y - a0)) / (1 + ax * ax); }
    inline T    getValue( const T* x ) const
                        { return getValue(*x); }

    T           a0, ax;
};

typedef LineParameters<float> LinePars;


template <class T>
mClass(Algo) PlaneParameters : public MathXYFunction<T,T>
{
public:
                PlaneParameters( T i0=0, T i1=0, T i2=0 )
                : a0(i0), ax(i1), ay(i2)        {}

    inline T    getValue( T x, T y ) const
                { return a0 + ax * x + ay * y; }
    inline T    getValue( const T* x ) const
                        { return getValue(x[0],x[1]); }

    T           a0, ax, ay;

};

typedef PlaneParameters<float> PlanePars;


mExpClass(Algo) LinStats2D
{
public:
                LinStats2D() : corrcoeff(0)     {}

    LinePars    lp;             
    LinePars    sd;             
    float       corrcoeff;      

    void        use(const float*,const float*,int nrpts);
    void        use(const Geom::Point2D<float>*,int nrpts);
};


mExpClass(Algo) LinStats3D
{
public:
                LinStats3D() : corrcoeff(0)     {}

    PlanePars   pp;             
    PlanePars   sd;             
    float       corrcoeff;      

};

mExpClass(Algo) Plane3DFit
{
public:
                        Plane3DFit();

    bool                compute(const TypeSet<Coord3>& points,Plane3& result);

protected:

    void                setScatterMatrix(double scattermatrix[3][3],
                                         int order[3]);
    void                tqli(double d[3],double e[3],double z[3][3]);
    void                tred2(double a[3][3],double d[3],double e[3]);

    TypeSet<Coord3>     points_;
    Coord3              centroid_;
};