mitkMovieGeneratorOpenCV.cpp
Go to the documentation of this file.
00001 /*========================================================================= 00002 00003 Program: Medical Imaging & Interaction Toolkit 00004 Language: C++ 00005 Date: $Date$ 00006 Version: $Revision$ 00007 00008 Copyright (c) German Cancer Research Center, Division of Medical and 00009 Biological Informatics. All rights reserved. 00010 See MITKCopyright.txt or https://www.mitk.org/copyright.html for details. 00011 00012 This software is distributed WITHOUT ANY WARRANTY; without even 00013 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 00014 PURPOSE. See the above copyright notices for more information. 00015 00016 =========================================================================*/ 00017 00018 #include "mitkMovieGeneratorOpenCV.h" 00019 #include <GL/gl.h> 00020 #include <iostream> 00021 00022 00023 mitk::MovieGeneratorOpenCV::MovieGeneratorOpenCV() 00024 { 00025 m_initialized = false; 00026 m_aviWriter = NULL; 00027 m_dwRate = 20; 00028 00029 m_FourCCCodec = NULL; 00030 } 00031 00032 00033 void mitk::MovieGeneratorOpenCV::SetFileName( const char *fileName ) 00034 { 00035 00036 m_sFile = fileName; 00037 } 00038 00039 bool mitk::MovieGeneratorOpenCV::InitGenerator() 00040 { 00041 m_width = m_renderer->GetRenderWindow()->GetSize()[0]; // changed from glGetIntegerv( GL_VIEWPORT, viewport ); 00042 m_height = m_renderer->GetRenderWindow()->GetSize()[1]; // due to sometimes strange dimensions 00043 00044 m_width -= 10; //remove colored boarders around renderwindows 00045 m_height -= 10; 00046 00047 m_width -= m_width % 4; // some video codecs have prerequisites to the image dimensions 00048 m_height -= m_height % 4; 00049 00050 00051 00052 m_currentFrame = cvCreateImage(cvSize(m_width,m_height),8,3); // creating image with widget size, 8 bit per pixel and 3 channel r,g,b 00053 m_currentFrame->origin = 1; // avoid building a video with bottom up 00054 00055 /* 00056 m_sFile = Name of the output video file. 00057 CV_FOURCC = 4-character code of codec used to compress the frames. For example, CV_FOURCC('P','I','M','1') is MPEG-1 codec, 00058 CV_FOURCC('M','J','P','G') is motion-jpeg codec etc. 00059 CV_FOURCC('P','I','M','1') = MPEG-1 codec 00060 CV_FOURCC('M','J','P','G') = motion-jpeg codec (does not work well) 00061 CV_FOURCC('M', 'P', '4', '2') = MPEG-4.2 codec 00062 CV_FOURCC('D', 'I', 'V', '3') = MPEG-4.3 codec 00063 CV_FOURCC('D', 'I', 'V', 'X') = MPEG-4 codec 00064 CV_FOURCC('U', '2', '6', '3') = H263 codec 00065 CV_FOURCC('I', '2', '6', '3') = H263I codec 00066 CV_FOURCC('F', 'L', 'V', '1') = FLV1 codec 00067 00068 List of FOURCC codes is available at http://msdn2.microsoft.com/en-us/library/ms867195.aspx 00069 00070 Under Win32 it is possible to pass -1 in order to choose compression 00071 method and additional compression parameters from dialog. 00072 m_dwRate = Framerate of the created video stream. 00073 frame_size Size of video frames. 00074 1 = If it is not zero, the encoder will expect and encode color frames, otherwise it will work with grayscale frames 00075 (the flag is currently supported on Windows only).*/ 00076 00077 if(m_FourCCCodec != NULL) 00078 { 00079 #ifdef WIN32 00080 m_aviWriter = cvCreateVideoWriter(m_sFile,CV_FOURCC(m_FourCCCodec[0],m_FourCCCodec[1],m_FourCCCodec[2], 00081 m_FourCCCodec[3]),m_dwRate,cvSize(m_width,m_height),1); //initializing video writer 00082 #else 00083 m_aviWriter = cvCreateVideoWriter(m_sFile,CV_FOURCC(m_FourCCCodec[0],m_FourCCCodec[1],m_FourCCCodec[2], 00084 m_FourCCCodec[3]),m_dwRate,cvSize(m_width,m_height)); //initializing video writer 00085 #endif 00086 } 00087 else 00088 { 00089 #ifdef WIN32 00090 m_aviWriter = cvCreateVideoWriter(m_sFile,-1,m_dwRate,cvSize(m_width,m_height),1); //initializing video writer 00091 #else 00092 m_aviWriter = cvCreateVideoWriter(m_sFile,CV_FOURCC('X','V','I','D'),m_dwRate,cvSize(m_width,m_height)); //initializing video writer 00093 #endif 00094 } 00095 00096 00097 if(!m_aviWriter) 00098 { 00099 std::cout << "errors initializing video writer...correct video file path? on linux: ffmpeg must be included in OpenCV."; 00100 return false; 00101 } 00102 00103 return true; 00104 } 00105 00106 00107 bool mitk::MovieGeneratorOpenCV::AddFrame( void *data ) 00108 { 00109 //cvSetImageData(m_currentFrame,data,m_width*3); 00110 memcpy(m_currentFrame->imageData,data,m_width*m_height*3); 00111 cvWriteFrame(m_aviWriter,m_currentFrame); 00112 return true; 00113 } 00114 00115 00116 bool mitk::MovieGeneratorOpenCV::TerminateGenerator() 00117 { 00118 if (m_aviWriter) 00119 { 00120 cvReleaseVideoWriter(&m_aviWriter); 00121 } 00122 return true; 00123 }
