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''' 31770b58fbSMatthew G. Knepley homeloc = os.path.join(os.getenv('HOME', '.'), 'datafiles') 32f1bfc634SSatish Balay parentloc = os.path.join(self.petscdir.dir,'..','datafiles') 339d310bb7SBarry Smith self.datafilespath = None 34f1bfc634SSatish Balay 35*5b6bfdb9SJed Brown if 'DATAFILESPATH' in self.framework.argDB: 36f1bfc634SSatish Balay if os.path.isdir(self.framework.argDB['DATAFILESPATH']) and os.path.isdir(os.path.join(self.framework.argDB['DATAFILESPATH'], 'matrices')): 379d310bb7SBarry Smith self.datafilespath = str(self.framework.argDB['DATAFILESPATH']) 389d310bb7SBarry Smith else: 399d310bb7SBarry Smith raise RuntimeError('Path given with option -DATAFILES='+self.framework.argDB['DATAFILESPATH']+' is not a valid datafiles directory') 40f1bfc634SSatish Balay elif os.path.isdir(homeloc) and os.path.isdir(os.path.join(homeloc,'matrices')): 41f1bfc634SSatish Balay self.datafilespath = homeloc 42f1bfc634SSatish Balay elif os.path.isdir(parentloc) and os.path.isdir(os.path.join(parentloc,'matrices')): 43f1bfc634SSatish Balay self.datafilespath = parentloc 44eb023ebfSBarry Smith elif os.path.isdir(os.path.join(self.petscdir.dir, '..', '..','Datafiles')) & os.path.isdir(os.path.join(self.petscdir.dir, '..','..', 'Datafiles', 'Matrices')): 45eb023ebfSBarry Smith self.datafilespath = os.path.join(self.petscdir.dir, '..','..', 'Datafiles') 469d310bb7SBarry Smith return 479d310bb7SBarry Smith 489d310bb7SBarry Smith def configure(self): 499d310bb7SBarry Smith self.executeTest(self.getDatafilespath) 509d310bb7SBarry Smith return 51