[python] Expose AxisQuery and fast csr to tiledbsoma to use C++ reind…

…exer

python-spec/src/somacore/__init__.py

@@ -24,6 +24,9 @@
 from .options import ResultOrder
 from .query import AxisColumnNames
 from .query import AxisQuery
+from .query import AxisQueryResult
+from .query import CSRAccumulator
+from .query import EagerIterator
 from .query import ExperimentAxisQuery
 
 try:
@@ -54,4 +57,7 @@
     "AxisColumnNames",
     "AxisQuery",
     "ExperimentAxisQuery",
+    "CSRAccumulator",
+    "EagerIterator",
+    "AxisQueryResult",
 )

python-spec/src/somacore/query/__init__.py

@@ -1,12 +1,28 @@
+from . import _eager_iter
+from . import _fast_csr
 from . import axis
 from . import query
 
 ExperimentAxisQuery = query.ExperimentAxisQuery
 AxisColumnNames = query.AxisColumnNames
+AxisQueryResult = query._AxisQueryResult
 AxisQuery = axis.AxisQuery
+# importing CSRAccumulator and EagerIterator to be used in tiledbsome
+# for overwriting theAxisQuery run with C++ re-indexer
+CSRAccumulator = _fast_csr._CSRAccumulator
+EagerIterator = _eager_iter.EagerIterator
+_accum_row_length = _fast_csr._accum_row_length
+_create_scipy_csr_matrix = _fast_csr._create_scipy_csr_matrix
+_select_dtype = _fast_csr._select_dtype
 
 __all__ = (
     "ExperimentAxisQuery",
     "AxisColumnNames",
     "AxisQuery",
+    "CSRAccumulator",
+    "EagerIterator",
+    "AxisQueryResult",
+    "_create_scipy_csr_matrix",
+    "_select_dtype",
+    "_accum_row_length",
 )