blob: e3398850373d2fd0af1d5cdd07d25a52a8204898 (
plain) (
blame)
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.tensor;
import com.google.common.annotations.Beta;
/**
* An address to a subset of a tensors' cells, specifying a label for some but not necessarily all of the tensors
* dimensions.
*
* @author bratseth
*/
// Implementation notes:
// - These are created in inner (though not inner-most) loops so they are implemented with minimal allocation.
// We also avoid non-essential error checking.
// - We can add support for string labels later without breaking the API
@Beta
public class PartialAddress {
// Two arrays which contains corresponding dimension=label pairs.
// The sizes of these are always equal.
private final String[] dimensionNames;
private final int[] labels;
private PartialAddress(Builder builder) {
this.dimensionNames = builder.dimensionNames;
this.labels = builder.labels;
builder.dimensionNames = null; // invalidate builder to safely take over array ownership
builder.labels = null;
}
/** Returns the int label of this dimension, or -1 if no label is specified for it */
int intLabel(String dimensionName) {
for (int i = 0; i < dimensionNames.length; i++)
if (dimensionNames[i].equals(dimensionName))
return labels[i];
return -1;
}
public static class Builder {
private String[] dimensionNames;
private int[] labels;
private int index = 0;
public Builder(int size) {
dimensionNames = new String[size];
labels = new int[size];
}
public void add(String dimensionName, int label) {
dimensionNames[index] = dimensionName;
labels[index] = label;
index++;
}
public PartialAddress build() {
return new PartialAddress(this);
}
}
}
|
