xref: /petsc/config/PETSc/Configure.py (revision ac199c897b92e43b66f3d89832efd26e2fbb07b0)
1import config.base
2
3import os
4import re
5
6class Configure(config.base.Configure):
7  def __init__(self, framework):
8    config.base.Configure.__init__(self, framework)
9    self.headerPrefix = 'PETSC'
10    self.substPrefix  = 'PETSC'
11    self.defineAutoconfMacros()
12    return
13
14  def __str2__(self):
15    desc = []
16    desc.append('xxx=========================================================================xxx')
17    desc.append('   Configure stage complete. Now build PETSc libraries with:')
18    desc.append('   make PETSC_DIR='+self.petscdir.dir+' PETSC_ARCH='+self.arch.arch+' all')
19    desc.append('xxx=========================================================================xxx')
20    return '\n'.join(desc)+'\n'
21
22  def setupHelp(self, help):
23    import nargs
24    help.addArgument('PETSc',  '-prefix=<dir>',                  nargs.Arg(None, '', 'Specifiy location to install PETSc (eg. /usr/local)'))
25    help.addArgument('Windows','-with-windows-graphics=<bool>',   nargs.ArgBool(None, 1,'Enable check for Windows Graphics'))
26    help.addArgument('PETSc', '-with-default-arch=<bool>',        nargs.ArgBool(None, 1, 'Allow using the last configured arch without setting PETSC_ARCH'))
27    help.addArgument('PETSc','-with-single-library=<bool>',       nargs.ArgBool(None, 1,'Put all PETSc code into the single -lpetsc library'))
28    help.addArgument('PETSc', '-with-iphone=<bool>',              nargs.ArgBool(None, 0, 'Build an iPhone version of PETSc'))
29    return
30
31  def setupDependencies(self, framework):
32    config.base.Configure.setupDependencies(self, framework)
33    self.setCompilers  = framework.require('config.setCompilers',      self)
34    self.arch          = framework.require('PETSc.utilities.arch',     self.setCompilers)
35    self.petscdir      = framework.require('PETSc.utilities.petscdir', self.setCompilers)
36    self.languages     = framework.require('PETSc.utilities.languages',self.setCompilers)
37    self.debugging     = framework.require('PETSc.utilities.debugging',self.setCompilers)
38    self.make          = framework.require('PETSc.utilities.Make',     self)
39    self.CHUD          = framework.require('PETSc.utilities.CHUD',     self)
40    self.compilers     = framework.require('config.compilers',         self)
41    self.types         = framework.require('config.types',             self)
42    self.headers       = framework.require('config.headers',           self)
43    self.functions     = framework.require('config.functions',         self)
44    self.libraries     = framework.require('config.libraries',         self)
45    if os.path.isdir(os.path.join('config', 'PETSc')):
46      for d in ['utilities', 'packages']:
47        for utility in os.listdir(os.path.join('config', 'PETSc', d)):
48          (utilityName, ext) = os.path.splitext(utility)
49          if not utilityName.startswith('.') and not utilityName.startswith('#') and ext == '.py' and not utilityName == '__init__':
50            utilityObj                    = self.framework.require('PETSc.'+d+'.'+utilityName, self)
51            utilityObj.headerPrefix       = self.headerPrefix
52            utilityObj.archProvider       = self.arch
53            utilityObj.languageProvider   = self.languages
54            utilityObj.installDirProvider = self.petscdir
55            setattr(self, utilityName.lower(), utilityObj)
56    self.qd    = framework.require('PETSc.packages.qd', self)
57    self.qd.archProvider      = self.arch
58    self.qd.precisionProvider = self.scalartypes
59    self.qd.installDirProvider= self.petscdir
60
61    for package in config.packages.all:
62      if not package == 'PETSc':
63        packageObj                    = framework.require('config.packages.'+package, self)
64        packageObj.archProvider       = self.arch
65        packageObj.languageProvider   = self.languages
66        packageObj.installDirProvider = self.petscdir
67        setattr(self, package.lower(), packageObj)
68    # Force blaslapack to depend on scalarType so precision is set before BlasLapack is built
69    framework.require('PETSc.utilities.scalarTypes', self.blaslapack)
70    self.blaslapack.precisionProvider = self.scalartypes
71
72    self.compilers.headerPrefix  = self.headerPrefix
73    self.types.headerPrefix      = self.headerPrefix
74    self.headers.headerPrefix    = self.headerPrefix
75    self.functions.headerPrefix  = self.headerPrefix
76    self.libraries.headerPrefix  = self.headerPrefix
77    self.blaslapack.headerPrefix = self.headerPrefix
78    self.mpi.headerPrefix        = self.headerPrefix
79    headersC = map(lambda name: name+'.h', ['dos', 'endian', 'fcntl', 'float', 'io', 'limits', 'malloc', 'pwd', 'search', 'strings',
80                                            'unistd', 'machine/endian', 'sys/param', 'sys/procfs', 'sys/resource',
81                                            'sys/systeminfo', 'sys/times', 'sys/utsname','string', 'stdlib','memory',
82                                            'sys/socket','sys/wait','netinet/in','netdb','Direct','time','Ws2tcpip','sys/types',
83                                            'WindowsX', 'cxxabi','float','ieeefp','stdint','fenv'])
84    functions = ['access', '_access', 'clock', 'drand48', 'getcwd', '_getcwd', 'getdomainname', 'gethostname', 'getpwuid',
85                 'gettimeofday', 'getwd', 'memalign', 'memmove', 'mkstemp', 'popen', 'PXFGETARG', 'rand', 'getpagesize',
86                 'readlink', 'realpath',  'sigaction', 'signal', 'sigset', 'nanosleep', 'usleep', 'sleep', '_sleep', 'socket',
87                 'times', 'gethostbyname', 'uname','snprintf','_snprintf','_fullpath','lseek','_lseek','time','fork','stricmp',
88                 'strcasecmp', 'bzero', 'dlopen', 'dlsym', 'dlclose', 'dlerror',
89                 '_intel_fast_memcpy','_intel_fast_memset']
90    libraries1 = [(['socket', 'nsl'], 'socket'), (['fpe'], 'handle_sigfpes')]
91    self.headers.headers.extend(headersC)
92    self.functions.functions.extend(functions)
93    self.libraries.libraries.extend(libraries1)
94    return
95
96  def defineAutoconfMacros(self):
97    self.hostMacro = 'dnl Version: 2.13\ndnl Variable: host_cpu\ndnl Variable: host_vendor\ndnl Variable: host_os\nAC_CANONICAL_HOST'
98    return
99
100  def Dump(self):
101    ''' Actually put the values into the configuration files '''
102    # eventually everything between -- should be gone
103#-----------------------------------------------------------------------------------------------------
104
105    # Sometimes we need C compiler, even if built with C++
106    self.setCompilers.pushLanguage('C')
107    self.addMakeMacro('CC_FLAGS',self.setCompilers.getCompilerFlags())
108    self.setCompilers.popLanguage()
109
110    # C preprocessor values
111    self.addMakeMacro('CPP_FLAGS',self.setCompilers.CPPFLAGS+self.CHUD.CPPFLAGS)
112
113    # compiler values
114    self.setCompilers.pushLanguage(self.languages.clanguage)
115    self.addMakeMacro('PCC',self.setCompilers.getCompiler())
116    self.addMakeMacro('PCC_FLAGS',self.setCompilers.getCompilerFlags())
117    self.setCompilers.popLanguage()
118    # .o or .obj
119    self.addMakeMacro('CC_SUFFIX','o')
120
121    # executable linker values
122    self.setCompilers.pushLanguage(self.languages.clanguage)
123    pcc_linker = self.setCompilers.getLinker()
124    self.addMakeMacro('PCC_LINKER',pcc_linker)
125    self.addMakeMacro('PCC_LINKER_FLAGS',self.setCompilers.getLinkerFlags())
126    self.setCompilers.popLanguage()
127    # '' for Unix, .exe for Windows
128    self.addMakeMacro('CC_LINKER_SUFFIX','')
129    self.addMakeMacro('PCC_LINKER_LIBS',self.libraries.toStringNoDupes(self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' '))+self.CHUD.LIBS)
130
131    if hasattr(self.compilers, 'FC'):
132      self.setCompilers.pushLanguage('FC')
133      # need FPPFLAGS in config/setCompilers
134      self.addDefine('HAVE_FORTRAN','1')
135      self.addMakeMacro('FPP_FLAGS',self.setCompilers.CPPFLAGS)
136
137      # compiler values
138      self.addMakeMacro('FC_FLAGS',self.setCompilers.getCompilerFlags())
139      self.setCompilers.popLanguage()
140      # .o or .obj
141      self.addMakeMacro('FC_SUFFIX','o')
142
143      # executable linker values
144      self.setCompilers.pushLanguage('FC')
145      # Cannot have NAG f90 as the linker - so use pcc_linker as fc_linker
146      fc_linker = self.setCompilers.getLinker()
147      if config.setCompilers.Configure.isNAG(fc_linker):
148        self.addMakeMacro('FC_LINKER',pcc_linker)
149      else:
150        self.addMakeMacro('FC_LINKER',fc_linker)
151      self.addMakeMacro('FC_LINKER_FLAGS',self.setCompilers.getLinkerFlags())
152      # apple requires this shared library linker flag on SOME versions of the os
153      if self.setCompilers.getLinkerFlags().find('-Wl,-commons,use_dylibs') > -1:
154        self.addMakeMacro('DARWIN_COMMONS_USE_DYLIBS',' -Wl,-commons,use_dylibs ')
155      self.setCompilers.popLanguage()
156
157      # F90 Modules
158      if self.setCompilers.fortranModuleIncludeFlag:
159        self.addMakeMacro('FC_MODULE_FLAG', self.setCompilers.fortranModuleIncludeFlag)
160      else: # for non-f90 compilers like g77
161        self.addMakeMacro('FC_MODULE_FLAG', '-I')
162      if self.setCompilers.fortranModuleIncludeFlag:
163        self.addMakeMacro('FC_MODULE_OUTPUT_FLAG', self.setCompilers.fortranModuleOutputFlag)
164    else:
165      self.addMakeMacro('FC','')
166
167    if hasattr(self.compilers, 'CUDAC'):
168      self.setCompilers.pushLanguage('CUDA')
169      self.addMakeMacro('CUDAC_FLAGS',self.setCompilers.getCompilerFlags())
170      self.setCompilers.popLanguage()
171
172    # shared library linker values
173    self.setCompilers.pushLanguage(self.languages.clanguage)
174    # need to fix BuildSystem to collect these separately
175    self.addMakeMacro('SL_LINKER',self.setCompilers.getLinker())
176    self.addMakeMacro('SL_LINKER_FLAGS','${PCC_LINKER_FLAGS}')
177    self.setCompilers.popLanguage()
178    # One of 'a', 'so', 'lib', 'dll', 'dylib' (perhaps others also?) depending on the library generator and architecture
179    # Note: . is not included in this macro, consistent with AR_LIB_SUFFIX
180    if self.setCompilers.sharedLibraryExt == self.setCompilers.AR_LIB_SUFFIX:
181      self.addMakeMacro('SL_LINKER_SUFFIX', '')
182      self.addDefine('SLSUFFIX','""')
183    else:
184      self.addMakeMacro('SL_LINKER_SUFFIX', self.setCompilers.sharedLibraryExt)
185      self.addDefine('SLSUFFIX','"'+self.setCompilers.sharedLibraryExt+'"')
186
187    #SL_LINKER_LIBS is currently same as PCC_LINKER_LIBS - so simplify
188    self.addMakeMacro('SL_LINKER_LIBS','${PCC_LINKER_LIBS}')
189    #self.addMakeMacro('SL_LINKER_LIBS',self.libraries.toStringNoDupes(self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' ')))
190
191#-----------------------------------------------------------------------------------------------------
192
193    # CONLY or CPP. We should change the PETSc makefiles to do this better
194    if self.languages.clanguage == 'C': lang = 'CONLY'
195    else: lang = 'CXXONLY'
196    self.addMakeMacro('PETSC_LANGUAGE',lang)
197
198    # real or complex
199    self.addMakeMacro('PETSC_SCALAR',self.scalartypes.scalartype)
200    # double or float
201    self.addMakeMacro('PETSC_PRECISION',self.scalartypes.precision)
202
203    if self.framework.argDB['with-batch']:
204      self.addMakeMacro('PETSC_WITH_BATCH','1')
205
206    # Test for compiler-specific macros that need to be defined.
207    if self.setCompilers.isCray('CC'):
208      self.addDefine('HAVE_CRAYC','1')
209
210#-----------------------------------------------------------------------------------------------------
211    if self.functions.haveFunction('gethostbyname') and self.functions.haveFunction('socket'):
212      self.addDefine('USE_SOCKET_VIEWER','1')
213
214#-----------------------------------------------------------------------------------------------------
215    # print include and lib for external packages
216    self.framework.packages.reverse()
217    includes = []
218    libs = []
219    for i in self.framework.packages:
220      if i.useddirectly:
221        self.addDefine('HAVE_'+i.PACKAGE, 1)  # ONLY list package if it is used directly by PETSc (and not only by another package)
222      if not isinstance(i.lib, list):
223        i.lib = [i.lib]
224      libs.extend(i.lib)
225      self.addMakeMacro(i.PACKAGE+'_LIB', self.libraries.toStringNoDupes(i.lib))
226      if hasattr(i,'include'):
227        if not isinstance(i.include,list):
228          i.include = [i.include]
229        includes.extend(i.include)
230        self.addMakeMacro(i.PACKAGE+'_INCLUDE',self.headers.toStringNoDupes(i.include))
231    self.addMakeMacro('PACKAGES_LIBS',self.libraries.toStringNoDupes(libs+self.libraries.math))
232    self.PACKAGES_LIBS = self.libraries.toStringNoDupes(libs+self.libraries.math)
233    self.addMakeMacro('PACKAGES_INCLUDES',self.headers.toStringNoDupes(includes))
234    self.PACKAGES_INCLUDES = self.headers.toStringNoDupes(includes)
235    if hasattr(self.compilers, 'FC'):
236      if self.compilers.fortranIsF90:
237        self.addMakeMacro('PACKAGES_MODULES_INCLUDES',self.headers.toStringModulesNoDupes(includes))
238
239    self.addMakeMacro('DESTDIR',self.installdir)
240    self.addDefine('LIB_DIR','"'+os.path.join(self.installdir,'lib')+'"')
241
242    if self.framework.argDB['with-single-library']:
243      # overrides the values set in conf/variables
244      self.addMakeMacro('LIBNAME','${INSTALL_LIB_DIR}/libpetsc.${AR_LIB_SUFFIX}')
245      self.addMakeMacro('PETSC_SYS_LIB_BASIC','-lpetsc')
246      self.addMakeMacro('PETSC_VEC_LIB_BASIC','-lpetsc')
247      self.addMakeMacro('PETSC_MAT_LIB_BASIC','-lpetsc')
248      self.addMakeMacro('PETSC_DM_LIB_BASIC','-lpetsc')
249      self.addMakeMacro('PETSC_KSP_LIB_BASIC','-lpetsc')
250      self.addMakeMacro('PETSC_SNES_LIB_BASIC','-lpetsc')
251      self.addMakeMacro('PETSC_TS_LIB_BASIC','-lpetsc')
252      self.addMakeMacro('PETSC_LIB_BASIC','-lpetsc')
253      self.addMakeMacro('PETSC_CONTRIB_BASIC','-lpetsc')
254      self.addMakeMacro('SHLIBS','libpetsc')
255      self.addDefine('USE_SINGLE_LIBRARY', '1')
256
257    if not os.path.exists(os.path.join(self.petscdir.dir,self.arch.arch,'lib')):
258      os.makedirs(os.path.join(self.petscdir.dir,self.arch.arch,'lib'))
259
260    # add a makefile entry for configure options
261    self.addMakeMacro('CONFIGURE_OPTIONS', self.framework.getOptionsString(['configModules', 'optionsModule']).replace('\"','\\"'))
262    return
263
264  def dumpConfigInfo(self):
265    import time
266    fd = file(os.path.join(self.arch.arch,'include','petscconfiginfo.h'),'w')
267    fd.write('static const char *petscconfigureruntime = "'+time.ctime(time.time())+'";\n')
268    fd.write('static const char *petscconfigureoptions = "'+self.framework.getOptionsString(['configModules', 'optionsModule']).replace('\"','\\"')+'";\n')
269    fd.close()
270    return
271
272  def dumpMachineInfo(self):
273    import platform
274    import time
275    import script
276    fd = file(os.path.join(self.arch.arch,'include','petscmachineinfo.h'),'w')
277    fd.write('static const char *petscmachineinfo = \"\\n\"\n')
278    fd.write('\"-----------------------------------------\\n\"\n')
279    fd.write('\"Libraries compiled on %s on %s \\n\"\n' % (time.ctime(time.time()), platform.node()))
280    fd.write('\"Machine characteristics: %s\\n\"\n' % (platform.platform()))
281    fd.write('\"Using PETSc directory: %s\\n\"\n' % (self.petscdir.dir))
282    fd.write('\"Using PETSc arch: %s\\n\"\n' % (self.arch.arch))
283    fd.write('\"-----------------------------------------\\n\";\n')
284    fd.write('static const char *petsccompilerinfo = \"\\n\"\n')
285    self.setCompilers.pushLanguage(self.languages.clanguage)
286    fd.write('\"Using C compiler: %s %s ${COPTFLAGS} ${CFLAGS}\\n\"\n' % (self.setCompilers.getCompiler(), self.setCompilers.getCompilerFlags()))
287    self.setCompilers.popLanguage()
288    if hasattr(self.compilers, 'FC'):
289      self.setCompilers.pushLanguage('FC')
290      fd.write('\"Using Fortran compiler: %s %s ${FOPTFLAGS} ${FFLAGS} %s\\n\"\n' % (self.setCompilers.getCompiler(), self.setCompilers.getCompilerFlags(), self.setCompilers.CPPFLAGS))
291      self.setCompilers.popLanguage()
292    fd.write('\"-----------------------------------------\\n\";\n')
293    fd.write('static const char *petsccompilerflagsinfo = \"\\n\"\n')
294    fd.write('\"Using include paths: %s %s %s\\n\"\n' % ('-I'+os.path.join(self.petscdir.dir, self.arch.arch, 'include'), '-I'+os.path.join(self.petscdir.dir, 'include'), self.PACKAGES_INCLUDES))
295    fd.write('\"-----------------------------------------\\n\";\n')
296    fd.write('static const char *petsclinkerinfo = \"\\n\"\n')
297    self.setCompilers.pushLanguage(self.languages.clanguage)
298    fd.write('\"Using C linker: %s\\n\"\n' % (self.setCompilers.getLinker()))
299    self.setCompilers.popLanguage()
300    if hasattr(self.compilers, 'FC'):
301      self.setCompilers.pushLanguage('FC')
302      fd.write('\"Using Fortran linker: %s\\n\"\n' % (self.setCompilers.getLinker()))
303      self.setCompilers.popLanguage()
304    fd.write('\"Using libraries: %s%s -L%s %s %s %s\\n\"\n' % (self.setCompilers.CSharedLinkerFlag, os.path.join(self.petscdir.dir, self.arch.arch, 'lib'), os.path.join(self.petscdir.dir, self.arch.arch, 'lib'), '-lpetscts -lpetscsnes -lpetscksp -lpetscdm -lpetscmat -lpetscvec -lpetscsys', self.PACKAGES_LIBS, self.libraries.toStringNoDupes(self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' '))+self.CHUD.LIBS))
305    fd.write('\"-----------------------------------------\\n\";\n')
306    fd.close()
307    return
308
309  def dumpCMakeConfig(self):
310    '''
311    Writes configuration-specific values to ${PETSC_ARCH}/conf/PETScConfig.cmake.
312    This file is private to PETSc and should not be included by third parties
313    (a suitable file can be produced later by CMake, but this is not it).
314    '''
315    def cmakeset(fd,key,val=True):
316      if val == True: val = 'YES'
317      if val == False: val = 'NO'
318      fd.write('set (' + key + ' ' + val + ')\n')
319    def ensurelist(a):
320      if isinstance(a,list):
321        return a
322      else:
323        return [a]
324    def libpath(lib):
325      'Returns a search path if that is what this item provides, else "" which will be cleaned out later'
326      if lib.startswith('-L'): return lib[2:]
327      if lib.startswith('-R'): return lib[2:]
328      if lib.startswith('-Wl,-rpath,'):
329        # This case occurs when an external package needs a specific system library that is normally provided by the compiler.
330        # In other words, the -L path is builtin to the wrapper or compiler, here we provide it so that CMake can locate the
331        # corresponding library.
332        return lib[len('-Wl,-rpath,'):]
333      if lib.startswith('-'): return ''
334      return os.path.dirname(lib)
335    def cleanlib(lib):
336      'Returns a library name if that is what this item provides, else "" which will be cleaned out later'
337      if lib.startswith('-l'):  return lib[2:]
338      if lib.startswith('-Wl') or lib.startswith('-L'): return ''
339      lib = os.path.splitext(os.path.basename(lib))[0]
340      if lib.startswith('lib'): return lib[3:]
341      return lib
342    def nub(lst):
343      unique = []
344      for elem in lst:
345        if elem not in unique and elem != '':
346          unique.append(elem)
347      return unique
348    def cmakeexpand(varname):
349      return r'"${' + varname + r'}"'
350    def uniqextend(list,new):
351      for x in ensurelist(new):
352        if x not in list:
353          list.append(x)
354    def notstandardinclude(path):
355      return path not in '/usr/include /usr/local/include'.split()
356    def writeMacroDefinitions(fd):
357      if self.mpi.usingMPIUni:
358        cmakeset(fd,'PETSC_HAVE_MPIUNI')
359      for pkg in self.framework.packages:
360        if pkg.useddirectly:
361          cmakeset(fd,'PETSC_HAVE_' + pkg.PACKAGE)
362      for name,val in self.functions.defines.items():
363        cmakeset(fd,'PETSC_'+name,val)
364      for dct in [self.defines, self.libraryoptions.defines]:
365        for k,v in dct.items():
366          if k.startswith('USE_'):
367            cmakeset(fd,'PETSC_' + k, v)
368      cmakeset(fd,'PETSC_USE_COMPLEX', self.scalartypes.scalartype == 'complex')
369      cmakeset(fd,'PETSC_USE_SCALAR_' + self.scalartypes.precision.upper())
370      cmakeset(fd,'PETSC_CLANGUAGE_'+self.languages.clanguage)
371      if hasattr(self.compilers, 'FC'):
372        cmakeset(fd,'PETSC_HAVE_FORTRAN')
373        if self.compilers.fortranIsF90:
374          cmakeset(fd,'PETSC_USING_F90')
375      if self.sharedlibraries.useShared:
376        cmakeset(fd,'BUILD_SHARED_LIBS')
377    def writeBuildFlags(fd):
378      lib_paths = []
379      lib_libs  = []
380      includes  = []
381      libvars   = []
382      for pkg in self.framework.packages:
383        libs = ensurelist(pkg.lib)
384        lib_paths.extend(map(libpath,libs))
385        lib_libs.extend(map(cleanlib,libs))
386        uniqextend(includes,pkg.include)
387      if self.libraries.math: lib_libs.extend(map(cleanlib,self.libraries.math))
388      if self.libraries.rt: lib_libs.extend(map(cleanlib,self.libraries.rt))
389      for libname in nub(lib_libs):
390        libvar = 'PETSC_' + libname.upper() + '_LIB'
391        addpath = ''
392        for lpath in nub(lib_paths):
393          addpath += '"' + str(lpath) + '" '
394        fd.write('find_library (' + libvar + ' ' + libname + ' HINTS ' + addpath + ')\n')
395        libvars.append(libvar)
396      fd.write('mark_as_advanced (' + ' '.join(libvars) + ')\n')
397      fd.write('set (PETSC_PACKAGE_LIBS ' + ' '.join(map(cmakeexpand,libvars)) + ')\n')
398      fd.write('set (PETSC_PACKAGE_INCLUDES ' + ' '.join(map(lambda i: '"'+i+'"',filter(notstandardinclude,includes))) + ')\n')
399    fd = open(os.path.join(self.arch.arch,'conf','PETScConfig.cmake'), 'w')
400    writeMacroDefinitions(fd)
401    writeBuildFlags(fd)
402    fd.close()
403    return
404
405  def dumpCMakeLists(self):
406    import sys
407    if sys.version_info >= (2,5):
408      sys.path.insert(0,os.path.join(self.petscdir.dir,'bin','maint'))
409      import cmakegen
410      del sys.path[0]
411      try:
412        cmakegen.main(self.petscdir.dir)
413      except (OSError), e:
414        self.framework.logPrint('Generating CMakeLists.txt failed:\n' + str(e))
415
416  def cmakeBoot(self):
417    import sys
418    if sys.version_info >= (2,5) and hasattr(self.cmake,'cmake'):
419      sys.path.insert(0,os.path.join(self.petscdir.dir,'bin','maint'))
420      import cmakeboot
421      del sys.path[0]
422      try:
423        cmakeboot.main(petscdir=self.petscdir.dir,petscarch=self.arch.arch,argDB=self.argDB,framework=self.framework,logPrint=self.framework.logPrint)
424      except (OSError), e:
425        self.framework.logPrint('Booting CMake in PETSC_ARCH failed:\n' + str(e))
426
427  def configurePrefetch(self):
428    '''Sees if there are any prefetch functions supported'''
429    if config.setCompilers.Configure.isSolaris() or self.framework.argDB['with-iphone'] or self.framework.argDB['with-cuda']:
430      self.addDefine('Prefetch(a,b,c)', ' ')
431      return
432    self.pushLanguage(self.languages.clanguage)
433    if self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch(v,(int)0);\n'):
434      self.addDefine('HAVE_XMMINTRIN_H', 1)
435      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const void*)(a),(int)(c))')
436    elif self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch((const char*)v,(int)0);\n'):
437      self.addDefine('HAVE_XMMINTRIN_H', 1)
438      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const char*)(a),(int)(c))')
439    elif self.checkLink('', 'void *v = 0;__builtin_prefetch(v,0,0);\n'):
440      self.addDefine('Prefetch(a,b,c)', '__builtin_prefetch((a),(b),(c))')
441    else:
442      self.addDefine('Prefetch(a,b,c)', ' ')
443    self.popLanguage()
444
445  def configureUnused(self):
446    '''Sees if __attribute((unused)) is supported'''
447    if self.framework.argDB['with-iphone'] or self.framework.argDB['with-cuda']:
448      self.addDefine('UNUSED', ' ')
449      return
450    self.pushLanguage(self.languages.clanguage)
451    if self.checkLink('__attribute((unused)) static int myfunc(void){ return 1;}', 'int i = myfunc();\n'):
452      self.addDefine('UNUSED', '__attribute((unused))')
453    else:
454      self.addDefine('UNUSED', ' ')
455    self.popLanguage()
456
457  def configureExpect(self):
458    '''Sees if the __builtin_expect directive is supported'''
459    self.pushLanguage(self.languages.clanguage)
460    if self.checkLink('', 'if (__builtin_expect(0,1)) return 1;'):
461      self.addDefine('HAVE_BUILTIN_EXPECT', 1)
462    self.popLanguage()
463
464  def configureIntptrt(self):
465    '''Determine what to use for uintptr_t'''
466    def staticAssertSizeMatchesVoidStar(inc,typename):
467      # The declaration is an error if either array size is negative.
468      # It should be okay to use an int that is too large, but it would be very unlikely for this to be the case
469      return self.checkCompile(inc, '#define SZ (sizeof(void*)-sizeof(%s))\nint type_is_too_large[SZ],type_is_too_small[-SZ];'%typename)
470    self.pushLanguage(self.languages.clanguage)
471    if self.checkCompile('#include <stdint.h>', 'int x; uintptr_t i = (uintptr_t)&x;'):
472      self.addDefine('UINTPTR_T', 'uintptr_t')
473    elif staticAssertSizeMatchesVoidStar('','unsigned long long'):
474      self.addDefine('UINTPTR_T', 'unsigned long long')
475    elif staticAssertSizeMatchesVoidStar('#include <stdlib.h>','size_t') or staticAssertSizeMatchesVoidStar('#include <string.h>', 'size_t'):
476      self.addDefine('UINTPTR_T', 'size_t')
477    elif staticAssertSizeMatchesVoidStar('','unsigned'):
478      self.addDefine('UINTPTR_T', 'unsigned')
479    self.popLanguage()
480
481  def configureInline(self):
482    '''Get a generic inline keyword, depending on the language'''
483    if self.languages.clanguage == 'C':
484      self.addDefine('STATIC_INLINE', self.compilers.cStaticInlineKeyword)
485      self.addDefine('RESTRICT', self.compilers.cRestrict)
486    elif self.languages.clanguage == 'Cxx':
487      self.addDefine('STATIC_INLINE', self.compilers.cxxStaticInlineKeyword)
488      self.addDefine('RESTRICT', self.compilers.cxxRestrict)
489    return
490
491  def configureSolaris(self):
492    '''Solaris specific stuff'''
493    if self.arch.hostOsBase.startswith('solaris'):
494      if os.path.isdir(os.path.join('/usr','ucblib')):
495        try:
496          flag = getattr(self.setCompilers, self.language[-1]+'SharedLinkerFlag')
497        except AttributeError:
498          flag = None
499        if flag is None:
500          self.compilers.LIBS += ' -L/usr/ucblib'
501        else:
502          self.compilers.LIBS += ' '+flag+'/usr/ucblib'
503    return
504
505  def configureLinux(self):
506    '''Linux specific stuff'''
507    if self.arch.hostOsBase == 'linux':
508      self.addDefine('HAVE_DOUBLE_ALIGN_MALLOC', 1)
509    return
510
511  def configureWin32(self):
512    '''Win32 non-cygwin specific stuff'''
513    kernel32=0
514    if self.libraries.add('Kernel32.lib','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
515      self.addDefine('HAVE_WINDOWS_H',1)
516      self.addDefine('HAVE_GETCOMPUTERNAME',1)
517      kernel32=1
518    elif self.libraries.add('kernel32','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
519      self.addDefine('HAVE_WINDOWS_H',1)
520      self.addDefine('HAVE_GETCOMPUTERNAME',1)
521      kernel32=1
522    if kernel32:
523      if self.framework.argDB['with-windows-graphics']:
524        self.addDefine('USE_WINDOWS_GRAPHICS',1)
525      if self.checkLink('#include <Windows.h>','LoadLibrary(0)'):
526        self.addDefine('HAVE_LOADLIBRARY',1)
527      if self.checkLink('#include <Windows.h>','GetProcAddress(0,0)'):
528        self.addDefine('HAVE_GETPROCADDRESS',1)
529      if self.checkLink('#include <Windows.h>','FreeLibrary(0)'):
530        self.addDefine('HAVE_FREELIBRARY',1)
531      if self.checkLink('#include <Windows.h>','GetLastError()'):
532        self.addDefine('HAVE_GETLASTERROR',1)
533      if self.checkLink('#include <Windows.h>','SetLastError(0)'):
534        self.addDefine('HAVE_SETLASTERROR',1)
535      if self.checkLink('#include <Windows.h>\n','QueryPerformanceCounter(0);\n'):
536        self.addDefine('USE_NT_TIME',1)
537    if self.libraries.add('Advapi32.lib','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
538      self.addDefine('HAVE_GET_USER_NAME',1)
539    elif self.libraries.add('advapi32','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
540      self.addDefine('HAVE_GET_USER_NAME',1)
541
542    if not self.libraries.add('User32.lib','GetDC',prototype='#include <Windows.h>',call='GetDC(0);'):
543      self.libraries.add('user32','GetDC',prototype='#include <Windows.h>',call='GetDC(0);')
544    if not self.libraries.add('Gdi32.lib','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);'):
545      self.libraries.add('gdi32','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);')
546
547    self.types.check('int32_t', 'int')
548    if not self.checkCompile('#include <sys/types.h>\n','uid_t u;\n'):
549      self.addTypedef('int', 'uid_t')
550      self.addTypedef('int', 'gid_t')
551    if not self.checkLink('#if defined(PETSC_HAVE_UNISTD_H)\n#include <unistd.h>\n#endif\n','int a=R_OK;\n'):
552      self.framework.addDefine('R_OK', '04')
553      self.framework.addDefine('W_OK', '02')
554      self.framework.addDefine('X_OK', '01')
555    if not self.checkLink('#include <sys/stat.h>\n','int a=0;\nif (S_ISDIR(a)){}\n'):
556      self.framework.addDefine('S_ISREG(a)', '(((a)&_S_IFMT) == _S_IFREG)')
557      self.framework.addDefine('S_ISDIR(a)', '(((a)&_S_IFMT) == _S_IFDIR)')
558    if self.checkCompile('#include <Windows.h>\n','LARGE_INTEGER a;\nDWORD b=a.u.HighPart;\n'):
559      self.addDefine('HAVE_LARGE_INTEGER_U',1)
560
561    # Windows requires a Binary file creation flag when creating/opening binary files.  Is a better test in order?
562    if self.checkCompile('#include <Windows.h>\n',''):
563      self.addDefine('HAVE_O_BINARY',1)
564
565    if self.compilers.CC.find('win32fe') >= 0:
566      self.addDefine('PATH_SEPARATOR','\';\'')
567      self.addDefine('DIR_SEPARATOR','\'\\\\\'')
568      self.addDefine('REPLACE_DIR_SEPARATOR','\'/\'')
569      self.addDefine('CANNOT_START_DEBUGGER',1)
570    else:
571      self.addDefine('PATH_SEPARATOR','\':\'')
572      self.addDefine('REPLACE_DIR_SEPARATOR','\'\\\\\'')
573      self.addDefine('DIR_SEPARATOR','\'/\'')
574    return
575
576#-----------------------------------------------------------------------------------------------------
577  def configureDefaultArch(self):
578    conffile = os.path.join('conf', 'petscvariables')
579    if self.framework.argDB['with-default-arch']:
580      fd = file(conffile, 'w')
581      fd.write('PETSC_ARCH='+self.arch.arch+'\n')
582      fd.write('PETSC_DIR='+self.petscdir.dir+'\n')
583      fd.write('include ${PETSC_DIR}/${PETSC_ARCH}/conf/petscvariables\n')
584      fd.close()
585      self.framework.actions.addArgument('PETSc', 'Build', 'Set default architecture to '+self.arch.arch+' in '+conffile)
586    elif os.path.isfile(conffile):
587      try:
588        os.unlink(conffile)
589      except:
590        raise RuntimeError('Unable to remove file '+conffile+'. Did a different user create it?')
591    return
592
593#-----------------------------------------------------------------------------------------------------
594  def configureScript(self):
595    '''Output a script in the conf directory which will reproduce the configuration'''
596    import nargs
597    import sys
598    scriptName = os.path.join(self.arch.arch,'conf', 'reconfigure-'+self.arch.arch+'.py')
599    args = dict([(nargs.Arg.parseArgument(arg)[0], arg) for arg in self.framework.clArgs])
600    if 'configModules' in args:
601      if nargs.Arg.parseArgument(args['configModules'])[1] == 'PETSc.Configure':
602        del args['configModules']
603    if 'optionsModule' in args:
604      if nargs.Arg.parseArgument(args['optionsModule'])[1] == 'PETSc.compilerOptions':
605        del args['optionsModule']
606    if not 'PETSC_ARCH' in args:
607      args['PETSC_ARCH'] = 'PETSC_ARCH='+str(self.arch.arch)
608    f = file(scriptName, 'w')
609    f.write('#!'+sys.executable+'\n')
610    f.write('if __name__ == \'__main__\':\n')
611    f.write('  import sys\n')
612    f.write('  import os\n')
613    f.write('  sys.path.insert(0, os.path.abspath(\'config\'))\n')
614    f.write('  import configure\n')
615    # pretty print repr(args.values())
616    f.write('  configure_options = [\n')
617    for itm in args.values():
618      f.write('    \''+str(itm)+'\',\n')
619    f.write('  ]\n')
620    f.write('  configure.petsc_configure(configure_options)\n')
621    f.close()
622    try:
623      os.chmod(scriptName, 0775)
624    except OSError, e:
625      self.framework.logPrint('Unable to make reconfigure script executable:\n'+str(e))
626    self.framework.actions.addArgument('PETSc', 'File creation', 'Created '+scriptName+' for automatic reconfiguration')
627    return
628
629  def configureInstall(self):
630    '''Setup the directories for installation'''
631    if self.framework.argDB['prefix']:
632      self.installdir = self.framework.argDB['prefix']
633      self.addMakeRule('shared_install','',['-@echo "Now to install the libraries do:"',\
634                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} install"',\
635                                              '-@echo "========================================="'])
636    else:
637      self.installdir = os.path.join(self.petscdir.dir,self.arch.arch)
638      self.addMakeRule('shared_install','',['-@echo "Now to check if the libraries are working do:"',\
639                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} test"',\
640                                              '-@echo "========================================="'])
641      return
642
643  def configureGCOV(self):
644    if self.framework.argDB['with-gcov']:
645      self.addDefine('USE_GCOV','1')
646    return
647
648  def configureFortranFlush(self):
649    if hasattr(self.compilers, 'FC'):
650      for baseName in ['flush','flush_']:
651        if self.libraries.check('', baseName, otherLibs = self.compilers.flibs, fortranMangle = 1):
652          self.addDefine('HAVE_'+baseName.upper(), 1)
653          return
654
655
656  def configure(self):
657    if not os.path.samefile(self.petscdir.dir, os.getcwd()):
658      raise RuntimeError('Wrong PETSC_DIR option specified: '+str(self.petscdir.dir) + '\n  Configure invoked in: '+os.path.realpath(os.getcwd()))
659    if self.framework.argDB['prefix'] and os.path.isdir(self.framework.argDB['prefix']) and os.path.samefile(self.framework.argDB['prefix'],self.petscdir.dir):
660      raise RuntimeError('Incorrect option --prefix='+self.framework.argDB['prefix']+' specified. It cannot be same as PETSC_DIR!')
661    self.framework.header          = os.path.join(self.arch.arch,'include','petscconf.h')
662    self.framework.cHeader         = os.path.join(self.arch.arch,'include','petscfix.h')
663    self.framework.makeMacroHeader = os.path.join(self.arch.arch,'conf','petscvariables')
664    self.framework.makeRuleHeader  = os.path.join(self.arch.arch,'conf','petscrules')
665    if self.libraries.math is None:
666      raise RuntimeError('PETSc requires a functional math library. Please send configure.log to petsc-maint@mcs.anl.gov.')
667    if self.languages.clanguage == 'Cxx' and not hasattr(self.compilers, 'CXX'):
668      raise RuntimeError('Cannot set C language to C++ without a functional C++ compiler.')
669    self.executeTest(self.configureInline)
670    self.executeTest(self.configurePrefetch)
671    self.executeTest(self.configureUnused)
672    self.executeTest(self.configureExpect);
673    self.executeTest(self.configureIntptrt);
674    self.executeTest(self.configureSolaris)
675    self.executeTest(self.configureLinux)
676    self.executeTest(self.configureWin32)
677    self.executeTest(self.configureDefaultArch)
678    self.executeTest(self.configureScript)
679    self.executeTest(self.configureInstall)
680    self.executeTest(self.configureGCOV)
681    self.executeTest(self.configureFortranFlush)
682    # dummy rules, always needed except for remote builds
683    self.addMakeRule('remote','')
684    self.addMakeRule('remoteclean','')
685
686    self.Dump()
687    self.dumpConfigInfo()
688    self.dumpMachineInfo()
689    self.dumpCMakeConfig()
690    self.dumpCMakeLists()
691    self.cmakeBoot()
692    self.framework.log.write('================================================================================\n')
693    self.logClear()
694    return
695