nladataprep.h

Go to the documentation of this file.

#ifndef nladataprep_h
#define nladataprep_h

/*+
________________________________________________________________________

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

-*/

#include "nlamod.h"
#include "nladesign.h"

class BinIDValueSet;
class PosVecDataSet;

mExpClass(NLA) NLADataPreparer
{
public:
                        NLADataPreparer( BinIDValueSet& bvs, int tc )
                            : bvs_(bvs), targetcol_(tc)         {}

    void                removeUndefs(bool targetonly=false);
    void                limitRange(const Interval<float>&);

    struct BalanceSetup
    {
                        BalanceSetup()
                        : nrclasses(0), nrptsperclss(0), noiselvl(0.01) {}
        int             nrclasses, nrptsperclss;
        float           noiselvl;
    };
    void                balance(const BalanceSetup&);

protected:

    BinIDValueSet&      bvs_;
    int                 targetcol_;

    void                addVecs(BinIDValueSet&,int,float,
                                const Interval<float>*);

};


#endif