Subversion Repositories slepc-dev

Rev

Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
808 dsic.upv.es!antodo 1 
import os
2 
import sys
912 dsic.upv.es!antodo 3 
import commands
808 dsic.upv.es!antodo 4 
 
912 dsic.upv.es!antodo 5 
import petscconf
923 dsic.upv.es!antodo 6 
import log
1071 slepc 7 
 
8 
def LinkWithOutput(functions,callbacks,flags):
1072 slepc 9 
  code = '#include "petsc.h"\n'
10 
  code += 'EXTERN_C_BEGIN\n'
860 dsic.upv.es!antodo 11 
  for f in functions:
1072 slepc 12 
    code += 'EXTERN int\n' + f + '();\n'
13 
  code += 'EXTERN_C_END\n'
860 dsic.upv.es!antodo 14 
 
808 dsic.upv.es!antodo 15 
  for c in callbacks:
1072 slepc 16 
    code += 'int '+ c + '() { return 0; } \n'
860 dsic.upv.es!antodo 17 
 
1072 slepc 18 
  code += 'int main() {\n'
808 dsic.upv.es!antodo 19 
  for f in functions:
1072 slepc 20 
    code += f + '();\n'
21 
  code += 'return 0;\n}\n'
22 
 
23 
  os.chdir('config')
24 
  cfile = open('checklink.c','w')
25 
  cfile.write(code)
808 dsic.upv.es!antodo 26 
  cfile.close()
912 dsic.upv.es!antodo 27 
  (result, output) = commands.getstatusoutput(petscconf.MAKE + ' checklink TESTFLAGS="'+str.join(' ',flags)+'"')
808 dsic.upv.es!antodo 28 
  os.chdir(os.pardir)
29 
  if result:
1072 slepc 30 
    return (0,code + output)
1071 slepc 31 
  else:
1072 slepc 32 
    return (1,code + output)  
1071 slepc 33 
 
34 
def Link(functions,callbacks,flags):
35 
  (result, output) = LinkWithOutput(functions,callbacks,flags)
1072 slepc 36 
  log.Write(output)
1071 slepc 37 
  return result
808 dsic.upv.es!antodo 38 
 
912 dsic.upv.es!antodo 39 
def FortranLink(functions,callbacks,flags):
1072 slepc 40 
  output =  '\n=== With linker flags: '+str.join(' ',flags)
1071 slepc 41 
 
808 dsic.upv.es!antodo 42 
  f = []
43 
  for i in functions:
44 
    f.append(i+'_')
45 
  c = []
46 
  for i in callbacks:
1071 slepc 47 
    c.append(i+'_')
48 
  (result, output1) = LinkWithOutput(f,c,flags)
1072 slepc 49 
  output1 = '\n====== With underscore Fortran names\n' + output1
50 
  if result: return ('UNDERSCORE',output1)
1071 slepc 51 
 
808 dsic.upv.es!antodo 52 
  f = []
53 
  for i in functions:
54 
    f.append(i.upper())
55 
  c = []
56 
  for i in callbacks:
57 
    c.append(i.upper())  
1071 slepc 58 
  (result, output2) = LinkWithOutput(f,c,flags)
1072 slepc 59 
  output2 = '\n====== With capital Fortran names\n' + output2
60 
  if result: return ('CAPS',output2)
808 dsic.upv.es!antodo 61 
 
1071 slepc 62 
  (result, output3) = LinkWithOutput(functions,callbacks,flags)
1072 slepc 63 
  output3 = '\n====== With unmodified Fortran names\n' + output3
64 
  if result: return ('STDCALL',output3)
1071 slepc 65 
 
1072 slepc 66 
  return ('',output + output1 + output2 + output3)
1071 slepc 67 
 
912 dsic.upv.es!antodo 68 
def GenerateGuesses(name):
808 dsic.upv.es!antodo 69 
  installdirs = ['/usr/local','/opt']
70 
  if 'HOME' in os.environ:
71 
    installdirs.insert(0,os.environ['HOME'])
72 
 
73 
  dirs = []
74 
  for i in installdirs:
75 
    dirs = dirs + [i + '/lib']
76 
    for d in [name,name.upper(),name.lower()]:
77 
      dirs = dirs + [i + '/' + d]
78 
      dirs = dirs + [i + '/' + d + '/lib']
79 
      dirs = dirs + [i + '/lib/' + d]
80 
 
860 dsic.upv.es!antodo 81 
  for d in dirs[:]:
808 dsic.upv.es!antodo 82 
    if not os.path.exists(d):
83 
      dirs.remove(d)
84 
  dirs = [''] + dirs
85 
  return dirs
86 
 
912 dsic.upv.es!antodo 87 
def FortranLib(conf,name,dirs,libs,functions,callbacks = []):
1072 slepc 88 
  log.Write('='*80)
923 dsic.upv.es!antodo 89 
  log.Println('Checking '+name+' library...')
912 dsic.upv.es!antodo 90 
 
1071 slepc 91 
  error = ''
92 
  mangling = ''
808 dsic.upv.es!antodo 93 
  for d in dirs:
94 
    for l in libs:
95 
      if d:
96 
        flags = ['-L' + d] + l
97 
      else:
98 
        flags = l
1071 slepc 99 
      (mangling, output) = FortranLink(functions,callbacks,flags)
100 
      error += output
808 dsic.upv.es!antodo 101 
      if mangling: break
102 
    if mangling: break    
103 
 
1072 slepc 104 
  if mangling:
105 
    log.Write(output);
106 
  else:
1071 slepc 107 
    log.Write(error);
912 dsic.upv.es!antodo 108 
    print 'ERROR: Unable to link with library',name
109 
    print 'ERROR: In directories',dirs
110 
    print 'ERROR: With flags',libs
1071 slepc 111 
    print 'ERROR: See "configure_log_' + petscconf.ARCH + '" file for details'
912 dsic.upv.es!antodo 112 
    sys.exit(1)
113 
 
808 dsic.upv.es!antodo 114 
 
115 
  conf.write('SLEPC_HAVE_' + name + ' = -DSLEPC_HAVE_' + name + ' -DSLEPC_' + name + '_HAVE_'+mangling+'\n')
116 
  conf.write(name + '_LIB = '+str.join(' ',flags)+'\n')
1071 slepc 117 
  return flags