ppapi: implement PPB_AudioEncoder

ppapi/proxy/audio_encoder_resource.h

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef PPAPI_PROXY_AUDIO_ENCODER_RESOURCE_H_
 #define PPAPI_PROXY_AUDIO_ENCODER_RESOURCE_H_
 
 #include "base/basictypes.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/scoped_vector.h"
 #include "ppapi/proxy/connection.h"
 #include "ppapi/proxy/plugin_resource.h"
+#include "ppapi/shared_impl/media_stream_buffer_manager.h"
 #include "ppapi/shared_impl/resource.h"
 #include "ppapi/thunk/ppb_audio_encoder_api.h"
 
 namespace ppapi {
 
 class TrackedCallback;
 
 namespace proxy {
 
-class SerializedHandle;
-class VideoFrameResource;
+class AudioBufferResource;
 
 class PPAPI_PROXY_EXPORT AudioEncoderResource
     : public PluginResource,
-      public thunk::PPB_AudioEncoder_API {
+      public thunk::PPB_AudioEncoder_API,
+      public ppapi::MediaStreamBufferManager::Delegate {
  public:
   AudioEncoderResource(Connection connection, PP_Instance instance);
   ~AudioEncoderResource() override;
 
   thunk::PPB_AudioEncoder_API* AsPPB_AudioEncoder_API() override;
 
  private:
-  // PPB_AduioEncoder_API implementation.
+  // MediaStreamBufferManager::Delegate implementation.
+  void OnNewBufferEnqueued() override {}
+
+  // PPB_AudioEncoder_API implementation.
   int32_t GetSupportedProfiles(
       const PP_ArrayOutput& output,
       const scoped_refptr<TrackedCallback>& callback) override;
   int32_t Initialize(uint32_t channels,
                      PP_AudioBuffer_SampleRate input_sample_rate,
                      PP_AudioBuffer_SampleSize input_sample_size,
                      PP_AudioProfile output_profile,
                      uint32_t initial_bitrate,
                      PP_HardwareAcceleration acceleration,
                      const scoped_refptr<TrackedCallback>& callback) override;
   int32_t GetNumberOfSamples() override;
   int32_t GetBuffer(PP_Resource* audio_buffer,
                     const scoped_refptr<TrackedCallback>& callback) override;
   int32_t Encode(PP_Resource audio_buffer,
                  const scoped_refptr<TrackedCallback>& callback) override;
   int32_t GetBitstreamBuffer(
       PP_AudioBitstreamBuffer* bitstream_buffer,
       const scoped_refptr<TrackedCallback>& callback) override;
   void RecycleBitstreamBuffer(
       const PP_AudioBitstreamBuffer* bitstream_buffer) override;
   void RequestBitrateChange(uint32_t bitrate) override;
   void Close() override;
 
+  // PluginResource implementation.
+  void OnReplyReceived(const ResourceMessageReplyParams& params,
+                       const IPC::Message& msg) override;
+
+  // Message handlers for the host's messages.
+  void OnPluginMsgGetSupportedProfilesReply(
+      const PP_ArrayOutput& output,
+      const ResourceMessageReplyParams& params,
+      const std::vector<PP_AudioProfileDescription>& profiles);
+  void OnPluginMsgInitializeReply(const ResourceMessageReplyParams& params,
+                                  int32_t number_of_samples,
+                                  int32_t audio_buffer_count,
+                                  int32_t audio_buffer_size,
+                                  int32_t bitstream_buffer_count,
+                                  int32_t bitstream_buffer_size);
+  void OnPluginMsgEncodeReply(const ResourceMessageReplyParams& params,
+                              int32_t buffer_id);
+  void OnPluginMsgBitstreamBufferReady(const ResourceMessageReplyParams& params,
+                                       int32_t buffer_id);
+  void OnPluginMsgNotifyError(const ResourceMessageReplyParams& params,
+                              int32_t error);
+
+  // Internal utility functions.
+  void NotifyError(int32_t error);
+  void TryGetAudioBuffer();
+  void TryWriteBitstreamBuffer();
+  void ReleaseBuffers();
+
+  int32_t encoder_last_error_;
+
+  bool initialized_;
+
+  uint32_t number_of_samples_;
+
+  using AudioBufferMap =
+      std::map<PP_Resource, scoped_refptr<AudioBufferResource>>;
+  AudioBufferMap audio_buffers_;
+
+  scoped_refptr<TrackedCallback> get_supported_profiles_callback_;
+  scoped_refptr<TrackedCallback> initialize_callback_;
+  scoped_refptr<TrackedCallback> get_buffer_callback_;
+  PP_Resource* get_buffer_data_;
+
+  using EncodeMap = std::map<int32_t, scoped_refptr<TrackedCallback>>;
+  EncodeMap encode_callbacks_;
+
+  scoped_refptr<TrackedCallback> get_bitstream_buffer_callback_;
+  PP_AudioBitstreamBuffer* get_bitstream_buffer_data_;
+
+  MediaStreamBufferManager audio_buffer_manager_;
+  MediaStreamBufferManager bitstream_buffer_manager_;
+
+  // Map of bitstream buffer pointers to buffer ids.
+  using BufferMap = std::map<void*, int32_t>;
+  BufferMap bitstream_buffer_map_;
+
   DISALLOW_COPY_AND_ASSIGN(AudioEncoderResource);
 };
 
 }  // namespace proxy
 }  // namespace ppapi
 
 #endif  // PPAPI_PROXY_AUDIO_ENCODER_RESOURCE_H_