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