hepmc2make.py

#!/usr/bin/env python3

""" Simple script that handles the installation of hepmc2
"""

import subprocess, os, sys
    
ver="2.06.11"
tarball = f"hepmc{ver}.tgz"
path = f"HepMC-{ver}"
libraryname = f"{path}/fio/libHepMCfio.la"
includename = f"{path}/HepMC/HepMCDefs.h"

def fetchTarball():
    """ fetch the hepmc2.06.11.tgz tarball """
    if not os.path.exists ( tarball ):
        cmd = f"wget https://smodels.github.io/downloads/tarballs/{tarball}"
        o = subprocess.getoutput ( cmd )
        print ( f"[hepmc2make] {cmd}: {o}" )

def explodeTarball():
    """ fetch the tarball, then explode it """
    fetchTarball()
    if os.path.exists ( path ):
        shutil.rmtree ( path )
    cmd = f"tar xzvf {tarball}"
    subprocess.getoutput ( cmd )

def makeHepmc2():
    """ fetch the tarball, then explode it, then make hepmc2 """
    explodeTarball()
    cmd = f"cd {path}; ./configure --with-momentum=GEV --with-length=CM"
    o = subprocess.getoutput ( cmd )
    print ( f"[hepmc2make] {cmd}: {o}" )
    cmd = f" cd {path}; make -j 2"
    o = subprocess.getoutput ( cmd )
    print ( f"[hepmc2make] {cmd}: {o}" )

def install():
    if os.path.exists ( libraryname ) and os.path.exists ( includename ):
        print ( f"[hepmc2make] not installing hepmc2: {libraryname} and {includename} exist" )
        return
    if os.path.exists ( path ):
        ## so path exists, but not the library. clean!
        clean()
    print ( "[hepmc2make] installing hepmc2 ..." )
    makeHepmc2()

def clean():
    import glob
    for file in glob.glob ( "*" ):
        if file not in [ "make.py", "Makefile" ]:
            cmd = "rm -rf %s" % file
            subprocess.getoutput ( cmd )

if __name__ == "__main__":
    import inspect
    D = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) 
    os.chdir ( D )
    if len(sys.argv)>1 and sys.argv[1]=="clean":
        clean()
        sys.exit()
    install()