[WebRTC] A real ScreenCapturer test

webrtc/modules/desktop_capture/color.cc

Index: webrtc/modules/desktop_capture/color.cc
diff --git a/webrtc/modules/desktop_capture/color.cc b/webrtc/modules/desktop_capture/color.cc
new file mode 100644
index 0000000000000000000000000000000000000000..c74a2b59f5bd47e6343328f603af994c9844f3ad
--- /dev/null
+++ b/webrtc/modules/desktop_capture/color.cc
@@ -0,0 +1,82 @@
+/*
+ *  Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "webrtc/modules/desktop_capture/color.h"
+
+#include <string.h>
+
+namespace webrtc {
+
+namespace {
+
+bool AlphaEquals(uint8_t i, uint8_t j) {
+  // On Linux and Windows 8 or early version, '0' was returned for alpha channel
+  // from capturer APIs, on Windows 10, '255' was returned. So a workaround is
+  // to treat 0 as 255.
+  return i == j || ((i == 0 || i == 255) && (j == 0 || j == 255));
+}
+
+}  // namespace
+
+Color::Color(uint8_t blue, uint8_t green, uint8_t red, uint8_t alpha) {
+  bgra[0] = blue;
+  bgra[1] = green;
+  bgra[2] = red;
+  bgra[3] = alpha;
+}
+
+Color::Color(uint8_t blue, uint8_t green, uint8_t red)
+    : Color(blue, green, red, 0xff) {}
+
+bool Color::operator==(const uint8_t* const bgra) const {
+  for (int i = 0; i < DesktopFrame::kBytesPerPixel - 1; i++) {
+    if (this->bgra[i] != bgra[i]) {
+      return false;
+    }
+  }
+  return AlphaEquals(this->bgra[DesktopFrame::kBytesPerPixel - 1],
+                     bgra[DesktopFrame::kBytesPerPixel - 1]);
+}
+
+bool Color::operator!=(const uint8_t* const bgra) const {
+  return !(*this == bgra);
+}
+
+bool Color::operator==(const Color& right) const {
+  return *this == right.bgra;
+}
+
+bool Color::operator!=(const Color& right) const {
+  return !(*this == right);
+}
+
+uint32_t Color::ToUInt32() const {
+  uint32_t result = 0;
+  memcpy(&result, bgra, DesktopFrame::kBytesPerPixel);
+  return result;
+}
+
+uint8_t Color::blue() const {
+  return bgra[0];
+}
+
+uint8_t Color::green() const {
+  return bgra[1];
+}
+
+uint8_t Color::red() const {
+  return bgra[2];
+}
+
+uint8_t Color::alpha() const {
+  return bgra[3];
+}
+
+}  // namespace webrtc