H5PartWrapperForPT.h

Go to the documentation of this file.

//
// Class H5PartWrapperForPT
//   A class that manages all calls to H5Part for the Parallel-T tracker.
//
// Copyright (c) 200x-2021, Paul Scherrer Institut, Villigen PSI, Switzerland
// All rights reserved
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#ifndef OPAL_H5PARTWRAPPERFORPT_H
#define OPAL_H5PARTWRAPPERFORPT_H
 
#include "OPALTypes.h"
#include "Structure/H5PartWrapper.h"
 
#include "H5hut.h"
 
class H5PartWrapperForPT : public H5PartWrapper {
public:
    H5PartWrapperForPT(const std::string& fileName, h5_int32_t flags = H5_O_WRONLY);
    H5PartWrapperForPT(
            const std::string& fileName, int restartStep, std::string sourceFile,
            h5_int32_t flags = H5_O_RDWR);
    virtual ~H5PartWrapperForPT();
 
    virtual void readHeader();
    virtual void readStep(PartBunch_t*, h5_ssize_t firstParticle, h5_ssize_t lastParticle);
 
    virtual void writeHeader();
    virtual void writeStep(
            PartBunch_t*, const std::map<std::string, double>& additionalStepAttributes,
            size_t particleContainerIndex = 0);
 
    virtual bool predecessorIsSameFlavour() const;
 
private:
    void readStepHeader(PartBunch_t*);
    void readStepData(PartBunch_t*, h5_ssize_t, h5_ssize_t);
 
    void writeStepHeader(
            PartBunch_t*, const std::map<std::string, double>&, size_t particleContainerIndex);
    void writeStepData(PartBunch_t*, size_t particleContainerIndex);
};
 
inline bool H5PartWrapperForPT::predecessorIsSameFlavour() const {
    return (predecessorOPALFlavour_m == "opal-t");
}
 
#endif  // OPAL_H5PARTWRAPPERFORPT_H