1 /*M/////////////////////////////////////////////////////////////////////////////////////// 2 // 3 // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. 4 // 5 // By downloading, copying, installing or using the software you agree to this license. 6 // If you do not agree to this license, do not download, install, 7 // copy or use the software. 8 // 9 // 10 // License Agreement 11 // For Open Source Computer Vision Library 12 // 13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved. 14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved. 15 // Copyright (C) 2013, OpenCV Foundation, all rights reserved. 16 // Third party copyrights are property of their respective owners. 17 // 18 // Redistribution and use in source and binary forms, with or without modification, 19 // are permitted provided that the following conditions are met: 20 // 21 // * Redistribution's of source code must retain the above copyright notice, 22 // this list of conditions and the following disclaimer. 23 // 24 // * Redistribution's in binary form must reproduce the above copyright notice, 25 // this list of conditions and the following disclaimer in the documentation 26 // and/or other materials provided with the distribution. 27 // 28 // * The name of the copyright holders may not be used to endorse or promote products 29 // derived from this software without specific prior written permission. 30 // 31 // This software is provided by the copyright holders and contributors "as is" and 32 // any express or implied warranties, including, but not limited to, the implied 33 // warranties of merchantability and fitness for a particular purpose are disclaimed. 34 // In no event shall the Intel Corporation or contributors be liable for any direct, 35 // indirect, incidental, special, exemplary, or consequential damages 36 // (including, but not limited to, procurement of substitute goods or services; 37 // loss of use, data, or profits; or business interruption) however caused 38 // and on any theory of liability, whether in contract, strict liability, 39 // or tort (including negligence or otherwise) arising in any way out of 40 // the use of this software, even if advised of the possibility of such damage. 41 // 42 //M*/ 43 44 #ifndef __VIDEO_DECODER_HPP__ 45 #define __VIDEO_DECODER_HPP__ 46 47 #include <nvcuvid.h> 48 49 #include "opencv2/core/private.cuda.hpp" 50 #include "opencv2/cudacodec.hpp" 51 52 namespace cv { namespace cudacodec { namespace detail 53 { 54 55 class VideoDecoder 56 { 57 public: VideoDecoder(const FormatInfo & videoFormat,CUvideoctxlock lock)58 VideoDecoder(const FormatInfo& videoFormat, CUvideoctxlock lock) : lock_(lock), decoder_(0) 59 { 60 create(videoFormat); 61 } 62 ~VideoDecoder()63 ~VideoDecoder() 64 { 65 release(); 66 } 67 68 void create(const FormatInfo& videoFormat); 69 void release(); 70 71 // Get the code-type currently used. codec() const72 cudaVideoCodec codec() const { return createInfo_.CodecType; } maxDecodeSurfaces() const73 unsigned long maxDecodeSurfaces() const { return createInfo_.ulNumDecodeSurfaces; } 74 frameWidth() const75 unsigned long frameWidth() const { return createInfo_.ulWidth; } frameHeight() const76 unsigned long frameHeight() const { return createInfo_.ulHeight; } 77 targetWidth() const78 unsigned long targetWidth() const { return createInfo_.ulTargetWidth; } targetHeight() const79 unsigned long targetHeight() const { return createInfo_.ulTargetHeight; } 80 chromaFormat() const81 cudaVideoChromaFormat chromaFormat() const { return createInfo_.ChromaFormat; } 82 decodePicture(CUVIDPICPARAMS * picParams)83 bool decodePicture(CUVIDPICPARAMS* picParams) 84 { 85 return cuvidDecodePicture(decoder_, picParams) == CUDA_SUCCESS; 86 } 87 mapFrame(int picIdx,CUVIDPROCPARAMS & videoProcParams)88 cuda::GpuMat mapFrame(int picIdx, CUVIDPROCPARAMS& videoProcParams) 89 { 90 CUdeviceptr ptr; 91 unsigned int pitch; 92 93 cuSafeCall( cuvidMapVideoFrame(decoder_, picIdx, &ptr, &pitch, &videoProcParams) ); 94 95 return cuda::GpuMat(targetHeight() * 3 / 2, targetWidth(), CV_8UC1, (void*) ptr, pitch); 96 } 97 unmapFrame(cuda::GpuMat & frame)98 void unmapFrame(cuda::GpuMat& frame) 99 { 100 cuSafeCall( cuvidUnmapVideoFrame(decoder_, (CUdeviceptr) frame.data) ); 101 frame.release(); 102 } 103 104 private: 105 CUvideoctxlock lock_; 106 CUVIDDECODECREATEINFO createInfo_; 107 CUvideodecoder decoder_; 108 }; 109 110 }}} 111 112 #endif // __VIDEO_DECODER_HPP__ 113