Method ImageDecoder

dali/operators/imgcodec/image_decoder.h:210–391

  explicit ImageDecoder(const OpSpec &spec) : StatelessOperator<Backend>(spec) {
    device_id_ = std::is_same<CPUBackend, Backend>::value ? CPU_ONLY_DEVICE_ID :
                                                            spec.GetArgument<int>("device_id");
    format_ = spec.GetArgument<DALIImageType>("output_type");
    dtype_ = spec.GetArgument<DALIDataType>("dtype");
    use_orientation_ = spec.GetArgument<bool>("adjust_orientation");
    max_batch_size_ = spec.GetArgument<int>("max_batch_size");
    num_threads_ = spec.GetArgument<int>("num_threads");
    GetDecoderSpecificArguments(spec);

    if (std::is_same<MixedBackend, Backend>::value) {
      thread_pool_ = std::make_unique<OldThreadPool>(num_threads_, device_id_,
                                                  spec.GetArgument<bool>("affine"), "MixedDecoder");
      if (spec_.HasArgument("cache_size"))
        cache_ = std::make_unique<CachedDecoderImpl>(spec_);
    }
    EnforceMinimumNvimgcodecVersion();

    // WAR for the nvImageCodec ROI/orientation contract bug: on affected versions, the decoder
    // validates the ROI against raw codestream dims while interpreting it in output (post-EXIF)
    // coords, breaking 90/270-rotated samples that carry a ROI. Disable orientation in the
    // decoder and apply it in the post-decode Convert; ROIs are translated to raw coords.
    roi_orient_war_ = use_orientation_ && !version_at_least(0, 9, 0);

    nvimgcodecDeviceAllocator_t *dev_alloc_ptr = nullptr;
    nvimgcodecPinnedAllocator_t *pinned_alloc_ptr = nullptr;
    if (device_id_ >= 0) {
      dev_alloc_.struct_type = NVIMGCODEC_STRUCTURE_TYPE_DEVICE_ALLOCATOR;
      dev_alloc_.struct_size = sizeof(nvimgcodecDeviceAllocator_t);
      dev_alloc_.struct_next = nullptr;
      dev_alloc_.device_malloc = static_dali_device_malloc;
      dev_alloc_.device_free = static_dali_device_free;
      dev_alloc_.device_ctx = mm::GetDefaultResource<mm::memory_kind::device>();
      dev_alloc_.device_mem_padding = decoder_params_.count("device_memory_padding") == 0 ?
                                          0 :
                                          spec.GetArgument<int64_t>("device_memory_padding");
      dev_alloc_ptr = &dev_alloc_;

      pinned_alloc_.struct_type = NVIMGCODEC_STRUCTURE_TYPE_PINNED_ALLOCATOR;
      pinned_alloc_.struct_size = sizeof(nvimgcodecPinnedAllocator_t);
      pinned_alloc_.struct_next = nullptr;
      pinned_alloc_.pinned_malloc = static_dali_pinned_malloc;
      pinned_alloc_.pinned_free = static_dali_pinned_free;
      pinned_alloc_.pinned_ctx = mm::GetDefaultResource<mm::memory_kind::pinned>();
      pinned_alloc_.pinned_mem_padding = decoder_params_.count("host_memory_padding") == 0 ?
                                             0 :
                                             spec.GetArgument<int64_t>("host_memory_padding");
      pinned_alloc_ptr = &pinned_alloc_;
    }

    nvimgcodecInstanceCreateInfo_t instance_create_info{
        NVIMGCODEC_STRUCTURE_TYPE_INSTANCE_CREATE_INFO, sizeof(nvimgcodecInstanceCreateInfo_t),
        nullptr};

    const char *log_lvl_env = std::getenv("DALI_NVIMGCODEC_LOG_LEVEL");
    int log_lvl = log_lvl_env ? clamp(atoi(log_lvl_env), 1, 5) : 2;

    instance_create_info.load_extension_modules = static_cast<int>(WITH_DYNAMIC_NVIMGCODEC_ENABLED);