HandleMgr.hpp

#ifndef HANDLEMGR_HPP
#define HANDLEMGR_HPP

#include <map>
#include <set>
#include <string>

#include "splash/DCException.hpp"
#include "splash/Dimensions.hpp"

namespace splash
{

    typedef hid_t H5Handle;

    static const H5Handle INVALID_HANDLE = -1;

    class HandleMgr
    {
    private:

        typedef struct
        {
            H5Handle handle;
            uint32_t ctr;
        } HandleCtrStr;

        typedef struct
        {
            uint32_t index;
            uint32_t ctr;
        } IndexCtrStr;

        typedef std::map<uint32_t, HandleCtrStr> HandleMap;

    public:
        enum FileNameScheme
        {
            FNS_MPI = 0, FNS_ITERATIONS, FNS_FULLNAME
        };

        // callback function types
        typedef void (*FileCreateCallback)(H5Handle handle, uint32_t index, void *userData);
        typedef void (*FileOpenCallback)(H5Handle handle, uint32_t index, void *userData);
        typedef void (*FileCloseCallback)(H5Handle handle, uint32_t index, void *userData);

        HandleMgr(uint32_t maxHandles, FileNameScheme fileNameScheme);

        virtual ~HandleMgr();

        void setFileNameScheme(FileNameScheme fileNameScheme) throw (DCException);

        void open(Dimensions mpiSize, const std::string baseFilename,
                hid_t fileAccProperties, unsigned flags) throw (DCException);

        void open(const std::string fullFilename,
                hid_t fileAccProperties, unsigned flags) throw (DCException);

        void close();

        H5Handle get(uint32_t index) throw (DCException);

        H5Handle get(Dimensions mpiPos) throw (DCException);

        void registerFileCreate(FileCreateCallback callback, void *userData);

        void registerFileOpen(FileOpenCallback callback, void *userData);

        void registerFileClose(FileCloseCallback callback, void *userData);

    private:
        uint32_t maxHandles;

        Dimensions mpiSize;
        std::string filename;
        FileNameScheme fileNameScheme;

        hid_t fileAccProperties;
        unsigned fileFlags;

        HandleMap handles;
        IndexCtrStr leastAccIndex;
        std::set<uint32_t> createdFiles;

        // callback handles
        FileCreateCallback fileCreateCallback;
        void *fileCreateUserData;
        FileOpenCallback fileOpenCallback;
        void *fileOpenUserData;
        FileCloseCallback fileCloseCallback;
        void *fileCloseUserData;

        static std::string getExceptionString(std::string func,
                std::string msg, const char *info);

        uint32_t indexFromPos(Dimensions& mpiPos);
        Dimensions posFromIndex(uint32_t index);
    };
}

#endif /* HANDLEMGR_HPP */