diff --git a/dicom2stl.py b/dicom2stl.py index fc50d72..6e9f5f4 100755 --- a/dicom2stl.py +++ b/dicom2stl.py @@ -48,7 +48,7 @@ def elapsedTime(start_time): print(" %4.3f seconds" % dt) -def loadVolume(fname, tempDir=None, verbose=False): +def loadVolume(fname, tempDir=None, verbose=0): modality = None zipFlag = False dirFlag = False @@ -85,7 +85,7 @@ def loadVolume(fname, tempDir=None, verbose=False): # if zipFlag: # Case for a zip file of images - if args.verbose: + if verbose: print("zip") if not tempDir: with tempfile.TemporaryDirectory() as defaultTempDir: @@ -120,8 +120,8 @@ def extract_int(file_path): fname = sorted(fname, key=extract_int) - if args.verbose: - if args.verbose > 1: + if verbose: + if verbose > 1: print("Reading images: ", fname) else: print( @@ -427,6 +427,9 @@ def dicom2stl(args): print("") -if __name__ == "__main__": +def main(): args = parseargs.parseargs() dicom2stl(args) + +if __name__ == "__main__": + main() diff --git a/pyproject.toml b/pyproject.toml index 5a5bb17..6a09e3c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -7,7 +7,7 @@ name = "dicom2stl" authors = [ { name="David Chen", email="dchen@mail.nih.gov" }, ] -description = "A script to extract an STL surface mesh from a DICOM volume image." +description = "A script to extract an iso-surface from a DICOM series to produce an STL mesh." readme = "README.md" requires-python = ">=3.8" classifiers = [