// Copyright (c) 2019, the R8 project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
package com.android.tools.r8.ir.optimize.string;

import com.android.tools.r8.NeverInline;

class StringConcatenationTestClass {
  @NeverInline
  public static void unusedBuilder() {
    StringBuilder builder = new StringBuilder();
    builder.append(4);
    builder.append(2);
    builder.toString();
  }

  @NeverInline
  public static void trivialSequence() {
    String x = "x";
    String y = "y";
    String z = "z";
    System.out.println(x + y + z);
  }

  @NeverInline
  public static void builderWithInitialValue() {
    StringBuilder builder = new StringBuilder("Hello");
    builder.append(",");
    builder.append("R8");
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void builderWithCapacity() {
    StringBuilder builder = new StringBuilder(3);
    builder.append(4);
    builder.append(2);
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void nonStringArgs() {
    StringBuilder builder = new StringBuilder();
    builder.append(4);
    builder.append(2);
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void typeConversion() {
    StringBuilder builder = new StringBuilder();
    float f = 0.14f;
    builder.append(f);
    builder.append(' ');
    int i = (int) f;
    builder.append(i);
    builder.append(' ');

    boolean b = false;
    builder.append(b);
    builder.append(' ');
    Object n = null;
    builder.append(n);
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void typeConversion_withPhis() {
    StringBuilder builder = new StringBuilder();
    float pi = 3.14f;
    float f = 0.14f;
    float phi = System.currentTimeMillis() > 0 ? pi : f;
    builder.append(phi);
    builder.append(' ');
    int i_phi = (int) phi;
    builder.append(i_phi);
    builder.append(' ');
    int another_i_phi = System.currentTimeMillis() > 0 ? (int) f : (int) pi;
    builder.append(another_i_phi);
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void nestedBuilders_appendBuilderItself() {
    StringBuilder b1 = new StringBuilder();
    b1.append("Hello");
    b1.append(",");
    StringBuilder b2 = new StringBuilder();
    b2.append("R");
    b2.append(8);
    b1.append(b2);
    System.out.println(b1.toString());
  }

  @NeverInline
  public static void nestedBuilders_appendBuilderResult() {
    StringBuilder b1 = new StringBuilder();
    b1.append("Hello");
    b1.append(",");
    StringBuilder b2 = new StringBuilder();
    b2.append("R");
    b2.append(8);
    b1.append(b2.toString());
    System.out.println(b1.toString());
  }

  @NeverInline
  public static void nestedBuilders_conditional() {
    StringBuilder b1 = new StringBuilder();
    StringBuilder b2 = new StringBuilder();
    StringBuilder b3 = new StringBuilder();
    b1.append("Hello,");
    if (System.currentTimeMillis() > 0) {
      b2.append("R");
      b2.append("8");
      b1.append(b2);
    } else {
      // To avoid canonicalization
      b3.append("D");
      b3.append(8);
      b1.append(b3.toString());
    }
    System.out.println(b1.toString());
  }

  @NeverInline
  public static void concatenatedBuilders_init() {
    StringBuilder b1 = new StringBuilder();
    b1.append("Hello,");
    b1.append("R");
    StringBuilder b2 = new StringBuilder(b1);
    b2.append(8);
    System.out.println(b2.toString());
  }

  @NeverInline
  public static void concatenatedBuilders_append() {
    StringBuilder b1 = new StringBuilder();
    b1.append("Hello,");
    b1.append("R");
    StringBuilder b2 = new StringBuilder();
    b2.append(b1);
    b2.append(8);
    System.out.println(b2.toString());
  }

  @NeverInline
  public static void concatenatedBuilders_conditional() {
    String result;
    StringBuilder b1 = new StringBuilder();
    b1.append("Hello,");
    if (System.currentTimeMillis() > 0) {
      StringBuilder b2 = new StringBuilder();
      b2.append(b1);
      b2.append("R");
      b2.append("8");
      result = b2.toString();
    } else {
      StringBuilder b3 = new StringBuilder();
      // To avoid canonicalization
      b3.append("D");
      b3.append(8);
      result = b3.toString();
    }
    System.out.println(result);
  }

  @NeverInline
  public static void simplePhi() {
    StringBuilder builder = new StringBuilder();
    builder.append("Hello");
    builder.append(",");
    System.out.println(builder.toString());

    if (System.currentTimeMillis() > 0) {
      builder.append("D");
    } else {
      builder.append("R");
    }
    builder.append(8);
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void phiAtInit() {
    StringBuilder builder =
        System.currentTimeMillis() > 0 ? new StringBuilder("Hello") : new StringBuilder("Hi");
    builder.append(",R8");
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void phiWithDifferentInits() {
    StringBuilder b1 = new StringBuilder("Hello");
    StringBuilder b2 = new StringBuilder("Hi");
    StringBuilder builder = System.currentTimeMillis() > 0 ? b1 : b2;
    builder.append(",R8");
    System.out.println(builder.toString());
  }

  @NeverInline
  public static void conditionalPhiWithoutAppend() {
    StringBuilder b = new StringBuilder("initial");
    String suffix = "suffix";
    if (!suffix.isEmpty()) {
      b.append(":").append(suffix);
    }
    System.out.println(b.toString());
  }

  @NeverInline
  public static void loop() {
    String r = "";
    for (int i = 0; i < 8; i++) {
      r = r + "na;";
    }
    System.out.println(r + "Batman!");
  }

  @NeverInline
  public static void loopWithBuilder() {
    StringBuilder builder = new StringBuilder();
    for (int i = 0; i < 8; i++) {
      builder.append("na;");
    }
    builder.append("Batman!");
    System.out.println(builder.toString());
  }

  public static void main(String[] args) {
    unusedBuilder();
    trivialSequence();
    builderWithInitialValue();
    builderWithCapacity();
    nonStringArgs();
    typeConversion();
    typeConversion_withPhis();
    nestedBuilders_appendBuilderItself();
    nestedBuilders_appendBuilderResult();
    nestedBuilders_conditional();
    concatenatedBuilders_init();
    concatenatedBuilders_append();
    concatenatedBuilders_conditional();
    simplePhi();
    phiAtInit();
    phiWithDifferentInits();
    conditionalPhiWithoutAppend();
    loop();
    loopWithBuilder();
  }
}