| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/copresence/handlers/audio/audio_directive_list.h" | 5 #include "components/copresence/handlers/audio/audio_directive_list.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "base/strings/string_util.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "media/base/audio_bus.h" | 10 #include "base/time/time.h" |
| 11 | |
| 12 namespace { | |
| 13 | |
| 14 // UrlSafe is defined as: | |
| 15 // '/' represented by a '_' and '+' represented by a '-' | |
| 16 // TODO(rkc): Move this processing to the whispernet wrapper. | |
| 17 std::string FromUrlSafe(std::string token) { | |
| 18 base::ReplaceChars(token, "-", "+", &token); | |
| 19 base::ReplaceChars(token, "_", "/", &token); | |
| 20 return token; | |
| 21 } | |
| 22 | |
| 23 const int kSampleExpiryTimeMs = 60 * 60 * 1000; // 60 minutes. | |
| 24 const int kMaxSamples = 10000; | |
| 25 | |
| 26 } // namespace | |
| 27 | 11 |
| 28 namespace copresence { | 12 namespace copresence { |
| 29 | 13 |
| 30 // Public methods. | 14 // Public methods. |
| 31 | 15 |
| 32 AudioDirective::AudioDirective() { | 16 AudioDirective::AudioDirective() { |
| 33 } | 17 } |
| 34 | 18 |
| 35 AudioDirective::AudioDirective(const std::string& token, | 19 AudioDirective::AudioDirective(const std::string& op_id, base::Time end_time) |
| 36 const std::string& op_id, | 20 : op_id(op_id), end_time(end_time) { |
| 37 base::Time end_time) | |
| 38 : token(token), op_id(op_id), end_time(end_time) { | |
| 39 } | 21 } |
| 40 | 22 |
| 41 AudioDirective::AudioDirective( | 23 AudioDirectiveList::AudioDirectiveList() { |
| 42 const std::string& token, | |
| 43 const std::string& op_id, | |
| 44 base::Time end_time, | |
| 45 const scoped_refptr<media::AudioBusRefCounted>& samples) | |
| 46 : token(token), op_id(op_id), end_time(end_time), samples(samples) { | |
| 47 } | |
| 48 | |
| 49 AudioDirective::~AudioDirective() { | |
| 50 } | |
| 51 | |
| 52 AudioDirectiveList::AudioDirectiveList( | |
| 53 const EncodeTokenCallback& encode_token_callback, | |
| 54 const base::Closure& token_added_callback, | |
| 55 bool use_audible_encoding) | |
| 56 : encode_token_callback_(encode_token_callback), | |
| 57 token_added_callback_(token_added_callback), | |
| 58 use_audible_encoding_(use_audible_encoding), | |
| 59 samples_cache_(base::TimeDelta::FromMilliseconds(kSampleExpiryTimeMs), | |
| 60 kMaxSamples) { | |
| 61 } | 24 } |
| 62 | 25 |
| 63 AudioDirectiveList::~AudioDirectiveList() { | 26 AudioDirectiveList::~AudioDirectiveList() { |
| 64 } | 27 } |
| 65 | 28 |
| 66 void AudioDirectiveList::AddTransmitDirective(const std::string& token, | 29 void AudioDirectiveList::AddDirective(const std::string& op_id, |
| 67 const std::string& op_id, | 30 base::TimeDelta ttl) { |
| 68 base::TimeDelta ttl) { | |
| 69 std::string valid_token = FromUrlSafe(token); | |
| 70 base::Time end_time = base::Time::Now() + ttl; | 31 base::Time end_time = base::Time::Now() + ttl; |
| 71 | 32 |
| 72 if (samples_cache_.HasKey(valid_token)) { | 33 // In case this op is already in the list, update it instead of adding |
| 73 active_transmit_tokens_.push(AudioDirective( | 34 // it again. |
| 74 valid_token, op_id, end_time, samples_cache_.GetValue(valid_token))); | 35 std::vector<AudioDirective>::iterator it = FindDirectiveByOpId(op_id); |
| 36 if (it != active_directives_.end()) { |
| 37 it->end_time = end_time; |
| 38 std::make_heap(active_directives_.begin(), |
| 39 active_directives_.end(), |
| 40 LatestFirstComparator()); |
| 75 return; | 41 return; |
| 76 } | 42 } |
| 77 | 43 |
| 78 // If an encode request for this token has been sent, don't send it again. | 44 active_directives_.push_back(AudioDirective(op_id, end_time)); |
| 79 if (pending_transmit_tokens_.find(valid_token) != | 45 std::push_heap(active_directives_.begin(), |
| 80 pending_transmit_tokens_.end()) { | 46 active_directives_.end(), |
| 81 return; | 47 LatestFirstComparator()); |
| 48 } |
| 49 |
| 50 void AudioDirectiveList::RemoveDirective(const std::string& op_id) { |
| 51 std::vector<AudioDirective>::iterator it = FindDirectiveByOpId(op_id); |
| 52 if (it != active_directives_.end()) |
| 53 active_directives_.erase(it); |
| 54 |
| 55 std::make_heap(active_directives_.begin(), |
| 56 active_directives_.end(), |
| 57 LatestFirstComparator()); |
| 58 } |
| 59 |
| 60 scoped_ptr<AudioDirective> AudioDirectiveList::GetActiveDirective() { |
| 61 // The top is always the instruction that is ending the latest. If that time |
| 62 // has passed, means all our previous instructions have expired too, hence |
| 63 // clear the list. |
| 64 if (!active_directives_.empty() && |
| 65 active_directives_.front().end_time < base::Time::Now()) { |
| 66 active_directives_.clear(); |
| 82 } | 67 } |
| 83 | 68 |
| 84 pending_transmit_tokens_[valid_token] = | 69 if (active_directives_.empty()) |
| 85 AudioDirective(valid_token, op_id, end_time); | 70 return make_scoped_ptr<AudioDirective>(NULL); |
| 86 // All whispernet callbacks will be cleared before we are destructed, so | 71 |
| 87 // unretained is safe to use here. | 72 return make_scoped_ptr(new AudioDirective(active_directives_.front())); |
| 88 encode_token_callback_.Run( | |
| 89 valid_token, | |
| 90 use_audible_encoding_, | |
| 91 base::Bind(&AudioDirectiveList::OnTokenEncoded, base::Unretained(this))); | |
| 92 } | 73 } |
| 93 | 74 |
| 94 void AudioDirectiveList::AddReceiveDirective(const std::string& op_id, | 75 std::vector<AudioDirective>::iterator AudioDirectiveList::FindDirectiveByOpId( |
| 95 base::TimeDelta ttl) { | 76 const std::string& op_id) { |
| 96 active_receive_tokens_.push( | 77 for (std::vector<AudioDirective>::iterator it = active_directives_.begin(); |
| 97 AudioDirective(std::string(), op_id, base::Time::Now() + ttl)); | 78 it != active_directives_.end(); |
| 98 } | 79 ++it) { |
| 99 | 80 if (it->op_id == op_id) |
| 100 scoped_ptr<AudioDirective> AudioDirectiveList::GetNextTransmit() { | 81 return it; |
| 101 return GetNextFromList(&active_transmit_tokens_); | |
| 102 } | |
| 103 | |
| 104 scoped_ptr<AudioDirective> AudioDirectiveList::GetNextReceive() { | |
| 105 return GetNextFromList(&active_receive_tokens_); | |
| 106 } | |
| 107 | |
| 108 scoped_ptr<AudioDirective> AudioDirectiveList::GetNextFromList( | |
| 109 AudioDirectiveQueue* list) { | |
| 110 CHECK(list); | |
| 111 | |
| 112 // Checks if we have any valid tokens at all (since the top of the list is | |
| 113 // always pointing to the token with the latest expiry time). If we don't | |
| 114 // have any valid tokens left, clear the list. | |
| 115 if (!list->empty() && list->top().end_time < base::Time::Now()) { | |
| 116 while (!list->empty()) | |
| 117 list->pop(); | |
| 118 } | 82 } |
| 119 | 83 return active_directives_.end(); |
| 120 if (list->empty()) | |
| 121 return make_scoped_ptr<AudioDirective>(NULL); | |
| 122 | |
| 123 return make_scoped_ptr(new AudioDirective(list->top())); | |
| 124 } | |
| 125 | |
| 126 void AudioDirectiveList::OnTokenEncoded( | |
| 127 const std::string& token, | |
| 128 bool /* audible */, | |
| 129 const scoped_refptr<media::AudioBusRefCounted>& samples) { | |
| 130 // We shouldn't re-encode a token if it's already in the cache. | |
| 131 DCHECK(!samples_cache_.HasKey(token)); | |
| 132 DVLOG(3) << "Token: " << token << " encoded."; | |
| 133 samples_cache_.Add(token, samples); | |
| 134 | |
| 135 // Copy the samples into their corresponding directive object and move | |
| 136 // that object into the active queue. | |
| 137 std::map<std::string, AudioDirective>::iterator it = | |
| 138 pending_transmit_tokens_.find(token); | |
| 139 | |
| 140 it->second.samples = samples; | |
| 141 active_transmit_tokens_.push(it->second); | |
| 142 pending_transmit_tokens_.erase(it); | |
| 143 | |
| 144 if (!token_added_callback_.is_null()) | |
| 145 token_added_callback_.Run(); | |
| 146 } | 84 } |
| 147 | 85 |
| 148 } // namespace copresence | 86 } // namespace copresence |
| OLD | NEW |