Added includeSelf flag to genSystemsInRange

eyeonus · Nov 3, 2014 · fff2ccd · fff2ccd
1 parent cce3309
commit fff2ccd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tradedb.py b/tradedb.py
@@ -533,7 +533,7 @@ def lookupSystemRelaxed(self, key):
             return system
 
 
-    def genSystemsInRange(self, system, ly):
+    def genSystemsInRange(self, system, ly, includeSelf=False):
         """
             Generator for systems within ly range of system using a
             lazily-populated, per-system cache.
@@ -542,6 +542,9 @@ def genSystemsInRange(self, system, ly):
 
         # Yield what we already have
         system = self.lookupSystem(system)
+        if includeSelf:
+            yield system, 0.
+
         cache = system._rangeCache
         if not cache:
             cache = system._rangeCache = System.RangeCache()