19d310bb7SBarry Smithfrom __future__ import generators 29d310bb7SBarry Smithimport config.base 39d310bb7SBarry Smithimport os 49d310bb7SBarry Smithimport re 59d310bb7SBarry Smith 69d310bb7SBarry Smithclass Configure(config.base.Configure): 79d310bb7SBarry Smith def __init__(self, framework): 89d310bb7SBarry Smith config.base.Configure.__init__(self, framework) 99d310bb7SBarry Smith self.headerPrefix = 'PETSC' 109d310bb7SBarry Smith self.substPrefix = 'PETSC' 119d310bb7SBarry Smith self.updated = 0 129d310bb7SBarry Smith self.strmsg = '' 139d310bb7SBarry Smith self.datafilespath = '' 149d310bb7SBarry Smith return 159d310bb7SBarry Smith 169d310bb7SBarry Smith def __str__(self): 179d310bb7SBarry Smith return self.strmsg 189d310bb7SBarry Smith 199d310bb7SBarry Smith def setupHelp(self, help): 209d310bb7SBarry Smith import nargs 219d310bb7SBarry Smith help.addArgument('PETSc', '-DATAFILESPATH=<dir>', nargs.Arg(None, None, 'Specifiy location of PETSc datafiles, e.g. test matrices')) 229d310bb7SBarry Smith return 239d310bb7SBarry Smith 249d310bb7SBarry Smith def setupDependencies(self, framework): 259d310bb7SBarry Smith config.base.Configure.setupDependencies(self, framework) 269d310bb7SBarry Smith self.petscdir = framework.require('PETSc.options.petscdir', self) 279d310bb7SBarry Smith return 289d310bb7SBarry Smith 299d310bb7SBarry Smith def getDatafilespath(self): 309d310bb7SBarry Smith '''Checks what DATAFILESPATH should be''' 319d310bb7SBarry Smith self.datafilespath = None 329d310bb7SBarry Smith if self.framework.argDB.has_key('DATAFILESPATH'): 339d310bb7SBarry Smith if os.path.isdir(self.framework.argDB['DATAFILESPATH']) & os.path.isdir(os.path.join(self.framework.argDB['DATAFILESPATH'], 'matrices')): 349d310bb7SBarry Smith self.datafilespath = str(self.framework.argDB['DATAFILESPATH']) 359d310bb7SBarry Smith else: 369d310bb7SBarry Smith raise RuntimeError('Path given with option -DATAFILES='+self.framework.argDB['DATAFILESPATH']+' is not a valid datafiles directory') 379d310bb7SBarry Smith elif os.path.isdir(os.path.join('/home','petsc','datafiles')) & os.path.isdir(os.path.join('/home','petsc','datafiles','matrices')): 389d310bb7SBarry Smith self.datafilespath = os.path.join('/home','petsc','datafiles') 399d310bb7SBarry Smith elif os.path.isdir(os.path.join(self.petscdir.dir, '..', 'datafiles')) & os.path.isdir(os.path.join(self.petscdir.dir, '..', 'datafiles', 'matrices')): 409d310bb7SBarry Smith self.datafilespath = os.path.join(self.petscdir.dir, '..', 'datafiles') 41*eb023ebfSBarry Smith elif os.path.isdir(os.path.join(self.petscdir.dir, '..', '..','Datafiles')) & os.path.isdir(os.path.join(self.petscdir.dir, '..','..', 'Datafiles', 'Matrices')): 42*eb023ebfSBarry Smith self.datafilespath = os.path.join(self.petscdir.dir, '..','..', 'Datafiles') 439d310bb7SBarry Smith if self.datafilespath is not None: 449d310bb7SBarry Smith self.addMakeMacro('DATAFILESPATH',self.datafilespath) 459d310bb7SBarry Smith return 469d310bb7SBarry Smith 479d310bb7SBarry Smith def configure(self): 489d310bb7SBarry Smith self.executeTest(self.getDatafilespath) 499d310bb7SBarry Smith return 50