FWLite/scripts/newFWLiteAna.py

0001 #! /usr/bin/env python3
0002
0003 from __future__ import print_function
0004 import optparse
0005 import os
0006 import sys
0007 import re
0008 import shutil
0009
0010 ccRE = re.compile (r'(\w+)\.cc')
0011
0012 def extractBuildFilePiece (buildfile, copy, target = 'dummy'):
0013     """Extracts necessary piece of the buildfile.  Returns empty
0014     string if not found."""
0015     try:
0016         build = open (buildfile, 'r')
0017     except:
0018         raise RuntimeError("Could not open BuildFile '%s' for reading. Aboring." \
0019               % buildfile)
0020     # make my regex
0021     startBinRE = re.compile (r'<\s*bin\s+name=(\S+)')
0022     endBinRE   = re.compile (r'<\s*/bin>')
0023     exeNameRE  = re.compile (r'(\w+)\.exe')
0024     ccName = os.path.basename (copy)
0025     match = ccRE.match (ccName)
0026     if match:
0027         ccName = match.group (1)
0028     retval = ''
0029     foundBin = False
0030     for line in build:
0031         # Are we in the middle of copying what we need?
0032         if foundBin:
0033             retval += line
0034         # Are we copying what we need and reach the end?
0035         if foundBin and endBinRE.search (line):
0036             # all done
0037             break
0038         # Is this the start of a bin line with the right name?
0039         match = startBinRE.search (line)
0040         if match:
0041             # strip of .exe if it's there
0042             exeName = match.group (1)
0043             exeMatch = exeNameRE.search (exeName)
0044             if exeMatch:
0045                 exeName = exeMatch.group (1)
0046             if exeName == ccName:
0047                 foundBin = True
0048                 line = re.sub (exeName, target, line)
0049                 retval = line
0050     build.close()
0051     return retval
0052
0053
0054 def createBuildFile (buildfile):
0055     """Creates a BuildFile if one doesn't already exist"""
0056     if os.path.exists (buildfile):
0057         return
0058     build = open (buildfile, 'w')
0059     build.write ("<!-- -*- XML -*- -->\n\n<environment>\n\n</environment>\n")
0060     build.close()
0061
0062
0063 def addBuildPiece (targetBuild, buildPiece):
0064     """Adds needed piece for new executable.  Returns true upon
0065     success."""
0066     backup = targetBuild + ".bak"
0067     shutil.copyfile (targetBuild, backup)
0068     oldBuild = open (backup, 'r')
0069     newBuild = open (targetBuild, 'w')
0070     environRE = re.compile (r'<\s*environment')
0071     success = False
0072     for line in oldBuild:
0073         newBuild.write (line)
0074         if environRE.search (line):
0075             newBuild.write ('\n\n')
0076             newBuild.write (buildPiece)
0077             success = True
0078     oldBuild.close()
0079     newBuild.close()
0080     return success
0081
0082
0083
0084 ########################
0085 ## ################## ##
0086 ## ## Main Program ## ##
0087 ## ################## ##
0088 ########################
0089
0090 if __name__ == '__main__':
0091     # We need CMSSW to already be setup
0092     base         = os.environ.get ('CMSSW_BASE')
0093     release_base = os.environ.get ('CMSSW_RELEASE_BASE')
0094     if not base or not release_base:
0095         print("Error: You must have already setup a CMSSW release.  Aborting.")
0096         sys.exit()
0097     # setup the options
0098     parser = optparse.OptionParser('usage: %prog [options] '
0099                                    'Package/SubPackage/name\n'
0100                                    'Creates new analysis code')
0101     parser.add_option ('--copy', dest='copy', type='string', default = 'blank',
0102                        help='Copies example. COPY should either be a file'
0103                        ' _or_ an example in PhysicsTools/FWLite/examples')
0104     parser.add_option ('--newPackage', dest='newPackage', action='store_true',
0105                        help='Will create Package/Subpackage folders if necessary')
0106     parser.add_option ('--toTest', dest='toTest', action='store_true',
0107                        help='Will create files in test/ instead of bin/')
0108     (options, args) = parser.parse_args()
0109     # get the name of the copy file and make sure we can find everything
0110     copy = options.copy
0111     if not re.search ('\.cc$', copy):
0112         copy += '.cc'
0113     buildCopy = os.path.dirname (copy)
0114     if len (buildCopy):
0115         buildCopy += '/BuildFile'
0116     else:
0117         buildCopy = 'BuildFile'
0118     found = False
0119     searchList = ['',
0120                   base + "/src/PhysicsTools/FWLite/examples/",
0121                   release_base+ "/src/PhysicsTools/FWLite/examples/"]
0122     fullName  = ''
0123     fullBuild = ''
0124     for where in searchList:
0125         name  = where + copy
0126         build = where + buildCopy
0127         # Is the copy file here
0128         if os.path.exists (name):
0129             # Yes.
0130             found = True
0131             # Is there a Buildfile too?
0132             if not os.path.exists (build):
0133                 print("Error: Found '%s', but no accompying " % name, \
0134                       "Buildfile '%s'. Aborting" % build)
0135                 sys.exit()
0136             fullName = name
0137             fullBuild = build
0138             break
0139     if not found:
0140         print("Error: Did not find '%s' to copy.  Aborting." % copy)
0141         sys.exit()
0142     if len (args) != 1:
0143         parser.print_usage()
0144         sys.exit()
0145     pieces = args[0].split('/')
0146     if len (pieces) != 3:
0147         print("Error: Need to provide 'Package/SubPackage/name'")
0148         sys.exit()
0149     target = pieces[2]
0150     match = ccRE.match (target)
0151     if match:
0152         target = match.group (1)
0153     buildPiece = extractBuildFilePiece (fullBuild, copy, target)
0154     if not buildPiece:
0155         print("Error: Could not extract necessary info from Buildfile. Aborting.")
0156         sys.exit()
0157     # print buildPiece
0158     firstDir  = base + '/src/' + pieces[0]
0159     secondDir = firstDir + '/' + pieces[1]
0160     if options.toTest:
0161         bin = '/test'
0162     else:
0163         bin = '/bin'
0164     if not os.path.exists (secondDir) and \
0165        not options.newPackage:
0166         raise RuntimeError("%s does not exist.  Use '--newPackage' to create" \
0167               % ("/".join (pieces[0:2]) ))
0168     dirList = [firstDir, secondDir, secondDir + bin]
0169     targetCC = dirList[2] + '/' + target + '.cc'
0170     targetBuild = dirList[2] + '/BuildFile'
0171     if os.path.exists (targetCC):
0172         print('Error: "%s" already exists.  Aborting.' % targetCC)
0173         sys.exit()
0174     # Start making directory structure
0175     for currDir in dirList:
0176         if not os.path.exists (currDir):
0177             os.mkdir (currDir)
0178     # copy the file over
0179     shutil.copyfile (fullName, targetCC)
0180     print("Copied:\n   %s\nto:\n   %s.\n" % (fullName, targetCC))
0181     createBuildFile (targetBuild)
0182     if extractBuildFilePiece (targetBuild, target):
0183         print("Buildfile already has '%s'.  Skipping" % target)
0184     else :
0185         # we don't already have a piece here
0186         if addBuildPiece (targetBuild, buildPiece):
0187             print("Added info to:\n   %s." % targetBuild)
0188         else:
0189             print("Unable to modify Buildfile.  Sorry.")