1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99
  100
  101
  102
  103
  104
  105
  106
  107
  108
  109
  110
  111
  112
  113
  114
  115
  116
  117
  118
  119
  120
  121


media / gpu / vaapi / vaapi_image_decode_accelerator_worker.h [blame]

// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_GPU_VAAPI_VAAPI_IMAGE_DECODE_ACCELERATOR_WORKER_H_
#define MEDIA_GPU_VAAPI_VAAPI_IMAGE_DECODE_ACCELERATOR_WORKER_H_

#include <stdint.h>

#include <memory>
#include <unordered_map>
#include <vector>

#include "base/containers/small_map.h"
#include "base/memory/scoped_refptr.h"
#include "base/sequence_checker.h"
#include "gpu/config/gpu_info.h"
#include "gpu/ipc/service/image_decode_accelerator_worker.h"

namespace base {
class SequencedTaskRunner;
}

namespace gfx {
class Size;
}

namespace media {

class VaapiImageDecoder;

using VaapiImageDecoderVector = std::vector<std::unique_ptr<VaapiImageDecoder>>;

using VaapiImageDecoderMap =
    base::small_map<std::unordered_map<gpu::ImageDecodeAcceleratorType,
                                       std::unique_ptr<VaapiImageDecoder>>>;

// This class uses the VAAPI to provide image decode acceleration. The
// interaction with the VAAPI is done on |decoder_task_runner_|.
class VaapiImageDecodeAcceleratorWorker
    : public gpu::ImageDecodeAcceleratorWorker {
 public:
  // Creates a VaapiImageDecodeAcceleratorWorker. Returns nullptr if no image
  // decode profiles are supported.
  static std::unique_ptr<VaapiImageDecodeAcceleratorWorker> Create();

  VaapiImageDecodeAcceleratorWorker(const VaapiImageDecodeAcceleratorWorker&) =
      delete;
  VaapiImageDecodeAcceleratorWorker& operator=(
      const VaapiImageDecodeAcceleratorWorker&) = delete;

  ~VaapiImageDecodeAcceleratorWorker() override;

  // gpu::ImageDecodeAcceleratorWorker implementation.
  gpu::ImageDecodeAcceleratorSupportedProfiles GetSupportedProfiles() override;
  void Decode(std::vector<uint8_t> encoded_data,
              const gfx::Size& output_size,
              CompletedDecodeCB decode_cb) override;

 private:
  friend struct std::default_delete<VaapiImageDecodeAcceleratorWorker>;

  friend class VaapiImageDecodeAcceleratorWorkerTest;

  VaapiImageDecodeAcceleratorWorker(
      VaapiImageDecoderVector decoders,
      gpu::ImageDecodeAcceleratorSupportedProfiles supported_profiles);

  // Calls the destructor of the VaapiImageDecodeAcceleratorWorker instance from
  // the |decoder_task_runner_|.
  void Destroy();

  // Returns a pointer to an Initialize()d VaapiImageDecoder that can be used to
  // decode |encoded_data|. If no suitable VaapiImageDecoder can be
  // Initialize()d, this method returns nullptr. Must be called on the
  // |decoder_task_runner_|.
  VaapiImageDecoder* GetInitializedDecoder(
      const std::vector<uint8_t>& encoded_data);

  // Tries to decode the image corresponding to |encoded_data| initializing the
  // appropriate VaapiImageDecoder instance in |decoders_| as needed. We defer
  // the initialization of the decoder until DecodeTask() so that initialization
  // occurs on the same sequence as decoding (the VaapiWrapper must be created
  // and used on the same sequence). |decode_cb| is called when finished or when
  // an error is encountered. |output_size_for_tracing| is only used for tracing
  // because we don't support decoding to scale. Must be called on the
  // |decoder_task_runner_|.
  void DecodeTask(
      std::vector<uint8_t> encoded_data,
      const gfx::Size& output_size_for_tracing,
      gpu::ImageDecodeAcceleratorWorker::CompletedDecodeCB decode_cb);

  // We delegate the decoding to the appropriate decoder in |decoders_| which
  // are used and destroyed on |decoder_task_runner_|.
  VaapiImageDecoderMap decoders_ GUARDED_BY_CONTEXT(decoder_sequence_checker_);
  const gpu::ImageDecodeAcceleratorSupportedProfiles supported_profiles_;

  scoped_refptr<base::SequencedTaskRunner> decoder_task_runner_;
  SEQUENCE_CHECKER(decoder_sequence_checker_);

  SEQUENCE_CHECKER(main_sequence_checker_);
  SEQUENCE_CHECKER(io_sequence_checker_);

  // WeakPtr factory for *|this| to post tasks to |decoder_task_runner_|.
  base::WeakPtrFactory<VaapiImageDecodeAcceleratorWorker>
      decoder_weak_this_factory_{this};
};

}  // namespace media

namespace std {

// Specializes std::default_delete to call Destroy().
template <>
struct default_delete<media::VaapiImageDecodeAcceleratorWorker> {
  void operator()(media::VaapiImageDecodeAcceleratorWorker* ptr) const;
};

}  // namespace std

#endif  // MEDIA_GPU_VAAPI_VAAPI_IMAGE_DECODE_ACCELERATOR_WORKER_H_