From 78ea94a642f5574da50218f6fd9d85cb9df5cac1 Mon Sep 17 00:00:00 2001
From: polarbean <harry.carey95@gmail.com>
Date: Wed, 21 Aug 2024 17:10:53 +0200
Subject: [PATCH] begin to introduce custom atlas function

---
 .gitignore      | 3 +++
 PyNutil/main.py | 5 +++++
 2 files changed, 8 insertions(+)

diff --git a/.gitignore b/.gitignore
index 7ad8120..57e17bc 100644
--- a/.gitignore
+++ b/.gitignore
@@ -167,3 +167,6 @@ test_data/tTA_2877_NOP_horizontal_final_2017.json
 fileForMessingAround.py
 
 outputs/
+test_data/
+test/
+PyNutil/metadata/annotation_volumes/
\ No newline at end of file
diff --git a/PyNutil/main.py b/PyNutil/main.py
index 57194a4..320776e 100644
--- a/PyNutil/main.py
+++ b/PyNutil/main.py
@@ -146,6 +146,11 @@ class PyNutil:
         print("atlas labels loaded ✅")
         return atlas_volume, atlas_labels
 
+    def load_custom_atlas(self, atlas_path, label_path):
+        atlas_volume = read_atlas_volume(atlas_path)
+        atlas_labels = pd.read_csv(label_path)
+        return atlas_volume, atlas_labels
+
     def get_coordinates(
         self, non_linear=True, method="all", object_cutoff=0, use_flat=False
     ):
-- 
GitLab