[llvm] r235898 - Make an RAII com initializer.

[Zachary Turner]

Author: zturner
Date: Mon Apr 27 12:19:26 2015
New Revision: 235898

URL: http://llvm.org/viewvc/llvm-project?rev=235898&view=rev
Log:
Make an RAII com initializer.

Differential Revision: http://reviews.llvm.org/D9267
Reviewed By: Aaron Ballman, David Majnemer

Added:
    llvm/trunk/include/llvm/Support/COM.h
    llvm/trunk/lib/Support/COM.cpp
    llvm/trunk/lib/Support/Unix/COM.inc
    llvm/trunk/lib/Support/Windows/COM.inc
Modified:
    llvm/trunk/lib/Support/CMakeLists.txt

Added: llvm/trunk/include/llvm/Support/COM.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/Support/COM.h?rev=235898&view=auto
==============================================================================
--- llvm/trunk/include/llvm/Support/COM.h (added)
+++ llvm/trunk/include/llvm/Support/COM.h Mon Apr 27 12:19:26 2015
@@ -0,0 +1,36 @@
+//===- llvm/Support/COM.h ---------------------------------------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+/// \file
+///
+/// Provides a library for accessing COM functionality of the Host OS.
+///
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_SUPPORT_COM_H
+#define LLVM_SUPPORT_COM_H
+
+namespace llvm {
+namespace sys {
+
+enum class COMThreadingMode { SingleThreaded, MultiThreaded };
+
+class InitializeCOMRAII {
+public:
+  explicit InitializeCOMRAII(COMThreadingMode Threading,
+                             bool SpeedOverMemory = false);
+  ~InitializeCOMRAII();
+
+private:
+  InitializeCOMRAII(const InitializeCOMRAII &) = delete;
+  void operator=(const InitializeCOMRAII &) = delete;
+};
+}
+}
+
+#endif

Modified: llvm/trunk/lib/Support/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Support/CMakeLists.txt?rev=235898&r1=235897&r2=235898&view=diff
==============================================================================
--- llvm/trunk/lib/Support/CMakeLists.txt (original)
+++ llvm/trunk/lib/Support/CMakeLists.txt Mon Apr 27 12:19:26 2015
@@ -37,6 +37,7 @@ add_llvm_library(LLVMSupport
   BlockFrequency.cpp
   BranchProbability.cpp
   circular_raw_ostream.cpp
+  COM.cpp
   CommandLine.cpp
   Compression.cpp
   ConvertUTF.c

Added: llvm/trunk/lib/Support/COM.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Support/COM.cpp?rev=235898&view=auto
==============================================================================
--- llvm/trunk/lib/Support/COM.cpp (added)
+++ llvm/trunk/lib/Support/COM.cpp Mon Apr 27 12:19:26 2015
@@ -0,0 +1,23 @@
+//===-- COM.cpp - Implement COM utility classes -----------------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+//  This file implements utility classes related to COM.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Support/COM.h"
+
+#include "llvm/Config/config.h"
+
+// Include the platform-specific parts of this class.
+#ifdef LLVM_ON_UNIX
+#include "Unix/COM.inc"
+#elif LLVM_ON_WIN32
+#include "Windows/COM.inc"
+#endif

Added: llvm/trunk/lib/Support/Unix/COM.inc
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Support/Unix/COM.inc?rev=235898&view=auto
==============================================================================
--- llvm/trunk/lib/Support/Unix/COM.inc (added)
+++ llvm/trunk/lib/Support/Unix/COM.inc Mon Apr 27 12:19:26 2015
@@ -0,0 +1,27 @@
+//===- llvm/Support/Unix/COM.inc - Unix COM Implementation -----*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements the Unix portion of COM support.
+//
+//===----------------------------------------------------------------------===//
+
+//===----------------------------------------------------------------------===//
+//=== WARNING: Implementation here must contain only generic UNIX code that
+//===          is guaranteed to work on *all* UNIX variants.
+//===----------------------------------------------------------------------===//
+
+namespace llvm {
+namespace sys {
+
+COMInitializer::COMInitializer(COMThreadingMode Threading,
+                               bool SpeedOverMemory) {}
+
+COMInitializer::~COMInitializer() {}
+}
+}

Added: llvm/trunk/lib/Support/Windows/COM.inc
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Support/Windows/COM.inc?rev=235898&view=auto
==============================================================================
--- llvm/trunk/lib/Support/Windows/COM.inc (added)
+++ llvm/trunk/lib/Support/Windows/COM.inc Mon Apr 27 12:19:26 2015
@@ -0,0 +1,37 @@
+//===- llvm/Support/Windows/COM.inc - Windows COM Implementation *- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements the Windows portion of COM support.
+//
+//===----------------------------------------------------------------------===//
+
+//===----------------------------------------------------------------------===//
+//=== WARNING: Implementation here must contain only Windows code.
+//===----------------------------------------------------------------------===//
+
+#include <objbase.h>
+
+namespace llvm {
+namespace sys {
+
+InitializeCOMRAII::InitializeCOMRAII(COMThreadingMode Threading,
+                                     bool SpeedOverMemory) {
+  DWORD Coinit = 0;
+  if (Threading == COMThreadingMode::SingleThreaded)
+    Coinit |= COINIT_APARTMENTTHREADED;
+  else
+    Coinit |= COINIT_MULTITHREADED;
+  if (SpeedOverMemory)
+    Coinit |= COINIT_SPEED_OVER_MEMORY;
+  ::CoInitializeEx(nullptr, Coinit);
+}
+
+InitializeCOMRAII::~InitializeCOMRAII() { ::CoUninitialize(); }
+}
+}