hwcontext_cuda.h 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. /*
  2. * This file is part of FFmpeg.
  3. *
  4. * FFmpeg is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU Lesser General Public
  6. * License as published by the Free Software Foundation; either
  7. * version 2.1 of the License, or (at your option) any later version.
  8. *
  9. * FFmpeg is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * Lesser General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU Lesser General Public
  15. * License along with FFmpeg; if not, write to the Free Software
  16. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17. */
  18. #ifndef AVUTIL_HWCONTEXT_CUDA_H
  19. #define AVUTIL_HWCONTEXT_CUDA_H
  20. #ifndef CUDA_VERSION
  21. #include <cuda.h>
  22. #endif
  23. #include "pixfmt.h"
  24. /**
  25. * @file
  26. * An API-specific header for AV_HWDEVICE_TYPE_CUDA.
  27. *
  28. * This API supports dynamic frame pools. AVHWFramesContext.pool must return
  29. * AVBufferRefs whose data pointer is a CUdeviceptr.
  30. */
  31. typedef struct AVCUDADeviceContextInternal AVCUDADeviceContextInternal;
  32. /**
  33. * This struct is allocated as AVHWDeviceContext.hwctx
  34. */
  35. typedef struct AVCUDADeviceContext {
  36. CUcontext cuda_ctx;
  37. CUstream stream;
  38. AVCUDADeviceContextInternal *internal;
  39. } AVCUDADeviceContext;
  40. /**
  41. * AVHWFramesContext.hwctx is currently not used
  42. */
  43. /**
  44. * @defgroup hwcontext_cuda Device context creation flags
  45. *
  46. * Flags for av_hwdevice_ctx_create.
  47. *
  48. * @{
  49. */
  50. /**
  51. * Use primary device context instead of creating a new one.
  52. */
  53. #define AV_CUDA_USE_PRIMARY_CONTEXT (1 << 0)
  54. /**
  55. * @}
  56. */
  57. #endif /* AVUTIL_HWCONTEXT_CUDA_H */