# emacs: -*- mode: python; py-indent-offset: 4; indent-tabs-mode: nil -*-# vi: set ft=python sts=4 ts=4 sw=4 et:## Copyright The NiPreps Developers <nipreps@gmail.com>## Licensed under the Apache License, Version 2.0 (the "License");# you may not use this file except in compliance with the License.# You may obtain a copy of the License at## http://www.apache.org/licenses/LICENSE-2.0## Unless required by applicable law or agreed to in writing, software# distributed under the License is distributed on an "AS IS" BASIS,# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.# See the License for the specific language governing permissions and# limitations under the License.## We support and encourage derived works from this project, please read# about our expectations at## https://www.nipreps.org/community/licensing/#"""NiFreeze runner."""frompathlibimportPathfromnifreeze.cli.parserimportparse_argsfromnifreeze.data.dmriimportDWIfromnifreeze.estimatorimportEstimator
[docs]defmain(argv=None)->None:""" Entry point. Returns ------- None """args=parse_args(argv)# Open the data with the given file pathdwi_dataset:DWI=DWI.from_filename(args.input_file)estimator:Estimator=Estimator()_=estimator.estimate(dwi_dataset,align_kwargs=args.align_config,models=args.models,omp_nthreads=args.nthreads,njobs=args.njobs,seed=args.seed,)# Set the output filename to be the same as the input filenameoutput_filename:str=Path(args.input_file).nameoutput_path:Path=Path(args.output_dir)/output_filename# Save the DWI dataset to the output pathdwi_dataset.to_filename(output_path)