com.yahoo.tensor.impl.TensorAddressAnyN Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vespajlib Show documentation

Library for use in Java components of Vespa. Shared code which do not fit anywhere else.

There is a newer version: 8.441.21

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

package com.yahoo.tensor.impl;

import com.yahoo.tensor.TensorAddress;

import java.util.Arrays;

import static java.lang.Math.abs;

/**
 * An n-dimensional address.
 *
 * @author baldersheim
 */
final class TensorAddressAnyN extends TensorAddressAny {

    private final long[] labels;

    TensorAddressAnyN(long[] labels) {
        if (labels.length < 1) throw new IllegalArgumentException("Need at least 1 label");
        this.labels = labels;
    }

    @Override public int size() { return labels.length; }

    @Override public long numericLabel(int i) { return labels[i]; }

    @Override
    public TensorAddress withLabel(int labelIndex, long label) {
        long[] copy = Arrays.copyOf(labels, labels.length);
        copy[labelIndex] = label;
        return new TensorAddressAnyN(copy);
    }

    @Override public int hashCode() {
        long hash = abs(labels[0]);
        for (int i = 0; i < size(); i++) {
            hash = hash | (abs(labels[i]) << (32 - Long.numberOfLeadingZeros(hash)));
        }
        return (int) hash;
    }

    @Override
    public boolean equals(Object o) {
        if (! (o instanceof TensorAddressAnyN any) || (size() != any.size())) return false;
        for (int i = 0; i < size(); i++) {
            if (labels[i] != any.labels[i]) return false;
        }
        return true;
    }

}