xref: /petsc/config/PETSc/Configure.py (revision a8a678f19bf6cbef4fcd1512d1aaab71af633e06)
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:
26*a8a678f1SSatish Balay      desc.append(' or (experimental with cmake):')
27ae937f1dSJed Brown      desc.append('   make -j4 -C '+os.path.join(self.petscdir.dir,self.arch.arch))
28*a8a678f1SSatish 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#-----------------------------------------------------------------------------------------------------
211f8833479SBarry Smith    if self.functions.haveFunction('gethostbyname') and self.functions.haveFunction('socket'):
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))
459356464bcSMatthew G Knepley    return
4608b0282a9SJed Brown
461b2843cf1SBarry Smith  def configurePrefetch(self):
462b2843cf1SBarry Smith    '''Sees if there are any prefetch functions supported'''
4633649974fSBarry Smith    if config.setCompilers.Configure.isSolaris() or self.framework.argDB['with-iphone'] or self.framework.argDB['with-cuda']:
46493f78423SSatish Balay      self.addDefine('Prefetch(a,b,c)', ' ')
46593f78423SSatish Balay      return
466ec284106SBarry Smith    self.pushLanguage(self.languages.clanguage)
46710699583SJed Brown    if self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch((const char*)v,_MM_HINT_NTA);\n'):
46850d8bf02SJed Brown      # The Intel Intrinsics manual [1] specifies the prototype
46950d8bf02SJed Brown      #
47050d8bf02SJed Brown      #   void _mm_prefetch(char const *a, int sel);
47150d8bf02SJed Brown      #
47250d8bf02SJed Brown      # but other vendors seem to insist on using subtly different
47350d8bf02SJed Brown      # prototypes, including void* for the pointer, and an enum for
47450d8bf02SJed Brown      # sel.  These are both reasonable changes, but negatively impact
47550d8bf02SJed Brown      # portability.
47650d8bf02SJed Brown      #
47750d8bf02SJed Brown      # [1] http://software.intel.com/file/6373
47850d8bf02SJed Brown      self.addDefine('HAVE_XMMINTRIN_H', 1)
47950d8bf02SJed Brown      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const char*)(a),(c))')
48050d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_NTA', '_MM_HINT_NTA')
48150d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T0',  '_MM_HINT_T0')
48250d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T1',  '_MM_HINT_T1')
48350d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T2',  '_MM_HINT_T2')
48450d8bf02SJed Brown    elif self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch(v,_MM_HINT_NTA);\n'):
48550d8bf02SJed Brown      self.addDefine('HAVE_XMMINTRIN_H', 1)
48650d8bf02SJed Brown      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const void*)(a),(c))')
48750d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_NTA', '_MM_HINT_NTA')
48850d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T0',  '_MM_HINT_T0')
48950d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T1',  '_MM_HINT_T1')
49050d8bf02SJed Brown      self.addDefine('PREFETCH_HINT_T2',  '_MM_HINT_T2')
49110699583SJed Brown    elif self.checkLink('', 'void *v = 0;__builtin_prefetch(v,0,0);\n'):
49210699583SJed Brown      # From GCC docs: void __builtin_prefetch(const void *addr,int rw,int locality)
49310699583SJed Brown      #
49410699583SJed Brown      #   The value of rw is a compile-time constant one or zero; one
49510699583SJed Brown      #   means that the prefetch is preparing for a write to the memory
49610699583SJed Brown      #   address and zero, the default, means that the prefetch is
49710699583SJed Brown      #   preparing for a read. The value locality must be a compile-time
49810699583SJed Brown      #   constant integer between zero and three. A value of zero means
49910699583SJed Brown      #   that the data has no temporal locality, so it need not be left
50010699583SJed Brown      #   in the cache after the access. A value of three means that the
50110699583SJed Brown      #   data has a high degree of temporal locality and should be left
50210699583SJed Brown      #   in all levels of cache possible. Values of one and two mean,
50310699583SJed Brown      #   respectively, a low or moderate degree of temporal locality.
50410699583SJed Brown      #
50510699583SJed Brown      # Here we adopt Intel's x86/x86-64 naming scheme for the locality
50610699583SJed Brown      # hints.  Using macros for these values in necessary since some
50710699583SJed Brown      # compilers require an enum.
50810699583SJed Brown      self.addDefine('Prefetch(a,b,c)', '__builtin_prefetch((a),(b),(c))')
50910699583SJed Brown      self.addDefine('PREFETCH_HINT_NTA', '0')
51010699583SJed Brown      self.addDefine('PREFETCH_HINT_T0',  '3')
51110699583SJed Brown      self.addDefine('PREFETCH_HINT_T1',  '2')
51210699583SJed Brown      self.addDefine('PREFETCH_HINT_T2',  '1')
513b2843cf1SBarry Smith    else:
514b2843cf1SBarry Smith      self.addDefine('Prefetch(a,b,c)', ' ')
5157d490b44SBarry Smith    self.popLanguage()
516b2843cf1SBarry Smith
5172400fdedSBarry Smith  def configureUnused(self):
5182400fdedSBarry Smith    '''Sees if __attribute((unused)) is supported'''
5192400fdedSBarry Smith    if self.framework.argDB['with-iphone'] or self.framework.argDB['with-cuda']:
5202400fdedSBarry Smith      self.addDefine('UNUSED', ' ')
5212400fdedSBarry Smith      return
5222400fdedSBarry Smith    self.pushLanguage(self.languages.clanguage)
5232400fdedSBarry Smith    if self.checkLink('__attribute((unused)) static int myfunc(void){ return 1;}', 'int i = myfunc();\n'):
5242400fdedSBarry Smith      self.addDefine('UNUSED', '__attribute((unused))')
5252400fdedSBarry Smith    else:
5262400fdedSBarry Smith      self.addDefine('UNUSED', ' ')
5272400fdedSBarry Smith    self.popLanguage()
5282400fdedSBarry Smith
5299800092aSJed Brown  def configureExpect(self):
5309800092aSJed Brown    '''Sees if the __builtin_expect directive is supported'''
5319800092aSJed Brown    self.pushLanguage(self.languages.clanguage)
5329800092aSJed Brown    if self.checkLink('', 'if (__builtin_expect(0,1)) return 1;'):
5339800092aSJed Brown      self.addDefine('HAVE_BUILTIN_EXPECT', 1)
5349800092aSJed Brown    self.popLanguage()
5359800092aSJed Brown
53653c77d0aSJed Brown  def configureFunctionName(self):
5371ec50b02SJed Brown    '''Sees if the compiler supports __func__ or a variant.  Falls back
5381ec50b02SJed Brown    on __FUNCT__ which PETSc source defines, but most users do not, thus
5391ec50b02SJed Brown    stack traces through user code are better when the compiler's
5401ec50b02SJed Brown    variant is used.'''
5411ec50b02SJed Brown    def getFunctionName(lang):
5421ec50b02SJed Brown      name = '__FUNCT__'
5431ec50b02SJed Brown      self.pushLanguage(lang)
54453c77d0aSJed Brown      if self.checkLink('', "if (__func__[0] != 'm') return 1;"):
5451ec50b02SJed Brown        name = '__func__'
54653c77d0aSJed Brown      elif self.checkLink('', "if (__FUNCTION__[0] != 'm') return 1;"):
5471ec50b02SJed Brown        name = '__FUNCTION__'
5481ec50b02SJed Brown      self.popLanguage()
5491ec50b02SJed Brown      return name
5501ec50b02SJed Brown    langs = []
551628773c9SSatish Balay
552628773c9SSatish Balay    self.addDefine('FUNCTION_NAME_C', getFunctionName('C'))
553628773c9SSatish Balay    if hasattr(self.compilers, 'CXX'):
554628773c9SSatish Balay      self.addDefine('FUNCTION_NAME_CXX', getFunctionName('Cxx'))
55512607bf0SSatish Balay    else:
55612607bf0SSatish Balay      self.addDefine('FUNCTION_NAME_CXX', '__FUNCT__')
55753c77d0aSJed Brown
558753ebd1dSJed Brown  def configureIntptrt(self):
559753ebd1dSJed Brown    '''Determine what to use for uintptr_t'''
560753ebd1dSJed Brown    def staticAssertSizeMatchesVoidStar(inc,typename):
561753ebd1dSJed Brown      # The declaration is an error if either array size is negative.
562753ebd1dSJed 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
563d26187a0SJed Brown      return self.checkCompile(inc, ('#define STATIC_ASSERT(cond) char negative_length_if_false[2*(!!(cond))-1]\n'
564979939cdSSatish Balay                                     + 'STATIC_ASSERT(sizeof(void*) == sizeof(%s));'%typename))
565753ebd1dSJed Brown    self.pushLanguage(self.languages.clanguage)
566753ebd1dSJed Brown    if self.checkCompile('#include <stdint.h>', 'int x; uintptr_t i = (uintptr_t)&x;'):
567753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'uintptr_t')
568753ebd1dSJed Brown    elif staticAssertSizeMatchesVoidStar('','unsigned long long'):
569753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'unsigned long long')
570753ebd1dSJed Brown    elif staticAssertSizeMatchesVoidStar('#include <stdlib.h>','size_t') or staticAssertSizeMatchesVoidStar('#include <string.h>', 'size_t'):
571753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'size_t')
572c82284b1SJed Brown    elif staticAssertSizeMatchesVoidStar('','unsigned long'):
573c82284b1SJed Brown      self.addDefine('UINTPTR_T', 'unsigned long')
5742d1b7972SSatish Balay    elif staticAssertSizeMatchesVoidStar('','unsigned'):
575753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'unsigned')
576d26187a0SJed Brown    else:
577d26187a0SJed Brown      raise RuntimeError('Could not find any unsigned integer type matching void*')
578753ebd1dSJed Brown    self.popLanguage()
579753ebd1dSJed Brown
580f8833479SBarry Smith  def configureInline(self):
581f8833479SBarry Smith    '''Get a generic inline keyword, depending on the language'''
582f8833479SBarry Smith    if self.languages.clanguage == 'C':
583f8833479SBarry Smith      self.addDefine('STATIC_INLINE', self.compilers.cStaticInlineKeyword)
584f8833479SBarry Smith      self.addDefine('RESTRICT', self.compilers.cRestrict)
585f8833479SBarry Smith    elif self.languages.clanguage == 'Cxx':
586f8833479SBarry Smith      self.addDefine('STATIC_INLINE', self.compilers.cxxStaticInlineKeyword)
587f8833479SBarry Smith      self.addDefine('RESTRICT', self.compilers.cxxRestrict)
588bfef2c86SBarry Smith
589bfef2c86SBarry Smith    if self.checkCompile('#include <dlfcn.h>\n void *ptr =  RTLD_DEFAULT;'):
590bfef2c86SBarry Smith      self.addDefine('RTLD_DEFAULT','1')
591f8833479SBarry Smith    return
592f8833479SBarry Smith
593f8833479SBarry Smith  def configureSolaris(self):
594f8833479SBarry Smith    '''Solaris specific stuff'''
595f8833479SBarry Smith    if os.path.isdir(os.path.join('/usr','ucblib')):
596f8833479SBarry Smith      try:
597f8833479SBarry Smith        flag = getattr(self.setCompilers, self.language[-1]+'SharedLinkerFlag')
598f8833479SBarry Smith      except AttributeError:
599f8833479SBarry Smith        flag = None
600f8833479SBarry Smith      if flag is None:
601f8833479SBarry Smith        self.compilers.LIBS += ' -L/usr/ucblib'
602f8833479SBarry Smith      else:
603f8833479SBarry Smith        self.compilers.LIBS += ' '+flag+'/usr/ucblib'
604f8833479SBarry Smith    return
605f8833479SBarry Smith
606f8833479SBarry Smith  def configureLinux(self):
607f8833479SBarry Smith    '''Linux specific stuff'''
6089f15855cSMatthew G Knepley    # TODO: Test for this by mallocing an odd number of floats and checking the address
609f8833479SBarry Smith    self.addDefine('HAVE_DOUBLE_ALIGN_MALLOC', 1)
610f8833479SBarry Smith    return
611f8833479SBarry Smith
612f8833479SBarry Smith  def configureWin32(self):
613f8833479SBarry Smith    '''Win32 non-cygwin specific stuff'''
614f8833479SBarry Smith    kernel32=0
615f8833479SBarry Smith    if self.libraries.add('Kernel32.lib','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
616f8833479SBarry Smith      self.addDefine('HAVE_WINDOWS_H',1)
617f8833479SBarry Smith      self.addDefine('HAVE_GETCOMPUTERNAME',1)
618f8833479SBarry Smith      kernel32=1
619f8833479SBarry Smith    elif self.libraries.add('kernel32','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
620f8833479SBarry Smith      self.addDefine('HAVE_WINDOWS_H',1)
621f8833479SBarry Smith      self.addDefine('HAVE_GETCOMPUTERNAME',1)
622f8833479SBarry Smith      kernel32=1
623f8833479SBarry Smith    if kernel32:
624eed94e11SSatish Balay      if self.framework.argDB['with-windows-graphics']:
625eed94e11SSatish Balay        self.addDefine('USE_WINDOWS_GRAPHICS',1)
626f8833479SBarry Smith      if self.checkLink('#include <Windows.h>','LoadLibrary(0)'):
627f8833479SBarry Smith        self.addDefine('HAVE_LOADLIBRARY',1)
628b50f6d9eSLisandro Dalcin      if self.checkLink('#include <Windows.h>','GetProcAddress(0,0)'):
629b50f6d9eSLisandro Dalcin        self.addDefine('HAVE_GETPROCADDRESS',1)
630b50f6d9eSLisandro Dalcin      if self.checkLink('#include <Windows.h>','FreeLibrary(0)'):
631b50f6d9eSLisandro Dalcin        self.addDefine('HAVE_FREELIBRARY',1)
632a21658a3SLisandro Dalcin      if self.checkLink('#include <Windows.h>','GetLastError()'):
633a21658a3SLisandro Dalcin        self.addDefine('HAVE_GETLASTERROR',1)
634a21658a3SLisandro Dalcin      if self.checkLink('#include <Windows.h>','SetLastError(0)'):
635a21658a3SLisandro Dalcin        self.addDefine('HAVE_SETLASTERROR',1)
636f8833479SBarry Smith      if self.checkLink('#include <Windows.h>\n','QueryPerformanceCounter(0);\n'):
637f8833479SBarry Smith        self.addDefine('USE_NT_TIME',1)
638f8833479SBarry Smith    if self.libraries.add('Advapi32.lib','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
639f8833479SBarry Smith      self.addDefine('HAVE_GET_USER_NAME',1)
640f8833479SBarry Smith    elif self.libraries.add('advapi32','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
641f8833479SBarry Smith      self.addDefine('HAVE_GET_USER_NAME',1)
642f8833479SBarry Smith
643f8833479SBarry Smith    if not self.libraries.add('User32.lib','GetDC',prototype='#include <Windows.h>',call='GetDC(0);'):
644f8833479SBarry Smith      self.libraries.add('user32','GetDC',prototype='#include <Windows.h>',call='GetDC(0);')
645f8833479SBarry Smith    if not self.libraries.add('Gdi32.lib','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);'):
646f8833479SBarry Smith      self.libraries.add('gdi32','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);')
647f8833479SBarry Smith
648f8833479SBarry Smith    self.types.check('int32_t', 'int')
649f8833479SBarry Smith    if not self.checkCompile('#include <sys/types.h>\n','uid_t u;\n'):
650f8833479SBarry Smith      self.addTypedef('int', 'uid_t')
651f8833479SBarry Smith      self.addTypedef('int', 'gid_t')
652f8833479SBarry Smith    if not self.checkLink('#if defined(PETSC_HAVE_UNISTD_H)\n#include <unistd.h>\n#endif\n','int a=R_OK;\n'):
653f8833479SBarry Smith      self.framework.addDefine('R_OK', '04')
654f8833479SBarry Smith      self.framework.addDefine('W_OK', '02')
655f8833479SBarry Smith      self.framework.addDefine('X_OK', '01')
656f8833479SBarry Smith    if not self.checkLink('#include <sys/stat.h>\n','int a=0;\nif (S_ISDIR(a)){}\n'):
657f8833479SBarry Smith      self.framework.addDefine('S_ISREG(a)', '(((a)&_S_IFMT) == _S_IFREG)')
658f8833479SBarry Smith      self.framework.addDefine('S_ISDIR(a)', '(((a)&_S_IFMT) == _S_IFDIR)')
659f8833479SBarry Smith    if self.checkCompile('#include <Windows.h>\n','LARGE_INTEGER a;\nDWORD b=a.u.HighPart;\n'):
660f8833479SBarry Smith      self.addDefine('HAVE_LARGE_INTEGER_U',1)
661f8833479SBarry Smith
662f8833479SBarry Smith    # Windows requires a Binary file creation flag when creating/opening binary files.  Is a better test in order?
66349cb358eSMatthew G Knepley    if self.checkCompile('#include <Windows.h>\n', 'int flags = O_BINARY;'):
664f8833479SBarry Smith      self.addDefine('HAVE_O_BINARY',1)
665f8833479SBarry Smith
666f8833479SBarry Smith    if self.compilers.CC.find('win32fe') >= 0:
667f8833479SBarry Smith      self.addDefine('PATH_SEPARATOR','\';\'')
668f8833479SBarry Smith      self.addDefine('DIR_SEPARATOR','\'\\\\\'')
669f8833479SBarry Smith      self.addDefine('REPLACE_DIR_SEPARATOR','\'/\'')
670f8833479SBarry Smith      self.addDefine('CANNOT_START_DEBUGGER',1)
671f8833479SBarry Smith    else:
672f8833479SBarry Smith      self.addDefine('PATH_SEPARATOR','\':\'')
673f8833479SBarry Smith      self.addDefine('REPLACE_DIR_SEPARATOR','\'\\\\\'')
674f8833479SBarry Smith      self.addDefine('DIR_SEPARATOR','\'/\'')
675bfef2c86SBarry Smith
676f8833479SBarry Smith    return
677f8833479SBarry Smith
678f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
679569865ddSSatish Balay  def configureDefaultArch(self):
680569865ddSSatish Balay    conffile = os.path.join('conf', 'petscvariables')
681569865ddSSatish Balay    if self.framework.argDB['with-default-arch']:
682569865ddSSatish Balay      fd = file(conffile, 'w')
683569865ddSSatish Balay      fd.write('PETSC_ARCH='+self.arch.arch+'\n')
684da93591fSBarry Smith      fd.write('PETSC_DIR='+self.petscdir.dir+'\n')
685569865ddSSatish Balay      fd.write('include ${PETSC_DIR}/${PETSC_ARCH}/conf/petscvariables\n')
686569865ddSSatish Balay      fd.close()
687569865ddSSatish Balay      self.framework.actions.addArgument('PETSc', 'Build', 'Set default architecture to '+self.arch.arch+' in '+conffile)
688569865ddSSatish Balay    elif os.path.isfile(conffile):
689569865ddSSatish Balay      try:
690569865ddSSatish Balay        os.unlink(conffile)
691569865ddSSatish Balay      except:
692569865ddSSatish Balay        raise RuntimeError('Unable to remove file '+conffile+'. Did a different user create it?')
693569865ddSSatish Balay    return
694569865ddSSatish Balay
695569865ddSSatish Balay#-----------------------------------------------------------------------------------------------------
696f8833479SBarry Smith  def configureScript(self):
697f8833479SBarry Smith    '''Output a script in the conf directory which will reproduce the configuration'''
698f8833479SBarry Smith    import nargs
699495bf1a9SSatish Balay    import sys
700495bf1a9SSatish Balay    scriptName = os.path.join(self.arch.arch,'conf', 'reconfigure-'+self.arch.arch+'.py')
701f8833479SBarry Smith    args = dict([(nargs.Arg.parseArgument(arg)[0], arg) for arg in self.framework.clArgs])
702f8833479SBarry Smith    if 'configModules' in args:
7031063a081SSatish Balay      if nargs.Arg.parseArgument(args['configModules'])[1] == 'PETSc.Configure':
704f8833479SBarry Smith        del args['configModules']
705f8833479SBarry Smith    if 'optionsModule' in args:
706c1486898SMatthew Knepley      if nargs.Arg.parseArgument(args['optionsModule'])[1] == 'PETSc.compilerOptions':
707f8833479SBarry Smith        del args['optionsModule']
708f8833479SBarry Smith    if not 'PETSC_ARCH' in args:
7091063a081SSatish Balay      args['PETSC_ARCH'] = 'PETSC_ARCH='+str(self.arch.arch)
710f8833479SBarry Smith    f = file(scriptName, 'w')
711495bf1a9SSatish Balay    f.write('#!'+sys.executable+'\n')
712f8833479SBarry Smith    f.write('if __name__ == \'__main__\':\n')
713f8833479SBarry Smith    f.write('  import sys\n')
7147561c02cSSatish Balay    f.write('  import os\n')
7157561c02cSSatish Balay    f.write('  sys.path.insert(0, os.path.abspath(\'config\'))\n')
716f8833479SBarry Smith    f.write('  import configure\n')
7171063a081SSatish Balay    # pretty print repr(args.values())
7181063a081SSatish Balay    f.write('  configure_options = [\n')
7198bec23c5SJed Brown    for itm in sorted(args.values()):
7201063a081SSatish Balay      f.write('    \''+str(itm)+'\',\n')
7211063a081SSatish Balay    f.write('  ]\n')
722f8833479SBarry Smith    f.write('  configure.petsc_configure(configure_options)\n')
723f8833479SBarry Smith    f.close()
724f8833479SBarry Smith    try:
725f8833479SBarry Smith      os.chmod(scriptName, 0775)
726f8833479SBarry Smith    except OSError, e:
727f8833479SBarry Smith      self.framework.logPrint('Unable to make reconfigure script executable:\n'+str(e))
728f8833479SBarry Smith    self.framework.actions.addArgument('PETSc', 'File creation', 'Created '+scriptName+' for automatic reconfiguration')
729f8833479SBarry Smith    return
730f8833479SBarry Smith
731f8833479SBarry Smith  def configureInstall(self):
732f8833479SBarry Smith    '''Setup the directories for installation'''
733f8833479SBarry Smith    if self.framework.argDB['prefix']:
734f8833479SBarry Smith      self.installdir = self.framework.argDB['prefix']
735824e893fSSatish Balay      self.addMakeRule('shared_install','',['-@echo "Now to install the libraries do:"',\
736824e893fSSatish Balay                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} install"',\
737315b77e6SSatish Balay                                              '-@echo "========================================="'])
738f8833479SBarry Smith    else:
739b559b537SSatish Balay      self.installdir = os.path.join(self.petscdir.dir,self.arch.arch)
740824e893fSSatish Balay      self.addMakeRule('shared_install','',['-@echo "Now to check if the libraries are working do:"',\
741824e893fSSatish Balay                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} test"',\
742315b77e6SSatish Balay                                              '-@echo "========================================="'])
743f8833479SBarry Smith      return
744f8833479SBarry Smith
745f8833479SBarry Smith  def configureGCOV(self):
746f8833479SBarry Smith    if self.framework.argDB['with-gcov']:
747f8833479SBarry Smith      self.addDefine('USE_GCOV','1')
748f8833479SBarry Smith    return
749f8833479SBarry Smith
750f8833479SBarry Smith  def configureFortranFlush(self):
751f8833479SBarry Smith    if hasattr(self.compilers, 'FC'):
752f8833479SBarry Smith      for baseName in ['flush','flush_']:
753f8833479SBarry Smith        if self.libraries.check('', baseName, otherLibs = self.compilers.flibs, fortranMangle = 1):
754f8833479SBarry Smith          self.addDefine('HAVE_'+baseName.upper(), 1)
755f8833479SBarry Smith          return
756f8833479SBarry Smith
757f8833479SBarry Smith
758f8833479SBarry Smith  def configure(self):
759f8833479SBarry Smith    if not os.path.samefile(self.petscdir.dir, os.getcwd()):
760f8833479SBarry Smith      raise RuntimeError('Wrong PETSC_DIR option specified: '+str(self.petscdir.dir) + '\n  Configure invoked in: '+os.path.realpath(os.getcwd()))
761550489e3SMatthew 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):
7623552d8fbSSatish Balay      raise RuntimeError('Incorrect option --prefix='+self.framework.argDB['prefix']+' specified. It cannot be same as PETSC_DIR!')
763c16c35a9SSatish 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)):
764c16c35a9SSatish Balay      raise RuntimeError('Incorrect option --prefix='+self.framework.argDB['prefix']+' specified. It cannot be same as PETSC_DIR/PETSC_ARCH!')
765f16c1317SJed Brown    self.framework.header          = os.path.join(self.arch.arch,'include','petscconf.h')
766f16c1317SJed Brown    self.framework.cHeader         = os.path.join(self.arch.arch,'include','petscfix.h')
767f16c1317SJed Brown    self.framework.makeMacroHeader = os.path.join(self.arch.arch,'conf','petscvariables')
768f16c1317SJed Brown    self.framework.makeRuleHeader  = os.path.join(self.arch.arch,'conf','petscrules')
769f8833479SBarry Smith    if self.libraries.math is None:
770f8833479SBarry Smith      raise RuntimeError('PETSc requires a functional math library. Please send configure.log to petsc-maint@mcs.anl.gov.')
771f8833479SBarry Smith    if self.languages.clanguage == 'Cxx' and not hasattr(self.compilers, 'CXX'):
772f8833479SBarry Smith      raise RuntimeError('Cannot set C language to C++ without a functional C++ compiler.')
773f8833479SBarry Smith    self.executeTest(self.configureInline)
774b2843cf1SBarry Smith    self.executeTest(self.configurePrefetch)
7752400fdedSBarry Smith    self.executeTest(self.configureUnused)
7769800092aSJed Brown    self.executeTest(self.configureExpect);
77753c77d0aSJed Brown    self.executeTest(self.configureFunctionName);
778753ebd1dSJed Brown    self.executeTest(self.configureIntptrt);
779f8833479SBarry Smith    self.executeTest(self.configureSolaris)
780f8833479SBarry Smith    self.executeTest(self.configureLinux)
781f8833479SBarry Smith    self.executeTest(self.configureWin32)
782569865ddSSatish Balay    self.executeTest(self.configureDefaultArch)
783f8833479SBarry Smith    self.executeTest(self.configureScript)
784f8833479SBarry Smith    self.executeTest(self.configureInstall)
785f8833479SBarry Smith    self.executeTest(self.configureGCOV)
786f8833479SBarry Smith    self.executeTest(self.configureFortranFlush)
787f8833479SBarry Smith    # dummy rules, always needed except for remote builds
788f8833479SBarry Smith    self.addMakeRule('remote','')
789f8833479SBarry Smith    self.addMakeRule('remoteclean','')
790f8833479SBarry Smith
791f8833479SBarry Smith    self.Dump()
792f8833479SBarry Smith    self.dumpConfigInfo()
7932a4161d9SMatthew G Knepley    self.dumpMachineInfo()
794511a6afcSJed Brown    self.dumpCMakeConfig()
7958b0282a9SJed Brown    self.dumpCMakeLists()
7968b0282a9SJed Brown    self.cmakeBoot()
797f8833479SBarry Smith    self.framework.log.write('================================================================================\n')
798f8833479SBarry Smith    self.logClear()
799f8833479SBarry Smith    return
800