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