tableconvimpl.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:          Jul 2006
________________________________________________________________________

-*/

#include "generalmod.h"
#include "tableconv.h"
#include "bufstringset.h"
#include "globexpr.h"
#include "od_iosfwd.h"
#include "uistring.h"


namespace Table
{

mExpClass(General) WSImportHandler : public ImportHandler
{
public:


                        WSImportHandler( od_istream& s )
                            : ImportHandler(s)
                            , insingqstring_(false)
                            , indoubqstring_(false)     {}

    State               add(char);
    const char*         getCol() const                  { return col_.buf(); }
    const char*         errMsg() const                  { return col_.buf(); }

    virtual void        newRow()                        {}

protected:

    bool                insingqstring_;
    bool                indoubqstring_;

};


mExpClass(General) CSVImportHandler : public ImportHandler
{
public:
                        CSVImportHandler( od_istream& s )
                            : ImportHandler(s)
                            , nlreplace_('\n')
                            , instring_(false)  {}

    State               add(char);
    const char*         getCol() const          { return col_.buf(); }
    const char*         errMsg() const          { return col_.buf(); }

    virtual void        newRow()                { instring_ = false; }

    char                nlreplace_;

protected:

    bool                instring_;

};


mExpClass(General) WSExportHandler : public ExportHandler
{
public:

    enum ColWSHandl     { None, Underscores, SingQuot, DoubQuot };

                        WSExportHandler( od_ostream& s,
                                         ColWSHandl w=Underscores )
                            : ExportHandler(s)
                            , colwshanld_(w)    {}

    bool                putRow(const BufferStringSet&,uiString&);

    ColWSHandl          colwshanld_;

protected:

    void                addVal(int col,const char*);

};


mExpClass(General) CSVExportHandler : public ExportHandler
{
public:
                        CSVExportHandler( od_ostream& s )
                            : ExportHandler(s)          {}

    bool                putRow(const BufferStringSet&,uiString&);

protected:

    void                addVal(int col,const char*);

};


mExpClass(General) SQLInsertExportHandler : public ExportHandler
{ mODTextTranslationClass(SQLInsertExportHandler);
public:

                        SQLInsertExportHandler( od_ostream& s )
                            : ExportHandler(s)
                            , startindex_(1)
                            , stepindex_(1)
                            , nrrows_(0)            {}

    bool                putRow(const BufferStringSet&,uiString&);

    BufferString        tblname_;       
    BufferStringSet     colnms_;        

    BufferString        indexcolnm_;    
    int                 startindex_;    
    int                 stepindex_;     

    BufferStringSet     extracolvals_;  
    BufferStringSet     extracolnms_;   

protected:

    void                addVal(int col,const char*);

    int                 nrrows_;
    bool                addindex_;
    int                 nrextracols_;

};


mExpClass(General) StartStopManipulator : public Converter::RowManipulator
{
public:
                StartStopManipulator()
                    : startdone_(false)
                    , count_(0)         {}

    struct Criterion
    {
        enum Type       { None, Records, Match };

                        Criterion( Type t=None )
                            : type_(t)
                            , count_(1)
                            , matchcolidx_(0)           {}

        Type            type_;
        int             count_;         
        GlobExpr        matchexpr_;
        int             matchcolidx_;   

    };

    Criterion   start_;
    Criterion   stop_;

    bool        accept(BufferStringSet&) const;

protected:

    mutable bool startdone_;
    mutable int  count_;

    void        updCount(const Criterion&,const BufferStringSet&) const;
    bool        isGEMatch(const Criterion&,const BufferStringSet&) const;

};


mExpClass(General) RecordMatcher : public Converter::RowManipulator
{
public:
                        RecordMatcher( bool a=true )
                            : any_(a)   {}

    bool                accept(BufferStringSet&) const;

    bool                any_;           
    bool                not_;           
    TypeSet<int>        ckcols_;        
    TypeSet<GlobExpr>   colvals_;       

protected:

    static const GlobExpr emptyge_;

};


mExpClass(General) DuplicateKeyRemover : public Converter::RowManipulator
{
public:
                        DuplicateKeyRemover()
                            : nrdone_(0), nrremoved_(0) {}

    bool                accept(BufferStringSet&) const;

    TypeSet<int>        keycols_; 

    int                 nrRemoved() const               { return nrremoved_; }

protected:

    mutable BufferStringSet     prevkeys_;
    mutable int                 nrdone_;
    mutable int                 nrremoved_;

    void                        setPrevKeys(const BufferStringSet&) const;

};

}; // namespace Table