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 <vector>
  31
  32
  33 class CUDAJ2KFrameEncoder : public J2KFrameEncoder
  34 {
  35 public:
  36         CUDAJ2KFrameEncoder();
  37
  38         std::vector<dcp::ArrayData> encode(DCPVideo const &) override;
  39         std::vector<dcp::ArrayData> flush() override;
  40         void log_thread_start() override;
  41
  42 private:
  43         nvjpeg2kEncoder_t _encoder_handle;
  44         nvjpeg2kEncodeState_t _encoder_state;
  45         nvjpeg2kEncodeParams_t _encoder_params;
  46
  47         class Frame
  48         {
  49         public:
  50                 Frame(DCPVideo const& vf);
  51                 Frame(Frame const& other) = delete;
  52                 Frame(Frame&& other);
  53                 ~Frame();
  54
  55                 Frame& operator=(Frame const& other) = delete;
  56
  57                 nvjpeg2kImage_t const* const device_image() const {
  58                         return &_device_image;
  59                 }
  60
  61         private:
  62                 uint8_t* _pixel_data_d[3];
  63                 size_t _pitch_in_bytes[3];
  64                 nvjpeg2kImage_t _device_image;
  65         };
  66
  67         std::vector<Frame> _batch;
  68         boost::optional<dcp::Size> _size;
  69         boost::optional<Resolution> _resolution;
  70 };
  71
  72
  73 #endif