def test_update_active_mask(self): vec = BoolVector(False, 10) self.assertTrue(vec.updateActiveMask("1-2,5")) self.assertTrue(vec[1]) self.assertTrue(vec[2]) self.assertTrue(vec[5]) self.assertFalse(vec[4]) vec = BoolVector(False, 10) self.assertTrue(vec.updateActiveMask("1-5,2,3")) 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(vec.updateActiveMask("5,6,7,15")) 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): 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)