Add opt::ir::Module::SetIdBound

diff --git a/source/opt/module.h b/source/opt/module.h
index 1d6c98e..732a6ce 100644 (file)
--- a/source/opt/module.h
+++ b/source/opt/module.h
@@ -62,6 +62,8 @@ class Module {
 
   // Sets the header to the given |header|.
   void SetHeader(const ModuleHeader& header) { header_ = header; }
+  // Sets the Id bound.
+  void SetIdBound(uint32_t bound) { header_.bound = bound; }
   // Appends a capability instruction to this module.
   inline void AddCapability(std::unique_ptr<Instruction> c);
   // Appends an extension instruction to this module.

diff --git a/test/opt/CMakeLists.txt b/test/opt/CMakeLists.txt
index fc3d102..c032215 100644 (file)
--- a/test/opt/CMakeLists.txt
+++ b/test/opt/CMakeLists.txt
@@ -78,3 +78,8 @@ add_spvtools_unittest(TARGET iterator
   SRCS test_iterator.cpp
   LIBS SPIRV-Tools-opt ${SPIRV_TOOLS}
 )
+
+add_spvtools_unittest(TARGET module
+  SRCS test_module.cpp
+  LIBS SPIRV-Tools-opt ${SPIRV_TOOLS}
+)

diff --git a/test/opt/test_module.cpp b/test/opt/test_module.cpp
new file mode 100644 (file)
index 0000000..700e0f5
--- /dev/null
+++ b/test/opt/test_module.cpp
@@ -0,0 +1,58 @@
+// Copyright (c) 2016 Google Inc.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a
+// copy of this software and/or associated documentation files (the
+// "Materials"), to deal in the Materials without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sublicense, and/or sell copies of the Materials, and to
+// permit persons to whom the Materials are furnished to do so, subject to
+// the following conditions:
+//
+// The above copyright notice and this permission notice shall be included
+// in all copies or substantial portions of the Materials.
+//
+// MODIFICATIONS TO THIS FILE MAY MEAN IT NO LONGER ACCURATELY REFLECTS
+// KHRONOS STANDARDS. THE UNMODIFIED, NORMATIVE VERSIONS OF KHRONOS
+// SPECIFICATIONS AND HEADER INFORMATION ARE LOCATED AT
+//    https://www.khronos.org/registry/
+//
+// THE MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+// IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+// TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+// MATERIALS OR THE USE OR OTHER DEALINGS IN THE MATERIALS.
+
+#include <vector>
+
+#include "opt/module.h"
+
+#include "gtest/gtest.h"
+
+namespace {
+
+using spvtools::ir::Module;
+
+uint32_t GetIdBound(const Module& m) {
+  std::vector<uint32_t> binary;
+  m.ToBinary(&binary, false);
+  // The 5-word header must always exist.
+  EXPECT_GE(5u, binary.size());
+  // The bound is the fourth word.
+  return binary[3];
+}
+
+TEST(ModuleTest, SetIdBound) {
+  Module m;
+  // It's initialized to 0.
+  EXPECT_EQ(0u, GetIdBound(m));
+
+  m.SetIdBound(19);
+  EXPECT_EQ(19u, GetIdBound(m));
+
+  m.SetIdBound(102);
+  EXPECT_EQ(102u, GetIdBound(m));
+}
+
+}  // anonymous namespace