def test_update_active_mask(self):
vec = BoolVector(False, 10)
self.assertTrue(BoolVector.updateActiveMask("1-2,5", vec))
self.assertTrue(vec[1])
self.assertTrue(vec[2])
self.assertTrue(vec[5])
self.assertFalse(vec[4])
vec = BoolVector(False, 10)
self.assertTrue(BoolVector.updateActiveMask("1-5,2,3", vec))
self.assertTrue(vec[1])
self.assertTrue(vec[2])
self.assertTrue(vec[3])
self.assertTrue(vec[4])
self.assertTrue(vec[5])
self.assertFalse(vec[0])
self.assertFalse(vec[6])
vec = BoolVector(False, 10)
self.assertTrue(BoolVector.updateActiveMask("5,6,7,15", vec))
self.assertTrue(vec[5])
self.assertTrue(vec[6])
self.assertTrue(vec[7])
self.assertFalse(vec[4])
self.assertFalse(vec[8])
self.assertEqual(len(vec), 16)
def test_update_active_mask(self):
vec = BoolVector(False, 10)
self.assertTrue(BoolVector.updateActiveMask("1-2,5", vec))
self.assertTrue(vec[1])
self.assertTrue(vec[2])
self.assertTrue(vec[5])
self.assertFalse(vec[4])
vec = BoolVector(False, 10)
self.assertTrue(BoolVector.updateActiveMask("1-5,2,3", vec))
self.assertTrue(vec[1])
self.assertTrue(vec[2])
self.assertTrue(vec[3])
self.assertTrue(vec[4])
self.assertTrue(vec[5])
self.assertFalse(vec[0])
self.assertFalse(vec[6])
vec = BoolVector(False, 10)
self.assertTrue(BoolVector.updateActiveMask("5,6,7,15", vec))
self.assertTrue(vec[5])
self.assertTrue(vec[6])
self.assertTrue(vec[7])
self.assertFalse(vec[4])
self.assertFalse(vec[8])
self.assertEqual(len(vec), 16)
def getActiveRealizationsMask(self):
mask = BoolVector(False, self.__realization_count)
mask.updateActiveMask(self.getValue(), mask)
# mask = BoolVector.active_mask(self.getValue())
if mask is None:
raise ValueError("Error while parsing range string!")
if len(mask) > self.__realization_count:
raise ValueError("Mask size changed %d != %d!" % (self.__realization_count, len(mask)))
return mask
def getActiveRealizationsMask(self):
mask = BoolVector(False, self.__realization_count)
mask.updateActiveMask(self.getValue())
# mask = BoolVector.active_mask(self.getValue())
if mask is None:
raise ValueError("Error while parsing range string!")
if len(mask) > self.__realization_count:
raise ValueError("Mask size changed %d != %d!" %
(self.__realization_count, len(mask)))
return mask
def getActiveRealizationsMask(self):
count = getRealizationCount()
mask = BoolVector(default_value=False, initial_size = count)
if not mask.updateActiveMask(self.getValue()):
raise ValueError('Error while parsing range string "%s"!' % self.getValue())
if len(mask) != count:
raise ValueError("Mask size changed %d != %d!" % (count, len(mask)))
return mask
def getActiveRealizationsMask(self):
count = getRealizationCount()
mask = BoolVector(default_value=False, initial_size=count)
if not mask.updateActiveMask(self.getValue()):
raise ValueError('Error while parsing range string "%s"!' %
self.getValue())
if len(mask) != count:
raise ValueError("Mask size changed %d != %d!" %
(count, len(mask)))
return mask
def load(self, args):
arguments = splitArguments(args)
if len(arguments) < 1:
self.lastCommandFailed("Loading requires a realization mask.")
return False
realization_count = self.ert().getEnsembleSize()
mask = BoolVector(False, realization_count)
mask_success = mask.updateActiveMask(arguments[0])
if not mask_success:
self.lastCommandFailed("The realization mask: '%s' is not valid." % arguments[0])
return False
fs = self.ert().getEnkfFsManager().getCurrentFileSystem()
self.ert().loadFromForwardModel(mask, 0, fs)
def load(self, args):
arguments = splitArguments(args)
if len(arguments) < 1:
self.lastCommandFailed("Loading requires a realization mask.")
return False
realization_count = self.ert().getEnsembleSize()
mask = BoolVector(False, realization_count)
mask_success = mask.updateActiveMask(arguments[0])
if not mask_success:
self.lastCommandFailed("The realization mask: '%s' is not valid." %
arguments[0])
return False
fs = self.ert().getEnkfFsManager().getCurrentFileSystem()
self.ert().loadFromForwardModel(mask, 0, fs)
