mitkBaseVideoProcessor.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef mitkBaseVideoProcessor_h
#define mitkBaseVideoProcessor_h

#include "niftkOpenCVExports.h"

#include <cv.h>
#include <highgui.h>
#include <cstdlib>
#include <iostream>
#include <itkObject.h>
#include <itkObjectFactory.h>
#include <mitkCommon.h>

namespace mitk {

class NIFTKOPENCV_EXPORT BaseVideoProcessor : public itk::Object
{

public:

  mitkClassMacroItkParent(BaseVideoProcessor, itk::Object)

  
  virtual void Initialize();

  virtual void Run() = 0;

protected:

  ~BaseVideoProcessor();
  BaseVideoProcessor(CvCapture *capture = NULL, CvVideoWriter *writer = NULL);
  BaseVideoProcessor(const std::string& inputFile, const std::string& outputFile);

  BaseVideoProcessor(const BaseVideoProcessor&); // Purposefully not implemented.
  BaseVideoProcessor& operator=(const BaseVideoProcessor&); // Purposefully not implemented.

  virtual CvSize GetOutputImageSize() = 0;

  IplImage* GetCurrentImage() const;

  IplImage* GrabNewImage();

  CvVideoWriter* GetWriter() const;

private:

  IplImage      *m_GrabbedImage;
  CvCapture     *m_Capture;
  CvVideoWriter *m_Writer;
  std::string    m_InputFileName;
  std::string    m_OutputFileName;
}; // end class

} // end namespace

#endif // MITKBASEVIDEOPROCESSOR_H