hlsl: Remove the sanitizer transform

Invoke the required transforms directly in the HLSL backend.

Change-Id: I9465fef375dd4dad6a91c1e7e16ede6401b9bfc0
Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/63800
Kokoro: Kokoro <noreply+kokoro@google.com>
Reviewed-by: Ben Clayton <bclayton@google.com>
diff --git a/src/BUILD.gn b/src/BUILD.gn
index 9ad333d..ccbbfd6 100644
--- a/src/BUILD.gn
+++ b/src/BUILD.gn
@@ -681,8 +681,6 @@
 
 libtint_source_set("libtint_hlsl_writer_src") {
   sources = [
-    "transform/hlsl.cc",
-    "transform/hlsl.h",
     "writer/hlsl/generator.cc",
     "writer/hlsl/generator.h",
     "writer/hlsl/generator_impl.cc",
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 8dd1896..51a8b91 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -486,8 +486,6 @@
 
 if(${TINT_BUILD_HLSL_WRITER})
   list(APPEND TINT_LIB_SRCS
-    transform/hlsl.cc
-    transform/hlsl.h
     writer/hlsl/generator.cc
     writer/hlsl/generator.h
     writer/hlsl/generator_impl.cc
@@ -986,9 +984,6 @@
   endif()
 
   if (${TINT_BUILD_HLSL_WRITER})
-    if(${TINT_BUILD_WGSL_READER} AND ${TINT_BUILD_WGSL_WRITER})
-      list(APPEND TINT_TEST_SRCS transform/hlsl_test.cc)
-    endif()
     list(APPEND TINT_TEST_SRCS
       writer/hlsl/generator_impl_array_accessor_test.cc
       writer/hlsl/generator_impl_assign_test.cc
diff --git a/src/transform/hlsl.cc b/src/transform/hlsl.cc
deleted file mode 100644
index 0fe35b5..0000000
--- a/src/transform/hlsl.cc
+++ /dev/null
@@ -1,97 +0,0 @@
-// Copyright 2021 The Tint Authors.
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-#include "src/transform/hlsl.h"
-
-#include <utility>
-
-#include "src/program_builder.h"
-#include "src/transform/add_empty_entry_point.h"
-#include "src/transform/calculate_array_length.h"
-#include "src/transform/canonicalize_entry_point_io.h"
-#include "src/transform/decompose_memory_access.h"
-#include "src/transform/external_texture_transform.h"
-#include "src/transform/fold_trivial_single_use_lets.h"
-#include "src/transform/inline_pointer_lets.h"
-#include "src/transform/loop_to_for_loop.h"
-#include "src/transform/manager.h"
-#include "src/transform/pad_array_elements.h"
-#include "src/transform/promote_initializers_to_const_var.h"
-#include "src/transform/simplify.h"
-#include "src/transform/zero_init_workgroup_memory.h"
-
-TINT_INSTANTIATE_TYPEINFO(tint::transform::Hlsl);
-TINT_INSTANTIATE_TYPEINFO(tint::transform::Hlsl::Config);
-
-namespace tint {
-namespace transform {
-
-Hlsl::Hlsl() = default;
-Hlsl::~Hlsl() = default;
-
-Output Hlsl::Run(const Program* in, const DataMap& inputs) {
-  Manager manager;
-  DataMap data;
-
-  auto* cfg = inputs.Get<Config>();
-
-  // Attempt to convert `loop`s into for-loops. This is to try and massage the
-  // output into something that will not cause FXC to choke or misbehave.
-  manager.Add<FoldTrivialSingleUseLets>();
-  manager.Add<LoopToForLoop>();
-
-  if (!cfg || !cfg->disable_workgroup_init) {
-    // ZeroInitWorkgroupMemory must come before CanonicalizeEntryPointIO as
-    // ZeroInitWorkgroupMemory may inject new builtin parameters.
-    manager.Add<ZeroInitWorkgroupMemory>();
-  }
-  manager.Add<CanonicalizeEntryPointIO>();
-  manager.Add<InlinePointerLets>();
-  // Simplify cleans up messy `*(&(expr))` expressions from InlinePointerLets.
-  manager.Add<Simplify>();
-  // DecomposeMemoryAccess must come after InlinePointerLets as we cannot take
-  // the address of calls to DecomposeMemoryAccess::Intrinsic. Must also come
-  // after Simplify, as we need to fold away the address-of and defers of
-  // `*(&(intrinsic_load()))` expressions.
-  manager.Add<DecomposeMemoryAccess>();
-  // CalculateArrayLength must come after DecomposeMemoryAccess, as
-  // DecomposeMemoryAccess special-cases the arrayLength() intrinsic, which
-  // will be transformed by CalculateArrayLength
-  manager.Add<CalculateArrayLength>();
-  manager.Add<ExternalTextureTransform>();
-  manager.Add<PromoteInitializersToConstVar>();
-  manager.Add<PadArrayElements>();
-  manager.Add<AddEmptyEntryPoint>();
-
-  data.Add<CanonicalizeEntryPointIO::Config>(
-      CanonicalizeEntryPointIO::ShaderStyle::kHlsl);
-  auto out = manager.Run(in, data);
-  if (!out.program.IsValid()) {
-    return out;
-  }
-
-  ProgramBuilder builder;
-  CloneContext ctx(&builder, &out.program);
-  // TODO(jrprice): Move the sanitizer into the backend.
-  ctx.Clone();
-  builder.SetTransformApplied(this);
-  return Output{Program(std::move(builder))};
-}
-
-Hlsl::Config::Config(bool disable_wi) : disable_workgroup_init(disable_wi) {}
-Hlsl::Config::Config(const Config&) = default;
-Hlsl::Config::~Config() = default;
-
-}  // namespace transform
-}  // namespace tint
diff --git a/src/transform/hlsl.h b/src/transform/hlsl.h
deleted file mode 100644
index ea8a27e..0000000
--- a/src/transform/hlsl.h
+++ /dev/null
@@ -1,63 +0,0 @@
-// Copyright 2021 The Tint Authors.
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-#ifndef SRC_TRANSFORM_HLSL_H_
-#define SRC_TRANSFORM_HLSL_H_
-
-#include "src/transform/transform.h"
-
-namespace tint {
-
-// Forward declarations
-class CloneContext;
-
-namespace transform {
-
-/// Hlsl is a transform used to sanitize a Program for use with the Hlsl writer.
-/// Passing a non-sanitized Program to the Hlsl writer will result in undefined
-/// behavior.
-class Hlsl : public Castable<Hlsl, Transform> {
- public:
-  /// Configuration options for the Hlsl sanitizer transform.
-  struct Config : public Castable<Data, transform::Data> {
-    /// Constructor
-    /// @param disable_workgroup_init `true` to disable workgroup memory zero
-    ///        initialization
-    explicit Config(bool disable_workgroup_init = false);
-
-    /// Copy constructor
-    Config(const Config&);
-
-    /// Destructor
-    ~Config() override;
-
-    /// Set to `true` to disable workgroup memory zero initialization
-    bool disable_workgroup_init = false;
-  };
-
-  /// Constructor
-  Hlsl();
-  ~Hlsl() override;
-
-  /// Runs the transform on `program`, returning the transformation result.
-  /// @param program the source program to transform
-  /// @param data optional extra transform-specific data
-  /// @returns the transformation result
-  Output Run(const Program* program, const DataMap& data = {}) override;
-};
-
-}  // namespace transform
-}  // namespace tint
-
-#endif  // SRC_TRANSFORM_HLSL_H_
diff --git a/src/transform/hlsl_test.cc b/src/transform/hlsl_test.cc
deleted file mode 100644
index 4389d7e..0000000
--- a/src/transform/hlsl_test.cc
+++ /dev/null
@@ -1,29 +0,0 @@
-// Copyright 2021 The Tint Authors.
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     http://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-#include "src/transform/hlsl.h"
-
-#include "src/transform/test_helper.h"
-
-namespace tint {
-namespace transform {
-namespace {
-
-using HlslTest = TransformTest;
-
-// TODO(jrprice): Remove this file when we remove the sanitizer transforms.
-
-}  // namespace
-}  // namespace transform
-}  // namespace tint
diff --git a/src/writer/hlsl/generator.cc b/src/writer/hlsl/generator.cc
index 259f0b6..d2ee601 100644
--- a/src/writer/hlsl/generator.cc
+++ b/src/writer/hlsl/generator.cc
@@ -14,7 +14,6 @@
 
 #include "src/writer/hlsl/generator.h"
 
-#include "src/transform/hlsl.h"
 #include "src/writer/hlsl/generator_impl.h"
 
 namespace tint {
@@ -28,27 +27,24 @@
 Result Generate(const Program* program, const Options& options) {
   Result result;
 
-  // Run the HLSL sanitizer.
-  transform::Hlsl sanitizer;
-  transform::DataMap transform_input;
-  transform_input.Add<transform::Hlsl::Config>(options.disable_workgroup_init);
-  auto output = sanitizer.Run(program, transform_input);
-  if (!output.program.IsValid()) {
+  // Sanitize the program.
+  auto sanitized_result = Sanitize(program, options.disable_workgroup_init);
+  if (!sanitized_result.program.IsValid()) {
     result.success = false;
-    result.error = output.program.Diagnostics().str();
+    result.error = sanitized_result.program.Diagnostics().str();
     return result;
   }
 
   // Generate the HLSL code.
-  auto impl = std::make_unique<GeneratorImpl>(&output.program);
+  auto impl = std::make_unique<GeneratorImpl>(&sanitized_result.program);
   result.success = impl->Generate();
   result.error = impl->error();
   result.hlsl = impl->result();
 
   // Collect the list of entry points in the sanitized program.
-  for (auto* func : output.program.AST().Functions()) {
+  for (auto* func : sanitized_result.program.AST().Functions()) {
     if (func->IsEntryPoint()) {
-      auto name = output.program.Symbols().NameFor(func->symbol());
+      auto name = sanitized_result.program.Symbols().NameFor(func->symbol());
       result.entry_points.push_back({name, func->pipeline_stage()});
     }
   }
diff --git a/src/writer/hlsl/generator_impl.cc b/src/writer/hlsl/generator_impl.cc
index d8e6dfd..c70e331 100644
--- a/src/writer/hlsl/generator_impl.cc
+++ b/src/writer/hlsl/generator_impl.cc
@@ -42,8 +42,19 @@
 #include "src/sem/storage_texture_type.h"
 #include "src/sem/struct.h"
 #include "src/sem/variable.h"
+#include "src/transform/add_empty_entry_point.h"
 #include "src/transform/calculate_array_length.h"
-#include "src/transform/hlsl.h"
+#include "src/transform/canonicalize_entry_point_io.h"
+#include "src/transform/decompose_memory_access.h"
+#include "src/transform/external_texture_transform.h"
+#include "src/transform/fold_trivial_single_use_lets.h"
+#include "src/transform/inline_pointer_lets.h"
+#include "src/transform/loop_to_for_loop.h"
+#include "src/transform/manager.h"
+#include "src/transform/pad_array_elements.h"
+#include "src/transform/promote_initializers_to_const_var.h"
+#include "src/transform/simplify.h"
+#include "src/transform/zero_init_workgroup_memory.h"
 #include "src/utils/defer.h"
 #include "src/utils/get_or_create.h"
 #include "src/utils/scoped_assignment.h"
@@ -102,19 +113,51 @@
 
 }  // namespace
 
+SanitizedResult Sanitize(const Program* in, bool disable_workgroup_init) {
+  transform::Manager manager;
+  transform::DataMap data;
+
+  // Attempt to convert `loop`s into for-loops. This is to try and massage the
+  // output into something that will not cause FXC to choke or misbehave.
+  manager.Add<transform::FoldTrivialSingleUseLets>();
+  manager.Add<transform::LoopToForLoop>();
+
+  if (!disable_workgroup_init) {
+    // ZeroInitWorkgroupMemory must come before CanonicalizeEntryPointIO as
+    // ZeroInitWorkgroupMemory may inject new builtin parameters.
+    manager.Add<transform::ZeroInitWorkgroupMemory>();
+  }
+  manager.Add<transform::CanonicalizeEntryPointIO>();
+  manager.Add<transform::InlinePointerLets>();
+  // Simplify cleans up messy `*(&(expr))` expressions from InlinePointerLets.
+  manager.Add<transform::Simplify>();
+  // DecomposeMemoryAccess must come after InlinePointerLets as we cannot take
+  // the address of calls to DecomposeMemoryAccess::Intrinsic. Must also come
+  // after Simplify, as we need to fold away the address-of and defers of
+  // `*(&(intrinsic_load()))` expressions.
+  manager.Add<transform::DecomposeMemoryAccess>();
+  // CalculateArrayLength must come after DecomposeMemoryAccess, as
+  // DecomposeMemoryAccess special-cases the arrayLength() intrinsic, which
+  // will be transformed by CalculateArrayLength
+  manager.Add<transform::CalculateArrayLength>();
+  manager.Add<transform::ExternalTextureTransform>();
+  manager.Add<transform::PromoteInitializersToConstVar>();
+  manager.Add<transform::PadArrayElements>();
+  manager.Add<transform::AddEmptyEntryPoint>();
+
+  data.Add<transform::CanonicalizeEntryPointIO::Config>(
+      transform::CanonicalizeEntryPointIO::ShaderStyle::kHlsl);
+
+  SanitizedResult result;
+  result.program = std::move(manager.Run(in, data).program);
+  return result;
+}
+
 GeneratorImpl::GeneratorImpl(const Program* program) : TextGenerator(program) {}
 
 GeneratorImpl::~GeneratorImpl() = default;
 
 bool GeneratorImpl::Generate() {
-  if (!builder_.HasTransformApplied<transform::Hlsl>()) {
-    diagnostics_.add_error(
-        diag::System::Writer,
-        "HLSL writer requires the transform::Hlsl sanitizer to have been "
-        "applied to the input program");
-    return false;
-  }
-
   const TypeInfo* last_kind = nullptr;
   size_t last_padding_line = 0;
 
diff --git a/src/writer/hlsl/generator_impl.h b/src/writer/hlsl/generator_impl.h
index 7ec64ff..17a6674 100644
--- a/src/writer/hlsl/generator_impl.h
+++ b/src/writer/hlsl/generator_impl.h
@@ -48,6 +48,18 @@
 namespace writer {
 namespace hlsl {
 
+/// The result of sanitizing a program for generation.
+struct SanitizedResult {
+  /// The sanitized program.
+  Program program;
+};
+
+/// Sanitize a program in preparation for generating HLSL.
+/// @param disable_workgroup_init `true` to disable workgroup memory zero
+/// @returns the sanitized program and any supplementary information
+SanitizedResult Sanitize(const Program* program,
+                         bool disable_workgroup_init = false);
+
 /// Implementation class for HLSL generator
 class GeneratorImpl : public TextGenerator {
  public:
diff --git a/src/writer/hlsl/generator_impl_test.cc b/src/writer/hlsl/generator_impl_test.cc
index cd01730..d4227f3 100644
--- a/src/writer/hlsl/generator_impl_test.cc
+++ b/src/writer/hlsl/generator_impl_test.cc
@@ -21,16 +21,6 @@
 
 using HlslGeneratorImplTest = TestHelper;
 
-TEST_F(HlslGeneratorImplTest, ErrorIfSanitizerNotRun) {
-  auto program = std::make_unique<Program>(std::move(*this));
-  GeneratorImpl gen(program.get());
-  EXPECT_FALSE(gen.Generate());
-  EXPECT_EQ(
-      gen.error(),
-      "error: HLSL writer requires the transform::Hlsl sanitizer to have been "
-      "applied to the input program");
-}
-
 TEST_F(HlslGeneratorImplTest, Generate) {
   Func("my_func", ast::VariableList{}, ty.void_(), ast::StatementList{},
        ast::DecorationList{});
diff --git a/src/writer/hlsl/test_helper.h b/src/writer/hlsl/test_helper.h
index 7daca49..7523c82 100644
--- a/src/writer/hlsl/test_helper.h
+++ b/src/writer/hlsl/test_helper.h
@@ -20,7 +20,6 @@
 #include <utility>
 
 #include "gtest/gtest.h"
-#include "src/transform/hlsl.h"
 #include "src/transform/manager.h"
 #include "src/transform/renamer.h"
 #include "src/writer/hlsl/generator_impl.h"
@@ -44,9 +43,6 @@
     if (gen_) {
       return *gen_;
     }
-    // Fake that the HLSL sanitizer has been applied, so that we can unit test
-    // the writer without it erroring.
-    SetTransformApplied<transform::Hlsl>();
     [&]() {
       ASSERT_TRUE(IsValid()) << "Builder program is not valid\n"
                              << diag::Formatter().format(Diagnostics());
@@ -60,7 +56,7 @@
     return *gen_;
   }
 
-  /// Builds the program, runs the program through the transform::Hlsl sanitizer
+  /// Builds the program, runs the program through the HLSL sanitizer
   /// and returns a GeneratorImpl from the sanitized program.
   /// @note The generator is only built once. Multiple calls to Build() will
   /// return the same GeneratorImpl without rebuilding.
@@ -80,13 +76,19 @@
           << formatter.format(program->Diagnostics());
     }();
 
+    auto sanitized_result = Sanitize(program.get());
+    [&]() {
+      ASSERT_TRUE(sanitized_result.program.IsValid())
+          << formatter.format(sanitized_result.program.Diagnostics());
+    }();
+
     transform::Manager transform_manager;
     transform::DataMap transform_data;
     transform_data.Add<transform::Renamer::Config>(
         transform::Renamer::Target::kHlslKeywords);
     transform_manager.Add<tint::transform::Renamer>();
-    transform_manager.Add<tint::transform::Hlsl>();
-    auto result = transform_manager.Run(program.get(), transform_data);
+    auto result =
+        transform_manager.Run(&sanitized_result.program, transform_data);
     [&]() {
       ASSERT_TRUE(result.program.IsValid())
           << formatter.format(result.program.Diagnostics());
diff --git a/test/BUILD.gn b/test/BUILD.gn
index 8d3ee1f..203b853 100644
--- a/test/BUILD.gn
+++ b/test/BUILD.gn
@@ -586,7 +586,6 @@
 
 tint_unittests_source_set("tint_unittests_hlsl_writer_src") {
   sources = [
-    "../src/transform/hlsl_test.cc",
     "../src/writer/hlsl/generator_impl_array_accessor_test.cc",
     "../src/writer/hlsl/generator_impl_assign_test.cc",
     "../src/writer/hlsl/generator_impl_binary_test.cc",