StreamReader.h

Go to the documentation of this file.

 
#ifndef __ROGUE_UTILITIES_FILEIO_STREAM_READER_H__
#define __ROGUE_UTILITIES_FILEIO_STREAM_READER_H__
#include "rogue/Directives.h"
 
#include <stdint.h>
 
#include <atomic>
#include <condition_variable>
#include <map>
#include <memory>
#include <mutex>
#include <string>
#include <thread>
 
#include "rogue/interfaces/stream/Master.h"
 
namespace rogue {
namespace utilities {
namespace fileio {
 
class StreamReader : public rogue::interfaces::stream::Master {
    // Base file name for indexed-file mode.
    std::string baseName_;
 
    int32_t fd_ = -1;
 
    // Current file index in indexed-file mode.
    uint32_t fdIdx_ = 0;
 
    // True while read thread is actively processing file data.
    bool active_ = false;
 
  protected:
    std::thread* readThread_ = nullptr;
    std::atomic<bool> threadEn_{false};
 
  private:
    // Worker thread entry point.
    void runThread();
 
    // Open next indexed file in sequence.
    bool nextFile();
 
    // Internal close helper.
    void intClose();
 
    // Condition signaled when activity state changes.
    std::condition_variable cond_;
 
    // Mutex for file/thread/activity state.
    std::mutex mtx_;
 
  public:
    static std::shared_ptr<rogue::utilities::fileio::StreamReader> create();
 
    static void setup_python();
 
    StreamReader();
 
    ~StreamReader();
 
    void open(std::string file);
 
    void close();
 
    bool isOpen();
 
    void closeWait();
 
    bool isActive();
};
 
// Convenience
typedef std::shared_ptr<rogue::utilities::fileio::StreamReader> StreamReaderPtr;
}  // namespace fileio
}  // namespace utilities
}  // namespace rogue
#endif