Refactor BlockDifference() to use MediaInitializer().

media/video/capture/screen/differ_block.cc

 // Copyright (c) 2012 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/video/capture/screen/differ_block.h"
 
 #include "base/cpu.h"
 #include "build/build_config.h"
 #include "media/video/capture/screen/differ_block_sse2.h"
 
 namespace media {
+typedef int (*BlockDifferenceProc)(const uint8*, const uint8*, int);
+static BlockDifferenceProc g_block_difference_proc_ = NULL;
 
 int BlockDifference_C(const uint8* image1, const uint8* image2, int stride) {
   int width_bytes = kBlockSize * kBytesPerPixel;
 
   for (int y = 0; y < kBlockSize; y++) {
     if (memcmp(image1, image2, width_bytes) != 0)
       return 1;
     image1 += stride;
     image2 += stride;
   }
   return 0;
 }
 
+void InitializeCPUSpecificBlockDifference() {
+  g_block_difference_proc_ = &BlockDifference_C;
+
+#if defined(ARCH_CPU_X86_FAMILY)
+#if defined(__SSE2__)
+  const bool kHasSSE2 = true;
+#else
+  const bool kHasSSE2 = base::CPU().has_sse2();
+#endif
+
+  if (kHasSSE2) {
+    if (kBlockSize == 32) {
+      g_block_difference_proc_ = &BlockDifference_SSE2_W32;
+    } else if (kBlockSize == 16) {
+      g_block_difference_proc_ = &BlockDifference_SSE2_W16;
+    }
+  }
+#endif
+}
+
 int BlockDifference(const uint8* image1, const uint8* image2, int stride) {
-  static int (*diff_proc)(const uint8*, const uint8*, int) = NULL;
-
-  if (!diff_proc) {
-#if defined(ARCH_CPU_ARM_FAMILY) || defined(ARCH_CPU_MIPS_FAMILY)
-    // For ARM and MIPS processors, always use C version.
-    // TODO(hclam): Implement a NEON version.
-    diff_proc = &BlockDifference_C;
-#else
-    base::CPU cpu;
-    // For x86 processors, check if SSE2 is supported.
-    if (cpu.has_sse2() && kBlockSize == 32) {
-      diff_proc = &BlockDifference_SSE2_W32;
-    } else if (cpu.has_sse2() && kBlockSize == 16) {
-      diff_proc = &BlockDifference_SSE2_W16;
-    } else {
-      diff_proc = &BlockDifference_C;
-    }
-#endif
-  }
-
-  return diff_proc(image1, image2, stride);
+  return g_block_difference_proc_(image1, image2, stride);
 }
 
 }  // namespace media