Move compressed_sparse_row.hpp to extension

cwida · Mar 26, 2024 · ec96039 · ec96039
1 parent b99e0d7
commit ec96039
Show file tree

Hide file tree

Showing 8 changed files with 36 additions and 7 deletions.
diff --git a/duckdb-pgq b/duckdb-pgq
diff --git a/src/common.cpp b/src/common.cpp
@@ -2,7 +2,7 @@
 
 #include "duckdb/execution/expression_executor.hpp"
 #include "duckdb/main/client_data.hpp"
-#include "duckdb/common/compressed_sparse_row.hpp"
+#include "duckpgq/compressed_sparse_row.hpp"
 
 #include <utility>
 #include "duckpgq_extension.hpp"

diff --git a/src/functions/scalar/csr_creation.cpp b/src/functions/scalar/csr_creation.cpp
@@ -3,7 +3,7 @@
 #include "duckdb/planner/expression/bound_function_expression.hpp"
 #include "duckpgq/common.hpp"
 #include "duckpgq/duckpgq_functions.hpp"
-#include "duckdb/common/compressed_sparse_row.hpp"
+#include "duckpgq/compressed_sparse_row.hpp"
 
 #include <cmath>
 #include <mutex>

diff --git a/src/functions/tablefunctions/pgq_scan.cpp b/src/functions/tablefunctions/pgq_scan.cpp
@@ -2,7 +2,7 @@
 #include "duckpgq/functions/tablefunctions/pgq_scan.hpp"
 #include "duckpgq/common.hpp"
 #include "duckpgq/duckpgq_functions.hpp"
-#include "duckdb/common/compressed_sparse_row.hpp"
+#include "duckpgq/compressed_sparse_row.hpp"
 #include "duckdb/parser/parsed_data/create_table_function_info.hpp"
 #include "duckdb/function/table_function.hpp"
 #include "duckdb/parser/parsed_data/create_property_graph_info.hpp"

diff --git a/src/include/duckpgq/common.hpp b/src/include/duckpgq/common.hpp
@@ -15,7 +15,7 @@
 #include "duckdb/execution/expression_executor.hpp"
 #include "duckdb/main/client_context.hpp"
 #include "duckdb/planner/expression/bound_function_expression.hpp"
-#include "duckdb/common/compressed_sparse_row.hpp"
+#include "duckpgq/compressed_sparse_row.hpp"
 #include "duckdb/parser/parsed_data/create_property_graph_info.hpp"
 
 namespace duckdb {

diff --git a/src/include/duckpgq/compressed_sparse_row.hpp b/src/include/duckpgq/compressed_sparse_row.hpp
@@ -0,0 +1,29 @@
+#pragma once
+
+#include "duckdb/common/atomic.hpp"
+#include "duckdb/execution/executor.hpp"
+
+namespace duckdb {
+class CSR {
+public:
+	CSR() {
+	}
+	~CSR() {
+		delete[] v;
+	}
+
+	atomic<int64_t> *v;
+
+	vector<int64_t> e;
+	vector<int64_t> edge_ids;
+
+	vector<int64_t> w;
+	vector<double> w_double;
+
+	bool initialized_v = false;
+	bool initialized_e = false;
+	bool initialized_w = false;
+
+	size_t vsize;
+};
+} // namespace duckdb
diff --git a/src/include/duckpgq/functions/tablefunctions/pgq_scan.hpp b/src/include/duckpgq/functions/tablefunctions/pgq_scan.hpp
@@ -6,7 +6,7 @@
  */
 
 #include "duckdb/function/table_function.hpp"
-#include "duckdb/common/compressed_sparse_row.hpp"
+#include "duckpgq/compressed_sparse_row.hpp"
 #include "duckpgq/common.hpp"
 #include "duckpgq/duckpgq_functions.hpp"
 #include "duckdb/common/types/value.hpp"

diff --git a/src/include/duckpgq_extension.hpp b/src/include/duckpgq_extension.hpp
@@ -6,7 +6,7 @@
 #include "duckdb/parser/query_node.hpp"
 #include "duckdb/parser/column_list.hpp"
 #include "duckdb/parser/simplified_token.hpp"
-#include "duckdb/common/compressed_sparse_row.hpp"
+#include "duckpgq/compressed_sparse_row.hpp"
 #include "duckdb/parser/parsed_data/create_property_graph_info.hpp"
 
 namespace duckdb {