Skip to content

Commit 059d09a

Browse files
mp911demp911de
committed
Introduce Vector abstraction.
Closes #3193 Original pull request: #3194
1 parent f86ee7f commit 059d09a

File tree

7 files changed

+831
-0
lines changed

7 files changed

+831
-0
lines changed

‎src/main/java/org/springframework/data/domain/DoubleVector.java

+118
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,118 @@
1+
/*
2+
* Copyright 2024 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.springframework.data.domain;
17+
18+
import java.util.Arrays;
19+
import java.util.Collection;
20+
21+
import org.springframework.util.ObjectUtils;
22+
23+
/**
24+
* {@link Vector} implementation based on {@code double} array.
25+
*
26+
* @author Mark Paluch
27+
* @since 3.5
28+
*/
29+
class DoubleVector implements Vector {
30+
31+
private final double[] v;
32+
33+
public DoubleVector(double[] v) {
34+
this.v = v;
35+
}
36+
37+
/**
38+
* Copy the given {@code double} array and wrap it within a Vector.
39+
*/
40+
static Vector copy(double[] v) {
41+
42+
double[] copy = new double[v.length];
43+
System.arraycopy(v, 0, copy, 0, copy.length);
44+
45+
return new DoubleVector(copy);
46+
}
47+
48+
/**
49+
* Copy the given numeric values and wrap within a Vector.
50+
*/
51+
static Vector copy(Collection<? extends Number> v) {
52+
53+
double[] copy = new double[v.size()];
54+
int i = 0;
55+
for (Number number : v) {
56+
copy[i++] = number.doubleValue();
57+
}
58+
59+
return new DoubleVector(copy);
60+
}
61+
62+
@Override
63+
public Class<Double> getType() {
64+
return Double.TYPE;
65+
}
66+
67+
@Override
68+
public Object getSource() {
69+
return v;
70+
}
71+
72+
@Override
73+
public int size() {
74+
return v.length;
75+
}
76+
77+
@Override
78+
public float[] toFloatArray() {
79+
80+
float[] copy = new float[this.v.length];
81+
for (int i = 0; i < this.v.length; i++) {
82+
copy[i] = (float) this.v[i];
83+
}
84+
85+
return copy;
86+
}
87+
88+
@Override
89+
public double[] toDoubleArray() {
90+
91+
double[] copy = new double[this.v.length];
92+
System.arraycopy(this.v, 0, copy, 0, copy.length);
93+
94+
return copy;
95+
}
96+
97+
@Override
98+
public boolean equals(Object o) {
99+
100+
if (this == o) {
101+
return true;
102+
}
103+
if (!(o instanceof DoubleVector that)) {
104+
return false;
105+
}
106+
return ObjectUtils.nullSafeEquals(v, that.v);
107+
}
108+
109+
@Override
110+
public int hashCode() {
111+
return Arrays.hashCode(v);
112+
}
113+
114+
@Override
115+
public String toString() {
116+
return "D" + Arrays.toString(v);
117+
}
118+
}

‎src/main/java/org/springframework/data/domain/FloatVector.java

+118
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,118 @@
1+
/*
2+
* Copyright 2024 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.springframework.data.domain;
17+
18+
import java.util.Arrays;
19+
import java.util.Collection;
20+
21+
import org.springframework.util.ObjectUtils;
22+
23+
/**
24+
* {@link Vector} implementation based on {@code float} array.
25+
*
26+
* @author Mark Paluch
27+
* @since 3.5
28+
*/
29+
class FloatVector implements Vector {
30+
31+
private final float[] v;
32+
33+
public FloatVector(float[] v) {
34+
this.v = v;
35+
}
36+
37+
/**
38+
* Copy the given {@code float} array and wrap it within a Vector.
39+
*/
40+
static Vector copy(float[] v) {
41+
42+
float[] copy = new float[v.length];
43+
System.arraycopy(v, 0, copy, 0, copy.length);
44+
45+
return new FloatVector(copy);
46+
}
47+
48+
/**
49+
* Copy the given numeric values and wrap within a Vector.
50+
*/
51+
static Vector copy(Collection<? extends Number> v) {
52+
53+
float[] copy = new float[v.size()];
54+
int i = 0;
55+
for (Number number : v) {
56+
copy[i++] = number.floatValue();
57+
}
58+
59+
return new FloatVector(copy);
60+
}
61+
62+
@Override
63+
public Class<Float> getType() {
64+
return Float.TYPE;
65+
}
66+
67+
@Override
68+
public Object getSource() {
69+
return v;
70+
}
71+
72+
@Override
73+
public int size() {
74+
return v.length;
75+
}
76+
77+
@Override
78+
public float[] toFloatArray() {
79+
80+
float[] copy = new float[this.v.length];
81+
System.arraycopy(this.v, 0, copy, 0, copy.length);
82+
83+
return copy;
84+
}
85+
86+
@Override
87+
public double[] toDoubleArray() {
88+
89+
double[] copy = new double[this.v.length];
90+
for (int i = 0; i < this.v.length; i++) {
91+
copy[i] = this.v[i];
92+
}
93+
94+
return copy;
95+
}
96+
97+
@Override
98+
public boolean equals(Object o) {
99+
100+
if (this == o) {
101+
return true;
102+
}
103+
if (!(o instanceof FloatVector that)) {
104+
return false;
105+
}
106+
return ObjectUtils.nullSafeEquals(v, that.v);
107+
}
108+
109+
@Override
110+
public int hashCode() {
111+
return Arrays.hashCode(v);
112+
}
113+
114+
@Override
115+
public String toString() {
116+
return "F" + Arrays.toString(v);
117+
}
118+
}

‎src/main/java/org/springframework/data/domain/NumberVector.java

+132
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,132 @@
1+
/*
2+
* Copyright 2024 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.springframework.data.domain;
17+
18+
import java.util.Arrays;
19+
import java.util.Collection;
20+
21+
import org.springframework.util.ObjectUtils;
22+
23+
/**
24+
* {@link Vector} implementation based on {@link Number} array.
25+
*
26+
* @author Mark Paluch
27+
* @since 3.5
28+
*/
29+
class NumberVector implements Vector {
30+
31+
private final Number[] v;
32+
33+
public NumberVector(Number[] v) {
34+
this.v = v;
35+
}
36+
37+
/**
38+
* Copy the given {@link Number} array and wrap it within a Vector.
39+
*/
40+
static Vector copy(Number[] v) {
41+
42+
Number[] copy = new Number[v.length];
43+
System.arraycopy(v, 0, copy, 0, copy.length);
44+
45+
return new NumberVector(copy);
46+
}
47+
48+
/**
49+
* Copy the given {@link Number} and wrap it within a Vector.
50+
*/
51+
static Vector copy(Collection<? extends Number> numbers) {
52+
53+
Number[] copy = new Number[numbers.size()];
54+
55+
int i = 0;
56+
for (Number number : numbers) {
57+
copy[i++] = number;
58+
}
59+
60+
return new NumberVector(copy);
61+
}
62+
63+
@Override
64+
public Class<? extends Number> getType() {
65+
66+
Class<?> candidate = null;
67+
for (Object val : v) {
68+
if (val != null) {
69+
if (candidate == null) {
70+
candidate = val.getClass();
71+
} else if (candidate != val.getClass()) {
72+
return Number.class;
73+
}
74+
}
75+
}
76+
return (Class<? extends Number>) candidate;
77+
}
78+
79+
@Override
80+
public Object getSource() {
81+
return v;
82+
}
83+
84+
@Override
85+
public int size() {
86+
return v.length;
87+
}
88+
89+
@Override
90+
public float[] toFloatArray() {
91+
92+
float[] copy = new float[this.v.length];
93+
for (int i = 0; i < this.v.length; i++) {
94+
copy[i] = this.v[i].floatValue();
95+
}
96+
97+
return copy;
98+
}
99+
100+
@Override
101+
public double[] toDoubleArray() {
102+
103+
double[] copy = new double[this.v.length];
104+
for (int i = 0; i < this.v.length; i++) {
105+
copy[i] = this.v[i].doubleValue();
106+
}
107+
108+
return copy;
109+
}
110+
111+
@Override
112+
public boolean equals(Object o) {
113+
114+
if (this == o) {
115+
return true;
116+
}
117+
if (!(o instanceof NumberVector that)) {
118+
return false;
119+
}
120+
return ObjectUtils.nullSafeEquals(v, that.v);
121+
}
122+
123+
@Override
124+
public int hashCode() {
125+
return Arrays.hashCode(v);
126+
}
127+
128+
@Override
129+
public String toString() {
130+
return "N" + Arrays.toString(v);
131+
}
132+
}

0 commit comments

Comments
 (0)