io.github.vmzakharov.ecdataframe.dataframe.DfObjectColumnStored 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.2.0

package io.github.vmzakharov.ecdataframe.dataframe;

import io.github.vmzakharov.ecdataframe.dsl.value.Value;
import org.eclipse.collections.api.list.ImmutableList;
import org.eclipse.collections.api.list.ListIterable;
import org.eclipse.collections.api.list.MutableList;
import org.eclipse.collections.api.set.Pool;
import org.eclipse.collections.impl.factory.Lists;
import org.eclipse.collections.impl.set.mutable.UnifiedSet;

abstract public class DfObjectColumnStored
extends DfObjectColumnAbstract
implements DfColumnStored
{
    private MutableList values = Lists.mutable.of();

    private Pool pool = null;

    public DfObjectColumnStored(DataFrame owner, String newName)
    {
        super(owner, newName);
    }

    public DfObjectColumnStored(DataFrame owner, String newName, ListIterable newValues)
    {
        super(owner, newName);
        this.values.addAllIterable(newValues);
    }

    @Override
    public Value getValue(int rowIndex)
    {
        if (this.isNull(rowIndex))
        {
            return Value.VOID;
        }

        return this.objectToValue(this.getTypedObject(rowIndex));
    }

    protected void addMyType(T anObject)
    {
        if (this.pool == null)
        {
            this.values.add(anObject);
        }
        else
        {
            this.values.add(this.pool.put(anObject));
        }
    }

    @Override
    public void enablePooling()
    {
        this.pool = new UnifiedSet<>();
    }

    @Override
    public void disablePooling()
    {
        this.pool = null;
    }

    @Override
    public int getSize()
    {
        return this.values.size();
    }

    @Override
    public void setObject(int rowIndex, Object anObject)
    {
        this.values.set(rowIndex, (T) anObject);
    }

    @Override
    public void addEmptyValue()
    {
        this.values.add(null);
    }

    @Override
    public Object getObject(int rowIndex)
    {
        return this.values.get(rowIndex);
    }

    @Override
    public T getTypedObject(int rowIndex)
    {
        return this.values.get(rowIndex);
    }

    @Override
    public boolean isNull(int rowIndex)
    {
        return this.values.get(rowIndex) == null;
    }

    @Override
    public ImmutableList toList()
    {
        return this.values.toImmutable();
    }

    @Override
    public void aggregateValueInto(int rowIndex, DfColumn sourceColumn, int sourceRowIndex, AggregateFunction aggregator)
    {
        T currentAggregatedValue = this.values.get(rowIndex);
        this.values.set(rowIndex,
                (T) aggregator.objectAccumulator(
                        currentAggregatedValue, aggregator.getObjectValue(sourceColumn, sourceRowIndex)));
    }

    @Override
    public void ensureInitialCapacity(int newCapacity)
    {
        this.values = Lists.mutable.withInitialCapacity(newCapacity);
    }

    @Override
    protected void addAllItems(ListIterable items)
    {
        this.values.addAllIterable(items);
    }
}