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
  41         boost::optional<dcp::ArrayData> encode(DCPVideo const &) override;
  42         void flush() override;
  43         void log_thread_start() override;
  44
  45 private:
  46         void encode_queue();
  47
  48         nvjpeg2kEncoder_t _encoder_handle;
  49         nvjpeg2kEncodeState_t _encoder_state;
  50         nvjpeg2kEncodeParams_t _encoder_params;
  51
  52         class Input
  53         {
  54         public:
  55                 Input(DCPVideo const& vf);
  56                 Input(Input const& other) = delete;
  57                 Input(Input&& other);
  58                 ~Input();
  59
  60                 Input& operator=(Input const& other) = delete;
  61
  62                 nvjpeg2kImage_t const* const device_image() const {
  63                         return &_device_image;
  64                 }
  65
  66                 int index() const {
  67                         return _index;
  68                 }
  69
  70                 Eyes eyes() const {
  71                         return _eyes;
  72                 }
  73
  74         private:
  75                 uint8_t* _pixel_data_d[3];
  76                 size_t _pitch_in_bytes[3];
  77                 nvjpeg2kImage_t _device_image;
  78                 int _index;
  79                 Eyes _eyes;
  80         };
  81
  82         boost::optional<dcp::Size> _size;
  83         boost::optional<Resolution> _resolution;
  84         cudaStream_t _stream;
  85 };
  86
  87
  88 #endif