cmstools/resolveIncludes.py at master · apana/cmstools · GitHub

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#!/usr/bin/env python2.4

import sys, os, getopt
import FWCore.ParameterSet.Config as cms

def main(argv) :
    """

    resolveIncludes

    resolve includes of a given parameter-set and writes out resolved parameter-set

    required parameters
    --parameter-set <name>                    :       name of input parameter-set

    optional parameters                       :
    --output-filename <name>                  :       name of output parameter-set (default parameter-set_resolved.cfg)
    --help (-h)                               :       help
    --debug (-d)                              :       debug statements

    """

    # default
    parameter_set = ''
    output_filename = ''

    try:
        opts, args = getopt.getopt(argv, "", ["help", "debug", "parameter-set=", "output-filename="])
    except getopt.GetoptError:
        print main.__doc__
        sys.exit(2)

    # check command line parameter
    for opt, arg in opts :
        if opt == "--help" :
            print main.__doc__
            sys.exit()
        elif opt == "--debug" :
            debug = 1
        elif opt == "--parameter-set" :
            parameter_set = arg
        elif opt == "--output-filename" :
            output_filename = arg

    if parameter_set == '' :
        print main.__doc__
        sys.exit(2)

    if output_filename == '' :
        output_filename = parameter_set.split('.cfg')[0] + '_resolved.cfg'

    process = cms.include(parameter_set)

    temp = process.dumpConfig()
    # temp = process.dumpPython()

    outputfile = file(output_filename,'w')

    outputfile.write(temp)

    print 'Wrote resolved parameter-set into',output_filename


if __name__ == '__main__' :
    main(sys.argv[1:])