Python BroadSweepAndPrune.query_point Exemples

Exemple #1

Fichier : collision_test.py Projet : Beliaar/bGrease

	def test_query_point_with_mask(self):
		from bGrease.collision import BroadSweepAndPrune, Pair
		world = TestWorld()
		coll = BroadSweepAndPrune()
		coll.set_world(world)
		set_entity = world.collision.set
		
		set_entity(1, 0, 0, 2, 2, into_mask=1)
		set_entity(2, 0, 0, 2, 2, into_mask=2)
		set_entity(3, 0, 0, 2, 2, into_mask=5)
		coll.step(0)
		self.assertEqual(coll.query_point(1, 1), set([1, 2, 3]))
		self.assertEqual(coll.query_point(1, 1, from_mask=1), set([1, 3]))
		self.assertEqual(coll.query_point(1, 1, from_mask=2), set([2]))
		self.assertEqual(coll.query_point(1, 1, from_mask=3), set([1, 2, 3]))
		self.assertEqual(coll.query_point(1, 1, from_mask=4), set([3]))
		self.assertEqual(coll.query_point(1, 1, from_mask=5), set([1, 3]))
		self.assertEqual(coll.query_point(1, 1, from_mask=8), set())

Exemple #2

Fichier : collision_test.py Projet : Beliaar/bGrease

	def test_query_point(self):
		from bGrease.collision import BroadSweepAndPrune, Pair
		world = TestWorld()
		coll = BroadSweepAndPrune()
		coll.set_world(world)
		set_entity = world.collision.set

		set_entity(1, -1, -1, 3, 1)
		set_entity(2, 4, 4, 8, 8)
		set_entity(3, 6, 6, 9, 9)

		# Queries before the first step should always return no hits
		self.assertEqual(coll.query_point(0, 0), set())
		coll.step(0)
		self.assertEqual(coll.query_point(0, 0), set([1]))
		self.assertEqual(coll.query_point([0, 0]), set([1]))

		set_entity(2, 4, 4, 8, 8)
		set_entity(3, 6, 6, 9, 9)
		world.collision.new_entities.add(2)
		world.collision.new_entities.add(3)
		coll.step(0)
		self.assertEqual(coll.query_point(0, 0), set([1]))
		self.assertEqual(coll.query_point([0, 0]), set([1]))
		self.assertEqual(coll.query_point(-1, 0), set([1]))
		self.assertEqual(coll.query_point(-1.0001, 0), set())
		self.assertEqual(coll.query_point(3, 0), set([1]))
		self.assertEqual(coll.query_point(3.0001, 0), set())
		self.assertEqual(coll.query_point(0, -1), set([1]))
		self.assertEqual(coll.query_point(0, -1.0001), set())
		self.assertEqual(coll.query_point(0, 1), set([1]))
		self.assertEqual(coll.query_point(0, 1.0001), set())
		self.assertEqual(coll.query_point(-1, -1), set([1]))
		self.assertEqual(coll.query_point(3, -1), set([1]))
		self.assertEqual(coll.query_point(3, 1), set([1]))
		self.assertEqual(coll.query_point(3, 1), set([1]))

		self.assertEqual(coll.query_point(5, 5), set([2]))
		self.assertEqual(coll.query_point([6, 7]), set([2, 3]))
		self.assertEqual(coll.query_point([7, 7]), set([2, 3]))
		self.assertEqual(coll.query_point([7, 8]), set([2, 3]))
		self.assertEqual(coll.query_point([8.5, 8.5]), set([3]))

		self.assertEqual(coll.query_point(-2, 0), set())
		self.assertEqual(coll.query_point(10, 5), set())
		self.assertEqual(coll.query_point(7, 10), set())
		self.assertEqual(coll.query_point(7, -10), set())
		self.assertEqual(coll.query_point(-200, 100), set())

Exemple #3

Fichier : collision_test.py Projet : Beliaar/bGrease

	def test_before_step(self):
		# Queries should be well behaved even before the controller is run
		from bGrease.collision import BroadSweepAndPrune
		coll = BroadSweepAndPrune()
		self.assertEqual(coll.collision_pairs, set())
		self.assertEqual(coll.query_point(0,0), set())