update Install.py for LATTE to use the argparse module

from __future__ import print_function

import sys,os,re,subprocess,shutil

sys.path.append('..')

from install_helpers import error,get_cpus,fullpath,which,geturl,checkmd5sum

from install_helpers import get_cpus,fullpath,geturl,checkmd5sum

from argparse import ArgumentParser

parser = ArgumentParser(prog='Install.py',

                        description="LAMMPS library build wrapper script")

# settings

version = '1.2.1'

suffix = 'gfortran'

# known checksums for different LATTE versions. used to validate the download.

checksums = { \

        '1.1.0' : '533635721ee222d0ed2925a18fb5b294', \

        '1.2.0' : '68bf0db879da5e068a71281020239ae7', \

        '1.2.1' : '85ac414fdada2d04619c8f936344df14', \

        }

# help message

                 or: python Install.py -m gfortran

                 or: python Install.py -v 1.2.1 -b

specify one or more options, order does not matter

  -b = download and build the LATTE library

  -p = specify folder of existing LATTE installation

  -m = copy Makefile.lammps.suffix to Makefile.lammps

  -v = set version of LATTE library to download and set up (default = 1.2.1)

Example:

make lib-latte args="-b -m gfortran"   # download/build in lib/latte

make lib-latte args="-p $HOME/latte"   # use existing LATTE installation

"""

# settings

version = '1.2.1'

pgroup = parser.add_mutually_exclusive_group()

pgroup.add_argument("-b", "--build", action="store_true",

                    help="download and build the Eigen3 library")

pgroup.add_argument("-p", "--path",

                    help="specify folder of existing Eigen installation")

parser.add_argument("-m", "--machine", choices=['gfortran','ifort','linalg','serial','mpi'],

                    help="suffix of a Makefile.lammps.* file used for linking LAMMPS with this library")

parser.add_argument("-v", "--version", default=version,

                    help="set version of Eigen to download and build (default: %s)" % version)

# known checksums for different LATTE versions. used to validate the download.

checksums = { \

        '1.1.0' : '533635721ee222d0ed2925a18fb5b294', \

        '1.2.0' : '68bf0db879da5e068a71281020239ae7', \

        '1.2.1' : '85ac414fdada2d04619c8f936344df14', \

        }

args = parser.parse_args()

# parse args

args = sys.argv[1:]

nargs = len(args)

if nargs == 0: error(help=help)

homepath = "."

buildflag = False

pathflag = False

suffixflag = False

linkflag = True

iarg = 0

while iarg < nargs:

  if args[iarg] == "-p":

    if iarg+2 > nargs: error(help=help)

    lattedir = fullpath(args[iarg+1])

    pathflag = True

    iarg += 2

  elif args[iarg] == "-b":

    buildflag = True

    iarg += 1

  elif args[iarg] == "-m":

    if iarg+2 > nargs: error(help=help)

    suffix = args[iarg+1]

    suffixflag = True

    iarg += 2

  elif args[iarg] == "-v":

    if iarg+2 > nargs: error(help=help)

    version = args[iarg+1]

    iarg += 2

  else: error(help=help)

# print help message and exit, if neither build nor path options are given

if args.build == False and not args.path:

  parser.print_help()

  sys.exit(help)

homedir = "LATTE-%s" % version

homepath = fullpath(".")

buildflag = args.build

pathflag = args.path != None

version = args.version

suffixflag = args.machine != None

suffix = args.machine

if (buildflag and pathflag):

    error("Cannot use -b and -p flag at the same time")

if (pathflag):

  lattedir = args.path

  if not os.path.isdir(lattedir): sys.exit("LATTE path %s does not exist" % lattedir)

  lattedir = fullpath(lattedir)

homedir = "LATTE-%s" % version

if buildflag:

  url = "https://github.com/lanl/LATTE/archive/v%s.tar.gz" % version

  lattepath = fullpath(homepath)

  lattedir = "%s/%s" % (lattepath,homedir)

if pathflag:

  if not os.path.isdir(lattedir): error("LATTE path does not exist")

# download and unpack LATTE tarball

if buildflag:

  # verify downloaded archive integrity via md5 checksum, if known.

  if version in checksums:

    if not checkmd5sum(checksums[version],'LATTE.tar.gz'):

      error("Checksum for LATTE library does not match")

      sys.exit("Checksum for LATTE library does not match")

  print("Unpacking LATTE ...")

  if os.path.exists(lattedir):

    txt = subprocess.check_output(cmd,stderr=subprocess.STDOUT,shell=True)

    print(txt.decode('UTF-8'))

  except subprocess.CalledProcessError as e:

    print("Make failed with:\n %s" % e.output.decode('UTF-8'))

    sys.exit(1)

    sys.exit("Make failed with:\n %s" % e.output.decode('UTF-8'))

# create 3 links in lib/latte to LATTE dirs

# do this -b or -p is set

if buildflag or pathflag:

print("Creating links to LATTE files")

if os.path.isfile("includelink") or os.path.islink("includelink"):

  os.remove("includelink")

# copy Makefile.lammps.suffix to Makefile.lammps

if suffixflag:

if suffixflag or not os.path.exists("Makefile.lammps"):

  print("Creating Makefile.lammps")

  if os.path.exists("Makefile.lammps.%s" % suffix):

    cmd = 'cp Makefile.lammps.%s Makefile.lammps' % suffix