FileStorageExtractor.h

Go to the documentation of this file.

#ifndef FILE_STORAGE_EXTRACTOR_H
#define FILE_STORAGE_EXTRACTOR_H

#include <Extractor.h>
#include <StorageManager.h>

#include <thread>
#include <mutex>

class FileStorageExtractor : public Extractor {
public:
    FileStorageExtractor(zmq::context_t & _ctx,
                         const PubSubEndpoint & _source,
                         const std::string & _identifier,
                         StorageManager & _storageManager,
                         int _maxFrames = -1,
                         const std::string & _metaData = "{}",
                         int _framesPerFileCycle = -1,
                         bool _cumulative = false,
                         bool _binning = false,
                         double _wantedFPS = 0);
    ~FileStorageExtractor();
    std::string type();
    std::string identifier();
    std::list<std::string> parameters();
private:
    void extractSetup();
    void extractImpl(unsigned int _width,
                     unsigned int _height,
                     unsigned int _bytesPerPixel,
                     uint64_t highResTime,
                     uint64_t wallTime,
                     zmq::message_t * sourceImage);
    void extractTeardown();
    
    void openFile();
    void closeFile();

    std::string s_identifier;
    StorageManager * storageManager;
    char filename[256];
    FILE* fil = nullptr;
    
    std::list<uint64_t> wallTimes;
    std::list<uint64_t> highResTimes;
    unsigned int width;
    unsigned int height;
    unsigned int bytesPerPixel;
    int framesWritten;
    int totalFramesWritten;
    int maxFrames;
    int framesPerFileCycle;
    bool cumulative;
    bool binning;
    std::string metaData;
    bool error;

    float * accLinear = nullptr;
    float * accQuadratic = nullptr;
    uint16_t * tmpdata = nullptr;

    std::thread * lastHeaderWriterThread = nullptr;
    std::mutex lastHeaderWriterThreadMutex;
};

#endif /* FILE_STORAGE_EXTRACTOR_H */