BoostDataSaver.h

Go to the documentation of this file.

/*
 * File:   BoostDataSaver.h
 * Author: winckler
 *
 * Created on October 22, 2014, 5:16 PM
 */

#ifndef BOOSTDATASAVER_H
#define BOOSTDATASAVER_H

// std
#include <fstream>
#include <iostream>
#include <string>
#include <type_traits>
#include <vector>

// boost
namespace boost {
namespace serialization {
class access;
}
}   // namespace boost
#include <boost/archive/binary_iarchive.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/serialization/vector.hpp>

// FairRoot
#include <FairMQMessage.h>

template<typename TPayload,
         typename TArchiveIn = boost::archive::binary_iarchive,
         typename TArchiveOut = TBoboost::archive::binary_oarchiveostOut>
class BoostDataSaver
{
  public:
    BoostDataSaver() {}
    virtual ~BoostDataSaver() {}

    virtual void InitOutputFile() {}

    void Write(std::ofstream& outfile, TPayload* objArr, long sizeArr = 1)
    {
        std::vector<TPayload> dataVector(objArr, objArr + sizeArr);
        TArchiveOut outArchive(outfile);
        outArchive << dataVector;
    }

    void Write(std::ofstream& outfile, std::vector<TPayload> dataVector)
    {
        TArchiveOut outArchive(outfile);
        outArchive << dataVector;
    }

    void Write(std::ofstream& outfile, FairMQMessage* msg)
    {
        std::vector<TPayload> objArr;
        std::string msgStr(static_cast<char*>(msg->GetData()), msg->GetSize());
        std::istringstream ibuffer(msgStr);
        TArchiveIn inputArchive(ibuffer);
        try {
            inputArchive >> objArr;
        } catch (boost::archive::archive_exception& e) {
            LOG(error) << e.what();
        }
        TArchiveOut outArchive(outfile);
        outArchive << objArr;
    }

    std::vector<std::vector<TPayload>> Read(std::ifstream& infile)
    {
        std::vector<std::vector<TPayload>> dataContainer;
        while (infile.peek() != EOF && !infile.eof()) {
            TArchiveIn inArchive(infile);
            std::vector<TPayload> temp;
            inArchive >> temp;
            dataContainer.push_back(temp);
        }
        return dataContainer;
    }
};

#endif /* BOOSTDATASAVER_H */