Add element_size function for Dataset elements

include/svs/core/data/simple.h

@@ -292,6 +292,9 @@ class SimpleData {
     /// Return the number of dimensions for each entry in the dataset.
     size_t dimensions() const { return getsize<1>(data_); }
 
+    /// Return The size in bytes of one vector: sizeof(element_type) * dimensions()
+    size_t element_size() const { return sizeof(element_type) * dimensions(); }
+
     ///
     /// @brief Return a constant handle to vector stored as position ``i``.
     ///
@@ -720,6 +723,8 @@ class SimpleData<T, Extent, Blocked<Alloc>> {
         }
     }
 
+    size_t element_size() const { return sizeof(element_type) * dimensions(); }
+
     const_value_type get_datum(size_t i) const {
         auto [block_id, data_id] = resolve(i);
         return getindex(blocks_, block_id).slice(data_id);

include/svs/quantization/scalar/scalar.h

@@ -390,6 +390,7 @@ class SQDataset {
 
     size_t size() const { return data_.size(); }
     size_t dimensions() const { return data_.dimensions(); }
+    size_t element_size() const { return sizeof(element_type) * dimensions(); }
 
     float get_scale() const { return scale_; }
     float get_bias() const { return bias_; }

tests/svs/core/data.cpp

@@ -88,3 +88,21 @@ CATCH_TEST_CASE("Data Loading/Saving", "[core][data]") {
         CATCH_REQUIRE(w == z);
     }
 }
+
+CATCH_TEST_CASE("Element Size", "[core][data]") {
+    CATCH_SECTION("Check element_size()") {
+        // Test with float, dynamic dimensions
+        auto float_data = svs::data::SimpleData<float, svs::Dynamic>(5, 10);
+        CATCH_REQUIRE(float_data.element_size() == sizeof(float) * 10);
+
+        // Test fixed dimensions with blocked storage
+        auto blocked_fixed = svs::data::BlockedData<int32_t, 64>(25, 64);
+        CATCH_REQUIRE(blocked_fixed.element_size() == sizeof(int32_t) * 64);
+
+        // Test element_size consistency across different instances
+        auto data1 = svs::data::SimpleData<float, svs::Dynamic>(10, 20);
+        auto data2 =
+            svs::data::SimpleData<float, svs::Dynamic>(50, 20); // Different size, same dims
+        CATCH_REQUIRE(data1.element_size() == data2.element_size());
+    }
+}