Generate trgeometry stbox-restriction operators via hex-WKB serialization

CMakeLists.txt

@@ -239,6 +239,13 @@ endif ()
 
 add_custom_target(build_all_plugins)
 
+# Declared here so add_compile_definitions reaches all sibling nes-* targets.
+# nes-plugins/CMakeLists.txt re-declares the same option (no-op when cached).
+option(NES_ENABLE_MEOS "Enable MEOS plugin (requires libmeos installed on the system)" ON)
+if(NES_ENABLE_MEOS)
+    add_compile_definitions(NES_ENABLE_MEOS)
+endif()
+
 # Add target for common lib, which contains a minimal set
 # of shared functionality used by all components of nes
 file(GLOB NES_DIRECTORIES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "nes-*")

flake.nix

@@ -54,6 +54,8 @@
           tbb
           python3
           openjdk21
+          paho-mqtt-c
+          paho-mqtt-cpp
         ]) ++ [ follyPkg antlr4Pkg ];
 
         antlr4Jar = pkgs.fetchurl {
@@ -244,6 +246,7 @@
             "-DNES_ENABLES_TESTS=ON"
             "-DCMAKE_MODULE_PATH=${libdwarfModule}/share/cmake/Modules"
             "-DANTLR4_JAR_LOCATION=${antlr4Jar}"
+            "-DNES_ENABLE_MEOS=OFF"
           ];
 
           enableParallelBuilding = true;
@@ -347,6 +350,7 @@
               "-DLLVM_DIR=${commonCmakeEnv.LLVM_DIR}"
               "-DANTLR4_JAR_LOCATION=${antlr4Jar}"
               "-DCMAKE_MODULE_PATH=${libdwarfModule}/share/cmake/Modules"
+              "-DNES_ENABLE_MEOS=OFF"
             ];
             shellHook = ''
               unset NES_PREBUILT_VCPKG_ROOT

nes-logical-operators/include/Functions/Meos/TrgeometryAtStboxLogicalFunction.hpp

@@ -0,0 +1,54 @@
+/*
+    Licensed under the Apache License, Version 2.0 (the "License");
+    you may not use this file except in compliance with the License.
+    You may obtain a copy of the License at
+
+        https://www.apache.org/licenses/LICENSE-2.0
+
+    Unless required by applicable law or agreed to in writing, software
+    distributed under the License is distributed on an "AS IS" BASIS,
+    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    See the License for the specific language governing permissions and
+    limitations under the License.
+*/
+
+#pragma once
+
+#include <Functions/LogicalFunction.hpp>
+#include <string_view>
+#include <utility>
+#include <vector>
+
+namespace NES {
+
+/**
+ * @brief Per-event trgeometry_at_stbox: a hex-WKB temporal, a static STBox, and a border flag -> hex-WKB temporal.
+ *
+ * Generated by tools/codegen/codegen_nebula.py from the MEOS function
+ * `trgeometry_at_stbox`. Per-event scalar operator following the
+ * TemporalEDWithinGeometry pattern.
+ */
+class TrgeometryAtStboxLogicalFunction : public LogicalFunctionConcept {
+public:
+    static constexpr std::string_view NAME = "TrgeometryAtStbox";
+
+    TrgeometryAtStboxLogicalFunction(LogicalFunction traj,
+                                          LogicalFunction arg0,
+                                          LogicalFunction arg1);
+
+    DataType getDataType() const override;
+    LogicalFunction withDataType(const DataType& dataType) const override;
+    std::vector<LogicalFunction> getChildren() const override;
+    LogicalFunction withChildren(const std::vector<LogicalFunction>& children) const override;
+    std::string_view getType() const override;
+    bool operator==(const LogicalFunctionConcept& rhs) const override;
+    std::string explain(ExplainVerbosity verbosity) const override;
+    LogicalFunction withInferredDataType(const Schema& schema) const override;
+    SerializableFunction serialize() const override;
+
+private:
+    DataType dataType;
+    std::vector<LogicalFunction> parameters;
+};
+
+} // namespace NES

nes-logical-operators/include/Functions/Meos/TrgeometryMinusStboxLogicalFunction.hpp

@@ -0,0 +1,54 @@
+/*
+    Licensed under the Apache License, Version 2.0 (the "License");
+    you may not use this file except in compliance with the License.
+    You may obtain a copy of the License at
+
+        https://www.apache.org/licenses/LICENSE-2.0
+
+    Unless required by applicable law or agreed to in writing, software
+    distributed under the License is distributed on an "AS IS" BASIS,
+    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    See the License for the specific language governing permissions and
+    limitations under the License.
+*/
+
+#pragma once
+
+#include <Functions/LogicalFunction.hpp>
+#include <string_view>
+#include <utility>
+#include <vector>
+
+namespace NES {
+
+/**
+ * @brief Per-event trgeometry_minus_stbox: a hex-WKB temporal, a static STBox, and a border flag -> hex-WKB temporal.
+ *
+ * Generated by tools/codegen/codegen_nebula.py from the MEOS function
+ * `trgeometry_minus_stbox`. Per-event scalar operator following the
+ * TemporalEDWithinGeometry pattern.
+ */
+class TrgeometryMinusStboxLogicalFunction : public LogicalFunctionConcept {
+public:
+    static constexpr std::string_view NAME = "TrgeometryMinusStbox";
+
+    TrgeometryMinusStboxLogicalFunction(LogicalFunction traj,
+                                          LogicalFunction arg0,
+                                          LogicalFunction arg1);
+
+    DataType getDataType() const override;
+    LogicalFunction withDataType(const DataType& dataType) const override;
+    std::vector<LogicalFunction> getChildren() const override;
+    LogicalFunction withChildren(const std::vector<LogicalFunction>& children) const override;
+    std::string_view getType() const override;
+    bool operator==(const LogicalFunctionConcept& rhs) const override;
+    std::string explain(ExplainVerbosity verbosity) const override;
+    LogicalFunction withInferredDataType(const Schema& schema) const override;
+    SerializableFunction serialize() const override;
+
+private:
+    DataType dataType;
+    std::vector<LogicalFunction> parameters;
+};
+
+} // namespace NES

nes-logical-operators/src/Functions/Meos/CMakeLists.txt

@@ -18,3 +18,5 @@ add_plugin(TemporalAIntersectsGeometry LogicalFunction nes-logical-operators Tem
 add_plugin(TemporalEContainsGeometry LogicalFunction nes-logical-operators TemporalEContainsGeometryLogicalFunction.cpp)
 add_plugin(TemporalEDWithinGeometry LogicalFunction nes-logical-operators TemporalEDWithinGeometryLogicalFunction.cpp)
 add_plugin(TemporalAtStBox LogicalFunction nes-logical-operators TemporalAtStBoxLogicalFunction.cpp)
+add_plugin(TrgeometryAtStbox LogicalFunction nes-logical-operators TrgeometryAtStboxLogicalFunction.cpp)
+add_plugin(TrgeometryMinusStbox LogicalFunction nes-logical-operators TrgeometryMinusStboxLogicalFunction.cpp)

nes-logical-operators/src/Functions/Meos/TrgeometryAtStboxLogicalFunction.cpp

@@ -0,0 +1,128 @@
+/*
+    Licensed under the Apache License, Version 2.0 (the "License");
+    you may not use this file except in compliance with the License.
+    You may obtain a copy of the License at
+
+        https://www.apache.org/licenses/LICENSE-2.0
+
+    Unless required by applicable law or agreed to in writing, software
+    distributed under the License is distributed on an "AS IS" BASIS,
+    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    See the License for the specific language governing permissions and
+    limitations under the License.
+*/
+
+#include <Functions/Meos/TrgeometryAtStboxLogicalFunction.hpp>
+
+#include <DataTypes/DataType.hpp>
+#include <DataTypes/DataTypeProvider.hpp>
+#include <DataTypes/Schema.hpp>
+#include <ErrorHandling.hpp>
+#include <LogicalFunctionRegistry.hpp>
+#include <Serialization/DataTypeSerializationUtil.hpp>
+#include <fmt/format.h>
+#include <SerializableVariantDescriptor.pb.h>
+
+namespace NES
+{
+
+TrgeometryAtStboxLogicalFunction::TrgeometryAtStboxLogicalFunction(LogicalFunction traj,
+                                          LogicalFunction arg0,
+                                          LogicalFunction arg1)
+    : dataType(DataTypeProvider::provideDataType(DataType::Type::VARSIZED))
+{
+    parameters.reserve(3);
+    parameters.push_back(std::move(traj));
+    parameters.push_back(std::move(arg0));
+    parameters.push_back(std::move(arg1));
+}
+
+DataType TrgeometryAtStboxLogicalFunction::getDataType() const
+{
+    return dataType;
+}
+
+LogicalFunction TrgeometryAtStboxLogicalFunction::withDataType(const DataType& newDataType) const
+{
+    auto copy = *this;
+    copy.dataType = newDataType;
+    return copy;
+}
+
+std::vector<LogicalFunction> TrgeometryAtStboxLogicalFunction::getChildren() const
+{
+    return parameters;
+}
+
+LogicalFunction TrgeometryAtStboxLogicalFunction::withChildren(const std::vector<LogicalFunction>& children) const
+{
+    PRECONDITION(children.size() == 3, "TrgeometryAtStboxLogicalFunction requires 3 children, but got {}", children.size());
+    auto copy = *this;
+    copy.parameters = children;
+    return copy;
+}
+
+std::string_view TrgeometryAtStboxLogicalFunction::getType() const
+{
+    return NAME;
+}
+
+bool TrgeometryAtStboxLogicalFunction::operator==(const LogicalFunctionConcept& rhs) const
+{
+    if (const auto* other = dynamic_cast<const TrgeometryAtStboxLogicalFunction*>(&rhs))
+    {
+        return parameters == other->parameters;
+    }
+    return false;
+}
+
+std::string TrgeometryAtStboxLogicalFunction::explain(ExplainVerbosity verbosity) const
+{
+    std::string args;
+    for (size_t index = 0; index < parameters.size(); ++index)
+    {
+        if (index > 0)
+        {
+            args += ", ";
+        }
+        args += parameters[index].explain(verbosity);
+    }
+    return fmt::format("{}({})", NAME, args);
+}
+
+LogicalFunction TrgeometryAtStboxLogicalFunction::withInferredDataType(const Schema& schema) const
+{
+    std::vector<LogicalFunction> newChildren;
+    newChildren.reserve(parameters.size());
+    for (const auto& child : parameters)
+    {
+        newChildren.emplace_back(child.withInferredDataType(schema));
+    }
+    return withChildren(newChildren);
+}
+
+SerializableFunction TrgeometryAtStboxLogicalFunction::serialize() const
+{
+    SerializableFunction proto;
+    proto.set_function_type(std::string(NAME));
+    DataTypeSerializationUtil::serializeDataType(dataType, proto.mutable_data_type());
+    for (const auto& child : parameters)
+    {
+        proto.add_children()->CopyFrom(child.serialize());
+    }
+    return proto;
+}
+
+LogicalFunctionRegistryReturnType LogicalFunctionGeneratedRegistrar::RegisterTrgeometryAtStboxLogicalFunction(
+    LogicalFunctionRegistryArguments arguments)
+{
+    PRECONDITION(arguments.children.size() == 3,
+                 "TrgeometryAtStboxLogicalFunction requires 3 children but got {}",
+                 arguments.children.size());
+    auto arg0 = std::move(arguments.children[0]);
+    auto arg1 = std::move(arguments.children[1]);
+    auto arg2 = std::move(arguments.children[2]);
+    return TrgeometryAtStboxLogicalFunction(std::move(arg0), std::move(arg1), std::move(arg2));
+}
+
+} // namespace NES