Commit 7400e12d authored by Adrian Nievergelt's avatar Adrian Nievergelt

include reconstruction files in thumbnailing

parent 388b9e45
......@@ -19,6 +19,8 @@ parser.add_argument('-i', '--input', nargs='?', default='.', help='Input MDOC fi
parser.add_argument('-d', '--inDir', nargs='?', default='.', help='If specified, automatically process entire directory of mrc files.')
parser.add_argument('-mi', '--maxImages', nargs='?', type=float, default=30, help='If more images than specified are in a file, make amontage with a subset of the frames present instead of separate files. Detault is 30 images, minimum 6.')
parser.add_argument('-sd', '--subDirs', action='store_true', help='Include subdirectories when in directory mode.')
parser.add_argument('-et', '--excludeTomograms', action='store_true', help='Do not process tomograms (.rec) files.')
# parser.add_argument('-g', '--graph', action='store_true', help='Store a graph of the corrected and uncorrected line intensities for debug purposes.')
args = parser.parse_args()
......@@ -41,9 +43,14 @@ if args.maxImages < 6:
if dirMode and args.inDir is not None and os.path.isdir(args.inDir):
inFileList = []
if args.excludeTomograms:
extFilter = ('.mrc')
else:
extFilter = ('.mrc', '.rec')
for (dirpath, dirnames, filenames) in os.walk(args.inDir):
filenames.sort()
newFileList = [os.path.join(dirpath, fName) for fName in filenames if fName.endswith('.mrc')]
newFileList = [os.path.join(dirpath, fName) for fName in filenames if fName.endswith(extFilter)]
inFileList.extend(newFileList)
if not args.subDirs:
break
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment