com.hazelcast.org.apache.calcite.adapter.enumerable.impl.WinAggResetContextImpl Maven / Gradle / Ivy

/*
 * 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 com.hazelcast.org.apache.calcite.adapter.enumerable.impl;

import com.hazelcast.org.apache.calcite.adapter.enumerable.WinAggResetContext;
import com.hazelcast.org.apache.calcite.linq4j.tree.BlockBuilder;
import com.hazelcast.org.apache.calcite.linq4j.tree.Expression;

import java.util.List;

/**
 * Implementation of
 * {@link com.hazelcast.org.apache.calcite.adapter.enumerable.WinAggResetContext}.
 */
public class WinAggResetContextImpl extends AggResetContextImpl
    implements WinAggResetContext {
  private final Expression index;
  private final Expression startIndex;
  private final Expression endIndex;
  private final Expression frameRowCount;
  private final Expression partitionRowCount;
  private final Expression hasRows;

  /**
   * Creates window aggregate reset context.
   * @param block code block that will contain the added initialization
   * @param accumulator accumulator variables that store the intermediate
   *                    aggregate state
   * @param index index of the current row in the partition
   * @param startIndex index of the very first row in partition
   * @param endIndex index of the very last row in partition
   * @param hasRows boolean expression that tells if the partition has rows
   * @param frameRowCount number of rows in the current frame
   * @param partitionRowCount number of rows in the current partition
   */
  public WinAggResetContextImpl(BlockBuilder block,
      List accumulator, Expression index,
      Expression startIndex, Expression endIndex,
      Expression hasRows,
      Expression frameRowCount, Expression partitionRowCount) {
    super(block, accumulator);
    this.index = index;
    this.startIndex = startIndex;
    this.endIndex = endIndex;
    this.frameRowCount = frameRowCount;
    this.partitionRowCount = partitionRowCount;
    this.hasRows = hasRows;
  }

  @Override public Expression index() {
    return index;
  }

  @Override public Expression startIndex() {
    return startIndex;
  }

  @Override public Expression endIndex() {
    return endIndex;
  }

  @Override public Expression hasRows() {
    return hasRows;
  }

  @Override public Expression getFrameRowCount() {
    return frameRowCount;
  }

  @Override public Expression getPartitionRowCount() {
    return partitionRowCount;
  }
}