19d310bb7SBarry Smith#!/usr/bin/env python 29d310bb7SBarry Smithfrom __future__ import generators 39d310bb7SBarry Smithimport user 49d310bb7SBarry Smithimport config.base 59d310bb7SBarry Smithimport os 69d310bb7SBarry Smith 79d310bb7SBarry Smithclass Configure(config.base.Configure): 89d310bb7SBarry Smith def __init__(self, framework): 99d310bb7SBarry Smith config.base.Configure.__init__(self, framework) 109d310bb7SBarry Smith self.headerPrefix = '' 119d310bb7SBarry Smith self.substPrefix = '' 129d310bb7SBarry Smith return 139d310bb7SBarry Smith 149d310bb7SBarry Smith def __str__(self): 159d310bb7SBarry Smith return '' 169d310bb7SBarry Smith 179d310bb7SBarry Smith def setupHelp(self, help): 189d310bb7SBarry Smith import nargs 199d310bb7SBarry Smith help.addArgument('PETSc', '-with-clean=<bool>', nargs.ArgBool(None, 0, 'Delete prior build files including externalpackages')) 209d310bb7SBarry Smith return 219d310bb7SBarry Smith 229d310bb7SBarry Smith def setupDependencies(self, framework): 239d310bb7SBarry Smith config.base.Configure.setupDependencies(self, framework) 249d310bb7SBarry Smith self.arch = framework.require('PETSc.options.arch', self) 259d310bb7SBarry Smith return 269d310bb7SBarry Smith 279d310bb7SBarry Smith def printSudoPasswordMessage(self,needsudo = 1): 289d310bb7SBarry Smith '''Prints a message that sudo password will be needed for installs of packages''' 299d310bb7SBarry Smith '''Packages like sowing and make that are never installed in an sudo location would pass 0 for needsudo''' 309d310bb7SBarry Smith if needsudo and self.installSudoMessage: 319d310bb7SBarry Smith self.logPrintBox(self.installSudoMessage) 329d310bb7SBarry Smith self.installSudoMessage = '' 339d310bb7SBarry Smith 349d310bb7SBarry Smith def setInstallDir(self): 359d310bb7SBarry Smith ''' setup installDir to either prefix or if that is not set to PETSC_DIR/PETSC_ARCH''' 369d310bb7SBarry Smith self.installSudo = '' 379d310bb7SBarry Smith self.installSudoMessage = '' 389d310bb7SBarry Smith if self.framework.argDB['prefix']: 399d310bb7SBarry Smith self.dir = self.framework.argDB['prefix'] 409d310bb7SBarry Smith try: 419d310bb7SBarry Smith os.makedirs(os.path.join(self.dir,'PETScTestDirectory')) 429d310bb7SBarry Smith os.rmdir(os.path.join(self.dir,'PETScTestDirectory')) 439d310bb7SBarry Smith except: 449d310bb7SBarry Smith self.installSudoMessage = 'You do not have write permissions to the --prefix directory '+self.dir+'\nYou will be prompted for the sudo password for any external package installs' 459d310bb7SBarry Smith self.installSudo = 'sudo ' 469d310bb7SBarry Smith else: 479d310bb7SBarry Smith self.dir = os.path.abspath(os.path.join(self.arch.arch)) 489d310bb7SBarry Smith self.confDir = os.path.abspath(os.path.join(self.arch.arch)) 499d310bb7SBarry Smith 509d310bb7SBarry Smith def configureInstallDir(self): 519d310bb7SBarry Smith '''Makes installDir subdirectories if it does not exist for both prefix install location and PETSc work install location''' 529d310bb7SBarry Smith dir = os.path.abspath(os.path.join(self.arch.arch)) 539d310bb7SBarry Smith if not os.path.exists(dir): 549d310bb7SBarry Smith os.makedirs(dir) 55*fe998a80SBarry Smith for i in ['include','lib','bin',os.path.join('lib','petsc-conf')]: 569d310bb7SBarry Smith newdir = os.path.join(dir,i) 579d310bb7SBarry Smith if not os.path.exists(newdir): 589d310bb7SBarry Smith os.makedirs(newdir) 599d310bb7SBarry Smith if os.path.isfile(self.framework.argDB.saveFilename): 609d310bb7SBarry Smith os.remove(self.framework.argDB.saveFilename) 61*fe998a80SBarry Smith confdir = os.path.join(dir,'lib','petsc-conf') 629d310bb7SBarry Smith self.framework.argDB.saveFilename = os.path.abspath(os.path.join(confdir, 'RDict.db')) 639d310bb7SBarry Smith self.framework.logPrint('Changed persistence directory to '+confdir) 649d310bb7SBarry Smith return 659d310bb7SBarry Smith 669d310bb7SBarry Smith def cleanInstallDir(self): 679d310bb7SBarry Smith import shutil 689d310bb7SBarry Smith if self.framework.argDB['with-clean'] and os.path.isdir(self.dir): 699d310bb7SBarry Smith self.logPrintBox('Warning: "with-clean" is specified. Removing all build files from '+ self.dir) 709d310bb7SBarry Smith shutil.rmtree(self.dir) 719d310bb7SBarry Smith return 729d310bb7SBarry Smith 739d310bb7SBarry Smith def saveReconfigure(self): 74*fe998a80SBarry Smith self.reconfigure_file = os.path.join(self.dir,'lib','petsc-conf','reconfigure-'+self.arch.arch+'.py') 759d310bb7SBarry Smith self.save_reconfigure_file = None 769d310bb7SBarry Smith if self.framework.argDB['with-clean'] and os.path.exists(self.reconfigure_file): 779d310bb7SBarry Smith self.save_reconfigure_file = '.save.reconfigure-'+self.arch.arch+'.py' 789d310bb7SBarry Smith try: 799d310bb7SBarry Smith if os.path.exists(self.save_reconfigure_file): os.unlink(self.save_reconfigure_file) 809d310bb7SBarry Smith os.rename(self.reconfigure_file,self.save_reconfigure_file) 819d310bb7SBarry Smith except Exception, e: 829d310bb7SBarry Smith self.save_reconfigure_file = None 839d310bb7SBarry Smith self.framework.logPrint('error in saveReconfigure(): '+ str(e)) 849d310bb7SBarry Smith return 859d310bb7SBarry Smith 869d310bb7SBarry Smith def restoreReconfigure(self): 879d310bb7SBarry Smith if self.framework.argDB['with-clean'] and self.save_reconfigure_file: 889d310bb7SBarry Smith try: 899d310bb7SBarry Smith os.rename(self.save_reconfigure_file,self.reconfigure_file) 909d310bb7SBarry Smith except Exception, e: 919d310bb7SBarry Smith self.framework.logPrint('error in restoreReconfigure(): '+ str(e)) 929d310bb7SBarry Smith return 939d310bb7SBarry Smith 949d310bb7SBarry Smith def configure(self): 959d310bb7SBarry Smith self.executeTest(self.setInstallDir) 969d310bb7SBarry Smith self.executeTest(self.saveReconfigure) 979d310bb7SBarry Smith self.executeTest(self.cleanInstallDir) 989d310bb7SBarry Smith self.executeTest(self.configureInstallDir) 999d310bb7SBarry Smith self.executeTest(self.restoreReconfigure) 1009d310bb7SBarry Smith return 101