cxSavingVideoRecorder.h

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.
                 
Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.
                 
CustusX is released under a BSD 3-Clause license.
                 
See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/
#ifndef CXSAVINGVIDEORECORDER_H
#define CXSAVINGVIDEORECORDER_H
 
#include "cxResourceExport.h"
 
#include <vector>
#include <QFile>
#include <QThread>
#include <QMutex>
 
#include "vtkForwardDeclarations.h"
#include "cxForwardDeclarations.h"
#include "cxData.h"
 
namespace cx
{
typedef boost::shared_ptr<class CachedImageDataContainer> CachedImageDataContainerPtr;
 
class cxResource_EXPORT VideoRecorderSaveThread : public QThread
{
        Q_OBJECT
public:
        VideoRecorderSaveThread(QObject* parent, QString saveFolder, QString prefix, bool compressed, bool writeColor);
        virtual ~VideoRecorderSaveThread();
        QString addData(TimeInfo timestamp, vtkImageDataPtr data);
        void stop();
        void cancel();
 
protected:
        struct DataType
        {
                TimeInfo mTimestamp;
                QString mImageFilename;
                vtkImageDataPtr mImage;
        };
        QString mSaveFolder;
        QString mPrefix;
        int mImageIndex;
        std::list<DataType> mPendingData;
        QMutex mMutex; 
        bool mStop;
        bool mCancel;
        QFile mTimestampsFile;
        bool mCompressed;
        bool mWriteColor;
        virtual void run();
 
        void writeQueue();
        bool openTimestampsFile();
        bool closeTimestampsFile();
        void write(DataType data);
        void writeTimeStampsFile(TimeInfo timeStamps);
};
 
class cxResource_EXPORT SavingVideoRecorder : public QObject
{
        Q_OBJECT
 
public:
        SavingVideoRecorder(VideoSourcePtr source, QString saveFolder, QString prefix, bool compressed, bool writeColor, FileManagerServicePtr filemanagerservice);
        virtual ~SavingVideoRecorder();
 
        virtual void startRecord();
        virtual void stopRecord();
        void cancel();
 
        CachedImageDataContainerPtr getImageData();
        std::vector<TimeInfo> getTimestamps();
        QString getSaveFolder() { return mSaveFolder; }
 
        void completeSave();
 
        VideoSourcePtr getSource() { return mSource; }
 
private slots:
        void newFrameSlot();
private:
        void deleteFolder(QString folder);
        CachedImageDataContainerPtr mImages;
        std::vector<TimeInfo> mTimestamps;
        QString mSaveFolder;
        QString mPrefix;
        VideoSourcePtr mSource;
        boost::shared_ptr<VideoRecorderSaveThread> mSaveThread;
 
};
 
typedef boost::shared_ptr<SavingVideoRecorder> SavingVideoRecorderPtr;
 
 
} // namespace cx
 
 
#endif // CXSAVINGVIDEORECORDER_H