Reformat code.

Author: JimClarke5

tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/op/generator/OpGenerator.java

@@ -1,24 +1,28 @@
 /*
-  Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+ Copyright 2021 The TensorFlow 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
+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
+    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.
- ==============================================================================
- */
+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 org.tensorflow.op.generator;
 
 import com.google.protobuf.InvalidProtocolBufferException;
 import com.squareup.javapoet.JavaFile;
 import com.squareup.javapoet.TypeSpec;
+import org.tensorflow.proto.framework.ApiDef;
+import org.tensorflow.proto.framework.OpDef;
+import org.tensorflow.proto.framework.OpList;
+
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
@@ -32,28 +36,25 @@
 import java.nio.file.attribute.BasicFileAttributes;
 import java.util.LinkedHashMap;
 import java.util.Map;
-import org.tensorflow.proto.framework.ApiDef;
-import org.tensorflow.proto.framework.OpDef;
-import org.tensorflow.proto.framework.OpList;
 
 public final class OpGenerator {
 
   private static final String LICENSE =
-      "/* Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n" +
-          "\n" +
-          "Licensed under the Apache License, Version 2.0 (the \"License\");\n" +
-          "you may not use this file except in compliance with the License.\n" +
-          "You may obtain a copy of the License at\n" +
-          "\n" +
-          "    http://www.apache.org/licenses/LICENSE-2.0\n" +
-          "\n" +
-          "Unless required by applicable law or agreed to in writing, software\n" +
-          "distributed under the License is distributed on an \"AS IS\" BASIS,\n" +
-          "WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n" +
-          "See the License for the specific language governing permissions and\n" +
-          "limitations under the License.\n" +
-          "=======================================================================*/" +
-          "\n";
+      "/* Copyright 2018 The TensorFlow Authors. All Rights Reserved.\n"
+          + "\n"
+          + "Licensed under the Apache License, Version 2.0 (the \"License\");\n"
+          + "you may not use this file except in compliance with the License.\n"
+          + "You may obtain a copy of the License at\n"
+          + "\n"
+          + "    http://www.apache.org/licenses/LICENSE-2.0\n"
+          + "\n"
+          + "Unless required by applicable law or agreed to in writing, software\n"
+          + "distributed under the License is distributed on an \"AS IS\" BASIS,\n"
+          + "WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n"
+          + "See the License for the specific language governing permissions and\n"
+          + "limitations under the License.\n"
+          + "=======================================================================*/"
+          + "\n";
 
   private static final String HELP_TEXT = "Args should be: <outputDir> <opDefFile> [base_package]";
 
@@ -62,9 +63,9 @@ public final class OpGenerator {
   /**
    * Args should be {@code <outputDir> <opDefFile> [base_package]}.
    *
-   * {@code base_package} is {@code org.tensorflow.op} by default.
-   * <p>
-   * <b>Will delete everything in {@code outputDir}.</b>
+   * <p>{@code base_package} is {@code org.tensorflow.op} by default.
+   *
+   * <p><b>Will delete everything in {@code outputDir}.</b>
    */
   public static void main(String[] args) {
 
@@ -106,19 +107,23 @@ public static void main(String[] args) {
 
     if (basePackage.exists()) {
       try {
-        Files.walkFileTree(basePackage.toPath(), new SimpleFileVisitor<Path>() {
-          @Override
-          public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
-            Files.delete(file);
-            return FileVisitResult.CONTINUE;
-          }
-
-          @Override
-          public FileVisitResult postVisitDirectory(Path dir, IOException exc) throws IOException {
-            Files.delete(dir);
-            return FileVisitResult.CONTINUE;
-          }
-        });
+        Files.walkFileTree(
+            basePackage.toPath(),
+            new SimpleFileVisitor<Path>() {
+              @Override
+              public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
+                  throws IOException {
+                Files.delete(file);
+                return FileVisitResult.CONTINUE;
+              }
+
+              @Override
+              public FileVisitResult postVisitDirectory(Path dir, IOException exc)
+                  throws IOException {
+                Files.delete(dir);
+                return FileVisitResult.CONTINUE;
+              }
+            });
       } catch (IOException ignored) {
 
       }
@@ -127,9 +132,7 @@ public FileVisitResult postVisitDirectory(Path dir, IOException exc) throws IOEx
     generate(outputDir, packageName, inputFile);
   }
 
-  /**
-   * Build the list of ops and api defs, then call {@link #generate(File, String, Map)}
-   */
+  /** Build the list of ops and api defs, then call {@link #generate(File, String, Map)} */
   private static void generate(File outputDir, String packageName, File opDefs) {
     OpList opList = null;
     try {
@@ -140,87 +143,121 @@ private static void generate(File outputDir, String packageName, File opDefs) {
       System.exit(1);
     } catch (IOException e) {
       throw new RuntimeException(
-          "Error parsing op def file " + opDefs + ", was it generated by op_export_main.cc (in tensorflow-core-api)?",
+          "Error parsing op def file "
+              + opDefs
+              + ", was it generated by op_export_main.cc (in tensorflow-core-api)?",
           e);
     }
     Map<OpDef, ApiDef> defs = new LinkedHashMap<>(opList.getOpCount());
 
     for (OpDef op : opList.getOpList()) {
       try {
         if (!op.getUnknownFields().hasField(API_DEF_FIELD_NUMBER)) {
-          throw new RuntimeException("No attached ApiDef for op " + op.getName() + ", was " + opDefs
-              + " generated by op_export_main.cc (in tensorflow-core-api)?  The op's ApiDef should be"
-              + " attached as unknown field " + API_DEF_FIELD_NUMBER + ".");
+          throw new RuntimeException(
+              "No attached ApiDef for op "
+                  + op.getName()
+                  + ", was "
+                  + opDefs
+                  + " generated by op_export_main.cc (in tensorflow-core-api)?  The op's ApiDef should be"
+                  + " attached as unknown field "
+                  + API_DEF_FIELD_NUMBER
+                  + ".");
         }
-        ApiDef api = ApiDef
-            .parseFrom(op.getUnknownFields().getField(API_DEF_FIELD_NUMBER).getLengthDelimitedList().get(0));
+        ApiDef api =
+            ApiDef.parseFrom(
+                op.getUnknownFields()
+                    .getField(API_DEF_FIELD_NUMBER)
+                    .getLengthDelimitedList()
+                    .get(0));
         defs.put(op, api);
       } catch (InvalidProtocolBufferException e) {
-        throw new RuntimeException("Could not parse attached ApiDef for op " + op.getName() + ", was " + opDefs
-            + " generated by op_export_main.cc (in tensorflow-core-api)?", e);
+        throw new RuntimeException(
+            "Could not parse attached ApiDef for op "
+                + op.getName()
+                + ", was "
+                + opDefs
+                + " generated by op_export_main.cc (in tensorflow-core-api)?",
+            e);
       }
     }
 
     generate(outputDir, packageName, defs);
   }
 
-  /**
-   * Generate all the ops that pass {@link ClassGenerator#canGenerateOp(OpDef, ApiDef)}.
-   */
+  /** Generate all the ops that pass {@link ClassGenerator#canGenerateOp(OpDef, ApiDef)}. */
   private static void generate(File outputDir, String basePackage, Map<OpDef, ApiDef> ops) {
-    ops.entrySet().stream().filter(e -> ClassGenerator.canGenerateOp(e.getKey(), e.getValue())).forEach((entry) -> {
-      entry.getValue().getEndpointList().forEach((endpoint) -> {
-
-        String name;
-        String pack;
-
-        int pos = endpoint.getName().lastIndexOf('.');
-        if (pos > -1) {
-          pack = endpoint.getName().substring(0, pos);
-          name = endpoint.getName().substring(pos + 1);
-        } else {
-          pack = "core";
-          name = endpoint.getName();
-        }
-
-        TypeSpec.Builder cls = TypeSpec.classBuilder(name);
-        try {
-          new ClassGenerator(
-              cls,
-              entry.getKey(),
-              entry.getValue(),
-              basePackage,
-              basePackage + "." + pack,
-              pack,
-              name,
-              endpoint).buildClass();
-        } catch (Exception e) {
-          throw new IllegalStateException("Failed to generate class for op " + entry.getKey().getName(), e);
-        }
-        TypeSpec spec = cls.build();
-
-        JavaFile file = JavaFile.builder(basePackage + "." + pack, spec)
-            .indent("  ")
-            .skipJavaLangImports(true)
-            .build();
-
-        File outputFile = new File(outputDir, basePackage.replace('.', '/') +
-            '/' + pack.replace('.', '/') + '/' + spec.name + ".java");
-        outputFile.getParentFile().mkdirs();
-        try {
-          StringBuilder builder = new StringBuilder();
-          builder.append(LICENSE + '\n');
-          builder.append("// This class has been generated, DO NOT EDIT!\n\n");
-          file.writeTo(builder);
-
-          Files.write(outputFile.toPath(), builder.toString().getBytes(StandardCharsets.UTF_8), StandardOpenOption.WRITE,
-              StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
-        } catch (IOException ioException) {
-          throw new IllegalStateException("Failed to write file " + outputFile, ioException);
-        }
-      });
-    });
+    ops.entrySet().stream()
+        .filter(e -> ClassGenerator.canGenerateOp(e.getKey(), e.getValue()))
+        .forEach(
+            (entry) -> {
+              entry
+                  .getValue()
+                  .getEndpointList()
+                  .forEach(
+                      (endpoint) -> {
+                        String name;
+                        String pack;
+
+                        int pos = endpoint.getName().lastIndexOf('.');
+                        if (pos > -1) {
+                          pack = endpoint.getName().substring(0, pos);
+                          name = endpoint.getName().substring(pos + 1);
+                        } else {
+                          pack = "core";
+                          name = endpoint.getName();
+                        }
+
+                        TypeSpec.Builder cls = TypeSpec.classBuilder(name);
+                        try {
+                          new ClassGenerator(
+                                  cls,
+                                  entry.getKey(),
+                                  entry.getValue(),
+                                  basePackage,
+                                  basePackage + "." + pack,
+                                  pack,
+                                  name,
+                                  endpoint)
+                              .buildClass();
+                        } catch (Exception e) {
+                          throw new IllegalStateException(
+                              "Failed to generate class for op " + entry.getKey().getName(), e);
+                        }
+                        TypeSpec spec = cls.build();
+
+                        JavaFile file =
+                            JavaFile.builder(basePackage + "." + pack, spec)
+                                .indent("  ")
+                                .skipJavaLangImports(true)
+                                .build();
+
+                        File outputFile =
+                            new File(
+                                outputDir,
+                                basePackage.replace('.', '/')
+                                    + '/'
+                                    + pack.replace('.', '/')
+                                    + '/'
+                                    + spec.name
+                                    + ".java");
+                        outputFile.getParentFile().mkdirs();
+                        try {
+                          StringBuilder builder = new StringBuilder();
+                          builder.append(LICENSE + '\n');
+                          builder.append("// This class has been generated, DO NOT EDIT!\n\n");
+                          file.writeTo(builder);
+
+                          Files.write(
+                              outputFile.toPath(),
+                              builder.toString().getBytes(StandardCharsets.UTF_8),
+                              StandardOpenOption.WRITE,
+                              StandardOpenOption.CREATE,
+                              StandardOpenOption.TRUNCATE_EXISTING);
+                        } catch (IOException ioException) {
+                          throw new IllegalStateException(
+                              "Failed to write file " + outputFile, ioException);
+                        }
+                      });
+            });
   }
-
-
 }