]> Creatis software - gdcm.git/blob - gdcmPython/demo/ReorganiseFiles.py
ENH : Entry fffe have never VR. Don't verbose useless messages any longer
[gdcm.git] / gdcmPython / demo / ReorganiseFiles.py
1 # Rename the dcm extension files of a given directory according
2 # to series and image number information extracted in each respective header.
3
4 from gdcmPython import *
5 import glob
6 import os
7 import sys
8
9 def PrintUsage():
10    print "Usage: python ReorganiseFiles.py <SourceDirectory>"
11    print "       <SourceDirectory> is the directory containing the source"
12    print "                         images (defaulted to invocation directory."
13
14 def Warning(DirName):
15    print "Warning: this script will rename files in the"
16    print "        ", DirName, " directory."
17    a = raw_input("Proceed ?[y/n]")
18    if a != "y":
19       sys.exit()
20
21 try:
22    SourceDirectory = sys.argv[1]
23 except IndexError:
24    SourceDirectory = os.getcwd()
25
26 if not os.path.isdir(SourceDirectory):
27    PrintUsage
28 Warning(SourceDirectory)
29 SourceFiles=glob.glob(os.path.join(SourceDirectory, "*.dcm"))
30
31 for file in SourceFiles:
32    header = gdcmHeader(file)
33    info = header.GetEntry()
34    try:
35       ImNum = info["Image Number"]
36       if len(ImNum) == 0 or ImNum == "gdcm::Unfound":
37          raise KeyError
38    except KeyError:
39       print "Skipped file ", file, " (unfound Image Number tag)."
40       continue
41    try:
42       SeNum = info["Series Number"]
43       if len(SeNum) == 0 or SeNum == "gdcm::Unfound":
44          raise KeyError
45    except KeyError:
46       print "Skipped file ", file, " (unfound Series Number tag)."
47       continue
48    DestFile = "image_%02d_%04d.dcm"
49    DestFile %= (int(info["Image Number"]), int(info["Series Number"]))
50    DestFile = os.path.join(SourceDirectory, DestFile)
51    # necessary to close the file descriptor to enable further rename.
52    del header
53    if os.path.isfile(DestFile):
54       print "Skipped file ", file, " (allready existing destination"
55       print "     file ", DestFile
56       continue
57    os.rename(file, DestFile)