00001 // ***************************************************************************** 00002 // 00003 // Copyright (c) 2017, Southwest Research Institute® (SwRI®) 00004 // All rights reserved. 00005 // 00006 // Redistribution and use in source and binary forms, with or without 00007 // modification, are permitted provided that the following conditions are met: 00008 // * Redistributions of source code must retain the above copyright 00009 // notice, this list of conditions and the following disclaimer. 00010 // * Redistributions in binary form must reproduce the above copyright 00011 // notice, this list of conditions and the following disclaimer in the 00012 // documentation and/or other materials provided with the distribution. 00013 // * Neither the name of Southwest Research Institute® (SwRI®) nor the 00014 // names of its contributors may be used to endorse or promote products 00015 // derived from this software without specific prior written permission. 00016 // 00017 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 00018 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 00019 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 00020 // ARE DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY 00021 // DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 00022 // (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 00023 // LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 00024 // ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 00025 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00026 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00027 // 00028 // ***************************************************************************** 00029 00030 #ifndef MAPVIZ_VIDEO_WRITER_H 00031 #define MAPVIZ_VIDEO_WRITER_H 00032 00033 #include <QObject> 00034 #include <QMutex> 00035 #include <QImage> 00036 00037 #include <boost/shared_ptr.hpp> 00038 00039 #ifndef Q_MOC_RUN 00040 #include <opencv2/highgui/highgui.hpp> 00041 #endif 00042 00043 namespace mapviz 00044 { 00045 class VideoWriter : public QObject 00046 { 00047 Q_OBJECT 00048 00049 public: 00050 VideoWriter() : 00051 video_mutex_(QMutex::Recursive) 00052 {} 00053 00054 bool initializeWriter(const std::string& directory, int width, int height); 00055 bool isRecording(); 00056 void stop(); 00057 00058 public Q_SLOTS: 00059 void processFrame(QImage frame); 00060 00061 private: 00062 int height_; 00063 int width_; 00064 QMutex video_mutex_; 00065 boost::shared_ptr<cv::VideoWriter> video_writer_; 00066 }; 00067 } 00068 00069 #endif //MAPVIZ_VIDEO_WRITER_H