Parcourir la source

param `root_dir_path` default $SUBJECTS_DIR

Fabian Peter Hammerle il y a 5 ans
Parent
commit
85f35258c9
2 fichiers modifiés avec 17 ajouts et 3 suppressions
  1. 13 0
      README.md
  2. 4 3
      freesurfer_volume_reader/__init__.py

+ 13 - 0
README.md

@@ -1,3 +1,16 @@
+## Usage
+
+```sh
+export SUBJECTS_DIR=/my/freesurfer/subjects
+freesurfer-volume-reader
+```
+
+or
+
+```sh
+freesurfer-volume-reader /my/freesurfer/subjects
+```
+
 ## Tests
 
 ```sh

+ 4 - 3
freesurfer_volume_reader/__init__.py

@@ -57,7 +57,6 @@ def read_hippocampal_volume_file_dataframe(volume_file_path: str) -> pandas.Data
 
 
 def main():
-    # TODO add description
     argparser = argparse.ArgumentParser(
         description='Read hippocampal volumes computed by Freesurfer'
                     '\nhttps://surfer.nmr.mgh.harvard.edu/fswiki/HippocampalSubfields')
@@ -66,8 +65,10 @@ def main():
                            help='default: %(default)s')
     argparser.add_argument('--output-format', choices=['csv'], default='csv',
                            help='default: %(default)s')
-    # TODO default to $SUBJECTS_DIR
-    argparser.add_argument('root_dir_path')
+    argparser.add_argument('root_dir_path',
+                           nargs='?' if 'SUBJECTS_DIR' in os.environ else 1,
+                           default=os.environ.get('SUBJECTS_DIR', None),
+                           help='default: $SUBJECTS_DIR (%(default)s)')
     args = argparser.parse_args()
     volume_file_paths = find_hippocampal_volume_files(
         root_dir_path=args.root_dir_path,