Unmute Tab Audio For Desktop Share

content/browser/media/capture/audio_mirroring_manager.cc

 // Copyright (c) 2013 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 "content/browser/media/capture/audio_mirroring_manager.h"
 
 #include <algorithm>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 63 matching lines @@
   DCHECK(thread_checker_.CalledOnValidThread());
   DCHECK(destination);
 
   // Insert an entry into the set of active mirroring sessions, if this is a
   // previously-unknown destination.
   if (std::find(sessions_.begin(), sessions_.end(), destination) ==
           sessions_.end()) {
     sessions_.push_back(destination);
   }
 
+  std::set<SourceFrameRef> candidates;
+  if (destination->IsDuplication()) {

[miu, 2016/04/21 00:15:24]

See comment I made in the .h file about the IsDuplication() method.  Here, is a
good example of where that led you to copy-and-paste code.

[qiangchen, 2016/04/28 00:00:56]

Done.

+    for (StreamRoutes::const_iterator it = routes_.begin(); it != routes_.end();
+         ++it) {
+      candidates.insert(it->source_render_frame);
+    }
+    if (!candidates.empty()) {
+      destination->QueryForMatches(
+          candidates,
+          base::Bind(&AudioMirroringManager::UpdateRoutesToDupDestination,
+                     base::Unretained(this), destination, false));
+    }
+    return;
+  }
+
   // Query the MirroringDestination to see which of the audio streams should be
   // diverted.
-  std::set<SourceFrameRef> candidates;
   for (StreamRoutes::const_iterator it = routes_.begin(); it != routes_.end();
        ++it) {
     if (!it->destination || it->destination == destination)
       candidates.insert(it->source_render_frame);
   }
   if (!candidates.empty()) {
     destination->QueryForMatches(
         candidates,
         base::Bind(&AudioMirroringManager::UpdateRoutesToDestination,
                    base::Unretained(this),
                    destination,
                    false));
   }
 }
 
 void AudioMirroringManager::StopMirroring(MirroringDestination* destination) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
-  // Stop diverting each audio stream in the mirroring session being stopped.
-  // Each stopped stream becomes a candidate to be diverted to another
-  // destination.
-  std::set<SourceFrameRef> redivert_candidates;
-  for (StreamRoutes::iterator it = routes_.begin(); it != routes_.end(); ++it) {
-    if (it->destination == destination) {
-      ChangeRoute(&(*it), NULL);
-      redivert_candidates.insert(it->source_render_frame);
+  if (destination->IsDuplication()) {
+    std::set<SourceFrameRef> matches;
+    UpdateRoutesToDupDestination(destination, false, matches);
+  } else {
+    // Stop diverting each audio stream in the mirroring session being stopped.
+    // Each stopped stream becomes a candidate to be diverted to another
+    // destination.
+    std::set<SourceFrameRef> redivert_candidates;
+    for (StreamRoutes::iterator it = routes_.begin(); it != routes_.end();
+         ++it) {
+      if (it->destination == destination) {
+        ChangeRoute(&(*it), NULL);
+        redivert_candidates.insert(it->source_render_frame);
+      }
     }
+    if (!redivert_candidates.empty())
+      InitiateQueriesToFindNewDestination(destination, redivert_candidates);
   }
-  if (!redivert_candidates.empty())
-    InitiateQueriesToFindNewDestination(destination, redivert_candidates);
 
   // Remove the entry from the set of active mirroring sessions.
   const Destinations::iterator dest_it =
       std::find(sessions_.begin(), sessions_.end(), destination);
   if (dest_it == sessions_.end()) {
     NOTREACHED();
     return;
   }
   sessions_.erase(dest_it);
 }
 
 void AudioMirroringManager::InitiateQueriesToFindNewDestination(
     MirroringDestination* old_destination,
     const std::set<SourceFrameRef>& candidates) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
   for (Destinations::const_iterator it = sessions_.begin();
        it != sessions_.end(); ++it) {
-    if (*it != old_destination) {
+    if (*it == old_destination)
+      continue;
+
+    if ((*it)->IsDuplication()) {
+      (*it)->QueryForMatches(
+          candidates,
+          base::Bind(&AudioMirroringManager::UpdateRoutesToDupDestination,
+                     base::Unretained(this), *it, true));
+    } else {
       (*it)->QueryForMatches(
           candidates,
           base::Bind(&AudioMirroringManager::UpdateRoutesToDestination,
                      base::Unretained(this),
                      *it,
                      true));
     }
   }
 }
 
@@ skipping 24 matching lines @@
       if (it->destination == destination) {
         ChangeRoute(&(*it), NULL);
         redivert_candidates.insert(it->source_render_frame);
       }
     }
   }
   if (!redivert_candidates.empty())
     InitiateQueriesToFindNewDestination(destination, redivert_candidates);
 }
 
+void AudioMirroringManager::UpdateRoutesToDupDestination(
+    MirroringDestination* destination,
+    bool add_only,
+    const std::set<SourceFrameRef>& matches) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+  if (std::find(sessions_.begin(), sessions_.end(), destination) ==
+      sessions_.end()) {
+    return;  // Query result callback invoked after StopMirroring().
+  }
+
+  for (StreamRoutes::iterator it = routes_.begin(); it != routes_.end(); ++it) {
+    if (matches.find(it->source_render_frame) != matches.end()) {
+      if (it->dup_destinations.count(destination) == 0) {
+        it->dup_destinations[destination] =
+            destination->AddPushInput(it->diverter->GetAudioParameters());
+        it->diverter->StartDuplicating(it->dup_destinations[destination]);
+      }
+
+    } else if (!add_only) {
+      if (it->dup_destinations.count(destination) > 0) {
+        it->diverter->StopDuplicating(it->dup_destinations[destination]);
+        it->dup_destinations.erase(destination);
+      }
+    }
+  }
+}
+
 // static
 void AudioMirroringManager::ChangeRoute(
     StreamRoutingState* route, MirroringDestination* new_destination) {
   if (route->destination == new_destination)
     return;  // No change.
 
   if (route->destination) {
     DVLOG(1) << "Stop diverting render_process_id:render_frame_id="
              << route->source_render_frame.first << ':'
              << route->source_render_frame.second
@@ skipping 18 matching lines @@
   : source_render_frame(source_frame),
     diverter(stream_diverter),
     destination(NULL) {}
 
 AudioMirroringManager::StreamRoutingState::StreamRoutingState(
     const StreamRoutingState& other) = default;
 
 AudioMirroringManager::StreamRoutingState::~StreamRoutingState() {}
 
 }  // namespace content