示例#1
0
	def __init__(self, star, orbit_number):
		self.star = star
		self.orbit_number = orbit_number
		self.name = Planet.names[orbit_number]
		self.type = utilities.choose_from_probability(Planet.star_type_probability_table[self.star.type])
		self.size = utilities.choose_from_probability(Planet.size_probability_table)
		self.richness = utilities.choose_from_probability(Planet.star_richness_probability_table[self.star.type])
		self.gravity = Planet.size_richness_gravity_table[self.size][self.richness]
		logger.debug("%s: %s %s %s %s", self.name, self.size, self.type, self.richness, self.gravity)
示例#2
0
	def add(self, star, orbit_number, type=None):
		if type is None:
			type = utilities.choose_from_probability(Orbitals.star_orbital_probability_table[star.type])

		if self.star_orbits.has_key(star):
			if orbit_number in self.star_orbits[star]:
				raise Exception, "already added orbit %d for star %s"%(orbit_number, star.name)
		else:
			self.star_orbits[star] = []
		self.star_orbits[star].append(orbit_number)

		if type == 'nothing':
			object = Empty(star, orbit_number)
		elif type == 'planet':
			object = Planet(star, orbit_number)
			self.planets.append(object)
		elif type == 'asteroid belt':
			object = AsteroidBelt(star, orbit_number)
			self.asteroid_belts.append(object)
		elif type == 'gas giant':
			object = GasGiant(star, orbit_number)
			self.gas_giants.append(object)
		else:
			raise Exception, 'Unknown orbital type: %s'%type

		return object
示例#3
0
	def generate_stars_and_black_holes(self, config):
		# generate coordinates for stars AND black holes in one pass
		# since they may not overlap
		object_coordinates = utilities.random_dispersed_coordinates(
			config.limits[0], config.limits[1], config.limits[2], config.limits[3],
			amount=config.star_count, dispersion=config.dispersion
		)

		self.stars = stars.Stars(self.orbitals)
		self.black_holes = black_holes.BlackHoles()

		for coordinate in object_coordinates:
			object_type = utilities.choose_from_probability(config.object_pool)

			if object_type == 'black hole':
				self.black_holes.add(coordinate)

			else:
				# object_type is color
				self.stars.add(coordinate, object_type)
示例#4
0
	def test_zeroed_key(self):
		"If a key has value zero, it should never be the return value."
		table = {'a': 0, 'b': 1}
		self.assertEqual(utilities.choose_from_probability(table), 'b')
示例#5
0
	def test_total_0(self):
		"If probabilities total zero, result should be the (alpha-sort) first probability."
		table = {'c': 0, 'a':0, 'b': 0}
		self.assertEqual(utilities.choose_from_probability(table), 'a')