Home Research Papers Posters Manuals Talks Download People Help Internal

libs/ck-libs/io/ckio.h

Go to the documentation of this file.

#ifndef CK_IO_H
#define CK_IO_H

#include <string>
#include <pup.h>
#include <ckcallback.h>

#include "CkIO.decl.h"

namespace Ck { namespace IO {
  struct Options {
    Options()
      : peStripe(0), writeStripe(0), activePEs(-1), basePE(-1), skipPEs(-1)
      { }

    size_t peStripe;
    size_t writeStripe;
    int activePEs;
    int basePE;
    int skipPEs;

    void pup(PUP::er &p) {
      p|peStripe;
      p|writeStripe;
      p|activePEs;
      p|basePE;
      p|skipPEs;
    }
  };

  class File;
  class Session;

  void open(std::string name, CkCallback opened, Options opts);

  void startSession(File file, size_t bytes, size_t offset,
                    CkCallback ready, CkCallback complete);

  void startSession(File file, size_t bytes, size_t offset, CkCallback ready,
                    const char *commitData, size_t commitBytes, size_t commitOffset,
                    CkCallback complete);

  void write(Session session, const char *data, size_t bytes, size_t offset);

  void close(File file, CkCallback closed);

  class File {
    int token;
    friend void startSession(File file, size_t bytes, size_t offset,
                             CkCallback ready, CkCallback complete);
    friend void startSession(File file, size_t bytes, size_t offset, CkCallback ready,
                             const char *commitData, size_t commitBytes, size_t commitOffset,
                             CkCallback complete);
    friend void close(File file, CkCallback closed);
    friend class FileReadyMsg;

  public:
    File(int token_) : token(token_) { }
    File() : token(-1) { }
    void pup(PUP::er &p) { p|token; }
  };

  class FileReadyMsg : public CMessage_FileReadyMsg {
  public:
    File file;
    FileReadyMsg(const File &tok) : file(tok) {}
  };

  namespace impl { class Manager; }

  class Session {
    int file;
    size_t bytes, offset;
    CkArrayID sessionID;
    friend class Ck::IO::impl::Manager;
  public:
    Session(int file_, size_t bytes_, size_t offset_,
            CkArrayID sessionID_)
      : file(file_), bytes(bytes_), offset(offset_), sessionID(sessionID_)
      { }
    Session() { }
    void pup(PUP::er &p) {
      p|file;
      p|bytes;
      p|offset;
      p|sessionID;
    }
  };

  class SessionReadyMsg : public CMessage_SessionReadyMsg {
  public:
    Session session;
    SessionReadyMsg(Session session_) : session(session_) { }
  };

}}
#endif

---