liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-06-06 05:51:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added casting to int[] [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-10-13 18:55:12 -07:00
parent 4765df1c50
commit 6221779f74
5 changed files with 44 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sql/vector--0.7.4--0.8.0.sql
View File

@@ -81,9 +81,15 @@ CREATE FUNCTION intvec(intvec, integer, boolean) RETURNS intvec
CREATE FUNCTION array_to_intvec(integer[], integer, boolean) RETURNS intvec
AS 'MODULE_PATHNAME' LANGUAGE C IMMUTABLE STRICT PARALLEL SAFE;
CREATE FUNCTION intvec_to_int(intvec, integer, boolean) RETURNS int[]
AS 'MODULE_PATHNAME' LANGUAGE C IMMUTABLE STRICT PARALLEL SAFE;
CREATE CAST (intvec AS intvec)
WITH FUNCTION intvec(intvec, integer, boolean) AS IMPLICIT;
CREATE CAST (intvec AS int[])
WITH FUNCTION intvec_to_int(intvec, integer, boolean) AS ASSIGNMENT;
CREATE CAST (integer[] AS intvec)
WITH FUNCTION array_to_intvec(integer[], integer, boolean) AS ASSIGNMENT;

6
sql/vector.sql
View File

@@ -735,11 +735,17 @@ CREATE FUNCTION intvec(intvec, integer, boolean) RETURNS intvec
CREATE FUNCTION array_to_intvec(integer[], integer, boolean) RETURNS intvec
AS 'MODULE_PATHNAME' LANGUAGE C IMMUTABLE STRICT PARALLEL SAFE;
CREATE FUNCTION intvec_to_int(intvec, integer, boolean) RETURNS int[]
AS 'MODULE_PATHNAME' LANGUAGE C IMMUTABLE STRICT PARALLEL SAFE;
-- intvec casts
CREATE CAST (intvec AS intvec)
WITH FUNCTION intvec(intvec, integer, boolean) AS IMPLICIT;
CREATE CAST (intvec AS int[])
WITH FUNCTION intvec_to_int(intvec, integer, boolean) AS ASSIGNMENT;
CREATE CAST (integer[] AS intvec)
WITH FUNCTION array_to_intvec(integer[], integer, boolean) AS ASSIGNMENT;

25
src/intvec.c
View File

@@ -415,6 +415,29 @@ array_to_intvec(PG_FUNCTION_ARGS)
PG_RETURN_POINTER(result);
}
/*
* Convert int vector to int[]
*/
FUNCTION_PREFIX PG_FUNCTION_INFO_V1(intvec_to_int);
Datum
intvec_to_int(PG_FUNCTION_ARGS)
{
IntVector *vec = PG_GETARG_INTVEC_P(0);
Datum *datums;
ArrayType *result;
datums = (Datum *) palloc(sizeof(Datum) * vec->dim);
for (int i = 0; i < vec->dim; i++)
datums[i] = Int32GetDatum((int) vec->x[i]);
result = construct_array(datums, vec->dim, INT4OID, sizeof(int32), true, TYPALIGN_INT);
pfree(datums);
PG_RETURN_POINTER(result);
}
/*
* Get the L2 distance between int vectors
*/
@@ -587,7 +610,7 @@ FUNCTION_PREFIX PG_FUNCTION_INFO_V1(intvec_vector_dims);
Datum
intvec_vector_dims(PG_FUNCTION_ARGS)
{
IntVector *a = PG_GETARG_INTVEC_P(0);
IntVector *a = PG_GETARG_INTVEC_P(0);
PG_RETURN_INT32(a->dim);
}

6
test/expected/cast.out
View File

@@ -140,6 +140,12 @@ SELECT '{1e-8,-1e-8}'::real[]::halfvec;
[0,-0]
(1 row)
SELECT '[1,2,3]'::intvec::int[];
int4
---------
{1,2,3}
(1 row)
SELECT '{1,2,3}'::int[]::intvec;
intvec
---------

2
test/sql/cast.sql
View File

@@ -38,6 +38,8 @@ SELECT '{1,2,3}'::real[]::halfvec(2);
SELECT '{65520,-65520}'::real[]::halfvec;
SELECT '{1e-8,-1e-8}'::real[]::halfvec;
SELECT '[1,2,3]'::intvec::int[];
SELECT '{1,2,3}'::int[]::intvec;
SELECT '{1,2,3}'::int[]::intvec(3);
SELECT '{1,2,3}'::int[]::intvec(2);