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 #include "precomp.hpp"
     45 
     46 #ifdef HAVE_NVCUVID
     47 
     48 using namespace cv;
     49 using namespace cv::cudacodec;
     50 using namespace cv::cudacodec::detail;
     51 
     52 cv::cudacodec::detail::CuvidVideoSource::CuvidVideoSource(const String& fname)
     53 {
     54     CUVIDSOURCEPARAMS params;
     55     std::memset(&params, 0, sizeof(CUVIDSOURCEPARAMS));
     56 
     57     // Fill parameter struct
     58     params.pUserData = this;                        // will be passed to data handlers
     59     params.pfnVideoDataHandler = HandleVideoData;   // our local video-handler callback
     60     params.pfnAudioDataHandler = 0;
     61 
     62     // now create the actual source
     63     CUresult cuRes = cuvidCreateVideoSource(&videoSource_, fname.c_str(), &params);
     64     if (cuRes == CUDA_ERROR_INVALID_SOURCE)
     65         throw std::runtime_error("");
     66     cuSafeCall( cuRes );
     67 
     68     CUVIDEOFORMAT vidfmt;
     69     cuSafeCall( cuvidGetSourceVideoFormat(videoSource_, &vidfmt, 0) );
     70 
     71     format_.codec = static_cast<Codec>(vidfmt.codec);
     72     format_.chromaFormat = static_cast<ChromaFormat>(vidfmt.chroma_format);
     73     format_.width = vidfmt.coded_width;
     74     format_.height = vidfmt.coded_height;
     75 }
     76 
     77 cv::cudacodec::detail::CuvidVideoSource::~CuvidVideoSource()
     78 {
     79     cuvidDestroyVideoSource(videoSource_);
     80 }
     81 
     82 FormatInfo cv::cudacodec::detail::CuvidVideoSource::format() const
     83 {
     84     return format_;
     85 }
     86 
     87 void cv::cudacodec::detail::CuvidVideoSource::start()
     88 {
     89     cuSafeCall( cuvidSetVideoSourceState(videoSource_, cudaVideoState_Started) );
     90 }
     91 
     92 void cv::cudacodec::detail::CuvidVideoSource::stop()
     93 {
     94     cuSafeCall( cuvidSetVideoSourceState(videoSource_, cudaVideoState_Stopped) );
     95 }
     96 
     97 bool cv::cudacodec::detail::CuvidVideoSource::isStarted() const
     98 {
     99     return (cuvidGetVideoSourceState(videoSource_) == cudaVideoState_Started);
    100 }
    101 
    102 bool cv::cudacodec::detail::CuvidVideoSource::hasError() const
    103 {
    104     return (cuvidGetVideoSourceState(videoSource_) == cudaVideoState_Error);
    105 }
    106 
    107 int CUDAAPI cv::cudacodec::detail::CuvidVideoSource::HandleVideoData(void* userData, CUVIDSOURCEDATAPACKET* packet)
    108 {
    109     CuvidVideoSource* thiz = static_cast<CuvidVideoSource*>(userData);
    110 
    111     return thiz->parseVideoData(packet->payload, packet->payload_size, (packet->flags & CUVID_PKT_ENDOFSTREAM) != 0);
    112 }
    113 
    114 #endif // HAVE_NVCUVID
    115