mitkMovieGeneratorOpenCV.h

Go to the documentation of this file.

/*============================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

============================================================================*/

#ifndef MovieGeneratorOpenCV_H_HEADER_INCLUDED
#define MovieGeneratorOpenCV_H_HEADER_INCLUDED

#include "mitkMovieGenerator.h"
#include <MitkOpenCVVideoSupportExports.h>
#include <memory.h>
#include <cstring>

// OpenCV includes
#include "cv.h"
#include "highgui.h"


namespace mitk {


class MITKOPENCVVIDEOSUPPORT_EXPORT MovieGeneratorOpenCV : public MovieGenerator
{

public:

  mitkClassMacro(MovieGeneratorOpenCV, MovieGenerator);
  itkFactorylessNewMacro(Self);
  itkCloneMacro(Self);

  void SetFileName( const char *fileName ) override;

  void SetFourCCCodec(char* codec)
  {
    m_FourCCCodec = codec;
  }

  void SetFrameRate(unsigned int rate) override;

  void SetRemoveColouredFrame(bool);
protected:

  MovieGeneratorOpenCV();

  bool InitGenerator() override;

  bool AddFrame( void *data ) override;

  bool TerminateGenerator() override;

  std::string m_sFile;

  int m_dwRate;


private:

  CvVideoWriter*    m_aviWriter;
  IplImage *        m_currentFrame;

  char *            m_FourCCCodec;

  bool m_RemoveColouredFrame;
};

} // namespace mitk

#endif /* MovieGeneratorOpenCV_H_HEADER_INCLUDED */