io.deephaven.engine.table.impl.partitioned.BaseTableTransformationColumn 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.engine.table.*;
import io.deephaven.engine.table.impl.MatchPair;
import io.deephaven.engine.table.impl.NoSuchColumnException;
import io.deephaven.engine.table.impl.select.SelectColumn;
import io.deephaven.engine.table.impl.sources.InMemoryColumnSource;
import io.deephaven.engine.table.impl.sources.SparseArrayColumnSource;
import org.jetbrains.annotations.NotNull;

import java.util.Collections;
import java.util.List;
import java.util.Map;

/**
 * Base {@link SelectColumn} implementation to wrap transformer functions for {@link PartitionedTable#transform} and
 * {@link PartitionedTable#partitionedTransform}.
 */
abstract class BaseTableTransformationColumn implements SelectColumn {

    BaseTableTransformationColumn() {}

    @Override
    public final Class getReturnedType() {
        return Table.class;
    }

    @Override
    public final List getColumnArrays() {
        return Collections.emptyList();
    }

    @NotNull
    @Override
    public final ColumnSource getLazyView() {
        return getDataView();
    }

    @Override
    public final MatchPair getMatchPair() {
        throw new UnsupportedOperationException();
    }

    @Override
    public final WritableColumnSource newDestInstance(final long size) {
        return SparseArrayColumnSource.getSparseMemoryColumnSource(size, Table.class);
    }

    @Override
    public final WritableColumnSource newFlatDestInstance(final long size) {
        return InMemoryColumnSource.getImmutableMemoryColumnSource(size, Table.class, null);
    }

    @Override
    public final boolean isRetain() {
        return false;
    }

    @Override
    public final boolean isStateless() {
        return true;
    }

    static ColumnSource getAndValidateInputColumnSource(
            @NotNull final String inputColumnName,
            @NotNull final Map> columnsOfInterest) {
        final ColumnSource inputColumnSource = columnsOfInterest.get(inputColumnName);
        if (inputColumnSource == null) {
            throw new NoSuchColumnException(columnsOfInterest.keySet(), inputColumnName);
        }
        if (!Table.class.isAssignableFrom(inputColumnSource.getType())) {
            throw new IllegalArgumentException("Input column \"" + inputColumnName
                    + "\" has unexpected type " + inputColumnSource.getType());
        }
        // noinspection unchecked
        return (ColumnSource) inputColumnSource;
    }

    static void validateInputColumnDefinition(
            @NotNull final String inputColumnName,
            @NotNull final Map> columnDefinitionMap) {
        final ColumnDefinition inputColumnDefinition = columnDefinitionMap.get(inputColumnName);
        if (inputColumnDefinition == null) {
            throw new NoSuchColumnException(columnDefinitionMap.keySet(), inputColumnName);
        }
        if (!Table.class.isAssignableFrom(inputColumnDefinition.getDataType())) {
            throw new IllegalArgumentException("Input column \"" + inputColumnName
                    + "\" has unexpected type " + inputColumnDefinition.getDataType());
        }
    }
}
    

    







    © 2015 - 2024 Weber Informatics LLC | Privacy Policy