File: convert.py

package info (click to toggle)
odil 0.13.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 5,476 kB
  • sloc: cpp: 55,982; python: 3,947; javascript: 460; xml: 182; makefile: 99; sh: 36
file content (49 lines) | stat: -rw-r--r-- 1,754 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import argparse
import sys

import odil

def main():
    parser = argparse.ArgumentParser(description="Convert a DICOM file")
    parser.add_argument("input")
    parser.add_argument("output")
    parser.add_argument("--transfer-syntax", "-t", help="Transfer syntax")
    parser.add_argument(
        "--item-length", "-i", choices=["Explicit", "Undefined"], 
        default="Explicit", help="Item length encoding")
    parser.add_argument(
        "--use-group-length", "-g", action="store_true", 
        help="Include group length tags")
    arguments = parser.parse_args()

    if arguments.transfer_syntax not in dir(odil.registry):
        parser.error("Unknown transfer syntax")
    uid = getattr(odil.registry, arguments.transfer_syntax)
    if uid not in odil.registry.uids_dictionary:
        parser.error("Unknown transfer syntax")
    if odil.registry.uids_dictionary[uid].type != "Transfer Syntax":
        parser.error("Not a transfer syntax")
    arguments.transfer_syntax = uid

    arguments.item_length = getattr(
        odil.Writer.ItemEncoding, "{}Length".format(arguments.item_length))

    convert(**arguments.__dict__)

def convert(input, output, transfer_syntax, item_length, use_group_length):
    with odil.open(input) as stream:
        header, data_set = odil.Reader.read_file(stream)

    to_remove = [
        "FileMetaInformationVersion",
        "MediaStorageSOPClassUID", "MediaStorageSOPInstanceUID", 
        "TransferSyntaxUID", 
        "ImplementationClassUID", "ImplementationVersionName"
    ]
    for name in to_remove:
        header.remove(getattr(odil.registry, name))

    odil.write(data_set, output, header, transfer_syntax, item_length, use_group_length)

if __name__ == "__main__":
    sys.exit(main())