first commit

utils/fgviewer/MEncoderCaptureOperation.hxx

@@ -0,0 +1,138 @@
+// MEncoderCaptureOperation.hxx -- capture video stream into mencoder
+//
+// Copyright (C) 2009 - 2012  Mathias Froehlich
+//
+// This program is free software; you can redistribute it and/or
+// modify it under the terms of the GNU General Public License as
+// published by the Free Software Foundation; either version 2 of the
+// License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+#ifndef MEncoderCaptureOperation_HXX
+#define MEncoderCaptureOperation_HXX
+
+#include <cstdio>
+#include <sstream>
+#include <string>
+
+#include <osgViewer/ViewerEventHandlers>
+
+/// Class to capture into a pipe driven mencoder.
+/// To integrate this into a viewer:
+///  MEncoderCaptureOperation* mencoderCaptureOperation = new MEncoderCaptureOperation("/tmp/fgviewer.avi", 60);
+///  osgViewer::ScreenCaptureHandler* c = new osgViewer::ScreenCaptureHandler(mencoderCaptureOperation, -1);
+///  viewer.addEventHandler(c);
+///  c->startCapture();
+
+namespace fgviewer {
+
+class MEncoderCaptureOperation : public osgViewer::ScreenCaptureHandler::CaptureOperation {
+public:
+    MEncoderCaptureOperation(const std::string& fileName = "video.avi", unsigned fps = 30) :
+        _fps(fps),
+        _fileName(fileName),
+        _options("-ovc lavc"),
+        _file(0),
+        _width(-1),
+        _height(-1)
+    { }
+    virtual ~MEncoderCaptureOperation()
+    { _close(); }
+
+    const std::string& getFileName() const
+    { return _fileName; }
+    void setFileName(const std::string& fileName)
+    { _fileName = fileName; }
+
+    unsigned getFramesPerSecond() const
+    { return _fps; }
+    void setFramesPerSecond(unsigned fps)
+    { _fps = fps; }
+
+    const std::string& getOptions() const
+    { return _options; }
+    void setOptions(const std::string& options)
+    { _options = options; }
+
+    virtual void operator()(const osg::Image& image, const unsigned int)
+    {
+        // Delay any action until we have a valid image
+        if (!image.valid())
+            return;
+
+        // Ensure an open file
+        if (!_file) {
+            // If the video was already opened and we got any error,
+            // do not reopen with the same name.
+            if (0 < _width)
+                return;
+            _width = image.s();
+            _height = image.t();
+            if (!_open())
+                return;
+        }
+        // Ensure we did not change dimensions
+        if (image.s() != _width)
+            return;
+        if (image.t() != _height)
+            return;
+
+        // Write upside down flipped image
+        for (int row = _height - 1; 0 <= row; --row) {
+            size_t ret = fwrite(image.data(0, row), 1, image.getRowSizeInBytes(), _file);
+            if (ret != image.getRowSizeInBytes())
+                return;
+        }
+    }
+
+private:
+    bool _open()
+    {
+        if (_file)
+            return false;
+        /// FIXME improve: adapt format to the format we get from the image
+        std::stringstream ss;
+        ss << "mencoder - -demuxer rawvideo -rawvideo fps="
+           << _fps << ":w=" << _width << ":h=" << _height
+           << ":format=rgb24 -o " << _fileName << " " << _options;
+#ifdef _WIN32
+        _file = _popen(ss.str().c_str(), "wb");
+#else
+        _file = popen(ss.str().c_str(), "w");
+#endif
+        return _file != 0;
+    }
+    void _close()
+    {
+        if (!_file)
+            return;
+#ifdef _WIN32
+        _pclose(_file);
+#else
+        pclose(_file);
+#endif
+        _file = 0;
+    }
+
+    /// Externally given:
+    unsigned _fps;
+    std::string _fileName;
+    std::string _options;
+
+    /// Internal determined
+    FILE* _file;
+    int _width;
+    int _height;
+};
+
+}
+
+#endif