xref: /petsc/config/PETSc/options/installDir.py (revision e5a36eccef3d6b83a2c625c30d0dfd5adb4001f2) 
1#!/usr/bin/env python
2from __future__ import generators
3import config.base
4import os
5
6class Configure(config.base.Configure):
7  def __init__(self, framework):
8    config.base.Configure.__init__(self, framework)
9    self.headerPrefix = ''
10    self.substPrefix  = ''
11    self.dir = ''
12    return
13
14  def __str__(self):
15    return ''
16
17  def setupHelp(self, help):
18    import nargs
19    help.addArgument('PETSc', '-with-clean=<bool>',         nargs.ArgBool(None, 0, 'Delete prior build files including externalpackages'))
20    return
21
22  def setupDependencies(self, framework):
23    config.base.Configure.setupDependencies(self, framework)
24    self.petscdir = framework.require('PETSc.options.petscdir', self)
25    self.arch     = framework.require('PETSc.options.arch', self)
26    return
27
28  def printSudoPasswordMessage(self,needsudo = 1):
29    '''Prints a message that sudo password will be needed for installs of packages'''
30    '''Packages like sowing and make that are never installed in an sudo location would pass 0 for needsudo'''
31    if needsudo and self.installSudoMessage:
32      self.logPrintBox(self.installSudoMessage)
33      self.installSudoMessage = ''
34
35  def setInstallDir(self):
36    ''' setup installDir to either prefix or if that is not set to PETSC_DIR/PETSC_ARCH'''
37    self.installSudo        = ''
38    self.installSudoMessage = ''
39    if self.framework.argDB['prefix']:
40      self.dir = os.path.abspath(os.path.expanduser(self.framework.argDB['prefix']))
41      self.petscDir = self.dir
42      self.petscArch = ''
43      try:
44        os.makedirs(os.path.join(self.dir,'PETScTestDirectory'))
45        os.rmdir(os.path.join(self.dir,'PETScTestDirectory'))
46      except:
47        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'
48        self.installSudo = 'sudo '
49    else:
50      self.dir = os.path.abspath(os.path.join(self.petscdir.dir, self.arch.arch))
51      self.petscDir = self.petscdir.dir
52      self.petscArch = self.arch.arch
53    self.addMakeMacro('PREFIXDIR',self.dir)
54    self.confDir = os.path.abspath(os.path.join(self.petscdir.dir, self.arch.arch))
55
56  def configureInstallDir(self):
57    '''Makes  installDir subdirectories if it does not exist for both prefix install location and PETSc work install location'''
58    dir = os.path.abspath(os.path.join(self.petscdir.dir, self.arch.arch))
59    if not os.path.exists(dir):
60      os.makedirs(dir)
61    for i in ['include','lib','bin',os.path.join('lib','petsc','conf')]:
62      newdir = os.path.join(dir,i)
63      if not os.path.exists(newdir):
64        os.makedirs(newdir)
65    if os.path.isfile(self.framework.argDB.saveFilename):
66      os.remove(self.framework.argDB.saveFilename)
67    confdir = os.path.join(dir,'lib','petsc','conf')
68    self.framework.argDB.saveFilename = os.path.abspath(os.path.join(confdir, 'RDict.db'))
69    self.logPrint('Changed persistence directory to '+confdir)
70    return
71
72  def cleanConfDir(self):
73    import shutil
74    if self.framework.argDB['with-clean'] and os.path.isdir(self.confDir):
75      self.logPrintBox('Warning: "with-clean" is specified. Removing all build files from '+ self.confDir)
76      shutil.rmtree(self.confDir)
77    return
78
79  def saveReconfigure(self):
80    self.reconfigure_file = os.path.join(self.dir,'lib','petsc','conf','reconfigure-'+self.arch.arch+'.py')
81    self.save_reconfigure_file = None
82    if self.framework.argDB['with-clean'] and os.path.exists(self.reconfigure_file):
83      self.save_reconfigure_file = '.save.reconfigure-'+self.arch.arch+'.py'
84      try:
85        if os.path.exists(self.save_reconfigure_file): os.unlink(self.save_reconfigure_file)
86        os.rename(self.reconfigure_file,self.save_reconfigure_file)
87      except Exception as e:
88        self.save_reconfigure_file = None
89        self.logPrint('error in saveReconfigure(): '+ str(e))
90    return
91
92  def restoreReconfigure(self):
93    if self.framework.argDB['with-clean'] and self.save_reconfigure_file:
94      try:
95        os.rename(self.save_reconfigure_file,self.reconfigure_file)
96      except Exception as e:
97        self.logPrint('error in restoreReconfigure(): '+ str(e))
98    return
99
100  def configure(self):
101    self.executeTest(self.setInstallDir)
102    self.executeTest(self.saveReconfigure)
103    self.executeTest(self.cleanConfDir)
104    self.executeTest(self.configureInstallDir)
105    self.executeTest(self.restoreReconfigure)
106    return
107