1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134
|
"""Unit tests for the CollisionPredicates class"""
# Copyright (C) 2014 Anders Logg and August Johansson
#
# This file is part of DOLFIN.
#
# DOLFIN is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# DOLFIN is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with DOLFIN. If not, see <http://www.gnu.org/licenses/>.
#
# First added: 2014-02-16
# Last changed: 2016-11-22
import pytest
from dolfin import *
from dolfin_utils.test import skip_in_parallel
import numpy as np
@skip_in_parallel
def create_mesh(a, b):
editor = MeshEditor()
mesh = Mesh()
editor.open(mesh, "interval", 1, 2)
editor.init_cells(1)
editor.init_vertices(2)
editor.add_cell(0, np.array( (0, 1), dtype='uint') )
editor.add_vertex(0, np.array( (a.x(), a.y()), dtype='float'))
editor.add_vertex(1, np.array( (b.x(), b.y()), dtype='float'))
editor.close()
return mesh;
@skip_in_parallel
def test_L_version_1():
mesh0 = create_mesh(Point(0., 0.), Point(1., 0.))
mesh1 = create_mesh(Point(0., 0.), Point(0., 1.))
cell0 = Cell(mesh0, 0)
cell1 = Cell(mesh1, 0)
assert cell0.collides(cell1) == True
@skip_in_parallel
def test_L_version_2():
# mesh0 = create_mesh(Point(np.finfo(np.float32).eps, 0.), Point(1., 0.))
# mesh0 = create_mesh(Point(eps(), 0.), Point(1., 0.))
mesh0 = create_mesh(Point(2.23e-15, 0.), Point(1., 0.))
mesh1 = create_mesh(Point(0., 0.), Point(0., 1.))
# print(mesh0.str(True))
# print(mesh1.str(True))
cell0 = Cell(mesh0, 0)
cell1 = Cell(mesh1, 0)
assert cell0.collides(cell1) == False
@skip_in_parallel
def test_L_version_3():
# mesh0 = create_mesh(Point(np.finfo(np.float32).eps, 0.), Point(1., 0.))
# mesh0 = create_mesh(Point(eps(), 0.), Point(1., 0.))
a = Point(2.23e-100, 0.) # assume shewchuk works
b = Point(1., 0.)
c = Point(0., 0.)
d = Point(0., 1.)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(a, b, c, d) == False
@skip_in_parallel
def test_aligned_version_1():
mesh0 = create_mesh(Point(0,0), Point(1,0))
mesh1 = create_mesh(Point(1,0), Point(2,0))
cell0 = Cell(mesh0, 0)
cell1 = Cell(mesh1, 0)
assert cell0.collides(cell1) == True
@skip_in_parallel
def test_aligned_version_2():
mesh0 = create_mesh(Point(0,0), Point(1,0))
mesh1 = create_mesh(Point(2,0), Point(3,0))
cell0 = Cell(mesh0, 0)
cell1 = Cell(mesh1, 0)
assert cell0.collides(cell1) == False
@skip_in_parallel
def test_collinear_1():
p0 = Point(0.05, 0.15)
p1 = Point(0.85, 0.95)
q0 = Point(0.2875, 0.3875)
q1 = Point(0.6125, 0.7125)
meshp = create_mesh(p0, p1)
meshq = create_mesh(q0, q1)
cellp = Cell(meshp, 0)
cellq = Cell(meshq, 0)
assert cellp.collides(cellq) == True
@skip_in_parallel
def test_collinear_2():
res = cpp.geometry.CollisionPredicates.collides_segment_segment_2d(Point(.5, .3),
Point(.5, .4),
Point(.5, .5),
Point(.5, .6))
assert not res
@skip_in_parallel
def test_segment_segment_2d():
# p0 is on segment q0-q1
p0 = Point(1e-30, 0)
p1 = Point(1, 2)
p2 = Point(2, 1)
q0 = Point(1, 0)
q1 = Point(0, 0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p1, q0, q1)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p1, p0, q0, q1)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p1, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p1, p0, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p2, q0, q1)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p2, p0, q0, q1)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p2, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p2, p0, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p1, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p1, p0, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p1, q0, q1)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p1, p0, q0, q1)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p2, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p2, p0, q1, q0)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p0, p2, q0, q1)
assert cpp.geometry.CollisionPredicates.collides_segment_segment_2d(p2, p0, q0, q1)
|