-
Notifications
You must be signed in to change notification settings - Fork 7
/
array_to_min.c
147 lines (124 loc) · 4.04 KB
/
array_to_min.c
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
Datum array_to_min(PG_FUNCTION_ARGS);
PG_FUNCTION_INFO_V1(array_to_min);
/**
* Returns a min from an (unsorted) array of numbers.
* by Paul A. Jungwirth
*/
Datum
array_to_min(PG_FUNCTION_ARGS)
{
// Our arguments:
ArrayType *vals;
// The array element type:
Oid valsType;
// The array element type widths for our input array:
int16 valsTypeWidth;
// The array element type "is passed by value" flags (not really used):
bool valsTypeByValue;
// The array element type alignment codes (not really used):
char valsTypeAlignmentCode;
// The array contents, as PostgreSQL "Datum" objects:
Datum *valsContent;
// List of "is null" flags for the array contents:
bool *valsNullFlags;
// The size of the input array:
int valsLength;
bool resultIsNull = true;
int i;
pgnum v;
if (PG_ARGISNULL(0)) {
ereport(ERROR, (errmsg("Null arrays not accepted")));
}
vals = PG_GETARG_ARRAYTYPE_P(0);
if (ARR_NDIM(vals) == 0) {
PG_RETURN_NULL();
}
if (ARR_NDIM(vals) > 1) {
ereport(ERROR, (errmsg("One-dimesional arrays are required")));
}
// Determine the array element types.
valsType = ARR_ELEMTYPE(vals);
if (valsType != INT2OID &&
valsType != INT4OID &&
valsType != INT8OID &&
valsType != FLOAT4OID &&
valsType != FLOAT8OID) {
ereport(ERROR, (errmsg("Min subject must be SMALLINT, INTEGER, BIGINT, REAL, or DOUBLE PRECISION values")));
}
valsLength = (ARR_DIMS(vals))[0];
get_typlenbyvalalign(valsType, &valsTypeWidth, &valsTypeByValue, &valsTypeAlignmentCode);
// Extract the array contents (as Datum objects).
deconstruct_array(vals, valsType, valsTypeWidth, valsTypeByValue, valsTypeAlignmentCode,
&valsContent, &valsNullFlags, &valsLength);
if (valsLength == 0) PG_RETURN_NULL();
// Compute the min.
switch (valsType) {
case INT2OID:
for (i = 0; i < valsLength; i++) {
if (valsNullFlags[i]) {
continue;
} else if (resultIsNull) {
v.i16 = DatumGetInt16(valsContent[i]);
resultIsNull = false;
} else if (DatumGetInt16(valsContent[i]) < v.i16) {
v.i16 = DatumGetInt16(valsContent[i]);
}
}
if (resultIsNull) PG_RETURN_NULL();
else PG_RETURN_INT16(v.i16);
case INT4OID:
for (i = 0; i < valsLength; i++) {
if (valsNullFlags[i]) {
continue;
} else if (resultIsNull) {
v.i32 = DatumGetInt32(valsContent[i]);
resultIsNull = false;
} else if (DatumGetInt32(valsContent[i]) < v.i32) {
v.i32 = DatumGetInt32(valsContent[i]);
}
}
if (resultIsNull) PG_RETURN_NULL();
else PG_RETURN_INT32(v.i32);
case INT8OID:
for (i = 0; i < valsLength; i++) {
if (valsNullFlags[i]) {
continue;
} else if (resultIsNull) {
v.i64 = DatumGetInt64(valsContent[i]);
resultIsNull = false;
} else if (DatumGetInt64(valsContent[i]) < v.i64) {
v.i64 = DatumGetInt64(valsContent[i]);
}
}
if (resultIsNull) PG_RETURN_NULL();
else PG_RETURN_INT64(v.i64);
case FLOAT4OID:
for (i = 0; i < valsLength; i++) {
if (valsNullFlags[i]) {
continue;
} else if (resultIsNull) {
v.f4 = DatumGetFloat4(valsContent[i]);
resultIsNull = false;
} else if (DatumGetFloat4(valsContent[i]) < v.f4) {
v.f4 = DatumGetFloat4(valsContent[i]);
}
}
if (resultIsNull) PG_RETURN_NULL();
else PG_RETURN_FLOAT4(v.f4);
case FLOAT8OID:
for (i = 0; i < valsLength; i++) {
if (valsNullFlags[i]) {
continue;
} else if (resultIsNull) {
v.f8 = DatumGetFloat8(valsContent[i]);
resultIsNull = false;
} else if (DatumGetFloat8(valsContent[i]) < v.f8) {
v.f8 = DatumGetFloat8(valsContent[i]);
}
}
if (resultIsNull) PG_RETURN_NULL();
else PG_RETURN_FLOAT8(v.f8);
default:
ereport(ERROR, (errmsg("Min subject must be SMALLINT, INTEGER, BIGINT, REAL, or DOUBLE PRECISION values")));
}
}