io.github.vmzakharov.ecdataframe.dataframe.aggregation.Max Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dataframe-ec Show documentation
A tabular data structure based on the Eclipse Collections framework
There is a newer version: 1.0.0
package io.github.vmzakharov.ecdataframe.dataframe.aggregation;

import io.github.vmzakharov.ecdataframe.dataframe.AggregateFunction;
import io.github.vmzakharov.ecdataframe.dataframe.DfDecimalColumn;
import io.github.vmzakharov.ecdataframe.dataframe.DfDoubleColumn;
import io.github.vmzakharov.ecdataframe.dataframe.DfFloatColumn;
import io.github.vmzakharov.ecdataframe.dataframe.DfIntColumn;
import io.github.vmzakharov.ecdataframe.dataframe.DfLongColumn;
import io.github.vmzakharov.ecdataframe.dataframe.DfObjectColumn;
import io.github.vmzakharov.ecdataframe.dsl.value.ValueType;
import org.eclipse.collections.api.list.ListIterable;
import org.eclipse.collections.impl.factory.Lists;

import java.math.BigDecimal;

import static io.github.vmzakharov.ecdataframe.dsl.value.ValueType.DECIMAL;
import static io.github.vmzakharov.ecdataframe.dsl.value.ValueType.DOUBLE;
import static io.github.vmzakharov.ecdataframe.dsl.value.ValueType.FLOAT;
import static io.github.vmzakharov.ecdataframe.dsl.value.ValueType.INT;
import static io.github.vmzakharov.ecdataframe.dsl.value.ValueType.LONG;

public class Max
extends AggregateFunction
{
    private static final ListIterable SUPPORTED_TYPES = Lists.immutable.of(INT, LONG, DOUBLE, FLOAT, DECIMAL);

    public Max(String newColumnName)
    {
        super(newColumnName);
    }

    public Max(String newColumnName, String newTargetColumnName)
    {
        super(newColumnName, newTargetColumnName);
    }

    @Override
    public ListIterable supportedSourceTypes()
    {
        return SUPPORTED_TYPES;
    }

    @Override
    public String getDescription()
    {
        return "Maximum numeric value";
    }

    @Override
    public Object applyToDoubleColumn(DfDoubleColumn doubleColumn)
    {
        return doubleColumn.asDoubleIterable().max();
    }

    @Override
    public Object applyToFloatColumn(DfFloatColumn floatColumn)
    {
        return floatColumn.asFloatIterable().max();
    }

    @Override
    public Object applyToLongColumn(DfLongColumn longColumn)
    {
        return longColumn.asLongIterable().max();
    }

    @Override
    public Object applyToIntColumn(DfIntColumn intColumn)
    {
        return intColumn.asIntIterable().max();
    }

    @Override
    public int intInitialValue()
    {
        return Integer.MIN_VALUE;
    }

    @Override
    public long longInitialValue()
    {
        return Long.MIN_VALUE;
    }

    @Override
    public double doubleInitialValue()
    {
        return -Double.MAX_VALUE;
    }

    @Override
    public float floatInitialValue()
    {
        return -Float.MAX_VALUE;
    }

    @Override
    public Object applyToObjectColumn(DfObjectColumn objectColumn)
    {
        return ((DfDecimalColumn) objectColumn).injectIntoBreakOnNulls(
                this.objectInitialValue(),
                (result, current) -> result.compareTo(current) > 0 ? result : current);
    }

    @Override
    public BigDecimal objectInitialValue()
    {
        return BigDecimal.valueOf(-Double.MAX_VALUE);
    }

    @Override
    protected long longAccumulator(long currentAggregate, long newValue)
    {
        return Math.max(currentAggregate, newValue);
    }

    @Override
    protected double doubleAccumulator(double currentAggregate, double newValue)
    {
        return Math.max(currentAggregate, newValue);
    }

    @Override
    protected float floatAccumulator(float currentAggregate, float newValue)
    {
        return Math.max(currentAggregate, newValue);
    }

    @Override
    protected int intAccumulator(int currentAggregate, int newValue)
    {
        return Math.max(currentAggregate, newValue);
    }

    @Override
    protected Object objectAccumulator(Object currentAggregate, Object newValue)
    {
        return ((BigDecimal) currentAggregate).compareTo((BigDecimal) newValue) < 0 ? newValue : currentAggregate;
    }
}