content: /*
   * Licensed to the Apache Software Foundation (ASF) under one or more
   * contributor license agreements.  See the NOTICE file distributed with
   * this work for additional information regarding copyright ownership.
   * The ASF licenses this file to you 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 org.apache.calcite.linq4j.test;
  
  import org.apache.calcite.linq4j.tree.BlockBuilder;
  import org.apache.calcite.linq4j.tree.BlockStatement;
  import org.apache.calcite.linq4j.tree.Expression;
  import org.apache.calcite.linq4j.tree.Expressions;
  import org.apache.calcite.linq4j.tree.ParameterExpression;
  import org.apache.calcite.linq4j.tree.Statement;
  
  /**
   * Base methods and constants for simplified Expression testing.
   */
  public final class BlockBuilderBase {
    private BlockBuilderBase() {}
  
    public static final Expression NULL = Expressions.constant(null);
    public static final Expression NULL_INTEGER =
        Expressions.constant(null, Integer.class);
    public static final Expression ONE = Expressions.constant(1);
    public static final Expression TWO = Expressions.constant(2);
    public static final Expression THREE = Expressions.constant(3);
    public static final Expression FOUR = Expressions.constant(4);
    public static final Expression TRUE = Expressions.constant(true);
    public static final Expression FALSE = Expressions.constant(false);
  
    public static final Expression TRUE_B =
        Expressions.field(null, Boolean.class, "TRUE");
    public static final Expression FALSE_B =
        Expressions.field(null, Boolean.class, "FALSE");
  
    public static String optimize(Expression expr) {
      return optimize(Expressions.return_(null, expr));
    }
  
    public static BlockStatement optimizeExpression(Expression expr) {
      return optimizeStatement(Expressions.return_(null, expr));
    }
  
    public static String optimize(Statement statement) {
      return optimizeStatement(statement).toString();
    }
  
    public static BlockStatement optimizeStatement(Statement statement) {
      BlockBuilder b = new BlockBuilder(true);
      if (!(statement instanceof BlockStatement)) {
        b.add(statement);
      } else {
        BlockStatement bs = (BlockStatement) statement;
        for (Statement stmt : bs.statements) {
          b.add(stmt);
        }
      }
      BlockStatement bs = b.toBlock();
      return bs;
    }
  
    public static ParameterExpression bool(String name) {
      return Expressions.parameter(boolean.class, name);
    }
  
    public static ParameterExpression int_(String name) {
      return Expressions.parameter(int.class, name);
    }
  
    public static ParameterExpression integer(String name) {
      return Expressions.parameter(Integer.class, name);
    }
  }
  