vespajlib/src/main/java/com/yahoo/tensor/impl/TensorAddressAny1.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

// 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;

/**
 * A one-dimensional address.
 *
 * @author baldersheim
 */
final class TensorAddressAny1 extends TensorAddressAny {

    private final int label;

    TensorAddressAny1(int label) { this.label = label; }

    @Override public int size() { return 1; }

    @Override
    public long numericLabel(int i) {
        if (i == 0) {
            return label;
        }
        throw new IndexOutOfBoundsException("Index is not zero: " + i);
    }

    @Override
    public TensorAddress withLabel(int labelIndex, long label) {
        if (labelIndex == 0) return new TensorAddressAny1(Convert.safe2Int(label));
        throw new IllegalArgumentException("No label " + labelIndex);
    }

    @Override public int hashCode() { return Math.abs(label); }

    @Override
    public boolean equals(Object o) {
        return (o instanceof TensorAddressAny1 any) && (label == any.label);
    }

}