xref: /petsc/config/PETSc/Configure.py (revision 3552d8fbef600cf64e7e1c7e2f3e017589ce6e10)
1f8833479SBarry Smithimport config.base
2f8833479SBarry Smith
3f8833479SBarry Smithimport os
4f8833479SBarry Smithimport re
5f8833479SBarry Smith
6f8833479SBarry Smithclass Configure(config.base.Configure):
7f8833479SBarry Smith  def __init__(self, framework):
8f8833479SBarry Smith    config.base.Configure.__init__(self, framework)
9f8833479SBarry Smith    self.headerPrefix = 'PETSC'
10f8833479SBarry Smith    self.substPrefix  = 'PETSC'
11f8833479SBarry Smith    self.defineAutoconfMacros()
12f8833479SBarry Smith    return
13f8833479SBarry Smith
147c939e48SSatish Balay  def __str2__(self):
157c939e48SSatish Balay    desc = []
167c939e48SSatish Balay    desc.append('xxx==========================================================================xxx')
177c939e48SSatish Balay    desc.append('   Configure stage complete. Now build PETSc libraries with:')
187c939e48SSatish Balay    desc.append('   make PETSC_DIR='+self.petscdir.dir+' PETSC_ARCH='+self.arch.arch+' all')
197c939e48SSatish Balay    desc.append('xxx==========================================================================xxx')
207c939e48SSatish Balay    return '\n'.join(desc)+'\n'
21f8833479SBarry Smith
22f8833479SBarry Smith  def setupHelp(self, help):
23f8833479SBarry Smith    import nargs
24f8833479SBarry Smith    help.addArgument('PETSc',  '-prefix=<path>',                  nargs.Arg(None, '', 'Specifiy location to install PETSc (eg. /usr/local)'))
25eed94e11SSatish Balay    help.addArgument('Windows','-with-windows-graphics=<bool>',   nargs.ArgBool(None, 1,'Enable check for Windows Graphics'))
26569865ddSSatish Balay    help.addArgument('PETSc', '-with-default-arch=<bool>',        nargs.ArgBool(None, 1, 'Allow using the last configured arch without setting PETSC_ARCH'))
2757cb31baSSatish Balay    help.addArgument('PETSc','-with-single-library=<bool>',       nargs.ArgBool(None, 1,'Put all PETSc code into the single -lpetsc library'))
28eed94e11SSatish Balay
29f8833479SBarry Smith    return
30f8833479SBarry Smith
31f8833479SBarry Smith  def setupDependencies(self, framework):
32f8833479SBarry Smith    config.base.Configure.setupDependencies(self, framework)
33f8833479SBarry Smith    self.setCompilers  = framework.require('config.setCompilers',      self)
34f8833479SBarry Smith    self.arch          = framework.require('PETSc.utilities.arch',     self.setCompilers)
35f8833479SBarry Smith    self.petscdir      = framework.require('PETSc.utilities.petscdir', self.setCompilers)
36f8833479SBarry Smith    self.languages     = framework.require('PETSc.utilities.languages',self.setCompilers)
37f8833479SBarry Smith    self.debugging     = framework.require('PETSc.utilities.debugging',self.setCompilers)
388397fe1aSBarry Smith    self.make          = framework.require('PETSc.utilities.Make',     self)
39a2b1cf82SBarry Smith    self.CHUD          = framework.require('PETSc.utilities.CHUD',     self)
40f8833479SBarry Smith    self.compilers     = framework.require('config.compilers',         self)
41f8833479SBarry Smith    self.types         = framework.require('config.types',             self)
42f8833479SBarry Smith    self.headers       = framework.require('config.headers',           self)
43f8833479SBarry Smith    self.functions     = framework.require('config.functions',         self)
44f8833479SBarry Smith    self.libraries     = framework.require('config.libraries',         self)
45f8833479SBarry Smith    if os.path.isdir(os.path.join('config', 'PETSc')):
46f8833479SBarry Smith      for d in ['utilities', 'packages']:
47f8833479SBarry Smith        for utility in os.listdir(os.path.join('config', 'PETSc', d)):
48f8833479SBarry Smith          (utilityName, ext) = os.path.splitext(utility)
49f8833479SBarry Smith          if not utilityName.startswith('.') and not utilityName.startswith('#') and ext == '.py' and not utilityName == '__init__':
50f8833479SBarry Smith            utilityObj                    = self.framework.require('PETSc.'+d+'.'+utilityName, self)
51f8833479SBarry Smith            utilityObj.headerPrefix       = self.headerPrefix
52f1ecedd2SMatthew Knepley            utilityObj.archProvider       = self.arch
53fdfda77fSMatthew Knepley            utilityObj.languageProvider   = self.languages
54fdfda77fSMatthew Knepley            utilityObj.installDirProvider = self.petscdir
55f8833479SBarry Smith            setattr(self, utilityName.lower(), utilityObj)
56e8793464SBarry Smith    self.qd    = framework.require('PETSc.packages.qd', self)
575389abc2SBarry Smith    self.qd.archProvider      = self.arch
585389abc2SBarry Smith    self.qd.precisionProvider = self.scalartypes
595389abc2SBarry Smith    self.qd.installDirProvider= self.petscdir
608cf378d1SBarry Smith
618cf378d1SBarry Smith    #force blaslapack to depend on scalarType so precision is set before BlasLapack is built
62f8833479SBarry Smith    self.blaslapack    = framework.require('config.packages.BlasLapack', self)
638cf378d1SBarry Smith    framework.require('PETSc.utilities.scalarTypes', self.blaslapack)
64f8833479SBarry Smith    self.blaslapack.archProvider      = self.arch
65f8833479SBarry Smith    self.blaslapack.precisionProvider = self.scalartypes
66f8833479SBarry Smith    self.blaslapack.installDirProvider= self.petscdir
678cf378d1SBarry Smith
68f8833479SBarry Smith    self.mpi           = framework.require('config.packages.MPI',        self)
69f8833479SBarry Smith    self.mpi.archProvider             = self.arch
70f8833479SBarry Smith    self.mpi.languageProvider         = self.languages
71f8833479SBarry Smith    self.mpi.installDirProvider       = self.petscdir
72f8833479SBarry Smith    self.umfpack       = framework.require('config.packages.UMFPACK',    self)
73f8833479SBarry Smith    self.umfpack.archProvider         = self.arch
74f8833479SBarry Smith    self.umfpack.languageProvider     = self.languages
75f8833479SBarry Smith    self.umfpack.installDirProvider   = self.petscdir
76d5b93044SMatthew Knepley    self.Boost         = framework.require('config.packages.Boost',      self)
77d5b93044SMatthew Knepley    self.Boost.archProvider           = self.arch
78d5b93044SMatthew Knepley    self.Boost.languageProvider       = self.languages
79d5b93044SMatthew Knepley    self.Boost.installDirProvider     = self.petscdir
80d5b93044SMatthew Knepley    self.Fiat          = framework.require('config.packages.Fiat',       self)
81d5b93044SMatthew Knepley    self.Fiat.archProvider            = self.arch
82d5b93044SMatthew Knepley    self.Fiat.languageProvider        = self.languages
83d5b93044SMatthew Knepley    self.Fiat.installDirProvider      = self.petscdir
847da8077dSMatthew Knepley    self.ExodusII      = framework.require('config.packages.ExodusII',   self)
857da8077dSMatthew Knepley    self.ExodusII.archProvider        = self.arch
867da8077dSMatthew Knepley    self.ExodusII.languageProvider    = self.languages
877da8077dSMatthew Knepley    self.ExodusII.installDirProvider  = self.petscdir
88f8833479SBarry Smith
89f8833479SBarry Smith    self.compilers.headerPrefix = self.headerPrefix
90f8833479SBarry Smith    self.types.headerPrefix     = self.headerPrefix
91f8833479SBarry Smith    self.headers.headerPrefix   = self.headerPrefix
92f8833479SBarry Smith    self.functions.headerPrefix = self.headerPrefix
93f8833479SBarry Smith    self.libraries.headerPrefix = self.headerPrefix
94f8833479SBarry Smith    self.blaslapack.headerPrefix = self.headerPrefix
95f8833479SBarry Smith    self.mpi.headerPrefix        = self.headerPrefix
96f8833479SBarry Smith    headersC = map(lambda name: name+'.h', ['dos', 'endian', 'fcntl', 'float', 'io', 'limits', 'malloc', 'pwd', 'search', 'strings',
97f8833479SBarry Smith                                            'unistd', 'machine/endian', 'sys/param', 'sys/procfs', 'sys/resource',
98b4bb646cSSatish Balay                                            'sys/systeminfo', 'sys/times', 'sys/utsname','string', 'stdlib','memory',
99f8833479SBarry Smith                                            'sys/socket','sys/wait','netinet/in','netdb','Direct','time','Ws2tcpip','sys/types',
100753ebd1dSJed Brown                                            'WindowsX', 'cxxabi','float','ieeefp','xmmintrin','stdint'])
101f8833479SBarry Smith    functions = ['access', '_access', 'clock', 'drand48', 'getcwd', '_getcwd', 'getdomainname', 'gethostname', 'getpwuid',
102f8833479SBarry Smith                 'gettimeofday', 'getwd', 'memalign', 'memmove', 'mkstemp', 'popen', 'PXFGETARG', 'rand', 'getpagesize',
103a6d0e24fSJed Brown                 'readlink', 'realpath',  'sigaction', 'signal', 'sigset', 'nanosleep', 'usleep', 'sleep', '_sleep', 'socket',
104a6d0e24fSJed Brown                 'times', 'gethostbyname', 'uname','snprintf','_snprintf','_fullpath','lseek','_lseek','time','fork','stricmp',
105a6d0e24fSJed Brown                 'strcasecmp', 'bzero', 'dlopen', 'dlsym', 'dlclose', 'dlerror',
10692157383SBarry Smith                 '_intel_fast_memcpy','_intel_fast_memset','_finite','_isnan']
107f8833479SBarry Smith    libraries1 = [(['socket', 'nsl'], 'socket'), (['fpe'], 'handle_sigfpes')]
108f8833479SBarry Smith    self.headers.headers.extend(headersC)
109f8833479SBarry Smith    self.functions.functions.extend(functions)
110f8833479SBarry Smith    self.libraries.libraries.extend(libraries1)
111f8833479SBarry Smith    return
112f8833479SBarry Smith
113f8833479SBarry Smith  def defineAutoconfMacros(self):
114f8833479SBarry Smith    self.hostMacro = 'dnl Version: 2.13\ndnl Variable: host_cpu\ndnl Variable: host_vendor\ndnl Variable: host_os\nAC_CANONICAL_HOST'
115f8833479SBarry Smith    return
116f8833479SBarry Smith
117f8833479SBarry Smith  def Dump(self):
118f8833479SBarry Smith    ''' Actually put the values into the configuration files '''
119f8833479SBarry Smith    # eventually everything between -- should be gone
120f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
121f8833479SBarry Smith
122f8833479SBarry Smith    # Sometimes we need C compiler, even if built with C++
123f8833479SBarry Smith    self.setCompilers.pushLanguage('C')
124f8833479SBarry Smith    self.addMakeMacro('CC_FLAGS',self.setCompilers.getCompilerFlags())
125f8833479SBarry Smith    self.setCompilers.popLanguage()
126f8833479SBarry Smith
127f8833479SBarry Smith    # C preprocessor values
128a2b1cf82SBarry Smith    self.addMakeMacro('CPP_FLAGS',self.setCompilers.CPPFLAGS+self.CHUD.CPPFLAGS)
129f8833479SBarry Smith
130f8833479SBarry Smith    # compiler values
131f8833479SBarry Smith    self.setCompilers.pushLanguage(self.languages.clanguage)
132f8833479SBarry Smith    self.addMakeMacro('PCC',self.setCompilers.getCompiler())
133f8833479SBarry Smith    self.addMakeMacro('PCC_FLAGS',self.setCompilers.getCompilerFlags())
134f8833479SBarry Smith    self.setCompilers.popLanguage()
135f8833479SBarry Smith    # .o or .obj
136f8833479SBarry Smith    self.addMakeMacro('CC_SUFFIX','o')
137f8833479SBarry Smith
138f8833479SBarry Smith    # executable linker values
139f8833479SBarry Smith    self.setCompilers.pushLanguage(self.languages.clanguage)
140f8833479SBarry Smith    pcc_linker = self.setCompilers.getLinker()
141f8833479SBarry Smith    self.addMakeMacro('PCC_LINKER',pcc_linker)
142c84a332bSSatish Balay    self.addMakeMacro('PCC_LINKER_FLAGS',self.setCompilers.getLinkerFlags())
143f8833479SBarry Smith    self.setCompilers.popLanguage()
144f8833479SBarry Smith    # '' for Unix, .exe for Windows
145f8833479SBarry Smith    self.addMakeMacro('CC_LINKER_SUFFIX','')
146a2b1cf82SBarry Smith    self.addMakeMacro('PCC_LINKER_LIBS',self.libraries.toStringNoDupes(self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' '))+self.CHUD.LIBS)
147f8833479SBarry Smith
148f8833479SBarry Smith    if hasattr(self.compilers, 'FC'):
149f8833479SBarry Smith      self.setCompilers.pushLanguage('FC')
150f8833479SBarry Smith      # need FPPFLAGS in config/setCompilers
151f8833479SBarry Smith      self.addDefine('HAVE_FORTRAN','1')
152f8833479SBarry Smith      self.addMakeMacro('FPP_FLAGS',self.setCompilers.CPPFLAGS)
153f8833479SBarry Smith
154f8833479SBarry Smith      # compiler values
155f8833479SBarry Smith      self.addMakeMacro('FC_FLAGS',self.setCompilers.getCompilerFlags())
156f8833479SBarry Smith      self.setCompilers.popLanguage()
157f8833479SBarry Smith      # .o or .obj
158f8833479SBarry Smith      self.addMakeMacro('FC_SUFFIX','o')
159f8833479SBarry Smith
160f8833479SBarry Smith      # executable linker values
161f8833479SBarry Smith      self.setCompilers.pushLanguage('FC')
162f8833479SBarry Smith      # Cannot have NAG f90 as the linker - so use pcc_linker as fc_linker
163f8833479SBarry Smith      fc_linker = self.setCompilers.getLinker()
164f8833479SBarry Smith      if config.setCompilers.Configure.isNAG(fc_linker):
165f8833479SBarry Smith        self.addMakeMacro('FC_LINKER',pcc_linker)
166f8833479SBarry Smith      else:
167f8833479SBarry Smith        self.addMakeMacro('FC_LINKER',fc_linker)
1686d53d35eSSatish Balay      self.addMakeMacro('FC_LINKER_FLAGS',self.setCompilers.getLinkerFlags())
1693feacd00SBarry Smith      # apple requires this shared library linker flag on SOME versions of the os
1703feacd00SBarry Smith      if self.setCompilers.getLinkerFlags().find('-Wl,-commons,use_dylibs') > -1:
1713feacd00SBarry Smith        self.addMakeMacro('DARWIN_COMMONS_USE_DYLIBS',' -Wl,-commons,use_dylibs ')
172bb82cf9cSSatish Balay      self.setCompilers.popLanguage()
1735d631499SMatthew Knepley
1745d631499SMatthew Knepley      # F90 Modules
1755d631499SMatthew Knepley      if self.setCompilers.fortranModuleIncludeFlag:
1765d631499SMatthew Knepley        self.addMakeMacro('FC_MODULE_FLAG', self.setCompilers.fortranModuleIncludeFlag)
177a324c51cSMatthew G Knepley      if self.setCompilers.fortranModuleIncludeFlag:
178a324c51cSMatthew G Knepley        self.addMakeMacro('FC_MODULE_OUTPUT_FLAG', self.setCompilers.fortranModuleOutputFlag)
179f8833479SBarry Smith    else:
180f8833479SBarry Smith      self.addMakeMacro('FC','')
181f8833479SBarry Smith
182f8833479SBarry Smith    # shared library linker values
183f8833479SBarry Smith    self.setCompilers.pushLanguage(self.languages.clanguage)
184f8833479SBarry Smith    # need to fix BuildSystem to collect these separately
185f8833479SBarry Smith    self.addMakeMacro('SL_LINKER',self.setCompilers.getLinker())
18670db8aa6SSatish Balay    self.addMakeMacro('SL_LINKER_FLAGS','${PCC_LINKER_FLAGS}')
187f8833479SBarry Smith    self.setCompilers.popLanguage()
188f8833479SBarry Smith    # One of 'a', 'so', 'lib', 'dll', 'dylib' (perhaps others also?) depending on the library generator and architecture
189f8833479SBarry Smith    # Note: . is not included in this macro, consistent with AR_LIB_SUFFIX
190f8833479SBarry Smith    if self.setCompilers.sharedLibraryExt == self.setCompilers.AR_LIB_SUFFIX:
191f8833479SBarry Smith      self.addMakeMacro('SL_LINKER_SUFFIX', '')
19246bc77b6SBarry Smith      self.addDefine('SLSUFFIX','""')
193f8833479SBarry Smith    else:
194f8833479SBarry Smith      self.addMakeMacro('SL_LINKER_SUFFIX', self.setCompilers.sharedLibraryExt)
19546bc77b6SBarry Smith      self.addDefine('SLSUFFIX','"'+self.setCompilers.sharedLibraryExt+'"')
196bb82cf9cSSatish Balay
197bb82cf9cSSatish Balay    #SL_LINKER_LIBS is currently same as PCC_LINKER_LIBS - so simplify
198bb82cf9cSSatish Balay    self.addMakeMacro('SL_LINKER_LIBS','${PCC_LINKER_LIBS}')
199bb82cf9cSSatish Balay    #self.addMakeMacro('SL_LINKER_LIBS',self.libraries.toStringNoDupes(self.compilers.flibs+self.compilers.cxxlibs+self.compilers.LIBS.split(' ')))
200bb82cf9cSSatish Balay
201f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
202f8833479SBarry Smith
203f8833479SBarry Smith    # CONLY or CPP. We should change the PETSc makefiles to do this better
204f8833479SBarry Smith    if self.languages.clanguage == 'C': lang = 'CONLY'
205f8833479SBarry Smith    else: lang = 'CXXONLY'
206f8833479SBarry Smith    self.addMakeMacro('PETSC_LANGUAGE',lang)
207f8833479SBarry Smith
208f8833479SBarry Smith    # real or complex
209f8833479SBarry Smith    self.addMakeMacro('PETSC_SCALAR',self.scalartypes.scalartype)
210f8833479SBarry Smith    # double or float
211f8833479SBarry Smith    self.addMakeMacro('PETSC_PRECISION',self.scalartypes.precision)
212f8833479SBarry Smith
213f8833479SBarry Smith    if self.framework.argDB['with-batch']:
214f8833479SBarry Smith      self.addMakeMacro('PETSC_WITH_BATCH','1')
215f8833479SBarry Smith
216f8833479SBarry Smith    # Test for compiler-specific macros that need to be defined.
217f8833479SBarry Smith    if self.setCompilers.isCray('CC'):
218f8833479SBarry Smith      self.addDefine('HAVE_CRAYC','1')
219f8833479SBarry Smith
220f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
221f8833479SBarry Smith    if self.functions.haveFunction('gethostbyname') and self.functions.haveFunction('socket'):
222f8833479SBarry Smith      self.addDefine('USE_SOCKET_VIEWER','1')
223f8833479SBarry Smith
224f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
225f8833479SBarry Smith    # print include and lib for external packages
226f8833479SBarry Smith    self.framework.packages.reverse()
227996b3231SBarry Smith    includes = []
228996b3231SBarry Smith    libs = []
229f8833479SBarry Smith    for i in self.framework.packages:
230f8833479SBarry Smith      self.addDefine('HAVE_'+i.PACKAGE, 1)
231f8833479SBarry Smith      if not isinstance(i.lib, list):
232f8833479SBarry Smith        i.lib = [i.lib]
233ac9e4c42SSatish Balay      libs.extend(i.lib)
23434cdeb2aSSatish Balay      self.addMakeMacro(i.PACKAGE+'_LIB', self.libraries.toStringNoDupes(i.lib))
235f8833479SBarry Smith      if hasattr(i,'include'):
236f8833479SBarry Smith        if not isinstance(i.include,list):
237f8833479SBarry Smith          i.include = [i.include]
238ac9e4c42SSatish Balay        includes.extend(i.include)
23934cdeb2aSSatish Balay        self.addMakeMacro(i.PACKAGE+'_INCLUDE',self.headers.toStringNoDupes(i.include))
240ac9e4c42SSatish Balay    self.addMakeMacro('PACKAGES_LIBS',self.libraries.toStringNoDupes(libs+self.libraries.math))
241ac9e4c42SSatish Balay    self.addMakeMacro('PACKAGES_INCLUDES',self.headers.toStringNoDupes(includes))
242cbd5cc15SBarry Smith    if hasattr(self.compilers, 'FC'):
243208c3fd5SBarry Smith      if self.compilers.fortranIsF90:
24408784ffaSBarry Smith        self.addMakeMacro('PACKAGES_MODULES_INCLUDES',self.headers.toStringModulesNoDupes(includes))
245f8833479SBarry Smith
246f8833479SBarry Smith    self.addMakeMacro('INSTALL_DIR',self.installdir)
247c6f99f23SBarry Smith    self.addDefine('LIB_DIR','"'+os.path.join(self.installdir,'lib')+'"')
248f8833479SBarry Smith
2490f3b21c2SBarry Smith    if self.framework.argDB['with-single-library']:
2500f3b21c2SBarry Smith      # overrides the values set in conf/variables
2510f3b21c2SBarry Smith      self.addMakeMacro('LIBNAME','${INSTALL_LIB_DIR}/libpetsc.${AR_LIB_SUFFIX}')
2520f3b21c2SBarry Smith      self.addMakeMacro('PETSC_SYS_LIB_BASIC','-lpetsc')
2530f3b21c2SBarry Smith      self.addMakeMacro('PETSC_VEC_LIB_BASIC','-lpetsc')
2540f3b21c2SBarry Smith      self.addMakeMacro('PETSC_MAT_LIB_BASIC','-lpetsc')
2550f3b21c2SBarry Smith      self.addMakeMacro('PETSC_DM_LIB_BASIC','-lpetsc')
2560f3b21c2SBarry Smith      self.addMakeMacro('PETSC_KSP_LIB_BASIC','-lpetsc')
2570f3b21c2SBarry Smith      self.addMakeMacro('PETSC_SNES_LIB_BASIC','-lpetsc')
2580f3b21c2SBarry Smith      self.addMakeMacro('PETSC_TS_LIB_BASIC','-lpetsc')
2590f3b21c2SBarry Smith      self.addMakeMacro('PETSC_LIB_BASIC','-lpetsc')
2600f3b21c2SBarry Smith      self.addMakeMacro('PETSC_CONTRIB_BASIC','-lpetsc')
26157cb31baSSatish Balay      self.addMakeMacro('SHLIBS','libpetsc')
262bb84e0fdSBarry Smith      self.addDefine('USE_SINGLE_LIBRARY', '1')
2630f3b21c2SBarry Smith
264f8833479SBarry Smith    if not os.path.exists(os.path.join(self.petscdir.dir,self.arch.arch,'lib')):
265f8833479SBarry Smith      os.makedirs(os.path.join(self.petscdir.dir,self.arch.arch,'lib'))
266f8833479SBarry Smith
267f8833479SBarry Smith    # add a makefile entry for configure options
268f8833479SBarry Smith    self.addMakeMacro('CONFIGURE_OPTIONS', self.framework.getOptionsString(['configModules', 'optionsModule']).replace('\"','\\"'))
269f8833479SBarry Smith    return
270f8833479SBarry Smith
271f8833479SBarry Smith  def dumpConfigInfo(self):
272f8833479SBarry Smith    import time
273f8833479SBarry Smith    fd = file(os.path.join(self.arch.arch,'include','petscconfiginfo.h'),'w')
274f8833479SBarry Smith    fd.write('static const char *petscconfigureruntime = "'+time.ctime(time.time())+'";\n')
275f8833479SBarry Smith    fd.write('static const char *petscconfigureoptions = "'+self.framework.getOptionsString(['configModules', 'optionsModule']).replace('\"','\\"')+'";\n')
276f8833479SBarry Smith    fd.close()
277f8833479SBarry Smith    return
278f8833479SBarry Smith
279b2843cf1SBarry Smith
280b2843cf1SBarry Smith  def configurePrefetch(self):
281b2843cf1SBarry Smith    '''Sees if there are any prefetch functions supported'''
282ec284106SBarry Smith    self.pushLanguage(self.languages.clanguage)
2830d79d417SBarry Smith    if self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch(v,(enum _mm_hint)0);\n'):
28447980e60SBarry Smith      self.addDefine('HAVE_XMMINTRIN_H', 1)
2856835f947SJed Brown      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const void*)(a),(enum _mm_hint)c)')
2861d9cd9bfSBarry Smith    elif self.checkLink('#include <xmmintrin.h>', 'void *v = 0;_mm_prefetch((const char*)v,(enum _mm_hint)0);\n'):
2871d9cd9bfSBarry Smith      self.addDefine('HAVE_XMMINTRIN_H', 1)
2881d9cd9bfSBarry Smith      self.addDefine('Prefetch(a,b,c)', '_mm_prefetch((const char*)(a),(enum _mm_hint)c)')
2891c3996cdSBarry Smith    elif self.checkLink('', 'void *v = 0;__builtin_prefetch(v,0,0);\n'):
290b2843cf1SBarry Smith      self.addDefine('Prefetch(a,b,c)', '__builtin_prefetch(a,b,c)')
291b2843cf1SBarry Smith    else:
292b2843cf1SBarry Smith      self.addDefine('Prefetch(a,b,c)', ' ')
2937d490b44SBarry Smith    self.popLanguage()
294b2843cf1SBarry Smith
2959800092aSJed Brown  def configureExpect(self):
2969800092aSJed Brown    '''Sees if the __builtin_expect directive is supported'''
2979800092aSJed Brown    self.pushLanguage(self.languages.clanguage)
2989800092aSJed Brown    if self.checkLink('', 'if (__builtin_expect(0,1)) return 1;'):
2999800092aSJed Brown      self.addDefine('HAVE_BUILTIN_EXPECT', 1)
3009800092aSJed Brown    self.popLanguage()
3019800092aSJed Brown
302753ebd1dSJed Brown  def configureIntptrt(self):
303753ebd1dSJed Brown    '''Determine what to use for uintptr_t'''
304753ebd1dSJed Brown    def staticAssertSizeMatchesVoidStar(inc,typename):
305753ebd1dSJed Brown      # The declaration is an error if either array size is negative.
306753ebd1dSJed 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
307753ebd1dSJed Brown      return self.checkCompile(inc, '#define SZ (sizeof(void*)-sizeof(%s))\nint type_is_too_large[SZ],type_is_too_small[-SZ];'%typename)
308753ebd1dSJed Brown    self.pushLanguage(self.languages.clanguage)
309753ebd1dSJed Brown    if self.checkCompile('#include <stdint.h>', 'int x; uintptr_t i = (uintptr_t)&x;'):
310753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'uintptr_t')
311753ebd1dSJed Brown    elif staticAssertSizeMatchesVoidStar('','unsigned long long'):
312753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'unsigned long long')
313753ebd1dSJed Brown    elif staticAssertSizeMatchesVoidStar('#include <stdlib.h>','size_t') or staticAssertSizeMatchesVoidStar('#include <string.h>', 'size_t'):
314753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'size_t')
3152d1b7972SSatish Balay    elif staticAssertSizeMatchesVoidStar('','unsigned'):
316753ebd1dSJed Brown      self.addDefine('UINTPTR_T', 'unsigned')
317753ebd1dSJed Brown    self.popLanguage()
318753ebd1dSJed Brown
319f8833479SBarry Smith  def configureInline(self):
320f8833479SBarry Smith    '''Get a generic inline keyword, depending on the language'''
321f8833479SBarry Smith    if self.languages.clanguage == 'C':
322f8833479SBarry Smith      self.addDefine('STATIC_INLINE', self.compilers.cStaticInlineKeyword)
323f8833479SBarry Smith      self.addDefine('RESTRICT', self.compilers.cRestrict)
324f8833479SBarry Smith    elif self.languages.clanguage == 'Cxx':
325f8833479SBarry Smith      self.addDefine('STATIC_INLINE', self.compilers.cxxStaticInlineKeyword)
326f8833479SBarry Smith      self.addDefine('RESTRICT', self.compilers.cxxRestrict)
327f8833479SBarry Smith    return
328f8833479SBarry Smith
329f8833479SBarry Smith  def configureSolaris(self):
330f8833479SBarry Smith    '''Solaris specific stuff'''
331f8833479SBarry Smith    if self.arch.hostOsBase.startswith('solaris'):
332f8833479SBarry Smith      if os.path.isdir(os.path.join('/usr','ucblib')):
333f8833479SBarry Smith        try:
334f8833479SBarry Smith          flag = getattr(self.setCompilers, self.language[-1]+'SharedLinkerFlag')
335f8833479SBarry Smith        except AttributeError:
336f8833479SBarry Smith          flag = None
337f8833479SBarry Smith        if flag is None:
338f8833479SBarry Smith          self.compilers.LIBS += ' -L/usr/ucblib'
339f8833479SBarry Smith        else:
340f8833479SBarry Smith          self.compilers.LIBS += ' '+flag+'/usr/ucblib'
341f8833479SBarry Smith    return
342f8833479SBarry Smith
343f8833479SBarry Smith  def configureLinux(self):
344f8833479SBarry Smith    '''Linux specific stuff'''
345f8833479SBarry Smith    if self.arch.hostOsBase == 'linux':
346f8833479SBarry Smith      self.addDefine('HAVE_DOUBLE_ALIGN_MALLOC', 1)
347f8833479SBarry Smith    return
348f8833479SBarry Smith
349f8833479SBarry Smith  def configureWin32(self):
350f8833479SBarry Smith    '''Win32 non-cygwin specific stuff'''
351f8833479SBarry Smith    kernel32=0
352f8833479SBarry Smith    if self.libraries.add('Kernel32.lib','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
353f8833479SBarry Smith      self.addDefine('HAVE_WINDOWS_H',1)
354f8833479SBarry Smith      self.addDefine('HAVE_GETCOMPUTERNAME',1)
355f8833479SBarry Smith      kernel32=1
356f8833479SBarry Smith    elif self.libraries.add('kernel32','GetComputerName',prototype='#include <Windows.h>', call='GetComputerName(NULL,NULL);'):
357f8833479SBarry Smith      self.addDefine('HAVE_WINDOWS_H',1)
358f8833479SBarry Smith      self.addDefine('HAVE_GETCOMPUTERNAME',1)
359f8833479SBarry Smith      kernel32=1
360f8833479SBarry Smith    if kernel32:
361eed94e11SSatish Balay      if self.framework.argDB['with-windows-graphics']:
362eed94e11SSatish Balay        self.addDefine('USE_WINDOWS_GRAPHICS',1)
363f8833479SBarry Smith      if self.checkLink('#include <Windows.h>','LoadLibrary(0)'):
364f8833479SBarry Smith        self.addDefine('HAVE_LOADLIBRARY',1)
365b50f6d9eSLisandro Dalcin      if self.checkLink('#include <Windows.h>','GetProcAddress(0,0)'):
366b50f6d9eSLisandro Dalcin        self.addDefine('HAVE_GETPROCADDRESS',1)
367b50f6d9eSLisandro Dalcin      if self.checkLink('#include <Windows.h>','FreeLibrary(0)'):
368b50f6d9eSLisandro Dalcin        self.addDefine('HAVE_FREELIBRARY',1)
369a21658a3SLisandro Dalcin      if self.checkLink('#include <Windows.h>','GetLastError()'):
370a21658a3SLisandro Dalcin        self.addDefine('HAVE_GETLASTERROR',1)
371a21658a3SLisandro Dalcin      if self.checkLink('#include <Windows.h>','SetLastError(0)'):
372a21658a3SLisandro Dalcin        self.addDefine('HAVE_SETLASTERROR',1)
373f8833479SBarry Smith      if self.checkLink('#include <Windows.h>\n','QueryPerformanceCounter(0);\n'):
374f8833479SBarry Smith        self.addDefine('USE_NT_TIME',1)
375f8833479SBarry Smith    if self.libraries.add('Advapi32.lib','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
376f8833479SBarry Smith      self.addDefine('HAVE_GET_USER_NAME',1)
377f8833479SBarry Smith    elif self.libraries.add('advapi32','GetUserName',prototype='#include <Windows.h>', call='GetUserName(NULL,NULL);'):
378f8833479SBarry Smith      self.addDefine('HAVE_GET_USER_NAME',1)
379f8833479SBarry Smith
380f8833479SBarry Smith    if not self.libraries.add('User32.lib','GetDC',prototype='#include <Windows.h>',call='GetDC(0);'):
381f8833479SBarry Smith      self.libraries.add('user32','GetDC',prototype='#include <Windows.h>',call='GetDC(0);')
382f8833479SBarry Smith    if not self.libraries.add('Gdi32.lib','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);'):
383f8833479SBarry Smith      self.libraries.add('gdi32','CreateCompatibleDC',prototype='#include <Windows.h>',call='CreateCompatibleDC(0);')
384f8833479SBarry Smith
385f8833479SBarry Smith    self.types.check('int32_t', 'int')
386f8833479SBarry Smith    if not self.checkCompile('#include <sys/types.h>\n','uid_t u;\n'):
387f8833479SBarry Smith      self.addTypedef('int', 'uid_t')
388f8833479SBarry Smith      self.addTypedef('int', 'gid_t')
389f8833479SBarry Smith    if not self.checkLink('#if defined(PETSC_HAVE_UNISTD_H)\n#include <unistd.h>\n#endif\n','int a=R_OK;\n'):
390f8833479SBarry Smith      self.framework.addDefine('R_OK', '04')
391f8833479SBarry Smith      self.framework.addDefine('W_OK', '02')
392f8833479SBarry Smith      self.framework.addDefine('X_OK', '01')
393f8833479SBarry Smith    if not self.checkLink('#include <sys/stat.h>\n','int a=0;\nif (S_ISDIR(a)){}\n'):
394f8833479SBarry Smith      self.framework.addDefine('S_ISREG(a)', '(((a)&_S_IFMT) == _S_IFREG)')
395f8833479SBarry Smith      self.framework.addDefine('S_ISDIR(a)', '(((a)&_S_IFMT) == _S_IFDIR)')
396f8833479SBarry Smith    if self.checkCompile('#include <Windows.h>\n','LARGE_INTEGER a;\nDWORD b=a.u.HighPart;\n'):
397f8833479SBarry Smith      self.addDefine('HAVE_LARGE_INTEGER_U',1)
398f8833479SBarry Smith
399f8833479SBarry Smith    # Windows requires a Binary file creation flag when creating/opening binary files.  Is a better test in order?
400f8833479SBarry Smith    if self.checkCompile('#include <Windows.h>\n',''):
401f8833479SBarry Smith      self.addDefine('HAVE_O_BINARY',1)
402f8833479SBarry Smith
403f8833479SBarry Smith    if self.compilers.CC.find('win32fe') >= 0:
404f8833479SBarry Smith      self.addDefine('PATH_SEPARATOR','\';\'')
405f8833479SBarry Smith      self.addDefine('DIR_SEPARATOR','\'\\\\\'')
406f8833479SBarry Smith      self.addDefine('REPLACE_DIR_SEPARATOR','\'/\'')
407f8833479SBarry Smith      self.addDefine('CANNOT_START_DEBUGGER',1)
408f8833479SBarry Smith    else:
409f8833479SBarry Smith      self.addDefine('PATH_SEPARATOR','\':\'')
410f8833479SBarry Smith      self.addDefine('REPLACE_DIR_SEPARATOR','\'\\\\\'')
411f8833479SBarry Smith      self.addDefine('DIR_SEPARATOR','\'/\'')
412f8833479SBarry Smith    return
413f8833479SBarry Smith
414f8833479SBarry Smith#-----------------------------------------------------------------------------------------------------
415569865ddSSatish Balay  def configureDefaultArch(self):
416569865ddSSatish Balay    conffile = os.path.join('conf', 'petscvariables')
417569865ddSSatish Balay    if self.framework.argDB['with-default-arch']:
418569865ddSSatish Balay      fd = file(conffile, 'w')
419569865ddSSatish Balay      fd.write('PETSC_ARCH='+self.arch.arch+'\n')
420da93591fSBarry Smith      fd.write('PETSC_DIR='+self.petscdir.dir+'\n')
421569865ddSSatish Balay      fd.write('include ${PETSC_DIR}/${PETSC_ARCH}/conf/petscvariables\n')
422569865ddSSatish Balay      fd.close()
423569865ddSSatish Balay      self.framework.actions.addArgument('PETSc', 'Build', 'Set default architecture to '+self.arch.arch+' in '+conffile)
424569865ddSSatish Balay    elif os.path.isfile(conffile):
425569865ddSSatish Balay      try:
426569865ddSSatish Balay        os.unlink(conffile)
427569865ddSSatish Balay      except:
428569865ddSSatish Balay        raise RuntimeError('Unable to remove file '+conffile+'. Did a different user create it?')
429569865ddSSatish Balay    return
430569865ddSSatish Balay
431569865ddSSatish Balay#-----------------------------------------------------------------------------------------------------
432f8833479SBarry Smith  def configureScript(self):
433f8833479SBarry Smith    '''Output a script in the conf directory which will reproduce the configuration'''
434f8833479SBarry Smith    import nargs
435495bf1a9SSatish Balay    import sys
436495bf1a9SSatish Balay    scriptName = os.path.join(self.arch.arch,'conf', 'reconfigure-'+self.arch.arch+'.py')
437f8833479SBarry Smith    args = dict([(nargs.Arg.parseArgument(arg)[0], arg) for arg in self.framework.clArgs])
438f8833479SBarry Smith    if 'configModules' in args:
4391063a081SSatish Balay      if nargs.Arg.parseArgument(args['configModules'])[1] == 'PETSc.Configure':
440f8833479SBarry Smith        del args['configModules']
441f8833479SBarry Smith    if 'optionsModule' in args:
442c1486898SMatthew Knepley      if nargs.Arg.parseArgument(args['optionsModule'])[1] == 'PETSc.compilerOptions':
443f8833479SBarry Smith        del args['optionsModule']
444f8833479SBarry Smith    if not 'PETSC_ARCH' in args:
4451063a081SSatish Balay      args['PETSC_ARCH'] = 'PETSC_ARCH='+str(self.arch.arch)
446f8833479SBarry Smith    f = file(scriptName, 'w')
447495bf1a9SSatish Balay    f.write('#!'+sys.executable+'\n')
448f8833479SBarry Smith    f.write('if __name__ == \'__main__\':\n')
449f8833479SBarry Smith    f.write('  import sys\n')
450f8833479SBarry Smith    f.write('  sys.path.insert(0, '+repr(os.path.join(self.petscdir.dir, 'config'))+')\n')
451f8833479SBarry Smith    f.write('  import configure\n')
4521063a081SSatish Balay    # pretty print repr(args.values())
4531063a081SSatish Balay    f.write('  configure_options = [\n')
4541063a081SSatish Balay    for itm in args.values():
4551063a081SSatish Balay      f.write('    \''+str(itm)+'\',\n')
4561063a081SSatish Balay    f.write('  ]\n')
457f8833479SBarry Smith    f.write('  configure.petsc_configure(configure_options)\n')
458f8833479SBarry Smith    f.close()
459f8833479SBarry Smith    try:
460f8833479SBarry Smith      os.chmod(scriptName, 0775)
461f8833479SBarry Smith    except OSError, e:
462f8833479SBarry Smith      self.framework.logPrint('Unable to make reconfigure script executable:\n'+str(e))
463f8833479SBarry Smith    self.framework.actions.addArgument('PETSc', 'File creation', 'Created '+scriptName+' for automatic reconfiguration')
464f8833479SBarry Smith    return
465f8833479SBarry Smith
466f8833479SBarry Smith  def configureInstall(self):
467f8833479SBarry Smith    '''Setup the directories for installation'''
468f8833479SBarry Smith    if self.framework.argDB['prefix']:
469f8833479SBarry Smith      self.installdir = self.framework.argDB['prefix']
47010b521ebSBarry Smith      self.addMakeRule('shared_nomesg_noinstall','')
471824e893fSSatish Balay      self.addMakeRule('shared_install','',['-@echo "Now to install the libraries do:"',\
472824e893fSSatish Balay                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} install"',\
473315b77e6SSatish Balay                                              '-@echo "========================================="'])
474f8833479SBarry Smith    else:
475b559b537SSatish Balay      self.installdir = os.path.join(self.petscdir.dir,self.arch.arch)
47610b521ebSBarry Smith      self.addMakeRule('shared_nomesg_noinstall','shared_nomesg')
477824e893fSSatish Balay      self.addMakeRule('shared_install','',['-@echo "Now to check if the libraries are working do:"',\
478824e893fSSatish Balay                                              '-@echo "make PETSC_DIR=${PETSC_DIR} PETSC_ARCH=${PETSC_ARCH} test"',\
479315b77e6SSatish Balay                                              '-@echo "========================================="'])
480f8833479SBarry Smith      return
481f8833479SBarry Smith
482f8833479SBarry Smith  def configureGCOV(self):
483f8833479SBarry Smith    if self.framework.argDB['with-gcov']:
484f8833479SBarry Smith      self.addDefine('USE_GCOV','1')
485f8833479SBarry Smith    return
486f8833479SBarry Smith
487f8833479SBarry Smith  def configureFortranFlush(self):
488f8833479SBarry Smith    if hasattr(self.compilers, 'FC'):
489f8833479SBarry Smith      for baseName in ['flush','flush_']:
490f8833479SBarry Smith        if self.libraries.check('', baseName, otherLibs = self.compilers.flibs, fortranMangle = 1):
491f8833479SBarry Smith          self.addDefine('HAVE_'+baseName.upper(), 1)
492f8833479SBarry Smith          return
493f8833479SBarry Smith
494f8833479SBarry Smith
495f8833479SBarry Smith  def configure(self):
496f8833479SBarry Smith    if not os.path.samefile(self.petscdir.dir, os.getcwd()):
497f8833479SBarry Smith      raise RuntimeError('Wrong PETSC_DIR option specified: '+str(self.petscdir.dir) + '\n  Configure invoked in: '+os.path.realpath(os.getcwd()))
498*3552d8fbSSatish Balay    if self.framework.argDB['prefix'] and os.path.samefile(self.framework.argDB['prefix'],self.petscdir.dir):
499*3552d8fbSSatish Balay      raise RuntimeError('Incorrect option --prefix='+self.framework.argDB['prefix']+' specified. It cannot be same as PETSC_DIR!')
500f8833479SBarry Smith    self.framework.header          = self.arch.arch+'/include/petscconf.h'
501f8833479SBarry Smith    self.framework.cHeader         = self.arch.arch+'/include/petscfix.h'
5029275508cSBarry Smith    self.framework.makeMacroHeader = self.arch.arch+'/conf/petscvariables'
503f8833479SBarry Smith    self.framework.makeRuleHeader  = self.arch.arch+'/conf/petscrules'
504f8833479SBarry Smith    if self.libraries.math is None:
505f8833479SBarry Smith      raise RuntimeError('PETSc requires a functional math library. Please send configure.log to petsc-maint@mcs.anl.gov.')
506f8833479SBarry Smith    if self.languages.clanguage == 'Cxx' and not hasattr(self.compilers, 'CXX'):
507f8833479SBarry Smith      raise RuntimeError('Cannot set C language to C++ without a functional C++ compiler.')
508f8833479SBarry Smith    self.executeTest(self.configureInline)
509b2843cf1SBarry Smith    self.executeTest(self.configurePrefetch)
5109800092aSJed Brown    self.executeTest(self.configureExpect);
511753ebd1dSJed Brown    self.executeTest(self.configureIntptrt);
512f8833479SBarry Smith    self.executeTest(self.configureSolaris)
513f8833479SBarry Smith    self.executeTest(self.configureLinux)
514f8833479SBarry Smith    self.executeTest(self.configureWin32)
515569865ddSSatish Balay    self.executeTest(self.configureDefaultArch)
516f8833479SBarry Smith    self.executeTest(self.configureScript)
517f8833479SBarry Smith    self.executeTest(self.configureInstall)
518f8833479SBarry Smith    self.executeTest(self.configureGCOV)
519f8833479SBarry Smith    self.executeTest(self.configureFortranFlush)
520f8833479SBarry Smith    # dummy rules, always needed except for remote builds
521f8833479SBarry Smith    self.addMakeRule('remote','')
522f8833479SBarry Smith    self.addMakeRule('remoteclean','')
523f8833479SBarry Smith
524f8833479SBarry Smith    self.Dump()
525f8833479SBarry Smith    self.dumpConfigInfo()
526f8833479SBarry Smith    self.framework.log.write('================================================================================\n')
527f8833479SBarry Smith    self.logClear()
528f8833479SBarry Smith    return
529