RecordInput.h

Go to the documentation of this file.

//
// Copyright 2020, Jefferson Science Associates, LLC.
// Subject to the terms in the LICENSE file found in the top-level directory.
//
// EPSCI Group
// Thomas Jefferson National Accelerator Facility
// 12000, Jefferson Ave, Newport News, VA 23606
// (757)-269-7100


#ifndef EVIO_6_0_RECORDINPUT_H
#define EVIO_6_0_RECORDINPUT_H


#include <iostream>
#include <fstream>
#include <memory>


#include "ByteOrder.h"
#include "ByteBuffer.h"
#include "RecordHeader.h"
#include "Compressor.h"
#include "EvioException.h"


namespace evio {


    class RecordInput {

    private:

        static const uint32_t DEFAULT_BUF_SIZE = 8 * 1024 * 1024;

        uint32_t nEntries = 0;

        uint32_t userHeaderOffset = 0;

        uint32_t eventsOffset = 0;

        uint32_t uncompressedEventsLength = 0;

        ByteOrder byteOrder {ByteOrder::ENDIAN_LOCAL};

        std::shared_ptr<RecordHeader> header;

        std::shared_ptr<ByteBuffer> dataBuffer;

        ByteBuffer recordBuffer;

        ByteBuffer headerBuffer;


    private:

        void allocate(size_t size);
        void setByteOrder(const ByteOrder & order);
        void showIndex() const;

    public:

        RecordInput();
        explicit RecordInput(const ByteOrder & order);
        RecordInput(const RecordInput & recordIn);
        RecordInput(RecordInput && srcRec) noexcept;

        ~RecordInput() = default;

        RecordInput & operator=(RecordInput&& other) noexcept;
        RecordInput & operator=(const RecordInput& other);

        std::shared_ptr<RecordHeader> getHeader();
        const ByteOrder & getByteOrder();
        std::shared_ptr<ByteBuffer> getUncompressedDataBuffer();

        bool hasIndex() const;
        bool hasUserHeader() const;

        std::shared_ptr<ByteBuffer> getEvent(std::shared_ptr<ByteBuffer> & buffer, uint32_t index, size_t bufOffset = 0);
        ByteBuffer & getEvent(ByteBuffer & buffer, uint32_t index, size_t bufOffset = 0);
        std::shared_ptr<uint8_t> getUserHeader();
        std::shared_ptr<ByteBuffer> getUserHeader(std::shared_ptr<ByteBuffer> & buffer, size_t bufOffset = 0);
        ByteBuffer & getUserHeader(ByteBuffer & buffer, size_t bufOffset = 0);

        std::shared_ptr<uint8_t> getEvent(uint32_t index, uint32_t * len);
        uint32_t getEventLength(uint32_t index) const;
        uint32_t getEntries() const;

        std::shared_ptr<RecordInput> getUserHeaderAsRecord(ByteBuffer & buffer,
                                                           size_t bufOffset);

        void readRecord(std::ifstream & file, size_t position);
        void readRecord(ByteBuffer & buffer, size_t offset);

        static uint32_t uncompressRecord(std::shared_ptr<ByteBuffer> & srcBuf, size_t srcOff,
                                               std::shared_ptr<ByteBuffer> & dstBuf,
                                               RecordHeader & hdr);
       static uint32_t uncompressRecord(ByteBuffer & srcBuf, size_t srcOff,
                                        ByteBuffer & dstBuf,
                                        RecordHeader & header);
    };

}
#endif //EVIO_6_0_RECORDINPUT_H