FrameGrabber.h

Go to the documentation of this file.

/*************************************************************************************
**
** Copyright 2007, MeVis Medical Solutions AG
**
** The user may use this file in accordance with the license agreement provided with
** the Software or, alternatively, in accordance with the terms contained in a
** written agreement between the user and MeVis Medical Solutions AG.
**
** For further information use the contact form at https://www.mevislab.de/contact
**
**************************************************************************************/
 
#ifndef FRAME_GRABBER_H
#define FRAME_GRABBER_H
 
 
#include "SoViewers.h"
 
#define FRAMEGRABBER_MAX_BYTES 1024*1024*50
 
#define FRAMEGRABBER_CROP 0xfffc
 
class SoSFString;
 
//=============================================================
/*
FrameGrabber is a helper class that handles the creation
of a movie by storing GL frames from a passed SoQtRenderArea
whenever requested. It stores the images in memory and writes
them on disk when requested or when too much memory is consumed.
It is used by \c SoRenderArea.
 
 
*/
//=============================================================
class SO_VIEWERS_CLASS_SPEC FrameGrabber
{
public:
  FrameGrabber();
  ~FrameGrabber();
 
  void cleanup(bool deleteFiles);
 
  void createAviWithBink(const char* filename,float framerate);
 
  void startRecording();
 
  void stopRecording() { _isRecording = false; }
 
  bool isRecording() { return _isRecording; }
 
  void grabFrame(void* renderarea, int mode=GL_BACK);
 
  void writeToFiles();
 
  void setMaximumBytes(long bytes) { _maxBytes = bytes; }
 
  long getMaximumBytes() { return _maxBytes; }
 
  int getDiskFrames() { return _diskCount; };
 
  int getMemFrames() { return static_cast<int>(_frames.size()); };
 
  void setStatusField(SoSFString* field, SoNode* owner) { _status = field; _statusNode = owner; };
 
  void updateStatus(const char* string);
 
  void  writeCurrentFrameToDisk(const char* name, void* renderarea ,int mode);
 
  MLuint8* grabGLFrame(void* renderarea , int& w, int& h, int mode = GL_BACK);
 
  // returns the filename template (without image number & .tif)
  std::string getFileNameTemplate();
 
private:
  void  writeToFile(const char* name, MLuint8* data, int width, int height);
 
  void  createUniqueDirectory();
 
  std::string intToString(int value);
 
  int _uniqueDirCount;
 
  int  _diskCount;
  int  _bufferCount;
 
  bool _dirCreated;
 
  bool _isRecording;
 
  std::string _tmpDir;
 
  int _width,_height;
 
  std::vector<MLuint8*> _frames;
 
  long _maxBytes;
  long _bytes;
 
  SoSFString* _status;
  SoNode*     _statusNode;
};
 
#endif