| // Copyright 2016 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. |
| |
| #include "media/gpu/ipc/service/media_gpu_channel_manager.h" |
| |
| #include <memory> |
| #include <utility> |
| |
| #include "gpu/ipc/service/gpu_channel.h" |
| #include "gpu/ipc/service/gpu_channel_manager.h" |
| #include "ipc/ipc_message_macros.h" |
| #include "ipc/param_traits_macros.h" |
| #include "media/gpu/ipc/service/gpu_video_decode_accelerator.h" |
| #include "media/gpu/ipc/service/gpu_video_encode_accelerator.h" |
| #include "media/gpu/ipc/service/media_gpu_channel.h" |
| |
| namespace media { |
| |
| MediaGpuChannelManager::MediaGpuChannelManager( |
| gpu::GpuChannelManager* channel_manager) |
| : channel_manager_(channel_manager) {} |
| |
| MediaGpuChannelManager::~MediaGpuChannelManager() {} |
| |
| void MediaGpuChannelManager::AddChannel(int32_t client_id) { |
| gpu::GpuChannel* gpu_channel = channel_manager_->LookupChannel(client_id); |
| DCHECK(gpu_channel); |
| base::UnguessableToken channel_token = base::UnguessableToken::Create(); |
| std::unique_ptr<MediaGpuChannel> media_gpu_channel( |
| new MediaGpuChannel(gpu_channel, channel_token)); |
| gpu_channel->SetUnhandledMessageListener(media_gpu_channel.get()); |
| media_gpu_channels_[client_id] = std::move(media_gpu_channel); |
| channel_to_token_[client_id] = channel_token; |
| token_to_channel_[channel_token] = client_id; |
| } |
| |
| void MediaGpuChannelManager::RemoveChannel(int32_t client_id) { |
| media_gpu_channels_.erase(client_id); |
| const auto it = channel_to_token_.find(client_id); |
| if (it != channel_to_token_.end()) { |
| token_to_channel_.erase(it->second); |
| channel_to_token_.erase(it); |
| } |
| } |
| |
| void MediaGpuChannelManager::DestroyAllChannels() { |
| media_gpu_channels_.clear(); |
| token_to_channel_.clear(); |
| channel_to_token_.clear(); |
| } |
| |
| gpu::GpuChannel* MediaGpuChannelManager::LookupChannel( |
| const base::UnguessableToken& channel_token) { |
| const auto it = token_to_channel_.find(channel_token); |
| if (it == token_to_channel_.end()) |
| return nullptr; |
| return channel_manager_->LookupChannel(it->second); |
| } |
| |
| } // namespace media |