nikdoof/dropbot
1
0
Fork 0
mirror of https://github.com/nikdoof/dropbot.git synced 2025-12-17 11:49:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add basic jump fatigue calculator into the map class.

Browse Source
This commit is contained in:
Andrew Williams
2015-04-02 12:07:10 +01:00
parent feb70dccf2
commit fd5b08227f
2 changed files with 103 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
dropbot/map.py
View File

@@ -52,6 +52,17 @@ isotope_usage = {
'blackops': 450,
}
fatiuge_bonus = {
'blackops': (.5, .5),
'covertops': (0, .5),
'forcerecon': (0, .5),
'blockaderunner': (.9, .95),
'industrial': (.9, 0),
'freighter': (.9, 0),
'jumpfreighter': (.9, 0),
'deepspacetransport': (.9, 0),
}
EVE_LY = 9460000000000000 # EVE's definition of a ly in KM
@@ -234,6 +245,34 @@ class Map(networkx.Graph):
ly = self.route_jump_distance(route)
return round(ly * base, 0)
def jump_fatigue(self, fatigue, source, destination, bonus=0, ship_class=None, jump_type='standard'):
"""Calculate the jump fatigue gained by jumping between two systems"""
if bonus == 0 and ship_class:
standard = covert = 0
if ship_class and ship_class in fatiuge_bonus:
standard, covert = fatiuge_bonus[ship_class]
if jump_type == 'standard':
bonus = standard
else:
bonus = covert
distance = self.system_distance(source, destination)
cooldown = max(fatigue / 10, 1 + (distance * (1-bonus)))
new_fatigue = min(60 * 24 * 30, max(fatigue, 10) * (1 + (distance * (1 - bonus))))
return round(cooldown, 2), round(new_fatigue, 2)
def route_jump_fatigue(self, route, fatigue, bonus=0, ship_class=None, jump_type='standard'):
"""Calculate the jump fatigue for the specified route"""
results = []
source = route.pop(0)
for target in route:
cooldown, new_fatigue = self.jump_fatigue(fatigue, source, target, bonus, ship_class, jump_type)
results.append({
'source': source, 'target': target, 'cooldown': cooldown, 'fatigue': new_fatigue,
})
source = target
fatigue = new_fatigue
return results
def neighbors_gate(self, system_id):
"""List systems that are connected to a system by gates"""
return self.neighbors(system_id)

65
tests/test_map.py
View File

@@ -8,18 +8,22 @@ class MapTestCase(TestCase):
self.map = Map.from_json(pkgutil.get_data('dropbot', 'data/map.json'))
def test_load_from_package_data(self):
"""Check the package data can be correctly loaded into the map"""
m = Map.from_json(pkgutil.get_data('dropbot', 'data/map.json'))
self.assertIsNotNone(m)
def test_get_system_name(self):
"""Test looking up system names from IDs"""
self.assertEquals(self.map.get_system_name(123), None)
self.assertEqual(self.map.get_system_name(30000142), 'Jita')
def test_get_system_id(self):
"""Test looking up system ID by name"""
self.assertEqual(self.map.get_system_id('Llamatron'), None)
self.assertEqual(self.map.get_system_id('Jita'), 30000142)
def test_get_systems(self):
"""Check partial matching of system names works correctly"""
self.assertEquals(len(self.map.get_systems('Jita')), 1)
self.assertEquals(len(self.map.get_systems('Ji')), 7)
self.assertEquals(len(self.map.get_systems('J')), 2760)
@@ -29,12 +33,14 @@ class MapTestCase(TestCase):
self.assertEquals(len(self.map.get_systems('JiTa')), 1)
def test_system_distance(self):
"""Test the distance calculator"""
self.assertEqual(self.map.system_distance(30000142, 30000144), 2.10268108033618)
self.assertEqual(self.map.system_distance(30000142, 30000222), 9.334275248404591)
self.assertEqual(self.map.system_distance(30000142, 30000536), 39.15289747780095)
self.assertRaises(Exception, self.map.system_distance, (1, 2))
def test_route_gate(self):
"""Test the gate routing system"""
r = self.map.route_gate(30001161, 30001198)
self.assertEqual(len(r), 9)
self.assertListEqual(r, [30001161, 30001158, 30001160, 30001154, 30001157, 30001155, 30001156, 30001162, 30001198])
@@ -48,6 +54,63 @@ class MapTestCase(TestCase):
def test_route_jump_isotopes(self):
pass
def test_jump_fatigue_standard(self):
"""Test jump fatigue calculator, 4.119ly jump, no bonus"""
sys1 = self.map.get_system_id('U-HVIX')
sys2 = self.map.get_system_id('V-IUEL')
cooldown, new_fatigue = self.map.jump_fatigue(0, sys1, sys2)
self.assertEqual(cooldown, 5.12)
self.assertEqual(new_fatigue, 51.19)
def test_jump_fatigue_covertops(self):
"""Test jump fatigue calculator, 4.119ly jump, covertops being bridged by titan"""
sys1 = self.map.get_system_id('U-HVIX')
sys2 = self.map.get_system_id('V-IUEL')
cooldown, new_fatigue = self.map.jump_fatigue(0, sys1, sys2, ship_class='covertops')
self.assertEqual(cooldown, 5.12)
self.assertEqual(new_fatigue, 51.19)
def test_jump_fatigue_covertops_covertcyno(self):
"""Test jump fatigue calculator, 4.119ly jump, covertops being bridged by blops"""
sys1 = self.map.get_system_id('U-HVIX')
sys2 = self.map.get_system_id('V-IUEL')
cooldown, new_fatigue = self.map.jump_fatigue(0, sys1, sys2, ship_class='covertops', jump_type='covert')
self.assertEqual(cooldown, 3.06)
self.assertEqual(new_fatigue, 30.59)
def test_route_jump_fatigue_covertops_covertcyno(self):
"""Test jump fatigue calculator, 4.119ly jump, covertops being bridged by blops"""
sys1 = self.map.get_system_id('U-HVIX')
sys2 = self.map.get_system_id('V-IUEL')
res = self.map.route_jump_fatigue([sys1, sys2], 0, ship_class='covertops', jump_type='covert')
self.assertListEqual(res, [
{'source': sys1, 'target': sys2, 'cooldown': 3.06, 'fatigue': 30.59,}
])
def test_route_jump_fatigue_covertops_covertcyno_two_jumps(self):
"""Test route jump fatigue calculator, covertops being bridged by blops for two jumps"""
sys1 = self.map.get_system_id('U-HVIX')
sys2 = self.map.get_system_id('RMOC-W')
sys3 = self.map.get_system_id('Podion')
res = self.map.route_jump_fatigue([sys1, sys2, sys3], 0, ship_class='covertops', jump_type='covert')
self.assertListEqual(res, [
{'source': sys1, 'target': sys2, 'cooldown': 4.77, 'fatigue': 47.72, },
{'source': sys2, 'target': sys3, 'cooldown': 4.77, 'fatigue': 222.91, },
])
def test_route_jump_fatigue_covertops_covertcyno_three_jumps(self):
"""Test route jump fatigue calculator, covertops being bridged by blops for three jumps"""
sys1 = self.map.get_system_id('U-HVIX')
sys2 = self.map.get_system_id('RMOC-W')
sys3 = self.map.get_system_id('Podion')
sys4 = self.map.get_system_id('Hothomouh')
res = self.map.route_jump_fatigue([sys1, sys2, sys3, sys4], 0, ship_class='covertops', jump_type='covert')
self.assertListEqual(res, [
{'source': sys1, 'target': sys2, 'cooldown': 4.77, 'fatigue': 47.72, },
{'source': sys2, 'target': sys3, 'cooldown': 4.77, 'fatigue': 222.91, },
{'source': sys3, 'target': sys4, 'cooldown': 22.29, 'fatigue': 786.39, },
])
def test_neighbors_gate(self):
pass
@@ -55,7 +118,7 @@ class MapTestCase(TestCase):
pass
def test_jump_bridge_addition(self):
# HED-GP to GE-8
"""Test addition of a jump bridge"""
self.assertGreater(len(self.map.route_gate(30001161, 30001198)), 2)
self.map.add_jumpbridge(30001161, 30001198)
r = self.map.route_gate(30001161, 30001198)