1
0
Fork 0
flightgear/utils/fgviewer/MEncoderCaptureOperation.hxx

139 lines
4 KiB
C++
Raw Normal View History

// 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