HepMCAnalysis: configure.py Source File

00001 #!/usr/bin/env python
00002 
00003 import sys
00004 import getopt
00005 import os 
00006 import string 
00007 
00008 # ---------------------------------------------------------------------- 
00009 # Default versions. They can be changed here, or (preferred) directly from 
00010 # the command line, e.g.: 
00011 #           source configure --hepmc-version=2.06.03
00012 # ---------------------------------------------------------------------- 
00013 versions = {
00014   'cascade-version':          '2.0.1',
00015   'clhep-version':            '1.9.4.7',
00016   'hepmc-version':            '2.03.11',
00017 #  'hepmc-version':            '2.06.03',
00018   'heppdt-version':           '3.04.01',
00019   'fastjet-version':          '2.4.2p5',
00020   'gsl-version':              '1.10',
00021   'pythia8-version':          '142',
00022   'pythia6-version':          '423.2',
00023   'alpgen-version':           '2.1.3d.2',
00024   'herwigpp-version':         '2.5.0',
00025   'herwig-version':           '6.520.2',
00026   'jimmy-version':            '4.31.3',
00027   'thepeg-version':           '1.7.0',
00028   'lhapdf-version':           '5.8.2',
00029   'tauola-version':           '28.121.2',
00030   'photos-version':           '215.5',
00031   'prefix':                   ''
00032 }
00033 
00034 # auxiliary stuff 
00035 colours = { 'default':   '',
00036             'blue':      '\x1b[01;34m',
00037             'cyan':      '\x1b[01;36m',
00038             'green':     '\x1b[01;32m',
00039             'red':       '\x1b[01;05;37;41m',
00040             'end':       '\x1b[00m' }
00041 
00042 # 
00043 envVarsFile   = open( 'setupEnvVariables.sh', 'w' )
00044 gccSetupFile  = open( 'setupGccVersion.sh', 'w' )
00045 
00046 # ---------------------------------------------------------------------- 
00047 def main():
00048   processOptions()    
00049  
00050   envVarsFile.write( '#!/bin/bash\n' ) 
00051   gccSetupFile.write( '#!/bin/bash\n' ) 
00052 
00053     # location's specific setup  
00054   location, curr_dir = detectLocation()
00055   envVarsFile.write( 'export MYCLASS_PATH=' + curr_dir + '\n' ) 
00056   location = location.rstrip( '\n' ) 
00057 
00058   if location == 'desy.de':
00059     prefix, sysname = initializeDesySetup()
00060   elif location == 'naf.desy.de':  
00061     prefix, sysname = initializeNafSetup()
00062   else:  
00063     prefix, sysname = initializeCernSetup()
00064 
00065   print "Prefix:", prefix, ", sysname:", sysname
00066   initializeStdLibs( prefix, sysname ) 
00067   
00068   print 'Done!' 
00069   
00070 # ---------------------------------------------------------------------- 
00071 # Process command line options  
00072 # ---------------------------------------------------------------------- 
00073 def processOptions():
00074   options     = ''
00075   longOptions = [] 
00076   for op in versions.keys():
00077     longOptions.append( op + '=' )    # options which require arguments need '='
00078   
00079     # add '--help' option
00080   longOptions.append( 'help' )
00081 
00082     # parse command line options
00083   try:
00084     optlist, args = getopt.getopt( sys.argv[1:], options, longOptions )
00085   except getopt.GetoptError, err:
00086     print str( err )
00087     print "For help use --help"
00088     sys.exit( 2 )
00089 
00090   for op, ar in optlist:
00091     if op.startswith( '--' ):   # remove '--' from the beginning
00092       op = op.strip( '-' )
00093     if op in versions.keys():
00094       versions[ op ] = ar
00095       print colours[ 'blue' ]+"Overwriting default option '", op, "' with value:", ar, colours['end']
00096       if op == 'hepmc-version':
00097         print colours['blue']+"Be aware that other libraries could be compiled with different version of HepMC!", colours['end']
00098     if op == 'help':
00099       showHelp()        
00100       sys.exit( 0 )
00101 
00102 # ---------------------------------------------------------------------- 
00103 # DESY specific setup
00104 # ---------------------------------------------------------------------- 
00105 def initializeDesySetup():
00106   print colours[ 'blue' ] + "[ Initialize DESY setup ]" + colours[ 'end' ]
00107   
00108   #sysNameInfo = os.popen( 'source check_platform_desy' ).readlines()
00109   #sysname = sysNameInfo[0]
00110   #print 'Systsssssem:', sysname,
00111   #envVarsFile.write( 'export SYSNAME=' + sysname + '\n' )
00112 
00113     # determine system's details
00114   sysArch = 0
00115   sysDsc  = ''
00116   sysName = ''
00117   sysCmd = os.popen( 'fs sysname' ).readline()
00118     # == SL4
00119   if contains( sysCmd, 'i586_rhel40' ) or contains( sysCmd, 'i386_linux26' ):
00120     sysArch = 32
00121     sysDsc  = 'SL4'
00122     sysName = 'i586_rhel40'
00123   elif contains( sysCmd, 'amd64_rhel40' ) or contains( sysCmd, 'amd64_linux26' ):
00124     sysArch = 64
00125     sysDsc  = 'SL4'
00126     sysName = 'amd64_rhel40'
00127     # == SL5
00128   if contains( sysCmd, 'i586_rhel50' ):
00129     sysArch = 32
00130     sysDsc  = 'SL5'
00131     sysName = 'i586_rhel50'
00132   elif contains( sysCmd, 'amd64_rhel50' ):
00133     sysArch = 64
00134     sysDsc  = 'SL5'
00135     sysName = 'amd64_rhel50'
00136 
00137   print 'System:', sysName, '(', sysDsc, '), architecture:', sysArch
00138   envVarsFile.write( 'export SYSNAME=' + sysName + '\n' )
00139 
00140     # SL4 specific settings
00141   if sysDsc == 'SL4':
00142     print 'Initialize SL4 settings'
00143     envVarsFile.write( 'export CXX="/usr/bin/g++"\n' )
00144     envVarsFile.write( 'export F77="/usr/bin/g77"\n' )
00145     envVarsFile.write( 'export FLIBS="-lfrtbegin -lg2c"\n' )
00146     # SL5 specific settings
00147   elif sysDsc == 'SL5':
00148     print 'Initialize SL5 settings'
00149     envVarsFile.write( 'export CXX="/usr/bin/g++44"\n' )
00150     envVarsFile.write( 'export F77="/usr/bin/gfortran44"\n' )
00151     envVarsFile.write( 'export FLIBS="-lgfortran -lgfortranbegin"\n' )
00152   else:
00153     print 'Unknown system name:',sysname,
00154  
00155     # prefix 
00156   if len( versions[ 'prefix' ] ) == 0:
00157     prefix = '/afs/desy.de/group/alliance/mcg/public'
00158   else:  
00159     prefix = versions[ 'prefix' ] 
00160   envVarsFile.write( 'export PREFIX=' + prefix + '\n' )
00161 
00162     # clhep
00163   envVarsFile.write( 'export CLHEPversion=' + versions[ 'clhep-version' ] + '\n' );
00164   clhepDir = prefix + '/CLHEP/' + versions[ 'clhep-version' ] + '/' + sysName   
00165   print 'CLHEP directory:', clhepDir
00166   envVarsFile.write( 'export CLHEPdir=' + clhepDir + '\n' );
00167  
00168     # cernlib 
00169   if sysDsc == 'SL5':
00170     envVarsFile.write( 'export CERNLIB=' + prefix + '/' + sysName + '/lib/cernlib/2006a/lib\n' )
00171   elif sysDsc == 'SL4':
00172     envVarsFile.write( 'export CERNLIB=/opt/products/cernlib/2005/lib\n' )
00173 
00174    # Root's initialization
00175   rootVersion = '5.22.00'    
00176   print 'ROOT version:', rootVersion
00177   envVarsFile.write( 'export ROOTver=' + rootVersion + '\n' )
00178   
00179   if sysName.find( 'amd64' ) != -1:
00180     rootSys = '/opt/products/root64/' + rootVersion
00181     rootLib = rootSys + '/lib64'
00182   else:
00183     rootSys = '/opt/products/root/' + rootVersion
00184     rootLib = rootSys + '/lib'
00185   envVarsFile.write( 'export ROOTSYS=' + rootSys + '\n' )
00186   envVarsFile.write( 'export ROOTLIB=' + rootLib + '\n' )
00187   envVarsFile.write( 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$ROOTLIB\n' )
00188   envVarsFile.write( 'export PATH=$ROOTSYS/bin:$PATH\n' )
00189       
00190   envVarsFile.write( 'export PDFPATH=' + prefix + '/MCGenerators/cascade/' + versions[ 'cascade-version' ] + '/data\n' )
00191 
00192   return prefix, sysName
00193 
00194 # ---------------------------------------------------------------------- 
00195 # NAF specific setup (TODO!)
00196 # ---------------------------------------------------------------------- 
00197 def initializeNafSetup():
00198   print '== Initialize NAF setup =='
00199 
00200   # determine system
00201   sysNameInfo = os.popen( 'source check_platform_desy' ).readlines()
00202   sysname = sysNameInfo[0]
00203   print 'System:', sysname,
00204   envVarsFile.write( 'export SYSNAME=' + sysname + '\n' )
00205 
00206   # prefix 
00207   if len( versions[ 'prefix' ] ) == 0:
00208     prefix = '/afs/desy.de/group/alliance/mcg/public'
00209   else:  
00210     prefix = versions[ 'prefix' ] 
00211   envVarsFile.write( 'export PREFIX=' + prefix + '\n' )
00212 
00213   # Root's initialization
00214   rootVersion = '5.22.00'    
00215   print 'ROOT version:', rootVersion
00216   envVarsFile.write( 'export ROOTver=' + rootVersion + '\n' )
00217   
00218   if sysname.find( 'amd64' ) != -1:
00219     rootSys = '/afs/naf.desy.de/products/root/' + rootVersion
00220     rootLib = rootSys + '/lib'
00221   else:
00222     print 'No 32 bit ROOT version at NAF!'
00223     sys.exit( 2 )
00224   envVarsFile.write( 'export ROOTSYS=' + rootSys + '\n' )
00225   envVarsFile.write( 'export ROOTLIB=' + rootLib + '\n' )
00226   envVarsFile.write( 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$ROOTLIB\n' )
00227   envVarsFile.write( 'export PATH=$ROOTSYS/bin:$PATH\n' )
00228       
00229   # clhep...
00230   envVarsFile.write( 'export CLHEPversion=' + versions[ 'clhep-version' ] + '\n' );
00231   clhepDir = prefix + '/CLHEP/' + versions[ 'clhep-version' ] + '/' + sysname   
00232   print 'CLHEP dir:', clhepDir,
00233   envVarsFile.write( 'export CLHEPdir=' + clhepDir + '\n' );
00234   
00235   envVarsFile.write( 'export CERNLIB=/opt/products/cernlib/2005/lib\n' )
00236   envVarsFile.write( 'export PDFPATH=' + prefix + '/MCGenerators/cascade/' + versions[ 'cascade-version' ] + '/data\n' )
00237 
00238   return prefix, sysname
00239 
00240 # ---------------------------------------------------------------------- 
00241 # CERN specific setup
00242 # ---------------------------------------------------------------------- 
00243 def initializeCernSetup():
00244   print colours[ 'blue' ] + "[ Initialize CERN setup ]" + colours[ 'end' ]
00245   
00246   # determine system
00247   sysNameInfo = os.popen( 'source check_platform' ).readlines()
00248   sysname = sysNameInfo[1].split()[2]
00249   #sysname = sysNameInfo[0]
00250   #sysname = sysname.replace( "\n", "" )
00251   print 'System:', sysname
00252   envVarsFile.write( 'export SYSNAME=' + sysname + '\n' )
00253 
00254   tmpTab = sysname.split( '-' )   # e.g. x86_64-slc5-gcc43-opt
00255   if len( tmpTab ) > 1:
00256     arch = tmpTab[ 0 ]                # e.g. x86_64
00257     slcVer = tmpTab[ 1 ]
00258     print 'Architecture:', arch
00259     print 'SLC version:', slcVer 
00260     if slcVer == 'slc5':
00261       ###gccSetup = '/afs/cern.ch/sw/lcg/external/gcc/4.3.2/' + arch + '/setup.sh\n'
00262       gccSetup = '/afs/cern.ch/sw/lcg/external/gcc/4.3.5/' + arch + '/setup.sh\n'
00263       print 'Gcc setup file:', gccSetup
00264       gccSetupFile.write( 'source %s' % gccSetup )
00265 #      envVarsFile.write( 'source /afs/cern.ch/sw/lcg/external/gcc/4.1.2/' + arch + '/setup.sh\n' )
00266 
00267   # determine GCC version
00268   gccVersion = os.popen( 'g++ --version' ).readline().split()[2]
00269   if gccVersion.startswith( '4' ):
00270     envVarsFile.write( 'export GFORTRAN=1\n' )
00271     print 'GFORTRAN = 1'
00272     envVarsFile.write( 'export F77="/usr/bin/gfortran44"\n' )
00273     envVarsFile.write( 'export FLIBS="-lgfortran -lgfortranbegin"\n' )
00274   else:
00275     print 'GFORTRAN = 0'
00276     envVarsFile.write( 'export F77="/usr/bin/g77"\n' )
00277     envVarsFile.write( 'export FLIBS="-lfrtbegin -lg2c"\n' )
00278   print 'GCC version:', gccVersion
00279  
00280   # prefix 
00281   if len( versions[ 'prefix' ] ) == 0:
00282     prefix = '/afs/cern.ch/sw/lcg/external'
00283   else:  
00284     prefix = versions[ 'prefix' ] 
00285   envVarsFile.write( 'export PREFIX=' + prefix + '\n' )
00286 
00287   # Root's initialization
00288 #  rootVersion = '5.27.04'    
00289   rootVersion = '5.28.00e'    
00290   print 'ROOT version:', rootVersion
00291   envVarsFile.write( 'export ROOTver=' + rootVersion + '\n' )
00292   rootSys = '/afs/cern.ch/sw/lcg/app/releases/ROOT/' + rootVersion + '/' + sysname + '/root'
00293   print 'ROOT path:', rootSys
00294   envVarsFile.write( 'export ROOTSYS=' + rootSys + '\n' )
00295   rootLib = rootSys + '/lib'
00296   print 'ROOT lib:', rootLib
00297   envVarsFile.write( 'export ROOTLIB=' + rootLib + '\n' )
00298   envVarsFile.write( 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:' + rootLib + '\n' )
00299   envVarsFile.write( 'export PATH=' + rootSys + '/bin:$PATH\n' )
00300 
00301   # clhep...
00302 #  versions[ 'clhep-version' ] = '2.0.4.4'
00303   envVarsFile.write( 'export CLHEPversion=' + versions[ 'clhep-version' ] + '\n' );
00304   clhepDir = prefix + '/clhep/' + versions[ 'clhep-version' ] + '/' + sysname   
00305   print 'CLHEP dir:', clhepDir
00306   envVarsFile.write( 'export CLHEPdir=' + clhepDir + '\n' );
00307   
00308   envVarsFile.write( 'export CERNLIB=' + prefix + '/cernlib/2006b/' + sysname + '/lib' + '\n' )
00309   envVarsFile.write( 'export PDFPATH=' + prefix + '/MCGenerators/cascade/' + versions[ 'cascade-version' ] + '/share/data\n' )
00310   return prefix, sysname
00311 
00312 # ---------------------------------------------------------------------- 
00313 # Setup environment variables
00314 # ---------------------------------------------------------------------- 
00315 def initializeStdLibs( prefix, sysname ):
00316   hepmc_suff = '' 
00317   pythia8_hepmc_suff = '' 
00318   if ( versions[ 'hepmc-version' ] == '2.06.03' ):      # use generators compiled with the new version of HepMC,
00319     pythia8_hepmc_suff = '_hepmc20603'                  # currently (Nov/2010) available only for Pythia8 at Cern...
00320   
00321   envVarsFile.write( 'export HepMCdir=' + prefix + '/HepMC/' + versions[ 'hepmc-version' ] + '/' + sysname + '\n' )
00322   envVarsFile.write( 'export HepPDTdir=' + prefix + '/HepPDT/' + versions[ 'heppdt-version'] + '/' + sysname + '\n' )
00323 
00324   envVarsFile.write( 'export FastJetdir=' + prefix + '/fastjet/' + versions[ 'fastjet-version'] + '/' + sysname + '\n' )
00325   envVarsFile.write( 'export GSLdir=' + prefix + '/GSL/' + versions[ 'gsl-version'] + '/' + sysname + '\n' )
00326   
00327   envVarsFile.write( 'export Pythia8dir=' + prefix + '/MCGenerators' + pythia8_hepmc_suff + '/pythia8/' + versions[ 'pythia8-version'] + '/' + sysname + '\n' )
00328   envVarsFile.write( 'export Pythia8DATA=' + prefix + '/MCGenerators' + pythia8_hepmc_suff + '/pythia8/' + versions[ 'pythia8-version'] + '/share/xmldoc\n' )
00329   
00330   envVarsFile.write( 'export Pythia6dir=' + prefix + '/MCGenerators' + hepmc_suff + '/pythia6/' + versions[ 'pythia6-version'] + '/' + sysname + '\n' )
00331   envVarsFile.write( 'export Alpgendir=' + prefix + '/MCGenerators' + hepmc_suff + '/alpgen/' + versions[ 'alpgen-version'] + '/' + sysname + '\n' )
00332   
00333   envVarsFile.write( 'export Herwigppdir=' + prefix + '/MCGenerators' + hepmc_suff + '/herwig++/' + versions[ 'herwigpp-version'] + '/' + sysname + '\n' )
00334   envVarsFile.write( 'export ThePEGdir=' + prefix + '/MCGenerators' + hepmc_suff + '/thepeg/' + versions[ 'thepeg-version'] + '/' + sysname + '\n' )
00335   envVarsFile.write( 'export Herwigdir=' + prefix + '/MCGenerators' + hepmc_suff + '/herwig/' + versions[ 'herwig-version'] + '/' + sysname + '\n' )
00336   envVarsFile.write( 'export Jimmydir=' + prefix + '/MCGenerators' + hepmc_suff + '/jimmy/' + versions[ 'jimmy-version'] + '/' + sysname + '\n' )
00337   envVarsFile.write( 'export LHAPDFdir=' + prefix + '/MCGenerators' + hepmc_suff + '/lhapdf/' + versions[ 'lhapdf-version'] + '/' + sysname + '\n' )
00338   envVarsFile.write( 'export LHAPDFsets=' + prefix + '/MCGenerators' + hepmc_suff + '/lhapdf/' + versions[ 'lhapdf-version'] + '/share/PDFsets\n' )
00339   envVarsFile.write( 'export LHAPATH=' + prefix + '/MCGenerators' + hepmc_suff + '/lhapdf/' + versions[ 'lhapdf-version'] + '/share/PDFsets\n' )
00340   envVarsFile.write( 'export Cascadedir=' + prefix + '/MCGenerators' + hepmc_suff + '/cascade/' + versions[ 'cascade-version'] + '/' + sysname + '\n' )
00341   envVarsFile.write( 'export Tauoladir=' + prefix + '/MCGenerators' + hepmc_suff + '/tauola/' + versions[ 'tauola-version'] + '/' + sysname + '\n' )
00342   envVarsFile.write( 'export Photosdir=' + prefix + '/MCGenerators' + hepmc_suff + '/photos/' + versions[ 'photos-version'] + '/' + sysname + '\n' )
00343 
00344     # additional environment variables
00345   envVarsFile.write( 'export Herwigppversion=' + versions[ 'herwigpp-version'] + '\n' )
00346   
00347     # add to LD_LIBRARY_PATH
00348   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Pythia8dir/lib\n" )
00349   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Pythia8DATA\n" )
00350   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Pythia6dir/lib\n" )
00351   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Alpgendir/lib\n" )
00352   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$GSLdir/lib/\n" )
00353   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HepMCdir/lib\n" )
00354   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HepPDTdir/lib\n" )
00355   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CLHEPdir/lib\n" )
00356   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$FastJetdir/lib\n" )
00357   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$LHAPDFdir/lib\n" )
00358   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Herwigppdir/lib/Herwig++\n" )
00359   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Herwigdir/lib\n" )
00360   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Jimmydir/lib\n" )
00361   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$ThePEGdir/lib/ThePEG\n" )
00362   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Cascadedir/lib\n" )
00363   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Tauoladir/lib\n" )
00364   envVarsFile.write( "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$Photosdir/lib\n" )
00365   envVarsFile.write( "export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$MYCLASS_PATH/lib\n" )
00366 
00367   return  
00368 
00369 # ---------------------------------------------------------------------- 
00370 def detectLocation():
00371   loc = os.popen( 'hostname -d' ).read()
00372   print 'Location:', loc,
00373   curr_dir = os.popen( 'pwd' ).read()
00374   print 'Current working directory:', curr_dir,
00375   return loc, curr_dir
00376 
00377 # ---------------------------------------------------------------------- 
00378 def contains( theString, theQueryValue ):
00379   return theString.find( theQueryValue ) > -1 
00380 
00381 # ---------------------------------------------------------------------- 
00382 def showHelp():
00383   print 'configure.py - HepMCAnalysis tool configuration script (Version: 0.1)\n'
00384   print 'Usage: source configure [arguments]\n'
00385   print 'Arguments:'
00386   for arg in versions.keys():
00387     print '  --' + arg 
00388   print '\nTo specify different version of a generator use command like:'
00389   print '  source configure --pythia8-version=135\n'
00390 
00391 # ---------------------------------------------------------------------- 
00392 # Execution starts here  
00393 # ---------------------------------------------------------------------- 
00394 if __name__ == "__main__":
00395   main()
00396   envVarsFile.close() 
00397   gccSetupFile.close()
00398