Skip to content

Generate the constructors of the temporal cell indexes #496

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
estebanzimanyi merged 1 commit into from
Oct 24, 2026
Merged

Generate the constructors of the temporal cell indexes #496

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions src/generated/generated_temporal_udfs.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -122,6 +122,7 @@ void RegisterGenerated_meos_h3_base_metrics(ExtensionLoader &loader);
void RegisterGenerated_meos_h3_comp(ExtensionLoader &loader);
void RegisterGenerated_meos_h3_comp_ever(ExtensionLoader &loader);
void RegisterGenerated_meos_h3_comp_temp(ExtensionLoader &loader);
void RegisterGenerated_meos_h3_constructor(ExtensionLoader &loader);
void RegisterGenerated_meos_h3_conversion(ExtensionLoader &loader);
void RegisterGenerated_meos_h3_edges(ExtensionLoader &loader);
void RegisterGenerated_meos_h3_hierarchy(ExtensionLoader &loader);
Expand Down Expand Up @@ -230,6 +231,7 @@ void RegisterGenerated_meos_quadbin_base_comp(ExtensionLoader &loader);
void RegisterGenerated_meos_quadbin_base_inout(ExtensionLoader &loader);
void RegisterGenerated_meos_quadbin_comp_ever(ExtensionLoader &loader);
void RegisterGenerated_meos_quadbin_comp_temp(ExtensionLoader &loader);
void RegisterGenerated_meos_quadbin_constructor(ExtensionLoader &loader);
void RegisterGenerated_meos_quadbin_conversion(ExtensionLoader &loader);
void RegisterGenerated_meos_quadbin_inout(ExtensionLoader &loader);
void RegisterGenerated_meos_quadbin_restrict(ExtensionLoader &loader);
Expand Down Expand Up @@ -265,6 +267,7 @@ void RegisterGenerated_meos_s2cell_base_comp(ExtensionLoader &loader);
void RegisterGenerated_meos_s2cell_base_inout(ExtensionLoader &loader);
void RegisterGenerated_meos_s2cell_comp_ever(ExtensionLoader &loader);
void RegisterGenerated_meos_s2cell_comp_temp(ExtensionLoader &loader);
void RegisterGenerated_meos_s2cell_constructor(ExtensionLoader &loader);
void RegisterGenerated_meos_s2cell_conversion(ExtensionLoader &loader);
void RegisterGenerated_meos_s2cell_inout(ExtensionLoader &loader);
void RegisterGenerated_meos_s2cell_restrict(ExtensionLoader &loader);
Expand Down Expand Up @@ -385,6 +388,7 @@ void RegisterGeneratedTemporalUdfs(ExtensionLoader &loader) {
RegisterGenerated_meos_h3_comp(loader);
RegisterGenerated_meos_h3_comp_ever(loader);
RegisterGenerated_meos_h3_comp_temp(loader);
RegisterGenerated_meos_h3_constructor(loader);
RegisterGenerated_meos_h3_conversion(loader);
RegisterGenerated_meos_h3_edges(loader);
RegisterGenerated_meos_h3_hierarchy(loader);
Expand Down Expand Up @@ -493,6 +497,7 @@ void RegisterGeneratedTemporalUdfs(ExtensionLoader &loader) {
RegisterGenerated_meos_quadbin_base_inout(loader);
RegisterGenerated_meos_quadbin_comp_ever(loader);
RegisterGenerated_meos_quadbin_comp_temp(loader);
RegisterGenerated_meos_quadbin_constructor(loader);
RegisterGenerated_meos_quadbin_conversion(loader);
RegisterGenerated_meos_quadbin_inout(loader);
RegisterGenerated_meos_quadbin_restrict(loader);
Expand Down Expand Up @@ -528,6 +533,7 @@ void RegisterGeneratedTemporalUdfs(ExtensionLoader &loader) {
RegisterGenerated_meos_s2cell_base_inout(loader);
RegisterGenerated_meos_s2cell_comp_ever(loader);
RegisterGenerated_meos_s2cell_comp_temp(loader);
RegisterGenerated_meos_s2cell_constructor(loader);
RegisterGenerated_meos_s2cell_conversion(loader);
RegisterGenerated_meos_s2cell_inout(loader);
RegisterGenerated_meos_s2cell_restrict(loader);
Expand Down
87 changes: 87 additions & 0 deletions src/generated/generated_temporal_udfs_14.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2856,6 +2856,86 @@ static void Gen_tne_ts2cell_s2cell(DataChunk &args, ExpressionState &, Vector &r
}


// ===== @ingroup meos_s2cell_constructor =====
static void Gen_ts2cellinst_make(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<timestamp_tz_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
Temporal *r = (Temporal *) ts2cellinst_make(static_cast<uint64_t>(in0[i]), DuckDBToMeosTimestamp(in1[i]).value);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}

static void Gen_ts2cellseq_from_base_tstzset(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<string_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
Set *p1 = BlobToSet(in1[i]);
Temporal *r = (Temporal *) ts2cellseq_from_base_tstzset(static_cast<uint64_t>(in0[i]), p1);
free(p1);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}

static void Gen_ts2cellseq_from_base_tstzspan(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<string_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
Span *p1 = BlobToSpan(in1[i]);
Temporal *r = (Temporal *) ts2cellseq_from_base_tstzspan(static_cast<uint64_t>(in0[i]), p1);
free(p1);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}

static void Gen_ts2cellseqset_from_base_tstzspanset(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<string_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
SpanSet *p1 = BlobToSpanSet(in1[i]);
Temporal *r = (Temporal *) ts2cellseqset_from_base_tstzspanset(static_cast<uint64_t>(in0[i]), p1);
free(p1);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}


// ===== @ingroup meos_s2cell_conversion =====
static void Gen_bigint_to_s2cell(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
Expand Down Expand Up @@ -3163,6 +3243,13 @@ void RegisterGenerated_meos_s2cell_comp_temp(ExtensionLoader &loader) {
RegisterSerializedScalarFunction(loader, ScalarFunction("tNotEqual", {S2cellTypes::ts2cell(), S2cellTypes::s2cell()}, TemporalTypes::tbool(), Gen_tne_ts2cell_s2cell));
}

void RegisterGenerated_meos_s2cell_constructor(ExtensionLoader &loader) {
RegisterSerializedScalarFunction(loader, ScalarFunction("ts2cell", {S2cellTypes::s2cell(), LogicalType::TIMESTAMP_TZ}, S2cellTypes::ts2cell(), Gen_ts2cellinst_make));
RegisterSerializedScalarFunction(loader, ScalarFunction("ts2cell", {S2cellTypes::s2cell(), SetTypes::tstzset()}, S2cellTypes::ts2cell(), Gen_ts2cellseq_from_base_tstzset));
RegisterSerializedScalarFunction(loader, ScalarFunction("ts2cell", {S2cellTypes::s2cell(), SpanTypes::tstzspan()}, S2cellTypes::ts2cell(), Gen_ts2cellseq_from_base_tstzspan));
RegisterSerializedScalarFunction(loader, ScalarFunction("ts2cell", {S2cellTypes::s2cell(), SpansetTypes::tstzspanset()}, S2cellTypes::ts2cell(), Gen_ts2cellseqset_from_base_tstzspanset));
}

void RegisterGenerated_meos_s2cell_conversion(ExtensionLoader &loader) {
RegisterSerializedScalarFunction(loader, ScalarFunction("s2cell", {LogicalType::BIGINT}, S2cellTypes::s2cell(), Gen_bigint_to_s2cell));
RegisterSerializedScalarFunction(loader, ScalarFunction("s2cell_recv", {LogicalType::BIGINT}, S2cellTypes::s2cell(), Gen_bigint_to_s2cell));
Expand Down
87 changes: 87 additions & 0 deletions src/generated/generated_temporal_udfs_2.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2072,6 +2072,86 @@ static void Gen_right_tspatial_stbox(DataChunk &args, ExpressionState &, Vector
}


// ===== @ingroup meos_h3_constructor =====
static void Gen_th3indexinst_make(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<timestamp_tz_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
Temporal *r = (Temporal *) th3indexinst_make(static_cast<uint64_t>(in0[i]), DuckDBToMeosTimestamp(in1[i]).value);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}

static void Gen_th3indexseq_from_base_tstzset(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<string_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
Set *p1 = BlobToSet(in1[i]);
Temporal *r = (Temporal *) th3indexseq_from_base_tstzset(static_cast<uint64_t>(in0[i]), p1);
free(p1);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}

static void Gen_th3indexseq_from_base_tstzspan(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<string_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
Span *p1 = BlobToSpan(in1[i]);
Temporal *r = (Temporal *) th3indexseq_from_base_tstzspan(static_cast<uint64_t>(in0[i]), p1);
free(p1);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}

static void Gen_th3indexseqset_from_base_tstzspanset(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
const idx_t rows = args.size();
for (idx_t c = 0; c < 2; c++) args.data[c].Flatten(rows);
auto in0 = FlatVector::GetData<int64_t>(args.data[0]);
auto in1 = FlatVector::GetData<string_t>(args.data[1]);
auto out = FlatVector::GetData<string_t>(result);
auto &outv = FlatVector::Validity(result);
for (idx_t i = 0; i < rows; i++) {
if (!FlatVector::Validity(args.data[0]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
if (!FlatVector::Validity(args.data[1]).RowIsValid(i)) { outv.SetInvalid(i); continue; }
SpanSet *p1 = BlobToSpanSet(in1[i]);
Temporal *r = (Temporal *) th3indexseqset_from_base_tstzspanset(static_cast<uint64_t>(in0[i]), p1);
free(p1);
if (!r) { outv.SetInvalid(i); continue; }
out[i] = TemporalToBlob(result, r);
}
if (rows == 1) result.SetVectorType(VectorType::CONSTANT_VECTOR);
}


// ===== @ingroup meos_h3_set_setops =====
static void Gen_contained_h3index_set(DataChunk &args, ExpressionState &, Vector &result) {
EnsureMeosThreadInitialized();
Expand Down Expand Up @@ -5551,6 +5631,13 @@ void RegisterGenerated_meos_geo_bbox_pos(ExtensionLoader &loader) {
RegisterSerializedScalarFunction(loader, ScalarFunction(">>", {PosechainTypes::tposechain(), StboxType::stbox()}, LogicalType::BOOLEAN, Gen_right_tspatial_stbox));
}

void RegisterGenerated_meos_h3_constructor(ExtensionLoader &loader) {
RegisterSerializedScalarFunction(loader, ScalarFunction("th3index", {H3indexTypes::h3index(), LogicalType::TIMESTAMP_TZ}, H3indexTypes::th3index(), Gen_th3indexinst_make));
RegisterSerializedScalarFunction(loader, ScalarFunction("th3index", {H3indexTypes::h3index(), SetTypes::tstzset()}, H3indexTypes::th3index(), Gen_th3indexseq_from_base_tstzset));
RegisterSerializedScalarFunction(loader, ScalarFunction("th3index", {H3indexTypes::h3index(), SpanTypes::tstzspan()}, H3indexTypes::th3index(), Gen_th3indexseq_from_base_tstzspan));
RegisterSerializedScalarFunction(loader, ScalarFunction("th3index", {H3indexTypes::h3index(), SpansetTypes::tstzspanset()}, H3indexTypes::th3index(), Gen_th3indexseqset_from_base_tstzspanset));
}

void RegisterGenerated_meos_h3_set_setops(ExtensionLoader &loader) {
RegisterSerializedScalarFunction(loader, ScalarFunction("setContained", {H3indexTypes::h3index(), SetTypes::h3indexset()}, LogicalType::BOOLEAN, Gen_contained_h3index_set));
RegisterSerializedScalarFunction(loader, ScalarFunction("<@", {H3indexTypes::h3index(), SetTypes::h3indexset()}, LogicalType::BOOLEAN, Gen_contained_h3index_set));
Expand Down
Loading
Loading