seiswrite.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:          27-1-98
________________________________________________________________________

-*/


#include "seisstor.h"
#include "fixedstring.h"
#include "uistring.h"
class SeisTrc;
class SeisPSWriter;
class Seis2DLinePutter;
class SeisStatsCollector;
namespace Threads { class ConditionVar; }
namespace PosInfo { class Line2DData; }


mExpClass(Seis) SeisTrcWriter : public SeisStoreAccess
{ mODTextTranslationClass(SeisTrcWriter);
public:

                        SeisTrcWriter(const IOObj*);
                        SeisTrcWriter(const char*,bool is_2d,bool is_ps);
                        ~SeisTrcWriter();
    virtual bool        close();

    bool                prepareWork(const SeisTrc&);
    virtual bool        put(const SeisTrc&);
    int                 nrWritten() const               { return nrwritten_; }

    bool                isMultiComp() const;
    bool                isMultiConn() const;
    IOPar&              auxPars()                       { return auxpars_; }

    Seis2DLinePutter*   linePutter()                    { return putter_; }
    const Seis2DLinePutter* linePutter() const          { return putter_; }

    SeisPSWriter*       psWriter()                      { return pswriter_; }
    const SeisPSWriter* psWriter() const                { return pswriter_; }

                        // 2D only
    void                setSelData(Seis::SelData*);
    void                setDataType( const char* dt )   { datatype_ = dt; }

    void                setCrFrom( const char* str )    { crfrom_ = str; }
    void                setCrUserInfo( const char* str ) { crusrinfo_ = str; }

protected:

    bool                prepared_;
    int                 nrtrcs_;
    int                 nrwritten_;
    int                 firstns_;
    SamplingData<float> firstsampling_;
    SeisStatsCollector& seisstatscollector_;
    IOPar&              auxpars_;

    void                init();
    void                startWork();
    void                writeCollectedStats() const;

    // PS only
    SeisPSWriter*       pswriter_;

    // 3D only
    Conn*               crConn(int,bool);
    bool                ensureRightConn(const SeisTrc&,bool);
    bool                start3DWrite(Conn*,const SeisTrc&);

    // 2D only
    Seis2DLinePutter*   putter_;
    PosInfo::Line2DData* linedata_;
    Pos::GeomID         prevgeomid_;
    BufferString        datatype_;
    bool                next2DLine();
    bool                put2D(const SeisTrc&);

    BufferString        crfrom_;
    BufferString        crusrinfo_;

};


mExpClass(Seis) SeisSequentialWriter
{ mODTextTranslationClass(SeisSequentialWriter);
public:
                        SeisSequentialWriter( SeisTrcWriter*, int buffsize=-1 );
                        ~SeisSequentialWriter();
    bool                announceTrace(const BinID&);
    bool                submitTrace( SeisTrc*, bool waitforbuffer=true );
    bool                submitGather( ObjectSet<SeisTrc>&,
                                      bool waitforbuffer=true );
    bool                finishWrite();
    uiString            errMsg() const { return errmsg_; }

protected:

    bool                        iterateBuffer(bool waitforbuffer);
    void                        reportWrite(const char*);
    void                        reportWrite(const uiString&);
    friend class                SeisSequentialWriterTask;

    SeisTrcWriter*              writer_;
    TypeSet<BinID>              announcedtraces_;
    Threads::ConditionVar&      lock_;
    ObjectSet<SeisTrc>          outputs_;
    const int                   maxbuffersize_;

    BinID                       latestbid_;

    int                         queueid_;
    uiString                    errmsg_;

};