Merge pull request #134 from brittonsmith/rdir

Set output directory properly for rockstar halo finder.

yt_astro_analysis/halo_analysis/halo_catalog/halo_catalog.py

@@ -115,7 +115,7 @@ def __init__(
         halo_field_type="all",
         finder_method=None,
         finder_kwargs=None,
-        output_dir="halo_catalogs",
+        output_dir=None,
     ):
 
         super().__init__()
@@ -134,6 +134,12 @@ def __init__(
             data_source = halos_ds.all_data()
         self.data_source = data_source
 
+        if output_dir is None:
+            if finder_method == "rockstar":
+                output_dir = finder_kwargs.get("outbase", "rockstar_halos")
+            else:
+                output_dir = "halo_catalogs"
+
         self.output_basedir = ensure_dir(output_dir)
         self.pipeline = AnalysisPipeline(output_dir=self.output_dir)
         self.quantities = self.pipeline.quantities