linq4j/src/test/java/org/apache/calcite/linq4j/test/BlockBuilderBase.java

/*
 * 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);
  }
}