def create_collection(self):
collection = CPolylineCollection()
p1 = CPolyline( name = "POLY1" , init_points = [(0,10) , (1,11) , (2,12)])
p2 = CPolyline( name = "POLY2" , init_points = [(0,100) , (10,110) , (20,120)])
collection.addPolyline( p1 )
collection.addPolyline( p2 )
tail = p1[-1]
self.assertEqual( tail , (2,12))
self.assertEqual(p1.getName() , "POLY1")
tail = p2[-1]
self.assertEqual( tail , (20,120))
self.assertEqual(p2.getName() , "POLY2")
return collection
def test_add_polyline(self):
pc = CPolylineCollection()
pl = pc.createPolyline( name = "TestP" )
self.assertTrue( isinstance(pl , CPolyline))
self.assertEqual(len(pc) , 1)
self.assertTrue( "TestP" in pc )
with self.assertRaises(IndexError):
pl = pc[2]
p0 = pc[0]
self.assertTrue( p0 == pl )
with self.assertRaises(KeyError):
pn = pc["missing"]
pn = pc["TestP"]
self.assertTrue( pn == pl )
px = CPolyline( name = "TestP")
with self.assertRaises(KeyError):
pc.addPolyline( px )
self.assertEqual(len(pc) , 1)
p2 = CPolyline( name = "Poly2")
pc.addPolyline( p2 )
self.assertEqual( len(pc) , 2 )
self.assertTrue( "Poly2" in pc )
l = []
for p in pc:
l.append(p)
self.assertEqual( len(pc) , 2 )
points = [(0,1) , (1,1)]
pc.addPolyline( points , name = "XYZ")
self.assertTrue( "XYZ" in pc )
def getNeighbours(self, polylines=None, connected_only=True):
"""
Will return a list of FaultBlock instances which are in direct
contact with this block.
"""
neighbour_id_list = IntVector()
if polylines is None:
polylines = CPolylineCollection()
self._get_neighbours(connected_only, polylines, neighbour_id_list)
parent_layer = self.getParentLayer()
neighbour_list = []
for id in neighbour_id_list:
neighbour_list.append(parent_layer.getBlock(id))
return neighbour_list
def test_neighbours2(self):
nx = 8
ny = 8
nz = 1
grid = EclGrid.createRectangular( (nx , ny , nz) , (1,1,1) )
layer = FaultBlockLayer( grid , 0 )
with TestAreaContext("python/FaultBlocks/neighbours"):
with open("faultblock.grdecl","w") as fileH:
fileH.write("FAULTBLK \n")
fileH.write("1 1 1 1 2 2 2 2 \n")
fileH.write("1 1 1 1 2 2 2 2 \n")
fileH.write("1 1 1 1 2 2 2 2 \n")
fileH.write("1 1 1 1 2 2 2 2 \n")
fileH.write("3 3 3 3 2 2 2 2 \n")
fileH.write("3 3 3 3 2 2 2 2 \n")
fileH.write("3 3 3 3 2 2 2 2 \n")
fileH.write("3 3 3 3 2 2 2 2 \n")
fileH.write("/\n")
kw = EclKW.read_grdecl(open("faultblock.grdecl") , "FAULTBLK" , ecl_type = EclTypeEnum.ECL_INT_TYPE)
with open("faults.grdecl" , "w") as f:
f.write("FAULTS\n")
f.write("\'FY\' 1 4 4 4 1 1 'Y' /\n")
f.write("\'FX\' 4 4 1 8 1 1 'X' /\n")
f.write("/")
faults = FaultCollection( grid , "faults.grdecl")
layer.loadKeyword( kw )
b1 = layer.getBlock( 1 )
b2 = layer.getBlock( 2 )
b3 = layer.getBlock( 3 )
nb = b1.getNeighbours()
self.assertTrue( b2 in nb )
self.assertTrue( b3 in nb )
polylines1 = CPolylineCollection()
p1 = polylines1.createPolyline(name="P1")
p1.addPoint(4,0)
p1.addPoint(4,4)
p1.addPoint(4,8)
nb = b1.getNeighbours( polylines = polylines1 )
self.assertFalse( b2 in nb )
self.assertTrue( b3 in nb )
polylines2 = CPolylineCollection()
p1 = polylines2.createPolyline(name="P2")
p1.addPoint(0,4)
p1.addPoint(4,4)
nb = b1.getNeighbours( polylines = polylines2 )
self.assertTrue( b2 in nb )
self.assertFalse( b3 in nb )
layer.addFaultBarrier( faults["FY"] )
nb = b1.getNeighbours()
self.assertTrue( b2 in nb )
self.assertFalse( b3 in nb )
layer.addFaultBarrier( faults["FX"] )
nb = b1.getNeighbours()
self.assertEqual( len(nb) , 0 )
def test_construction(self):
pc = CPolylineCollection()
self.assertEqual(len(pc) , 0)