Home | History | Annotate | Download | only in src 
      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:
     58     VideoDecoder(const FormatInfo& videoFormat, CUvideoctxlock lock) : lock_(lock), decoder_(0)
     59     {
     60         create(videoFormat);
     61     }
     62 
     63     ~VideoDecoder()
     64     {
     65         release();
     66     }
     67 
     68     void create(const FormatInfo& videoFormat);
     69     void release();
     70 
     71     // Get the code-type currently used.
     72     cudaVideoCodec codec() const { return createInfo_.CodecType; }
     73     unsigned long maxDecodeSurfaces() const { return createInfo_.ulNumDecodeSurfaces; }
     74 
     75     unsigned long frameWidth() const { return createInfo_.ulWidth; }
     76     unsigned long frameHeight() const { return createInfo_.ulHeight; }
     77 
     78     unsigned long targetWidth() const { return createInfo_.ulTargetWidth; }
     79     unsigned long targetHeight() const { return createInfo_.ulTargetHeight; }
     80 
     81     cudaVideoChromaFormat chromaFormat() const { return createInfo_.ChromaFormat; }
     82 
     83     bool decodePicture(CUVIDPICPARAMS* picParams)
     84     {
     85         return cuvidDecodePicture(decoder_, picParams) == CUDA_SUCCESS;
     86     }
     87 
     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 
     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