seisblockswriter.h

Go to the documentation of this file.

#pragma once

/*
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Bert
 Date:          Apr 2017
________________________________________________________________________

*/

#include "seisblocksdata.h"
#include "seistrc.h"
#include "filepath.h"
#include "uistring.h"
#include "manobjectset.h"

class Task;
class LinScaler;
class SeisTrc;
namespace Pos { class IdxPairDataSet; }
namespace PosInfo { class CubeData; }


namespace Seis
{
namespace Blocks
{

class Block;

mExpClass(Seis) Writer : public DataStorage
{ mODTextTranslationClass(Seis::Blocks::Writer);
public:

                        Writer(const SurvGeom* sg=0);
                        ~Writer();

    void                setBasePath(const File::Path&);
    void                setFileNameBase(const char*);
    void                setCubeName(const char*);
    void                setFPRep(OD::FPDataRepType);
    void                setScaler(const LinScaler*);
    void                addComponentName(const char*);
    void                addAuxInfo(const char* key,const IOPar&);

    BufferString        dirName() const;
    BufferString        mainFileName() const;

    uiRetVal            add(const SeisTrc&);
    Task*               finisher();

    typedef std::pair<IdxType,float>    ZEvalPos;
    typedef TypeSet<ZEvalPos>           ZEvalPosSet;

    struct Column
    {
        typedef ManagedObjectSet<Block>    BlockSet;

                        Column(const Dimensions&,int nrcomps);
                        ~Column();

        Block&          firstBlock()    { return *blocksets_.first()->first(); }
        const Block&    firstBlock() const
                                        { return *blocksets_.first()->first(); }
        void            retireAll();
        void            getDefArea(SampIdx&,Dimensions&) const;

        const Dimensions dims_;
        ObjectSet<BlockSet> blocksets_; // one set per component
        bool**          visited_;
        int             nruniquevisits_;
    };

    inline int          nrColumnBlocks() const
                        { return zevalpositions_.size(); }

protected:

    File::Path          basepath_;
    BufferString        filenamebase_;
    BufferString        cubename_;
    OD::FPDataRepType   specfprep_;
    OD::FPDataRepType   usefprep_;
    LinScaler*          scaler_;
    BufferStringSet     compnms_;
    ObjectSet<IOPar>    auxiops_;

    bool                needreset_;
    const int           nrpospercolumn_;
    int                 nrcomponents_;
    bool                isfinished_;
    Interval<IdxType>   globzidxrg_;
    ObjectSet<ZEvalPosSet> zevalpositions_;
    Pos::IdxPairDataSet& columns_;


    void                setEmpty();
    void                resetZ(const Interval<float>&);
    bool                removeExisting(const char*,uiRetVal&) const;
    bool                prepareWrite(uiRetVal&);
    void                add2Block(Block&,const ZEvalPosSet&,SampIdx,
                                    const SeisTrc&,int);
    Column*             getColumn(const GlobIdx&);
    Column*             mkNewColumn(const GlobIdx&);
    bool                isCompletionVisit(Column&,const SampIdx&) const;
    void                writeColumn(Column&,uiRetVal&);
    bool                writeColumnHeader(od_ostream&,const Column&,
                                    const SampIdx&,const Dimensions&) const;
    bool                writeBlock(od_ostream&,Block&,SampIdx,Dimensions);
    void                writeMainFile(uiRetVal&);
    bool                writeMainFileData(od_ostream&);
    void                scanPositions( PosInfo::CubeData& cubedata,
                            Interval<IdxType>&,Interval<IdxType>&,
                            Interval<int>&,Interval<int>&,
                            Interval<double>&,Interval<double>&);

    friend class        ColumnWriter;
    friend class        WriterFinisher;

};


} // namespace Blocks

} // namespace Seis