#!/usr/bin/env python
import os,re,sys,string,subprocess

from optparse import OptionParser, Option

# -----------------------------
class extendableOption(Option):
# -----------------------------
# used for definition of new option parser action 'extend', which enables to take multiple option arguments
# taken from online tutorial http://docs.python.org/library/optparse.html
  
  ACTIONS = Option.ACTIONS + ("extend",)
  STORE_ACTIONS = Option.STORE_ACTIONS + ("extend",)
  TYPED_ACTIONS = Option.TYPED_ACTIONS + ("extend",)
  ALWAYS_TYPED_ACTIONS = Option.ALWAYS_TYPED_ACTIONS + ("extend",)

  def take_action(self, action, dest, opt, value, values, parser):
    if action == "extend":
      lvalue = value.split(",")
      values.ensure_value(dest, []).extend(lvalue)
    else:
      Option.take_action(self, action, dest, opt, value, values, parser)



########################################################
# MAIN
########################################################

parser = OptionParser(option_class=extendableOption, usage='%prog options', description = """
Sets up the pre and post processing tools of DAMASK

""" + string.replace('$Id$','\n','\\n')
)

compilers = ['ifort','gfortran']

parser.add_option('--with-fc',               dest='compiler', type='string', \
                                             help='name of F90 [%default]')
parser.add_option('--with-fftw-dir',         dest='fftwroot', type='string', \
                                             help='root location of FFTW [%default]')
parser.add_option('--with-blas-lapack-dir',  dest='blasdir', type='string', \
                                             help='root location of lapack [%default]')
parser.add_option('--with-marc-dir',         dest='marcdir', type='string', \
                                             help='root location of MSC.Marc [%default]')
parser.add_option('--with-blas-type',        dest='blastype', type='string', \
                                             help='root location of LAPACK [%default]')
parser.add_option('--with-spectral-options',   dest='spectraloptions', type='string', \
                                             help='root location of LAPACK [%default]')

compiler = os.getenv('F90')
if compiler == None: compiler ={True:'ifort',False:'gfortran'}[\
                                subprocess.call(['which', 'ifort'],\
                                stdout=subprocess.PIPE, stderr=subprocess.PIPE) == 0]
parser.set_defaults(compiler=compiler)
parser.set_defaults(blastype='LAPACK')
parser.set_defaults(fftwroot="/usr/local")
parser.set_defaults(blasdir="/usr")
parser.set_defaults(marcdir="/msc")
(options,filenames) = parser.parse_args()
options.blastype=options.blastype.upper()
options.compiler=options.compiler.lower()
blastypes=['LAPACK','ACML','IMKL']
compilers=['gfortran','ifort']
if options.blastype not in blastypes: print('Unknown BLAS type selected')
if options.compiler not in compilers: print('Unknown compiler type selected')

f=open('installation/options','r')
f2=open('installation/options.tmp','w')
for line in f.readlines():
  if line[0:8]=='FFTWROOT': line='FFTWROOT='+options.fftwroot+'\n'
  if line[0:3]=='MSC':      line='MSC='+options.marcdir+'\n'
  if line[0:3]=='F90':      line='F90='+options.compiler+'\n'
  if options.blastype == 'LAPACK':
    if line[0:10]=='LAPACKROOT': line='LAPACKROOT='+options.blasdir+'\n'
  if options.blastype == 'ACML':
    if line[0:8]=='ACMLROOT': line='ACMLROOT='+options.blasdir+'\n'
  if options.blastype == 'IMKL':
    if line[0:8]=='IMKLROOT': line='IMKLROOT='+options.blasdir+'\n'
  f2.write(line)
for spectralOption in options.spectraloptions.split():
  f2.write(spectralOption+'\n')
f.close()
f2.close()
os.rename('installation/options.tmp','installation/options')