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);
        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