From 10fb09eca27991e878bee4a4c63d0ecd6b4a44b3 Mon Sep 17 00:00:00 2001
From: Griffin Smith <root@gws.fyi>
Date: Sun, 28 Jul 2019 21:16:20 -0400
Subject: [PATCH] Turns out, collect is the answer

---
 src/types/entity_map.rs | 23 ++++++++---------------
 1 file changed, 8 insertions(+), 15 deletions(-)

diff --git a/src/types/entity_map.rs b/src/types/entity_map.rs
index 0f224d15e..bec16cdab 100644
--- a/src/types/entity_map.rs
+++ b/src/types/entity_map.rs
@@ -42,21 +42,14 @@ impl<A> EntityMap<A> {
 
     /// Returns a list of all entities at the given position
     pub fn at<'a>(&'a self, pos: Position) -> Vec<&'a A> {
-        // self.by_position.get(&pos).iter().flat_map(|eids| {
-        //     eids.iter()
-        //         .map(|eid| self.by_id.get(eid).expect(BY_POS_INVARIANT))
-        // })
-        // gross.
-        match self.by_position.get(&pos) {
-            None => Vec::new(),
-            Some(eids) => {
-                let mut res = Vec::new();
-                for eid in eids {
-                    res.push(self.by_id.get(eid).expect(BY_POS_INVARIANT));
-                }
-                res
-            }
-        }
+        self.by_position
+            .get(&pos)
+            .iter()
+            .flat_map(|eids| {
+                eids.iter()
+                    .map(|eid| self.by_id.get(eid).expect(BY_POS_INVARIANT))
+            })
+            .collect()
     }
 
     /// Remove all entities at the given position