Swap shared_ptr with const reference, hide internal columns when open…

…ing an array for read unless explicity specify them
single-cell-data · Nov 1, 2024 · b77de49 · b77de49
1 parent fbac212
commit b77de49
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 9 deletions.
diff --git a/libtiledbsoma/src/soma/soma_array.cc b/libtiledbsoma/src/soma/soma_array.cc
@@ -250,6 +250,25 @@ void SOMAArray::reset(
 
     if (!column_names.empty()) {
         mq_->select_columns(column_names);
+    } else {
+        // Hide internal columns if any
+        std::vector<std::string> columns;
+        for (const auto& dim : this->tiledb_schema()->domain().dimensions()) {
+            columns.push_back(dim.name());
+        }
+
+        for (size_t i = 0; i < this->tiledb_schema()->attribute_num(); ++i) {
+            columns.push_back(this->tiledb_schema()->attribute(i).name());
+        }
+        auto is_internal = [](std::string name) {
+            return name.rfind(SOMA_GEOMETRY_DIMENSION_PREFIX, 0) == 0;
+        };
+
+        auto internal_end = std::remove_if(
+            columns.begin(), columns.end(), is_internal);
+        columns.erase(internal_end, columns.end());
+
+        mq_->select_columns(columns);
     }
 
     switch (result_order) {

diff --git a/libtiledbsoma/src/utils/arrow_adapter.cc b/libtiledbsoma/src/utils/arrow_adapter.cc
@@ -392,7 +392,7 @@ std::unique_ptr<ArrowSchema> ArrowAdapter::arrow_schema_from_tiledb_array(
                 } else if constexpr (std::is_same_v<T, Attribute>) {
                     child = arrow_schema->children[i] =
                         arrow_schema_from_tiledb_attribute(
-                            arg, ctx, tiledb_array)
+                            arg, *ctx, *tiledb_array)
                             .release();
                 }
             },
@@ -425,9 +425,7 @@ std::unique_ptr<ArrowSchema> ArrowAdapter::arrow_schema_from_tiledb_dimension(
 }
 
 std::unique_ptr<ArrowSchema> ArrowAdapter::arrow_schema_from_tiledb_attribute(
-    Attribute& attribute,
-    std::shared_ptr<Context> ctx,
-    std::shared_ptr<Array> tiledb_array) {
+    Attribute& attribute, const Context& ctx, const Array& tiledb_array) {
     std::unique_ptr<ArrowSchema> arrow_schema = std::make_unique<ArrowSchema>();
     arrow_schema->format = strdup(
         ArrowAdapter::to_arrow_format(attribute.type()).data());
@@ -454,10 +452,10 @@ std::unique_ptr<ArrowSchema> ArrowAdapter::arrow_schema_from_tiledb_attribute(
         arrow_schema->name));
 
     auto enmr_name = AttributeExperimental::get_enumeration_name(
-        *ctx, attribute);
+        ctx, attribute);
     if (enmr_name.has_value()) {
         auto enmr = ArrayExperimental::get_enumeration(
-            *ctx, *tiledb_array, attribute.name());
+            ctx, tiledb_array, attribute.name());
         auto dict = (ArrowSchema*)malloc(sizeof(ArrowSchema));
         dict->format = strdup(
             ArrowAdapter::to_arrow_format(enmr.type(), false).data());

diff --git a/libtiledbsoma/src/utils/arrow_adapter.h b/libtiledbsoma/src/utils/arrow_adapter.h
@@ -243,9 +243,7 @@ class ArrowAdapter {
      * @return ArrowSchema
      */
     static std::unique_ptr<ArrowSchema> arrow_schema_from_tiledb_attribute(
-        Attribute& attribute,
-        std::shared_ptr<Context> ctx,
-        std::shared_ptr<Array> tiledb_array);
+        Attribute& attribute, const Context& ctx, const Array& tiledb_array);
 
     /**
      * @brief Get members of the TileDB Schema in the form of a PlatformConfig