Remove TODOs

src/grid.rs

@@ -2,7 +2,7 @@
 mod builder;
 pub mod local_grid;
 
-pub use local_grid::{SingleElementGrid, SingleElementGridBorrowed, SingleElementGridBuilder};
+pub use local_grid::{SingleElementGrid, SingleElementGridBuilder};
 
 use local_grid::LocalGrid;
 

src/grid/local_grid.rs

@@ -7,7 +7,7 @@ use crate::{
 };
 
 mod single_element;
-pub use single_element::{SingleElementGrid, SingleElementGridBorrowed, SingleElementGridBuilder};
+pub use single_element::{SingleElementGrid, SingleElementGridBuilder};
 
 /// Grid entity
 #[derive(Debug)]

src/grid/local_grid/single_element.rs

@@ -1,8 +1,6 @@
 //! Single element grid
-mod borrowed;
 mod builder;
 mod grid;
 
-pub use borrowed::SingleElementGridBorrowed;
 pub use builder::SingleElementGridBuilder;
 pub use grid::SingleElementGrid;

src/grid/local_grid/single_element/grid.rs

@@ -277,12 +277,9 @@ impl<T: RealScalar, E: FiniteElement<CellType = ReferenceCellType, T = T>> Grid
     }
 
     fn entity_from_id(&self, dim: usize, id: usize) -> Option<Self::Entity<'_>> {
-        // TODO: store a HashMap rather than doing a find every time?
-        self.topology.ids[dim].as_ref().map(|ids| {
-            ids.iter()
-                .find(|&i| *i == id)
-                .map(|i| self.entity(dim, *i))?
-        })?
+        self.topology.ids_to_indices[dim]
+            .get(&id)
+            .map(|i| self.entity(dim, *i))?
     }
 
     fn geometry_map(

src/lib.rs

@@ -10,6 +10,4 @@ pub mod topology;
 pub mod traits;
 pub mod types;
 
-pub use grid::{
-    ParallelGridImpl, SingleElementGrid, SingleElementGridBorrowed, SingleElementGridBuilder,
-};
+pub use grid::{ParallelGridImpl, SingleElementGrid, SingleElementGridBuilder};