src/lib/cuda_j2k_frame_encoder.h

   1 /*
   2     Copyright (C) 2022 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     DCP-o-matic is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21
  22 #ifndef DCPOMATIC_CUDA_J2K_FRAME_ENCODER
  23 #define DCPOMATIC_CUDA_J2K_FRAME_ENCODER
  24
  25
  26 #include "j2k_frame_encoder.h"
  27 #include "types.h"
  28 #include <dcp/types.h>
  29 #include <nvjpeg2k.h>
  30 #include <boost/thread/condition.hpp>
  31 #include <boost/thread/mutex.hpp>
  32 #include <map>
  33 #include <vector>
  34
  35
  36 class CUDAJ2KFrameEncoder : public J2KFrameEncoder
  37 {
  38 public:
  39         CUDAJ2KFrameEncoder();
  40         ~CUDAJ2KFrameEncoder();
  41
  42         boost::optional<dcp::ArrayData> encode(DCPVideo const &) override;
  43         void flush() override;
  44         void log_thread_start() override;
  45
  46 private:
  47         void encode_queue();
  48
  49         nvjpeg2kEncoder_t _encoder_handle;
  50         nvjpeg2kEncodeState_t _encoder_state;
  51         nvjpeg2kEncodeParams_t _encoder_params;
  52
  53         class Input
  54         {
  55         public:
  56                 Input(DCPVideo const& vf, cudaStream_t stream);
  57                 Input(Input const& other) = delete;
  58                 Input(Input&& other);
  59                 ~Input();
  60
  61                 Input& operator=(Input const& other) = delete;
  62
  63                 nvjpeg2kImage_t const* const device_image() const {
  64                         return &_device_image;
  65                 }
  66
  67                 int index() const {
  68                         return _index;
  69                 }
  70
  71                 Eyes eyes() const {
  72                         return _eyes;
  73                 }
  74
  75         private:
  76                 std::shared_ptr<dcp::OpenJPEGImage> _xyz;
  77                 uint8_t* _pixel_data_h[3];
  78                 uint8_t* _pixel_data_d[3];
  79                 size_t _pitch_in_bytes[3];
  80                 nvjpeg2kImage_t _device_image;
  81                 int _index;
  82                 Eyes _eyes;
  83         };
  84
  85         boost::optional<dcp::Size> _size;
  86         boost::optional<Resolution> _resolution;
  87         cudaStream_t _stream;
  88 };
  89
  90
  91 #endif