1 files changed, 35 insertions, 0 deletions
diff --git a/stockton-types/src/ent_map.rs b/stockton-types/src/ent_map.rs
new file mode 100644
index 0000000..213e481
--- /dev/null
+++ b/stockton-types/src/ent_map.rs
@@ -0,0 +1,35 @@
+// Copyright (C) Oscar Shrimpton 2019
+
+// This program is free software: you can redistribute it and/or modify it
+// under the terms of the GNU General Public License as published by the Free
+// Software Foundation, either version 3 of the License, or (at your option)
+// any later version.
+
+// This program is distributed in the hope that it will be useful, but WITHOUT
+// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+// more details.
+
+// You should have received a copy of the GNU General Public License along
+// with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+/// Convenience function for creating mappers for `World::new()`.
+#[macro_export]
+macro_rules! ent_map {
+	( $ ( $name:expr => $type:ident [ $( $key:expr => $target:ident ),* ] ),* ) => {
+		{
+			use stockton_bsp::lumps::entities::Entity as BSPEntity;	
+			use stockton_types::Entity;	
+			|ent: &BSPEntity| -> Box<dyn Entity> {
+				$(
+					if ent.attributes["classname"] == $name {
+						return Box::new($type {
+							$( $target : ent.attributes[$key].into() ),*
+						});
+					}
+				);*
+				panic!("Unrecognised Entity type: {:?}", ent);
+			}
+		}
+	}
+}
+\ No newline at end of file