xref: /petsc/config/PETSc/Configure.py (revision f7b66a64efd3f489320840c4e2d70252719390c8)
1f8833479SBarry Smithimport config.base
2f8833479SBarry Smith
3f8833479SBarry Smithimport os
4f8833479SBarry Smithimport re
5f8833479SBarry Smith
68bec23c5SJed Brown# The sorted() builtin is not available with python-2.3
78bec23c5SJed Browntry: sorted
88bec23c5SJed Brownexcept NameError:
98bec23c5SJed Brown  def sorted(lst):
108bec23c5SJed Brown    lst.sort()
118bec23c5SJed Brown    return lst
128bec23c5SJed Brown
13f8833479SBarry Smithclass Configure(config.base.Configure):
14f8833479SBarry Smith  def __init__(self, framework):
15f8833479SBarry Smith    config.base.Configure.__init__(self, framework)
16f8833479SBarry Smith    self.headerPrefix = 'PETSC'
17f8833479SBarry Smith    self.substPrefix  = 'PETSC'
18f8833479SBarry Smith    return
19f8833479SBarry Smith
207c939e48SSatish Balay  def __str2__(self):
217c939e48SSatish Balay    desc = []
22a0022257SSatish Balay    desc.append('xxx=========================================================================xxx')
237c939e48SSatish Balay    desc.append(' Configure stage complete. Now build PETSc libraries with:')
247c939e48SSatish Balay    desc.append('   make PETSC_DIR='+self.petscdir.dir+' PETSC_ARCH='+self.arch.arch+' all')
25ae937f1dSJed Brown    if self.cmakeboot_success:
26a8a678f1SSatish Balay      desc.append(' or (experimental with cmake):')
27ae937f1dSJed Brown      desc.append('   make -j4 -C '+os.path.join(self.petscdir.dir,self.arch.arch))
28a8a678f1SSatish Balay    desc.append(' or (experimental with python):')
29c77b0880SMatthew G Knepley    desc.append('   PETSC_DIR='+self.petscdir.dir+' PETSC_ARCH='+self.arch.arch+' ./config/builder.py')
30a0022257SSatish Balay    desc.append('xxx=========================================================================xxx')
317c939e48SSatish Balay    return '\n'.join(desc)+'\n'
32f8833479SBarry Smith
33f8833479SBarry Smith  def setupHelp(self, help):
34f8833479SBarry Smith    import nargs
35ce0b2093SBarry Smith    help.addArgument('PETSc',  '-prefix=<dir>',                  nargs.Arg(None, '', 'Specifiy location to install PETSc (eg. /usr/local)'))
36eed94e11SSatish Balay    help.addArgument('Windows','-with-windows-graphics=<bool>',   nargs.ArgBool(None, 1,'Enable check for Windows Graphics'))
37569865ddSSatish Balay    help.addArgument('PETSc', '-with-default-arch=<bool>',        nargs.ArgBool(None, 1, 'Allow using the last configured arch without setting PETSC_ARCH'))
3857cb31baSSatish Balay    help.addArgument('PETSc','-with-single-library=<bool>',       nargs.ArgBool(None, 1,'Put all PETSc code into the single -lpetsc library'))
39ce0b2093SBarry Smith    help.addArgument('PETSc', '-with-iphone=<bool>',              nargs.ArgBool(None, 0, 'Build an iPhone version of PETSc'))
40f8833479SBarry Smith    return
41f8833479SBarry Smith
42f8833479SBarry Smith  def setupDependencies(self, framework):
43f8833479SBarry Smith    config.base.Configure.setupDependencies(self, framework)
44f8833479SBarry Smith    self.setCompilers  = framework.require('config.setCompilers',      self)
45f8833479SBarry Smith    self.arch          = framework.require('PETSc.utilities.arch',     self.setCompilers)
46f8833479SBarry Smith    self.petscdir      = framework.require('PETSc.utilities.petscdir', self.setCompilers)
47f8833479SBarry Smith    self.languages     = framework.require('PETSc.utilities.languages',self.setCompilers)
48f8833479SBarry Smith    self.debugging     = framework.require('PETSc.utilities.debugging',self.setCompilers)
49a2b1cf82SBarry Smith    self.CHUD          = framework.require('PETSc.utilities.CHUD',     self)
50f8833479SBarry Smith    self.compilers     = framework.require('config.compilers',         self)
51f8833479SBarry Smith    self.types         = framework.require('config.types',             self)
52f8833479SBarry Smith    self.headers       = framework.require('config.headers',           self)
53f8833479SBarry Smith    self.functions     = framework.require('config.functions',         self)
54f8833479SBarry Smith    self.libraries     = framework.require('config.libraries',         self)
55f8833479SBarry Smith    if os.path.isdir(os.path.join('config', 'PETSc')):
56f8833479SBarry Smith      for d in ['utilities', 'packages']:
57f8833479SBarry Smith        for utility in os.listdir(os.path.join('config', 'PETSc', d)):
58f8833479SBarry Smith          (utilityName, ext) = os.path.splitext(utility)
59f8833479SBarry Smith          if not utilityName.startswith('.') and not utilityName.startswith('#') and ext == '.py' and not utilityName == '__init__':
60f8833479SBarry Smith            utilityObj                    = self.framework.require('PETSc.'+d+'.'+utilityName, self)
61f8833479SBarry Smith            utilityObj.headerPrefix       = self.headerPrefix
62f1ecedd2SMatthew Knepley            utilityObj.archProvider       = self.arch
63fdfda77fSMatthew Knepley            utilityObj.languageProvider   = self.languages
64fdfda77fSMatthew Knepley            utilityObj.installDirProvider = self.petscdir
65f8833479SBarry Smith            setattr(self, utilityName.lower(), utilityObj)
668cf378d1SBarry Smith
67d37554e4SMatthew G Knepley    for package in config.packages.all:
68d37554e4SMatthew G Knepley      if not package == 'PETSc':
69d37554e4SMatthew G Knepley        packageObj                    = framework.require('config.packages.'+package, self)
70d37554e4SMatthew G Knepley        packageObj.archProvider       = self.arch
71d37554e4SMatthew G Knepley        packageObj.languageProvider   = self.languages
72d37554e4SMatthew G Knepley        packageObj.installDirProvider = self.petscdir
73d37554e4SMatthew G Knepley        setattr(self, package.lower(), packageObj)
74d37554e4SMatthew G Knepley    # Force blaslapack to depend on scalarType so precision is set before BlasLapack is built
758cf378d1SBarry Smith    framework.require('PETSc.utilities.scalarTypes', self.blaslapack)
76f8833479SBarry Smith    self.blaslapack.precisionProvider = self.scalartypes
77f8833479SBarry Smith
78f8833479SBarry Smith    self.compilers.headerPrefix  = self.headerPrefix
79f8833479SBarry Smith    self.types.headerPrefix      = self.headerPrefix
80f8833479SBarry Smith    self.headers.headerPrefix    = self.headerPrefix
81f8833479SBarry Smith    self.functions.headerPrefix  = self.headerPrefix
82f8833479SBarry Smith    self.libraries.headerPrefix  = self.headerPrefix
83f8833479SBarry Smith    self.blaslapack.headerPrefix = self.headerPrefix
84f8833479SBarry Smith    self.mpi.headerPrefix        = self.headerPrefix
85f8833479SBarry Smith    headersC = map(lambda name: name+'.h', ['dos', 'endian', 'fcntl', 'float', 'io', 'limits', 'malloc', 'pwd', 'search', 'strings',
86f8833479SBarry Smith                                            'unistd', 'machine/endian', 'sys/param', 'sys/procfs', 'sys/resource',
87b4bb646cSSatish Balay                                            'sys/systeminfo', 'sys/times', 'sys/utsname','string', 'stdlib','memory',
88f8833479SBarry Smith                                            'sys/socket','sys/wait','netinet/in','netdb','Direct','time','Ws2tcpip','sys/types',
89b014e56cSJed Brown                                            'WindowsX', 'cxxabi','float','ieeefp','stdint','fenv'])
90f8833479SBarry Smith    functions = ['access', '_access', 'clock', 'drand48', 'getcwd', '_getcwd', 'getdomainname', 'gethostname', 'getpwuid',
91f8833479SBarry Smith                 'gettimeofday', 'getwd', 'memalign', 'memmove', 'mkstemp', 'popen', 'PXFGETARG', 'rand', 'getpagesize',
92a6d0e24fSJed Brown                 'readlink', 'realpath',  'sigaction', 'signal', 'sigset', 'nanosleep', 'usleep', 'sleep', '_sleep', 'socket',
93a6d0e24fSJed Brown                 'times', 'gethostbyname', 'uname','snprintf','_snprintf','_fullpath','lseek','_lseek','time','fork','stricmp',
94a6d0e24fSJed Brown                 'strcasecmp', 'bzero', 'dlopen', 'dlsym', 'dlclose', 'dlerror',
9564ffe704SBarry Smith                 '_intel_fast_memcpy','_intel_fast_memset']
96f8833479SBarry Smith    libraries1 = [(['socket', 'nsl'], 'socket'), (['fpe'], 'handle_sigfpes')]
97f8833479SBarry Smith    self.headers.headers.extend(headersC)
98f8833479SBarry Smith    self.functions.functions.extend(functions)
99f8833479SBarry Smith    self.libraries.libraries.extend(libraries1)
1007d421530SBarry Smith
101f8833479SBarry Smith    return
102f8833479SBarry Smith
103f8833479SBarry Smith  def Dump(self):
104f8833479SBarry Smith    ''' Actually put the values into the configuration files '''
105f8833479SBarry Smith    # eventually everything between -- should be gone
106f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
107f8833479SBarry Smith
108f8833479SBarry Smith    # Sometimes we need C compiler, even if built with C++
109f8833479SBarry Smith    self.setCompilers.pushLanguage('C')
110f8833479SBarry Smith    self.addMakeMacro('CC_FLAGS',self.setCompilers.getCompilerFlags())
111f8833479SBarry Smith    self.setCompilers.popLanguage()
112f8833479SBarry Smith
113f8833479SBarry Smith    # C preprocessor values
114a2b1cf82SBarry Smith    self.addMakeMacro('CPP_FLAGS',self.setCompilers.CPPFLAGS+self.CHUD.CPPFLAGS)
115f8833479SBarry Smith
116f8833479SBarry Smith    # compiler values
117f8833479SBarry Smith    self.setCompilers.pushLanguage(self.languages.clanguage)
118f8833479SBarry Smith    self.addMakeMacro('PCC',self.setCompilers.getCompiler())
119f8833479SBarry Smith    self.addMakeMacro('PCC_FLAGS',self.setCompilers.getCompilerFlags())
120f8833479SBarry Smith    self.setCompilers.popLanguage()
121f8833479SBarry Smith    # .o or .obj
122f8833479SBarry Smith    self.addMakeMacro('CC_SUFFIX','o')
123f8833479SBarry Smith
124f8833479SBarry Smith    # executable linker values
125f8833479SBarry Smith    self.setCompilers.pushLanguage(self.languages.clanguage)
126f8833479SBarry Smith    pcc_linker = self.setCompilers.getLinker()
127f8833479SBarry Smith    self.addMakeMacro('PCC_LINKER',pcc_linker)
128c84a332bSSatish Balay    self.addMakeMacro('PCC_LINKER_FLAGS',self.setCompilers.getLinkerFlags())
129f8833479SBarry Smith    self.setCompilers.popLanguage()
130f8833479SBarry Smith    # '' for Unix, .exe for Windows
131f8833479SBarry Smith    self.addMakeMacro('CC_LINKER_SUFFIX','')
132f8833479SBarry Smith
133f8833479SBarry Smith    if hasattr(self.compilers, 'FC'):
134f8833479SBarry Smith      self.setCompilers.pushLanguage('FC')
135f8833479SBarry Smith      # need FPPFLAGS in config/setCompilers
136f8833479SBarry Smith      self.addDefine('HAVE_FORTRAN','1')
137f8833479SBarry Smith      self.addMakeMacro('FPP_FLAGS',self.setCompilers.CPPFLAGS)
138f8833479SBarry Smith
139f8833479SBarry Smith      # compiler values
140f8833479SBarry Smith      self.addMakeMacro('FC_FLAGS',self.setCompilers.getCompilerFlags())
141f8833479SBarry Smith      self.setCompilers.popLanguage()
142f8833479SBarry Smith      # .o or .obj
143f8833479SBarry Smith      self.addMakeMacro('FC_SUFFIX','o')
144f8833479SBarry Smith
145f8833479SBarry Smith      # executable linker values
146f8833479SBarry Smith      self.setCompilers.pushLanguage('FC')
147f8833479SBarry Smith      # Cannot have NAG f90 as the linker - so use pcc_linker as fc_linker
148f8833479SBarry Smith      fc_linker = self.setCompilers.getLinker()
149f8833479SBarry Smith      if config.setCompilers.Configure.isNAG(fc_linker):
150f8833479SBarry Smith        self.addMakeMacro('FC_LINKER',pcc_linker)
151f8833479SBarry Smith      else:
152f8833479SBarry Smith        self.addMakeMacro('FC_LINKER',fc_linker)
1536d53d35eSSatish Balay      self.addMakeMacro('FC_LINKER_FLAGS',self.setCompilers.getLinkerFlags())
1543feacd00SBarry Smith      # apple requires this shared library linker flag on SOME versions of the os
1553feacd00SBarry Smith      if self.setCompilers.getLinkerFlags().find('-Wl,-commons,use_dylibs') > -1:
1563feacd00SBarry Smith        self.addMakeMacro('DARWIN_COMMONS_USE_DYLIBS',' -Wl,-commons,use_dylibs ')
157bb82cf9cSSatish Balay      self.setCompilers.popLanguage()
1585d631499SMatthew Knepley
1595d631499SMatthew Knepley      # F90 Modules
1605d631499SMatthew Knepley      if self.setCompilers.fortranModuleIncludeFlag:
1615d631499SMatthew Knepley        self.addMakeMacro('FC_MODULE_FLAG', self.setCompilers.fortranModuleIncludeFlag)
1626ddd6694SSatish Balay      else: # for non-f90 compilers like g77
1636ddd6694SSatish Balay        self.addMakeMacro('FC_MODULE_FLAG', '-I')
164a324c51cSMatthew G Knepley      if self.setCompilers.fortranModuleIncludeFlag:
165a324c51cSMatthew G Knepley        self.addMakeMacro('FC_MODULE_OUTPUT_FLAG', self.setCompilers.fortranModuleOutputFlag)
166f8833479SBarry Smith    else:
167f8833479SBarry Smith      self.addMakeMacro('FC','')
168f8833479SBarry Smith
16946a3958fSBarry Smith    if hasattr(self.compilers, 'CUDAC'):
1707ff2890cSSatish Balay      self.setCompilers.pushLanguage('CUDA')
171d93a25ecSSatish Balay      self.addMakeMacro('CUDAC_FLAGS',self.setCompilers.getCompilerFlags())
1727ff2890cSSatish Balay      self.setCompilers.popLanguage()
1737ff2890cSSatish Balay
174f8833479SBarry Smith    # shared library linker values
175f8833479SBarry Smith    self.setCompilers.pushLanguage(self.languages.clanguage)
176f8833479SBarry Smith    # need to fix BuildSystem to collect these separately
177f8833479SBarry Smith    self.addMakeMacro('SL_LINKER',self.setCompilers.getLinker())
17870db8aa6SSatish Balay    self.addMakeMacro('SL_LINKER_FLAGS','${PCC_LINKER_FLAGS}')
179f8833479SBarry Smith    self.setCompilers.popLanguage()
180f8833479SBarry Smith    # One of 'a', 'so', 'lib', 'dll', 'dylib' (perhaps others also?) depending on the library generator and architecture
181f8833479SBarry Smith    # Note: . is not included in this macro, consistent with AR_LIB_SUFFIX
182f8833479SBarry Smith    if self.setCompilers.sharedLibraryExt == self.setCompilers.AR_LIB_SUFFIX:
183f8833479SBarry Smith      self.addMakeMacro('SL_LINKER_SUFFIX', '')
18446bc77b6SBarry Smith      self.addDefine('SLSUFFIX','""')
185f8833479SBarry Smith    else:
186f8833479SBarry Smith      self.addMakeMacro('SL_LINKER_SUFFIX', self.setCompilers.sharedLibraryExt)
18746bc77b6SBarry Smith      self.addDefine('SLSUFFIX','"'+self.setCompilers.sharedLibraryExt+'"')
188bb82cf9cSSatish Balay
18923e93537SBarry Smith    self.addMakeMacro('SL_LINKER_LIBS','${PETSC_EXTERNAL_LIB_BASIC}')
190bb82cf9cSSatish Balay
191f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
192f8833479SBarry Smith
193f8833479SBarry Smith    # CONLY or CPP. We should change the PETSc makefiles to do this better
194f8833479SBarry Smith    if self.languages.clanguage == 'C': lang = 'CONLY'
195f8833479SBarry Smith    else: lang = 'CXXONLY'
196f8833479SBarry Smith    self.addMakeMacro('PETSC_LANGUAGE',lang)
197f8833479SBarry Smith
198f8833479SBarry Smith    # real or complex
199f8833479SBarry Smith    self.addMakeMacro('PETSC_SCALAR',self.scalartypes.scalartype)
200f8833479SBarry Smith    # double or float
201f8833479SBarry Smith    self.addMakeMacro('PETSC_PRECISION',self.scalartypes.precision)
202f8833479SBarry Smith
203f8833479SBarry Smith    if self.framework.argDB['with-batch']:
204f8833479SBarry Smith      self.addMakeMacro('PETSC_WITH_BATCH','1')
205f8833479SBarry Smith
206f8833479SBarry Smith    # Test for compiler-specific macros that need to be defined.
207b409243cSBarry Smith    if self.setCompilers.isCrayVector('CC'):
208b409243cSBarry Smith      self.addDefine('HAVE_CRAY_VECTOR','1')
209f8833479SBarry Smith
210f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
211df1a78b3SMatthew G Knepley    if self.functions.haveFunction('gethostbyname') and self.functions.haveFunction('socket') and self.headers.haveHeader('netinet/in.h'):
212f8833479SBarry Smith      self.addDefine('USE_SOCKET_VIEWER','1')
213f8833479SBarry Smith
214f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
215a6cc6bb1SBarry Smith    # print include and lib for makefiles
216f8833479SBarry Smith    self.framework.packages.reverse()
217a6cc6bb1SBarry Smith    includes = [os.path.join(self.petscdir.dir,'include'),os.path.join(self.petscdir.dir,self.arch.arch,'include')]
218996b3231SBarry Smith    libs = []
219f8833479SBarry Smith    for i in self.framework.packages:
220898a086dSBarry Smith      if i.useddirectly:
221898a086dSBarry Smith        self.addDefine('HAVE_'+i.PACKAGE, 1)  # ONLY list package if it is used directly by PETSc (and not only by another package)
222f8833479SBarry Smith      if not isinstance(i.lib, list):
223f8833479SBarry Smith        i.lib = [i.lib]
224ac9e4c42SSatish Balay      libs.extend(i.lib)
22534cdeb2aSSatish Balay      self.addMakeMacro(i.PACKAGE+'_LIB', self.libraries.toStringNoDupes(i.lib))
226f8833479SBarry Smith      if hasattr(i,'include'):
227f8833479SBarry Smith        if not isinstance(i.include,list):
228f8833479SBarry Smith          i.include = [i.include]
229ac9e4c42SSatish Balay        includes.extend(i.include)
23034cdeb2aSSatish Balay        self.addMakeMacro(i.PACKAGE+'_INCLUDE',self.headers.toStringNoDupes(i.include))
2312df986feSBarry Smith    if self.framework.argDB['with-single-library']:
232ea820d49SSatish Balay      self.addMakeMacro('PETSC_WITH_EXTERNAL_LIB',self.libraries.toStringNoDupes(['-L'+os.path.join(self.petscdir.dir,self.arch.arch,'lib'),' -lpetsc']+libs+self.libraries.math+self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' '))+self.CHUD.LIBS)
23323e93537SBarry Smith    self.addMakeMacro('PETSC_EXTERNAL_LIB_BASIC',self.libraries.toStringNoDupes(libs+self.libraries.math+self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' '))+self.CHUD.LIBS)
23423e93537SBarry Smith    self.PETSC_EXTERNAL_LIB_BASIC = self.libraries.toStringNoDupes(libs+self.libraries.math+self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' '))+self.CHUD.LIBS
235a6cc6bb1SBarry Smith    self.addMakeMacro('PETSC_CC_INCLUDES',self.headers.toStringNoDupes(includes))
236a6cc6bb1SBarry Smith    self.PETSC_CC_INCLUDES = self.headers.toStringNoDupes(includes)
237cbd5cc15SBarry Smith    if hasattr(self.compilers, 'FC'):
238208c3fd5SBarry Smith      if self.compilers.fortranIsF90:
23943a63bfbSSatish Balay        self.addMakeMacro('PETSC_FC_INCLUDES',self.headers.toStringNoDupes(includes,includes))
24030d43657SSatish Balay      else:
24130d43657SSatish Balay        self.addMakeMacro('PETSC_FC_INCLUDES',self.headers.toStringNoDupes(includes))
242f8833479SBarry Smith
2431b1e03dfSSatish Balay    self.addMakeMacro('DESTDIR',self.installdir)
244c6f99f23SBarry Smith    self.addDefine('LIB_DIR','"'+os.path.join(self.installdir,'lib')+'"')
245f8833479SBarry Smith
2460f3b21c2SBarry Smith    if self.framework.argDB['with-single-library']:
2470f3b21c2SBarry Smith      # overrides the values set in conf/variables
2480f3b21c2SBarry Smith      self.addMakeMacro('LIBNAME','${INSTALL_LIB_DIR}/libpetsc.${AR_LIB_SUFFIX}')
24957cb31baSSatish Balay      self.addMakeMacro('SHLIBS','libpetsc')
250bccf1c12SBarry Smith      self.addMakeMacro('PETSC_LIB_BASIC','-lpetsc')
251797063a9SSatish Balay      self.addMakeMacro('PETSC_KSP_LIB_BASIC','-lpetsc')
252797063a9SSatish Balay      self.addMakeMacro('PETSC_TS_LIB_BASIC','-lpetsc')
253bb84e0fdSBarry Smith      self.addDefine('USE_SINGLE_LIBRARY', '1')
2542df986feSBarry Smith      if self.sharedlibraries.useShared:
255ea820d49SSatish Balay        self.addMakeMacro('PETSC_SYS_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
256ea820d49SSatish Balay        self.addMakeMacro('PETSC_VEC_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
257ea820d49SSatish Balay        self.addMakeMacro('PETSC_MAT_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
258ea820d49SSatish Balay        self.addMakeMacro('PETSC_DM_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
259ea820d49SSatish Balay        self.addMakeMacro('PETSC_KSP_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
260ea820d49SSatish Balay        self.addMakeMacro('PETSC_SNES_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
261ea820d49SSatish Balay        self.addMakeMacro('PETSC_TS_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
262fdb87e33SJed Brown        self.addMakeMacro('PETSC_CHARACTERISTIC_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
263ea820d49SSatish Balay        self.addMakeMacro('PETSC_LIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
264ea820d49SSatish Balay        self.addMakeMacro('PETSC_CONTRIB','${C_SH_LIB_PATH} ${PETSC_WITH_EXTERNAL_LIB}')
2652df986feSBarry Smith      else:
266ea820d49SSatish Balay        self.addMakeMacro('PETSC_SYS_LIB','${PETSC_WITH_EXTERNAL_LIB}')
267ea820d49SSatish Balay        self.addMakeMacro('PETSC_VEC_LIB','${PETSC_WITH_EXTERNAL_LIB}')
268ea820d49SSatish Balay        self.addMakeMacro('PETSC_MAT_LIB','${PETSC_WITH_EXTERNAL_LIB}')
269ea820d49SSatish Balay        self.addMakeMacro('PETSC_DM_LIB','${PETSC_WITH_EXTERNAL_LIB}')
270ea820d49SSatish Balay        self.addMakeMacro('PETSC_KSP_LIB','${PETSC_WITH_EXTERNAL_LIB}')
271ea820d49SSatish Balay        self.addMakeMacro('PETSC_SNES_LIB','${PETSC_WITH_EXTERNAL_LIB}')
272ea820d49SSatish Balay        self.addMakeMacro('PETSC_TS_LIB','${PETSC_WITH_EXTERNAL_LIB}')
273fdb87e33SJed Brown        self.addMakeMacro('PETSC_CHARACTERISTIC_LIB','${PETSC_WITH_EXTERNAL_LIB}')
274ea820d49SSatish Balay        self.addMakeMacro('PETSC_LIB','${PETSC_WITH_EXTERNAL_LIB}')
275ea820d49SSatish Balay        self.addMakeMacro('PETSC_CONTRIB','${PETSC_WITH_EXTERNAL_LIB}')
2760f3b21c2SBarry Smith
277f8833479SBarry Smith    if not os.path.exists(os.path.join(self.petscdir.dir,self.arch.arch,'lib')):
278f8833479SBarry Smith      os.makedirs(os.path.join(self.petscdir.dir,self.arch.arch,'lib'))
279f8833479SBarry Smith
280f8833479SBarry Smith    # add a makefile entry for configure options
281f8833479SBarry Smith    self.addMakeMacro('CONFIGURE_OPTIONS', self.framework.getOptionsString(['configModules', 'optionsModule']).replace('\"','\\"'))
282f8833479SBarry Smith    return
283f8833479SBarry Smith
284f8833479SBarry Smith  def dumpConfigInfo(self):
285f8833479SBarry Smith    import time
286f8833479SBarry Smith    fd = file(os.path.join(self.arch.arch,'include','petscconfiginfo.h'),'w')
287f8833479SBarry Smith    fd.write('static const char *petscconfigureruntime = "'+time.ctime(time.time())+'";\n')
288f8833479SBarry Smith    fd.write('static const char *petscconfigureoptions = "'+self.framework.getOptionsString(['configModules', 'optionsModule']).replace('\"','\\"')+'";\n')
289f8833479SBarry Smith    fd.close()
290f8833479SBarry Smith    return
291f8833479SBarry Smith
2922a4161d9SMatthew G Knepley  def dumpMachineInfo(self):
2932a4161d9SMatthew G Knepley    import platform
2942a4161d9SMatthew G Knepley    import time
29540373944SSatish Balay    import script
2962a4161d9SMatthew G Knepley    fd = file(os.path.join(self.arch.arch,'include','petscmachineinfo.h'),'w')
2972a4161d9SMatthew G Knepley    fd.write('static const char *petscmachineinfo = \"\\n\"\n')
2982a4161d9SMatthew G Knepley    fd.write('\"-----------------------------------------\\n\"\n')
2992a4161d9SMatthew G Knepley    fd.write('\"Libraries compiled on %s on %s \\n\"\n' % (time.ctime(time.time()), platform.node()))
30060acdfe7SSatish Balay    fd.write('\"Machine characteristics: %s\\n\"\n' % (platform.platform()))
30160acdfe7SSatish Balay    fd.write('\"Using PETSc directory: %s\\n\"\n' % (self.petscdir.dir))
30260acdfe7SSatish Balay    fd.write('\"Using PETSc arch: %s\\n\"\n' % (self.arch.arch))
303cdec380aSBarry Smith    fd.write('\"-----------------------------------------\\n\";\n')
3042a4161d9SMatthew G Knepley    fd.write('static const char *petsccompilerinfo = \"\\n\"\n')
3052a4161d9SMatthew G Knepley    self.setCompilers.pushLanguage(self.languages.clanguage)
30660acdfe7SSatish Balay    fd.write('\"Using C compiler: %s %s ${COPTFLAGS} ${CFLAGS}\\n\"\n' % (self.setCompilers.getCompiler(), self.setCompilers.getCompilerFlags()))
3072a4161d9SMatthew G Knepley    self.setCompilers.popLanguage()
3088782282cSMatthew G Knepley    if hasattr(self.compilers, 'FC'):
3092a4161d9SMatthew G Knepley      self.setCompilers.pushLanguage('FC')
31060acdfe7SSatish Balay      fd.write('\"Using Fortran compiler: %s %s ${FOPTFLAGS} ${FFLAGS} %s\\n\"\n' % (self.setCompilers.getCompiler(), self.setCompilers.getCompilerFlags(), self.setCompilers.CPPFLAGS))
3112a4161d9SMatthew G Knepley      self.setCompilers.popLanguage()
312cdec380aSBarry Smith    fd.write('\"-----------------------------------------\\n\";\n')
3132a4161d9SMatthew G Knepley    fd.write('static const char *petsccompilerflagsinfo = \"\\n\"\n')
314a6cc6bb1SBarry Smith    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.PETSC_CC_INCLUDES))
315cdec380aSBarry Smith    fd.write('\"-----------------------------------------\\n\";\n')
3162a4161d9SMatthew G Knepley    fd.write('static const char *petsclinkerinfo = \"\\n\"\n')
3172a4161d9SMatthew G Knepley    self.setCompilers.pushLanguage(self.languages.clanguage)
31860acdfe7SSatish Balay    fd.write('\"Using C linker: %s\\n\"\n' % (self.setCompilers.getLinker()))
3192a4161d9SMatthew G Knepley    self.setCompilers.popLanguage()
3208782282cSMatthew G Knepley    if hasattr(self.compilers, 'FC'):
3212a4161d9SMatthew G Knepley      self.setCompilers.pushLanguage('FC')
32260acdfe7SSatish Balay      fd.write('\"Using Fortran linker: %s\\n\"\n' % (self.setCompilers.getLinker()))
3232a4161d9SMatthew G Knepley      self.setCompilers.popLanguage()
32423e93537SBarry Smith    fd.write('\"Using libraries: %s%s -L%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.PETSC_EXTERNAL_LIB_BASIC))
325cdec380aSBarry Smith    fd.write('\"-----------------------------------------\\n\";\n')
3262a4161d9SMatthew G Knepley    fd.close()
3272a4161d9SMatthew G Knepley    return
328b2843cf1SBarry Smith
329511a6afcSJed Brown  def dumpCMakeConfig(self):
330511a6afcSJed Brown    '''
33134ed7027SBarry Smith    Writes configuration-specific values to ${PETSC_ARCH}/conf/PETScConfig.cmake.
332511a6afcSJed Brown    This file is private to PETSc and should not be included by third parties
333511a6afcSJed Brown    (a suitable file can be produced later by CMake, but this is not it).
334511a6afcSJed Brown    '''
335511a6afcSJed Brown    def cmakeset(fd,key,val=True):
336511a6afcSJed Brown      if val == True: val = 'YES'
337511a6afcSJed Brown      if val == False: val = 'NO'
338511a6afcSJed Brown      fd.write('set (' + key + ' ' + val + ')\n')
339511a6afcSJed Brown    def ensurelist(a):
340826d9344SJed Brown      if isinstance(a,list):
341826d9344SJed Brown        return a
342826d9344SJed Brown      else:
343826d9344SJed Brown        return [a]
344511a6afcSJed Brown    def libpath(lib):
345511a6afcSJed Brown      'Returns a search path if that is what this item provides, else "" which will be cleaned out later'
3461b1c0b30SJed Brown      if not isinstance(lib,str): return ''
347511a6afcSJed Brown      if lib.startswith('-L'): return lib[2:]
348511a6afcSJed Brown      if lib.startswith('-R'): return lib[2:]
349511a6afcSJed Brown      if lib.startswith('-Wl,-rpath,'):
350511a6afcSJed Brown        # This case occurs when an external package needs a specific system library that is normally provided by the compiler.
351511a6afcSJed Brown        # In other words, the -L path is builtin to the wrapper or compiler, here we provide it so that CMake can locate the
352511a6afcSJed Brown        # corresponding library.
353511a6afcSJed Brown        return lib[len('-Wl,-rpath,'):]
354511a6afcSJed Brown      if lib.startswith('-'): return ''
355511a6afcSJed Brown      return os.path.dirname(lib)
356511a6afcSJed Brown    def cleanlib(lib):
357511a6afcSJed Brown      'Returns a library name if that is what this item provides, else "" which will be cleaned out later'
35842e8629dSMatthew G Knepley      if not isinstance(lib,str): return ''
359511a6afcSJed Brown      if lib.startswith('-l'):  return lib[2:]
360511a6afcSJed Brown      if lib.startswith('-Wl') or lib.startswith('-L'): return ''
361511a6afcSJed Brown      lib = os.path.splitext(os.path.basename(lib))[0]
362511a6afcSJed Brown      if lib.startswith('lib'): return lib[3:]
363511a6afcSJed Brown      return lib
364511a6afcSJed Brown    def nub(lst):
36506e8c1ddSJed Brown      'Return a list containing the first occurrence of each unique element'
366511a6afcSJed Brown      unique = []
367511a6afcSJed Brown      for elem in lst:
368511a6afcSJed Brown        if elem not in unique and elem != '':
369511a6afcSJed Brown          unique.append(elem)
370511a6afcSJed Brown      return unique
37150937898SJed Brown    try: reversed # reversed was added in Python-2.4
37250937898SJed Brown    except NameError:
37350937898SJed Brown      def reversed(lst): return lst[::-1]
37406e8c1ddSJed Brown    def nublast(lst):
37506e8c1ddSJed Brown      'Return a list containing the last occurrence of each unique entry in a list'
37650937898SJed Brown      return reversed(nub(reversed(lst)))
377511a6afcSJed Brown    def cmakeexpand(varname):
378511a6afcSJed Brown      return r'"${' + varname + r'}"'
379582751aaSJed Brown    def uniqextend(lst,new):
380511a6afcSJed Brown      for x in ensurelist(new):
381582751aaSJed Brown        if x not in lst:
382582751aaSJed Brown          lst.append(x)
383511a6afcSJed Brown    def notstandardinclude(path):
384511a6afcSJed Brown      return path not in '/usr/include /usr/local/include'.split()
385511a6afcSJed Brown    def writeMacroDefinitions(fd):
386511a6afcSJed Brown      if self.mpi.usingMPIUni:
387511a6afcSJed Brown        cmakeset(fd,'PETSC_HAVE_MPIUNI')
388511a6afcSJed Brown      for pkg in self.framework.packages:
389511a6afcSJed Brown        if pkg.useddirectly:
390511a6afcSJed Brown          cmakeset(fd,'PETSC_HAVE_' + pkg.PACKAGE)
391511a6afcSJed Brown      for name,val in self.functions.defines.items():
392511a6afcSJed Brown        cmakeset(fd,'PETSC_'+name,val)
393511a6afcSJed Brown      for dct in [self.defines, self.libraryoptions.defines]:
394511a6afcSJed Brown        for k,v in dct.items():
395511a6afcSJed Brown          if k.startswith('USE_'):
396511a6afcSJed Brown            cmakeset(fd,'PETSC_' + k, v)
397511a6afcSJed Brown      cmakeset(fd,'PETSC_USE_COMPLEX', self.scalartypes.scalartype == 'complex')
398ce63c4c1SBarry Smith      cmakeset(fd,'PETSC_USE_REAL_' + self.scalartypes.precision.upper())
399511a6afcSJed Brown      cmakeset(fd,'PETSC_CLANGUAGE_'+self.languages.clanguage)
400511a6afcSJed Brown      if hasattr(self.compilers, 'FC'):
401511a6afcSJed Brown        cmakeset(fd,'PETSC_HAVE_FORTRAN')
402511a6afcSJed Brown        if self.compilers.fortranIsF90:
403511a6afcSJed Brown          cmakeset(fd,'PETSC_USING_F90')
404511a6afcSJed Brown      if self.sharedlibraries.useShared:
405511a6afcSJed Brown        cmakeset(fd,'BUILD_SHARED_LIBS')
406511a6afcSJed Brown    def writeBuildFlags(fd):
40706e8c1ddSJed Brown      def extendby(lib):
40806e8c1ddSJed Brown        libs = ensurelist(lib)
40906e8c1ddSJed Brown        lib_paths.extend(map(libpath,libs))
41006e8c1ddSJed Brown        lib_libs.extend(map(cleanlib,libs))
41106e8c1ddSJed Brown        uniqextend(includes,pkg.include)
412511a6afcSJed Brown      lib_paths = []
413511a6afcSJed Brown      lib_libs  = []
414511a6afcSJed Brown      includes  = []
415511a6afcSJed Brown      libvars   = []
416511a6afcSJed Brown      for pkg in self.framework.packages:
41706e8c1ddSJed Brown        extendby(pkg.lib)
41806e8c1ddSJed Brown      extendby(self.libraries.math)
41906e8c1ddSJed Brown      extendby(self.libraries.rt)
42006e8c1ddSJed Brown      extendby(self.compilers.flibs)
42106e8c1ddSJed Brown      extendby(self.compilers.cxxlibs)
42206e8c1ddSJed Brown      extendby(self.compilers.LIBS.split())
42306e8c1ddSJed Brown      for libname in nublast(lib_libs):
424511a6afcSJed Brown        libvar = 'PETSC_' + libname.upper() + '_LIB'
4254c0032a9SSatish Balay        addpath = ''
42606e8c1ddSJed Brown        for lpath in nublast(lib_paths):
4274c0032a9SSatish Balay          addpath += '"' + str(lpath) + '" '
4284c0032a9SSatish Balay        fd.write('find_library (' + libvar + ' ' + libname + ' HINTS ' + addpath + ')\n')
429511a6afcSJed Brown        libvars.append(libvar)
430511a6afcSJed Brown      fd.write('mark_as_advanced (' + ' '.join(libvars) + ')\n')
431511a6afcSJed Brown      fd.write('set (PETSC_PACKAGE_LIBS ' + ' '.join(map(cmakeexpand,libvars)) + ')\n')
432511a6afcSJed Brown      fd.write('set (PETSC_PACKAGE_INCLUDES ' + ' '.join(map(lambda i: '"'+i+'"',filter(notstandardinclude,includes))) + ')\n')
433511a6afcSJed Brown    fd = open(os.path.join(self.arch.arch,'conf','PETScConfig.cmake'), 'w')
434511a6afcSJed Brown    writeMacroDefinitions(fd)
435511a6afcSJed Brown    writeBuildFlags(fd)
436511a6afcSJed Brown    fd.close()
437511a6afcSJed Brown    return
438511a6afcSJed Brown
4398b0282a9SJed Brown  def dumpCMakeLists(self):
4408b0282a9SJed Brown    import sys
4418b0282a9SJed Brown    if sys.version_info >= (2,5):
4428b0282a9SJed Brown      import cmakegen
4438b0282a9SJed Brown      try:
4448b0282a9SJed Brown        cmakegen.main(self.petscdir.dir)
4458b0282a9SJed Brown      except (OSError), e:
4468b0282a9SJed Brown        self.framework.logPrint('Generating CMakeLists.txt failed:\n' + str(e))
4478b0282a9SJed Brown
4488b0282a9SJed Brown  def cmakeBoot(self):
4498b0282a9SJed Brown    import sys
450ae937f1dSJed Brown    self.cmakeboot_success = False
4510b7111d2SJed Brown    if sys.version_info >= (2,5) and hasattr(self.cmake,'cmake'):
452356464bcSMatthew G Knepley      try:
4538b0282a9SJed Brown        import cmakeboot
454ae937f1dSJed Brown        self.cmakeboot_success = cmakeboot.main(petscdir=self.petscdir.dir,petscarch=self.arch.arch,argDB=self.argDB,framework=self.framework,log=self.framework.log)
4558b0282a9SJed Brown      except (OSError), e:
4568b0282a9SJed Brown        self.framework.logPrint('Booting CMake in PETSC_ARCH failed:\n' + str(e))
457356464bcSMatthew G Knepley      except (ImportError, KeyError), e:
458356464bcSMatthew G Knepley        self.framework.logPrint('Importing cmakeboot failed:\n' + str(e))
459*f7b66a64SJed Brown      if self.cmakeboot_success and not hasattr(self.compilers, 'CUDAC'): # Our CMake build does not support CUDA at this time
460*f7b66a64SJed Brown        self.addMakeMacro('PETSC_BUILD_USING_CMAKE',1)
461356464bcSMatthew G Knepley    return
4628b0282a9SJed Brown
463b2843cf1SBarry Smith  def configurePrefetch(self):
464b2843cf1SBarry Smith    '''Sees if there are any prefetch functions supported'''
4653649974fSBarry Smith    if config.setCompilers.Configure.isSolaris() or self.framework.argDB['with-iphone'] or self.framework.argDB['with-cuda']:
46693f78423SSatish Balay      self.addDefine('Prefetch(a,b,c)', ' ')
46793f78423SSatish Balay      return
468ec284106SBarry Smith    self.pushLanguage(self.languages.clanguage)
46910699583SJed Brown    if self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch((const char*)v,_MM_HINT_NTA);\n'):
47050d8bf02SJed Brown      # The Intel Intrinsics manual [1] specifies the prototype
47150d8bf02SJed Brown      #
47250d8bf02SJed Brown      #   void _mm_prefetch(char const *a, int sel);
47350d8bf02SJed Brown      #
47450d8bf02SJed Brown      # but other vendors seem to insist on using subtly different
47550d8bf02SJed Brown      # prototypes, including void* for the pointer, and an enum for
47650d8bf02SJed Brown      # sel.  These are both reasonable changes, but negatively impact
47750d8bf02SJed Brown      # portability.
47850d8bf02SJed Brown      #
47950d8bf02SJed Brown      # [1] http://software.intel.com/file/6373
48050d8bf02SJed Brown      self.addDefine('HAVE_XMMINTRIN_H', 1)
48150d8bf02SJed Brown      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const char*)(a),(c))')
48250d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_NTA', '_MM_HINT_NTA')
48350d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T0',  '_MM_HINT_T0')
48450d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T1',  '_MM_HINT_T1')
48550d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T2',  '_MM_HINT_T2')
48650d8bf02SJed Brown    elif self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch(v,_MM_HINT_NTA);\n'):
48750d8bf02SJed Brown      self.addDefine('HAVE_XMMINTRIN_H', 1)
48850d8bf02SJed Brown      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const void*)(a),(c))')
48950d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_NTA', '_MM_HINT_NTA')
49050d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T0',  '_MM_HINT_T0')
49150d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T1',  '_MM_HINT_T1')
49250d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T2',  '_MM_HINT_T2')
49310699583SJed Brown    elif self.checkLink('', 'void *v = 0;__builtin_prefetch(v,0,0);\n'):
49410699583SJed Brown      # From GCC docs: void __builtin_prefetch(const void *addr,int rw,int locality)
49510699583SJed Brown      #
49610699583SJed Brown      #   The value of rw is a compile-time constant one or zero; one
49710699583SJed Brown      #   means that the prefetch is preparing for a write to the memory
49810699583SJed Brown      #   address and zero, the default, means that the prefetch is
49910699583SJed Brown      #   preparing for a read. The value locality must be a compile-time
50010699583SJed Brown      #   constant integer between zero and three. A value of zero means
50110699583SJed Brown      #   that the data has no temporal locality, so it need not be left
50210699583SJed Brown      #   in the cache after the access. A value of three means that the
50310699583SJed Brown      #   data has a high degree of temporal locality and should be left
50410699583SJed Brown      #   in all levels of cache possible. Values of one and two mean,
50510699583SJed Brown      #   respectively, a low or moderate degree of temporal locality.
50610699583SJed Brown      #
50710699583SJed Brown      # Here we adopt Intel's x86/x86-64 naming scheme for the locality
50810699583SJed Brown      # hints.  Using macros for these values in necessary since some
50910699583SJed Brown      # compilers require an enum.
51010699583SJed Brown      self.addDefine('Prefetch(a,b,c)', '__builtin_prefetch((a),(b),(c))')
51110699583SJed Brown      self.addDefine('PREFETCH_HINT_NTA', '0')
51210699583SJed Brown      self.addDefine('PREFETCH_HINT_T0',  '3')
51310699583SJed Brown      self.addDefine('PREFETCH_HINT_T1',  '2')
51410699583SJed Brown      self.addDefine('PREFETCH_HINT_T2',  '1')
515b2843cf1SBarry Smith    else:
516b2843cf1SBarry Smith      self.addDefine('Prefetch(a,b,c)', ' ')
5177d490b44SBarry Smith    self.popLanguage()
518b2843cf1SBarry Smith
5192400fdedSBarry Smith  def configureUnused(self):
5202400fdedSBarry Smith    '''Sees if __attribute((unused)) is supported'''
5212400fdedSBarry Smith    if self.framework.argDB['with-iphone'] or self.framework.argDB['with-cuda']:
5222400fdedSBarry Smith      self.addDefine('UNUSED', ' ')
5232400fdedSBarry Smith      return
5242400fdedSBarry Smith    self.pushLanguage(self.languages.clanguage)
5252400fdedSBarry Smith    if self.checkLink('__attribute((unused)) static int myfunc(void){ return 1;}', 'int i = myfunc();\n'):
5262400fdedSBarry Smith      self.addDefine('UNUSED', '__attribute((unused))')
5272400fdedSBarry Smith    else:
5282400fdedSBarry Smith      self.addDefine('UNUSED', ' ')
5292400fdedSBarry Smith    self.popLanguage()
5302400fdedSBarry Smith
5319800092aSJed Brown  def configureExpect(self):
5329800092aSJed Brown    '''Sees if the __builtin_expect directive is supported'''
5339800092aSJed Brown    self.pushLanguage(self.languages.clanguage)
5349800092aSJed Brown    if self.checkLink('', 'if (__builtin_expect(0,1)) return 1;'):
5359800092aSJed Brown      self.addDefine('HAVE_BUILTIN_EXPECT', 1)
5369800092aSJed Brown    self.popLanguage()
5379800092aSJed Brown
53853c77d0aSJed Brown  def configureFunctionName(self):
5391ec50b02SJed Brown    '''Sees if the compiler supports __func__ or a variant.  Falls back
5401ec50b02SJed Brown    on __FUNCT__ which PETSc source defines, but most users do not, thus
5411ec50b02SJed Brown    stack traces through user code are better when the compiler's
5421ec50b02SJed Brown    variant is used.'''
5431ec50b02SJed Brown    def getFunctionName(lang):
5441ec50b02SJed Brown      name = '__FUNCT__'
5451ec50b02SJed Brown      self.pushLanguage(lang)
54653c77d0aSJed Brown      if self.checkLink('', "if (__func__[0] != 'm') return 1;"):
5471ec50b02SJed Brown        name = '__func__'
54853c77d0aSJed Brown      elif self.checkLink('', "if (__FUNCTION__[0] != 'm') return 1;"):
5491ec50b02SJed Brown        name = '__FUNCTION__'
5501ec50b02SJed Brown      self.popLanguage()
5511ec50b02SJed Brown      return name
5521ec50b02SJed Brown    langs = []
553628773c9SSatish Balay
554628773c9SSatish Balay    self.addDefine('FUNCTION_NAME_C', getFunctionName('C'))
555628773c9SSatish Balay    if hasattr(self.compilers, 'CXX'):
556628773c9SSatish Balay      self.addDefine('FUNCTION_NAME_CXX', getFunctionName('Cxx'))
55712607bf0SSatish Balay    else:
55812607bf0SSatish Balay      self.addDefine('FUNCTION_NAME_CXX', '__FUNCT__')
55953c77d0aSJed Brown
560753ebd1dSJed Brown  def configureIntptrt(self):
561753ebd1dSJed Brown    '''Determine what to use for uintptr_t'''
562753ebd1dSJed Brown    def staticAssertSizeMatchesVoidStar(inc,typename):
563753ebd1dSJed Brown      # The declaration is an error if either array size is negative.
564753ebd1dSJed Brown      # It should be okay to use an int that is too large, but it would be very unlikely for this to be the case
565d26187a0SJed Brown      return self.checkCompile(inc, ('#define STATIC_ASSERT(cond) char negative_length_if_false[2*(!!(cond))-1]\n'
566979939cdSSatish Balay                                     + 'STATIC_ASSERT(sizeof(void*) == sizeof(%s));'%typename))
567753ebd1dSJed Brown    self.pushLanguage(self.languages.clanguage)
568753ebd1dSJed Brown    if self.checkCompile('#include <stdint.h>', 'int x; uintptr_t i = (uintptr_t)&x;'):
569753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'uintptr_t')
570753ebd1dSJed Brown    elif staticAssertSizeMatchesVoidStar('','unsigned long long'):
571753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'unsigned long long')
572753ebd1dSJed Brown    elif staticAssertSizeMatchesVoidStar('#include <stdlib.h>','size_t') or staticAssertSizeMatchesVoidStar('#include <string.h>', 'size_t'):
573753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'size_t')
574c82284b1SJed Brown    elif staticAssertSizeMatchesVoidStar('','unsigned long'):
575c82284b1SJed Brown      self.addDefine('UINTPTR_T', 'unsigned long')
5762d1b7972SSatish Balay    elif staticAssertSizeMatchesVoidStar('','unsigned'):
577753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'unsigned')
578d26187a0SJed Brown    else:
579d26187a0SJed Brown      raise RuntimeError('Could not find any unsigned integer type matching void*')
580753ebd1dSJed Brown    self.popLanguage()
581753ebd1dSJed Brown
582f8833479SBarry Smith  def configureInline(self):
583f8833479SBarry Smith    '''Get a generic inline keyword, depending on the language'''
584f8833479SBarry Smith    if self.languages.clanguage == 'C':
585f8833479SBarry Smith      self.addDefine('STATIC_INLINE', self.compilers.cStaticInlineKeyword)
586f8833479SBarry Smith      self.addDefine('RESTRICT', self.compilers.cRestrict)
587f8833479SBarry Smith    elif self.languages.clanguage == 'Cxx':
588f8833479SBarry Smith      self.addDefine('STATIC_INLINE', self.compilers.cxxStaticInlineKeyword)
589f8833479SBarry Smith      self.addDefine('RESTRICT', self.compilers.cxxRestrict)
590bfef2c86SBarry Smith
591bfef2c86SBarry Smith    if self.checkCompile('#include <dlfcn.h>\n void *ptr =  RTLD_DEFAULT;'):
592bfef2c86SBarry Smith      self.addDefine('RTLD_DEFAULT','1')
593f8833479SBarry Smith    return
594f8833479SBarry Smith
595f8833479SBarry Smith  def configureSolaris(self):
596f8833479SBarry Smith    '''Solaris specific stuff'''
597f8833479SBarry Smith    if os.path.isdir(os.path.join('/usr','ucblib')):
598f8833479SBarry Smith      try:
599f8833479SBarry Smith        flag = getattr(self.setCompilers, self.language[-1]+'SharedLinkerFlag')
600f8833479SBarry Smith      except AttributeError:
601f8833479SBarry Smith        flag = None
602f8833479SBarry Smith      if flag is None:
603f8833479SBarry Smith        self.compilers.LIBS += ' -L/usr/ucblib'
604f8833479SBarry Smith      else:
605f8833479SBarry Smith        self.compilers.LIBS += ' '+flag+'/usr/ucblib'
606f8833479SBarry Smith    return
607f8833479SBarry Smith
608f8833479SBarry Smith  def configureLinux(self):
609f8833479SBarry Smith    '''Linux specific stuff'''
6109f15855cSMatthew G Knepley    # TODO: Test for this by mallocing an odd number of floats and checking the address
611f8833479SBarry Smith    self.addDefine('HAVE_DOUBLE_ALIGN_MALLOC', 1)
612f8833479SBarry Smith    return
613f8833479SBarry Smith
614f8833479SBarry Smith  def configureWin32(self):
615f8833479SBarry Smith    '''Win32 non-cygwin specific stuff'''
616f8833479SBarry Smith    kernel32=0
617f8833479SBarry Smith    if self.libraries.add('Kernel32.lib','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
618f8833479SBarry Smith      self.addDefine('HAVE_WINDOWS_H',1)
619f8833479SBarry Smith      self.addDefine('HAVE_GETCOMPUTERNAME',1)
620f8833479SBarry Smith      kernel32=1
621f8833479SBarry Smith    elif self.libraries.add('kernel32','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
622f8833479SBarry Smith      self.addDefine('HAVE_WINDOWS_H',1)
623f8833479SBarry Smith      self.addDefine('HAVE_GETCOMPUTERNAME',1)
624f8833479SBarry Smith      kernel32=1
625f8833479SBarry Smith    if kernel32:
626eed94e11SSatish Balay      if self.framework.argDB['with-windows-graphics']:
627eed94e11SSatish Balay        self.addDefine('USE_WINDOWS_GRAPHICS',1)
628f8833479SBarry Smith      if self.checkLink('#include <Windows.h>','LoadLibrary(0)'):
629f8833479SBarry Smith        self.addDefine('HAVE_LOADLIBRARY',1)
630b50f6d9eSLisandro Dalcin      if self.checkLink('#include <Windows.h>','GetProcAddress(0,0)'):
631b50f6d9eSLisandro Dalcin        self.addDefine('HAVE_GETPROCADDRESS',1)
632b50f6d9eSLisandro Dalcin      if self.checkLink('#include <Windows.h>','FreeLibrary(0)'):
633b50f6d9eSLisandro Dalcin        self.addDefine('HAVE_FREELIBRARY',1)
634a21658a3SLisandro Dalcin      if self.checkLink('#include <Windows.h>','GetLastError()'):
635a21658a3SLisandro Dalcin        self.addDefine('HAVE_GETLASTERROR',1)
636a21658a3SLisandro Dalcin      if self.checkLink('#include <Windows.h>','SetLastError(0)'):
637a21658a3SLisandro Dalcin        self.addDefine('HAVE_SETLASTERROR',1)
638f8833479SBarry Smith      if self.checkLink('#include <Windows.h>\n','QueryPerformanceCounter(0);\n'):
639f8833479SBarry Smith        self.addDefine('USE_NT_TIME',1)
640f8833479SBarry Smith    if self.libraries.add('Advapi32.lib','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
641f8833479SBarry Smith      self.addDefine('HAVE_GET_USER_NAME',1)
642f8833479SBarry Smith    elif self.libraries.add('advapi32','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
643f8833479SBarry Smith      self.addDefine('HAVE_GET_USER_NAME',1)
644f8833479SBarry Smith
645f8833479SBarry Smith    if not self.libraries.add('User32.lib','GetDC',prototype='#include <Windows.h>',call='GetDC(0);'):
646f8833479SBarry Smith      self.libraries.add('user32','GetDC',prototype='#include <Windows.h>',call='GetDC(0);')
647f8833479SBarry Smith    if not self.libraries.add('Gdi32.lib','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);'):
648f8833479SBarry Smith      self.libraries.add('gdi32','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);')
649f8833479SBarry Smith
650f8833479SBarry Smith    self.types.check('int32_t', 'int')
651f8833479SBarry Smith    if not self.checkCompile('#include <sys/types.h>\n','uid_t u;\n'):
652f8833479SBarry Smith      self.addTypedef('int', 'uid_t')
653f8833479SBarry Smith      self.addTypedef('int', 'gid_t')
654f8833479SBarry Smith    if not self.checkLink('#if defined(PETSC_HAVE_UNISTD_H)\n#include <unistd.h>\n#endif\n','int a=R_OK;\n'):
655f8833479SBarry Smith      self.framework.addDefine('R_OK', '04')
656f8833479SBarry Smith      self.framework.addDefine('W_OK', '02')
657f8833479SBarry Smith      self.framework.addDefine('X_OK', '01')
658f8833479SBarry Smith    if not self.checkLink('#include <sys/stat.h>\n','int a=0;\nif (S_ISDIR(a)){}\n'):
659f8833479SBarry Smith      self.framework.addDefine('S_ISREG(a)', '(((a)&_S_IFMT) == _S_IFREG)')
660f8833479SBarry Smith      self.framework.addDefine('S_ISDIR(a)', '(((a)&_S_IFMT) == _S_IFDIR)')
661f8833479SBarry Smith    if self.checkCompile('#include <Windows.h>\n','LARGE_INTEGER a;\nDWORD b=a.u.HighPart;\n'):
662f8833479SBarry Smith      self.addDefine('HAVE_LARGE_INTEGER_U',1)
663f8833479SBarry Smith
664f8833479SBarry Smith    # Windows requires a Binary file creation flag when creating/opening binary files.  Is a better test in order?
66549cb358eSMatthew G Knepley    if self.checkCompile('#include <Windows.h>\n', 'int flags = O_BINARY;'):
666f8833479SBarry Smith      self.addDefine('HAVE_O_BINARY',1)
667f8833479SBarry Smith
668f8833479SBarry Smith    if self.compilers.CC.find('win32fe') >= 0:
669f8833479SBarry Smith      self.addDefine('PATH_SEPARATOR','\';\'')
670f8833479SBarry Smith      self.addDefine('DIR_SEPARATOR','\'\\\\\'')
671f8833479SBarry Smith      self.addDefine('REPLACE_DIR_SEPARATOR','\'/\'')
672f8833479SBarry Smith      self.addDefine('CANNOT_START_DEBUGGER',1)
673f8833479SBarry Smith    else:
674f8833479SBarry Smith      self.addDefine('PATH_SEPARATOR','\':\'')
675f8833479SBarry Smith      self.addDefine('REPLACE_DIR_SEPARATOR','\'\\\\\'')
676f8833479SBarry Smith      self.addDefine('DIR_SEPARATOR','\'/\'')
677bfef2c86SBarry Smith
678f8833479SBarry Smith    return
679f8833479SBarry Smith
680f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
681569865ddSSatish Balay  def configureDefaultArch(self):
682569865ddSSatish Balay    conffile = os.path.join('conf', 'petscvariables')
683569865ddSSatish Balay    if self.framework.argDB['with-default-arch']:
684569865ddSSatish Balay      fd = file(conffile, 'w')
685569865ddSSatish Balay      fd.write('PETSC_ARCH='+self.arch.arch+'\n')
686da93591fSBarry Smith      fd.write('PETSC_DIR='+self.petscdir.dir+'\n')
687569865ddSSatish Balay      fd.write('include ${PETSC_DIR}/${PETSC_ARCH}/conf/petscvariables\n')
688569865ddSSatish Balay      fd.close()
689569865ddSSatish Balay      self.framework.actions.addArgument('PETSc', 'Build', 'Set default architecture to '+self.arch.arch+' in '+conffile)
690569865ddSSatish Balay    elif os.path.isfile(conffile):
691569865ddSSatish Balay      try:
692569865ddSSatish Balay        os.unlink(conffile)
693569865ddSSatish Balay      except:
694569865ddSSatish Balay        raise RuntimeError('Unable to remove file '+conffile+'. Did a different user create it?')
695569865ddSSatish Balay    return
696569865ddSSatish Balay
697569865ddSSatish Balay#-----------------------------------------------------------------------------------------------------
698f8833479SBarry Smith  def configureScript(self):
699f8833479SBarry Smith    '''Output a script in the conf directory which will reproduce the configuration'''
700f8833479SBarry Smith    import nargs
701495bf1a9SSatish Balay    import sys
702495bf1a9SSatish Balay    scriptName = os.path.join(self.arch.arch,'conf', 'reconfigure-'+self.arch.arch+'.py')
703f8833479SBarry Smith    args = dict([(nargs.Arg.parseArgument(arg)[0], arg) for arg in self.framework.clArgs])
704f8833479SBarry Smith    if 'configModules' in args:
7051063a081SSatish Balay      if nargs.Arg.parseArgument(args['configModules'])[1] == 'PETSc.Configure':
706f8833479SBarry Smith        del args['configModules']
707f8833479SBarry Smith    if 'optionsModule' in args:
708c1486898SMatthew Knepley      if nargs.Arg.parseArgument(args['optionsModule'])[1] == 'PETSc.compilerOptions':
709f8833479SBarry Smith        del args['optionsModule']
710f8833479SBarry Smith    if not 'PETSC_ARCH' in args:
7111063a081SSatish Balay      args['PETSC_ARCH'] = 'PETSC_ARCH='+str(self.arch.arch)
712f8833479SBarry Smith    f = file(scriptName, 'w')
713495bf1a9SSatish Balay    f.write('#!'+sys.executable+'\n')
714f8833479SBarry Smith    f.write('if __name__ == \'__main__\':\n')
715f8833479SBarry Smith    f.write('  import sys\n')
7167561c02cSSatish Balay    f.write('  import os\n')
7177561c02cSSatish Balay    f.write('  sys.path.insert(0, os.path.abspath(\'config\'))\n')
718f8833479SBarry Smith    f.write('  import configure\n')
7191063a081SSatish Balay    # pretty print repr(args.values())
7201063a081SSatish Balay    f.write('  configure_options = [\n')
7218bec23c5SJed Brown    for itm in sorted(args.values()):
7221063a081SSatish Balay      f.write('    \''+str(itm)+'\',\n')
7231063a081SSatish Balay    f.write('  ]\n')
724f8833479SBarry Smith    f.write('  configure.petsc_configure(configure_options)\n')
725f8833479SBarry Smith    f.close()
726f8833479SBarry Smith    try:
727f8833479SBarry Smith      os.chmod(scriptName, 0775)
728f8833479SBarry Smith    except OSError, e:
729f8833479SBarry Smith      self.framework.logPrint('Unable to make reconfigure script executable:\n'+str(e))
730f8833479SBarry Smith    self.framework.actions.addArgument('PETSc', 'File creation', 'Created '+scriptName+' for automatic reconfiguration')
731f8833479SBarry Smith    return
732f8833479SBarry Smith
733f8833479SBarry Smith  def configureInstall(self):
734f8833479SBarry Smith    '''Setup the directories for installation'''
735f8833479SBarry Smith    if self.framework.argDB['prefix']:
736f8833479SBarry Smith      self.installdir = self.framework.argDB['prefix']
737824e893fSSatish Balay      self.addMakeRule('shared_install','',['-@echo "Now to install the libraries do:"',\
738824e893fSSatish Balay                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} install"',\
739315b77e6SSatish Balay                                              '-@echo "========================================="'])
740f8833479SBarry Smith    else:
741b559b537SSatish Balay      self.installdir = os.path.join(self.petscdir.dir,self.arch.arch)
742824e893fSSatish Balay      self.addMakeRule('shared_install','',['-@echo "Now to check if the libraries are working do:"',\
743824e893fSSatish Balay                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} test"',\
744315b77e6SSatish Balay                                              '-@echo "========================================="'])
745f8833479SBarry Smith      return
746f8833479SBarry Smith
747f8833479SBarry Smith  def configureGCOV(self):
748f8833479SBarry Smith    if self.framework.argDB['with-gcov']:
749f8833479SBarry Smith      self.addDefine('USE_GCOV','1')
750f8833479SBarry Smith    return
751f8833479SBarry Smith
752f8833479SBarry Smith  def configureFortranFlush(self):
753f8833479SBarry Smith    if hasattr(self.compilers, 'FC'):
754f8833479SBarry Smith      for baseName in ['flush','flush_']:
755f8833479SBarry Smith        if self.libraries.check('', baseName, otherLibs = self.compilers.flibs, fortranMangle = 1):
756f8833479SBarry Smith          self.addDefine('HAVE_'+baseName.upper(), 1)
757f8833479SBarry Smith          return
758f8833479SBarry Smith
759f8833479SBarry Smith
760f8833479SBarry Smith  def configure(self):
761f8833479SBarry Smith    if not os.path.samefile(self.petscdir.dir, os.getcwd()):
762f8833479SBarry Smith      raise RuntimeError('Wrong PETSC_DIR option specified: '+str(self.petscdir.dir) + '\n  Configure invoked in: '+os.path.realpath(os.getcwd()))
763550489e3SMatthew G Knepley    if self.framework.argDB['prefix'] and os.path.isdir(self.framework.argDB['prefix']) and os.path.samefile(self.framework.argDB['prefix'],self.petscdir.dir):
7643552d8fbSSatish Balay      raise RuntimeError('Incorrect option --prefix='+self.framework.argDB['prefix']+' specified. It cannot be same as PETSC_DIR!')
765c16c35a9SSatish Balay    if self.framework.argDB['prefix'] and os.path.isdir(self.framework.argDB['prefix']) and os.path.samefile(self.framework.argDB['prefix'],os.path.join(self.petscdir.dir,self.arch.arch)):
766c16c35a9SSatish Balay      raise RuntimeError('Incorrect option --prefix='+self.framework.argDB['prefix']+' specified. It cannot be same as PETSC_DIR/PETSC_ARCH!')
767f16c1317SJed Brown    self.framework.header          = os.path.join(self.arch.arch,'include','petscconf.h')
768f16c1317SJed Brown    self.framework.cHeader         = os.path.join(self.arch.arch,'include','petscfix.h')
769f16c1317SJed Brown    self.framework.makeMacroHeader = os.path.join(self.arch.arch,'conf','petscvariables')
770f16c1317SJed Brown    self.framework.makeRuleHeader  = os.path.join(self.arch.arch,'conf','petscrules')
771f8833479SBarry Smith    if self.libraries.math is None:
772f8833479SBarry Smith      raise RuntimeError('PETSc requires a functional math library. Please send configure.log to petsc-maint@mcs.anl.gov.')
773f8833479SBarry Smith    if self.languages.clanguage == 'Cxx' and not hasattr(self.compilers, 'CXX'):
774f8833479SBarry Smith      raise RuntimeError('Cannot set C language to C++ without a functional C++ compiler.')
775f8833479SBarry Smith    self.executeTest(self.configureInline)
776b2843cf1SBarry Smith    self.executeTest(self.configurePrefetch)
7772400fdedSBarry Smith    self.executeTest(self.configureUnused)
7789800092aSJed Brown    self.executeTest(self.configureExpect);
77953c77d0aSJed Brown    self.executeTest(self.configureFunctionName);
780753ebd1dSJed Brown    self.executeTest(self.configureIntptrt);
781f8833479SBarry Smith    self.executeTest(self.configureSolaris)
782f8833479SBarry Smith    self.executeTest(self.configureLinux)
783f8833479SBarry Smith    self.executeTest(self.configureWin32)
784569865ddSSatish Balay    self.executeTest(self.configureDefaultArch)
785f8833479SBarry Smith    self.executeTest(self.configureScript)
786f8833479SBarry Smith    self.executeTest(self.configureInstall)
787f8833479SBarry Smith    self.executeTest(self.configureGCOV)
788f8833479SBarry Smith    self.executeTest(self.configureFortranFlush)
789f8833479SBarry Smith    # dummy rules, always needed except for remote builds
790f8833479SBarry Smith    self.addMakeRule('remote','')
791f8833479SBarry Smith    self.addMakeRule('remoteclean','')
792f8833479SBarry Smith
793f8833479SBarry Smith    self.Dump()
794f8833479SBarry Smith    self.dumpConfigInfo()
7952a4161d9SMatthew G Knepley    self.dumpMachineInfo()
796511a6afcSJed Brown    self.dumpCMakeConfig()
7978b0282a9SJed Brown    self.dumpCMakeLists()
7988b0282a9SJed Brown    self.cmakeBoot()
799f8833479SBarry Smith    self.framework.log.write('================================================================================\n')
800f8833479SBarry Smith    self.logClear()
801f8833479SBarry Smith    return
802