Prepare dynamic LIPO transitions.

Add a dynamic equivalent for LIPO_COLLECTOR transition.

Rename LipoDataTransition (which handles DATA transition) to DisableLipoTransition. A future change will add a corresponding EnableLipoTransition (which will model TARGET_CONFIG_FOR_LIPO).

PiperOrigin-RevId: 153611898

2 files changed, 69 insertions, 26 deletions

diff --git a/src/main/java/com/google/devtools/build/lib/rules/cpp/transitions/LipoDataTransition.java b/src/main/java/com/google/devtools/build/lib/rules/cpp/transitions/DisableLipoTransition.java
index d9b9ae06dc..c6314971c2 100644
--- a/src/main/java/com/google/devtools/build/lib/rules/cpp/transitions/LipoDataTransition.java
+++ b/src/main/java/com/google/devtools/build/lib/rules/cpp/transitions/DisableLipoTransition.java
@@ -14,51 +14,39 @@
 
 package com.google.devtools.build.lib.rules.cpp.transitions;
 
-import com.google.devtools.build.lib.analysis.config.BuildConfiguration;
 import com.google.devtools.build.lib.analysis.config.BuildOptions;
 import com.google.devtools.build.lib.analysis.config.PatchTransition;
 import com.google.devtools.build.lib.rules.cpp.CppOptions;
-import com.google.devtools.build.lib.rules.cpp.FdoSupport;
-import com.google.devtools.build.lib.view.config.crosstool.CrosstoolConfig;
+import com.google.devtools.build.lib.view.config.crosstool.CrosstoolConfig.LipoMode;
 
 /**
- * Dynamic transition that turns off LIPO/FDO settings.
+ * Configuration transition that turns off LIPO/FDO settings.
+ *
+ * <p>Has no effect on non-LIPO-enabled configurations or the LIPO context collector
+ * configuration.
  *
- * <p>This is suitable, for example, when visiting data dependencies of a C++ rule built with LIPO.
  */
-public final class LipoDataTransition implements PatchTransition {
-  public static final LipoDataTransition INSTANCE = new LipoDataTransition();
+public final class DisableLipoTransition implements PatchTransition {
+  public static final DisableLipoTransition INSTANCE = new DisableLipoTransition();
 
-  private LipoDataTransition() {}
+  private DisableLipoTransition() {}
 
   @Override
   public BuildOptions apply(BuildOptions options) {
-    if (options.get(BuildConfiguration.Options.class).isHost) {
-      return options;
-    }
-
     // If this target and its transitive closure don't have C++ options, there's no
     // LIPO context to change.
     if (!options.contains(CppOptions.class)) {
       return options;
     }
-
     CppOptions cppOptions = options.get(CppOptions.class);
-    if (cppOptions.getLipoMode() == CrosstoolConfig.LipoMode.OFF) {
-      return options;
-    }
 
-    options = options.clone();
-    cppOptions = options.get(CppOptions.class);
-
-    // Once autoFdoLipoData is on, it stays on (through all future transitions).
-    if (!cppOptions.getAutoFdoLipoData() && cppOptions.getFdoOptimize() != null) {
-      cppOptions.autoFdoLipoDataForBuild = FdoSupport.isAutoFdo(cppOptions.getFdoOptimize());
+    if (cppOptions.getLipoMode() == LipoMode.OFF || cppOptions.isLipoContextCollector()) {
+      return options;
     }
-    cppOptions.lipoModeForBuild = CrosstoolConfig.LipoMode.OFF;
-    cppOptions.fdoInstrumentForBuild = null;
-    cppOptions.fdoOptimizeForBuild = null;
-    return options;
+    BuildOptions lipoDisabledOptions = options.clone();
+    lipoDisabledOptions.get(CppOptions.class).lipoConfigurationState =
+        CppOptions.LipoConfigurationState.IGNORE_LIPO;
+    return lipoDisabledOptions;
   }
 
   @Override
diff --git a/src/main/java/com/google/devtools/build/lib/rules/cpp/transitions/LipoContextCollectorTransition.java b/src/main/java/com/google/devtools/build/lib/rules/cpp/transitions/LipoContextCollectorTransition.java
new file mode 100644
index 0000000000..1259c7534c
--- /dev/null
+++ b/src/main/java/com/google/devtools/build/lib/rules/cpp/transitions/LipoContextCollectorTransition.java
@@ -0,0 +1,55 @@
+// Copyright 2017 The Bazel Authors. All rights reserved.
+//
+// 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.
+
+package com.google.devtools.build.lib.rules.cpp.transitions;
+
+import com.google.devtools.build.lib.analysis.config.BuildOptions;
+import com.google.devtools.build.lib.analysis.config.PatchTransition;
+import com.google.devtools.build.lib.rules.cpp.CppOptions;
+import com.google.devtools.build.lib.rules.cpp.FdoSupport;
+
+/**
+ * Configuration transition that enters "LIPO context collector" mode on a
+ * "LIPO optimization"-enabled input configuration.
+ *
+ * <p>See {@link FdoSupport} for details.
+ */
+public class LipoContextCollectorTransition implements PatchTransition {
+  public static final LipoContextCollectorTransition INSTANCE =
+      new LipoContextCollectorTransition();
+
+  private LipoContextCollectorTransition() {}
+
+  @Override
+  public BuildOptions apply(BuildOptions options) {
+    // If this target and its transitive closure don't have C++ options, there's no
+    // LIPO context to change.
+    if (!options.contains(CppOptions.class)) {
+      return options;
+    }
+    CppOptions cppOptions = options.get(CppOptions.class);
+    if (!cppOptions.isLipoOptimization()) {
+      return options;
+    }
+    BuildOptions collectorOptions = options.clone();
+    collectorOptions.get(CppOptions.class).lipoConfigurationState =
+        CppOptions.LipoConfigurationState.LIPO_CONTEXT_COLLECTOR;
+    return collectorOptions;
+  }
+
+  @Override
+  public boolean defaultsToSelf() {
+    return false;
+  }
+}