io.deephaven.engine.table.impl.partitioned.BiTableTransformationColumn Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of deephaven-engine-table Show documentation
Engine Table: Implementation and closely-coupled utilities
There is a newer version: 0.37.1
Show newest version
/**
 * Copyright (c) 2016-2022 Deephaven Data Labs and Patent Pending
 */
package io.deephaven.engine.table.impl.partitioned;

import io.deephaven.chunk.ChunkType;
import io.deephaven.chunk.ObjectChunk;
import io.deephaven.chunk.WritableChunk;
import io.deephaven.chunk.WritableObjectChunk;
import io.deephaven.chunk.attributes.Values;
import io.deephaven.engine.rowset.RowSequence;
import io.deephaven.engine.rowset.TrackingRowSet;
import io.deephaven.engine.table.*;
import io.deephaven.engine.table.impl.select.Formula;
import io.deephaven.engine.table.impl.select.SelectColumn;
import io.deephaven.engine.table.impl.sources.ViewColumnSource;
import io.deephaven.engine.context.ExecutionContext;
import io.deephaven.util.SafeCloseable;
import org.jetbrains.annotations.NotNull;

import java.util.List;
import java.util.Map;
import java.util.function.BinaryOperator;

/**
 * {@link SelectColumn} implementation to wrap transformer functions for {@link PartitionedTable#partitionedTransform
 * partitioned transformations}.
 */
class BiTableTransformationColumn extends BaseTableTransformationColumn {

    private final String inputOutputColumnName;
    private final String secondInputColumnName;
    private final BinaryOperator transformer;
    private final ExecutionContext executionContext;

    private ColumnSource inputColumnSource1;
    private ColumnSource inputColumnSource2;

    BiTableTransformationColumn(
            @NotNull final String inputOutputColumnName,
            @NotNull final String secondInputColumnName,
            final ExecutionContext executionContext,
            @NotNull final BinaryOperator transformer) {
        this.inputOutputColumnName = inputOutputColumnName;
        this.secondInputColumnName = secondInputColumnName;
        this.executionContext = executionContext;
        this.transformer = transformer;
    }

    @Override
    public List initInputs(
            @NotNull final TrackingRowSet rowSet,
            @NotNull final Map> columnsOfInterest) {
        inputColumnSource1 = getAndValidateInputColumnSource(inputOutputColumnName, columnsOfInterest);
        inputColumnSource2 = getAndValidateInputColumnSource(secondInputColumnName, columnsOfInterest);
        return getColumns();
    }

    @Override
    public List initDef(@NotNull final Map> columnDefinitionMap) {
        validateInputColumnDefinition(inputOutputColumnName, columnDefinitionMap);
        validateInputColumnDefinition(secondInputColumnName, columnDefinitionMap);
        return getColumns();
    }

    @Override
    public List getColumns() {
        return List.of(inputOutputColumnName, secondInputColumnName);
    }

    @NotNull
    @Override
    public ColumnSource getDataView() {
        return new ViewColumnSource<>(Table.class, new OutputFormula(), true);
    }

    @Override
    public String getName() {
        return inputOutputColumnName;
    }

    @Override
    public SelectColumn copy() {
        return new BiTableTransformationColumn(inputOutputColumnName, secondInputColumnName,
                executionContext, transformer);
    }

    private final class OutputFormulaFillContext implements Formula.FillContext {

        private final ChunkSource.GetContext inputColumnSource1GetContext;
        private final ChunkSource.GetContext inputColumnSource2GetContext;

        private OutputFormulaFillContext(final int chunkCapacity) {
            inputColumnSource1GetContext = inputColumnSource1.makeGetContext(chunkCapacity);
            inputColumnSource2GetContext = inputColumnSource2.makeGetContext(chunkCapacity);
        }

        @Override
        public void close() {
            inputColumnSource1GetContext.close();
            inputColumnSource2GetContext.close();
        }
    }

    private final class OutputFormula extends Formula {

        private OutputFormula() {
            super(null);
        }

        @Override
        public Object get(final long rowKey) {
            try (final SafeCloseable ignored = executionContext == null ? null : executionContext.open()) {
                return transformer.apply(inputColumnSource1.get(rowKey), inputColumnSource2.get(rowKey));
            }
        }

        @Override
        public Object getPrev(final long rowKey) {
            try (final SafeCloseable ignored = executionContext == null ? null : executionContext.open()) {
                return transformer.apply(inputColumnSource1.getPrev(rowKey), inputColumnSource2.getPrev(rowKey));
            }
        }

        @Override
        protected ChunkType getChunkType() {
            return ChunkType.Object;
        }

        @Override
        public FillContext makeFillContext(final int chunkCapacity) {
            return new OutputFormulaFillContext(chunkCapacity);
        }

        @Override
        public void fillChunk(
                @NotNull final FillContext context,
                @NotNull final WritableChunk destination,
                @NotNull final RowSequence rowSequence) {
            final ObjectChunk source1 = inputColumnSource1.getChunk(
                    ((OutputFormulaFillContext) context).inputColumnSource1GetContext, rowSequence).asObjectChunk();
            final ObjectChunk source2 = inputColumnSource2.getChunk(
                    ((OutputFormulaFillContext) context).inputColumnSource2GetContext, rowSequence).asObjectChunk();
            transformAndFill(source1, source2, destination);
        }

        @Override
        public void fillPrevChunk(
                @NotNull final FillContext context,
                @NotNull final WritableChunk destination,
                @NotNull final RowSequence rowSequence) {
            final ObjectChunk source1 = inputColumnSource1.getPrevChunk(
                    ((OutputFormulaFillContext) context).inputColumnSource1GetContext, rowSequence).asObjectChunk();
            final ObjectChunk source2 = inputColumnSource2.getPrevChunk(
                    ((OutputFormulaFillContext) context).inputColumnSource2GetContext, rowSequence).asObjectChunk();
            transformAndFill(source1, source2, destination);
        }

        private void transformAndFill(
                @NotNull final ObjectChunk source1,
                @NotNull final ObjectChunk source2,
                @NotNull final WritableChunk destination) {
            final WritableObjectChunk typedDestination = destination.asWritableObjectChunk();
            final int size = source1.size();
            typedDestination.setSize(size);
            try (final SafeCloseable ignored = executionContext == null ? null : executionContext.open()) {
                for (int ii = 0; ii < size; ++ii) {
                    typedDestination.set(ii, transformer.apply(source1.get(ii), source2.get(ii)));
                }
            }
        }
    }
}
    

    







    © 2015 - 2024 Weber Informatics LLC | Privacy Policy