Subversion Repositories slepc-dev

#

#  - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

#  SLEPc - Scalable Library for Eigenvalue Problem Computations

#  Copyright (c) 2002-2009, Universidad Politecnica de Valencia, Spain

#

#  This file is part of SLEPc.

#     

#  SLEPc is free software: you can redistribute it and/or modify it under  the

#  terms of version 3 of the GNU Lesser General Public License as published by

#  the Free Software Foundation.

#

#  SLEPc  is  distributed in the hope that it will be useful, but WITHOUT  ANY 

#  WARRANTY;  without even the implied warranty of MERCHANTABILITY or  FITNESS 

#  FOR  A  PARTICULAR PURPOSE. See the GNU Lesser General Public  License  for 

#  more details.

#

#  You  should have received a copy of the GNU Lesser General  Public  License

#  along with SLEPc. If not, see <http://www.gnu.org/licenses/>.

#  - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

#

import os

import sys

import commands

import petscconf

import log

def LinkWithOutput(functions,callbacks,flags):

  code = '#include "petscksp.h"\n'

  code += 'EXTERN_C_BEGIN\n'

  for f in functions:

    code += 'EXTERN int\n' + f + '();\n'

  code += 'EXTERN_C_END\n'

  for c in callbacks:

    code += 'int '+ c + '() { return 0; } \n'

  code += 'int main() {\n'

  code += 'PetscInitialize(PETSC_NULL,PETSC_NULL,PETSC_NULL,PETSC_NULL);\n'

  code += 'VecCreate(PETSC_NULL,PETSC_NULL);\n'

  code += 'MatCreate(PETSC_NULL,PETSC_NULL);\n'

  code += 'KSPCreate(PETSC_NULL,PETSC_NULL);\n'

  for f in functions:

    code += f + '();\n'

  code += 'return 0;\n}\n'

  os.chdir('config')

  cfile = open('checklink.c','w')

  cfile.write(code)

  cfile.close()

  (result, output) = commands.getstatusoutput(petscconf.MAKE + ' checklink TESTFLAGS="'+str.join(' ',flags)+'"')

  if os.path.exists('checklink.o'):

    os.unlink('checklink.o')

  os.chdir(os.pardir)

  if result:

    return (0,code + output)

  else:

    return (1,code + output)  

def Link(functions,callbacks,flags):

  (result, output) = LinkWithOutput(functions,callbacks,flags)

  log.Write(output)

  return result

def FortranLink(functions,callbacks,flags):

  output =  '\n=== With linker flags: '+str.join(' ',flags)

  f = []

  for i in functions:

    f.append(i+'_')

  c = []

  for i in callbacks:

    c.append(i+'_')

  (result, output1) = LinkWithOutput(f,c,flags)

  output1 = '\n====== With underscore Fortran names\n' + output1

  if result: return ('UNDERSCORE',output1)

  f = []

  for i in functions:

    f.append(i.upper())

  c = []

  for i in callbacks:

    c.append(i.upper())  

  (result, output2) = LinkWithOutput(f,c,flags)

  output2 = '\n====== With capital Fortran names\n' + output2

  if result: return ('CAPS',output2)

  (result, output3) = LinkWithOutput(functions,callbacks,flags)

  output3 = '\n====== With unmodified Fortran names\n' + output3

  if result: return ('STDCALL',output3)

  return ('',output + output1 + output2 + output3)

def GenerateGuesses(name):

  installdirs = ['/usr/local','/opt']

  if 'HOME' in os.environ:

    installdirs.insert(0,os.environ['HOME'])

  dirs = []

  for i in installdirs:

    dirs = dirs + [i + '/lib']

    for d in [name,name.upper(),name.lower()]:

      dirs = dirs + [i + '/' + d]

      dirs = dirs + [i + '/' + d + '/lib']

      dirs = dirs + [i + '/lib/' + d]

  for d in dirs[:]:

    if not os.path.exists(d):

      dirs.remove(d)

  dirs = [''] + dirs

  return dirs

def FortranLib(conf,name,dirs,libs,functions,callbacks = []):

  log.Write('='*80)

  log.Println('Checking '+name+' library...')

  error = ''

  mangling = ''

  for d in dirs:

    for l in libs:

      if d:

        flags = ['-L' + d] + l

      else:

        flags = l

      (mangling, output) = FortranLink(functions,callbacks,flags)

      error += output

      if mangling: break

    if mangling: break    

  if mangling:

    log.Write(output)

  else:

    log.Write(error)

    log.Println('ERROR: Unable to link with library '+ name)

    print 'ERROR: In directories',dirs

    print 'ERROR: With flags',libs,

    log.Exit('')

  conf.write('SLEPC_HAVE_' + name + ' = -DSLEPC_HAVE_' + name + ' -DSLEPC_' + name + '_HAVE_'+mangling+'\n')

  conf.write(name + '_LIB = '+str.join(' ',flags)+'\n')

  return flags