1 files changed, 39 insertions, 7 deletions

diff --git a/src/lisp/prelude.rs b/src/lisp/prelude.rs
index d6e1874..7cb46ff 100644
--- a/src/lisp/prelude.rs
+++ b/src/lisp/prelude.rs
@@ -42,8 +42,8 @@ macro_rules! type_match {
             let mut temp_vec = vec![];
             $(
                 temp_vec.push(matches!(&$args[$range], $kind));
-            )+
-            temp_vec.iter().all(|&t| t)
+             )+
+                temp_vec.iter().all(|&t| t)
         }
     };
     ($args:expr, $($range:expr => $kind:pat),+) => {
@@ -54,8 +54,8 @@ macro_rules! type_match {
                 for arg in arg_range {
                     temp_vec.push(matches!(arg, $kind));
                 }
-            )+
-            temp_vec.iter().all(|&t| t)
+             )+
+                temp_vec.iter().all(|&t| t)
         }
     }
 }
@@ -139,6 +139,17 @@ pub fn new_env() -> Result<Environment, LispError> {
         }
     });
 
+    primitive!(env, Arity::Exact(1), "integer", |args, _| {
+        if type_match!(args, 0 => LispExpr::Number(_)) {
+            Ok(LispExpr::Number(match args[0].unwrap_number() {
+                LispNumber::Float(f) => LispNumber::Integer(f as i64),
+                s => s,
+            }))
+        } else {
+            Err(EvalError::TypeMismatch.into())
+        }
+    });
+
     primitive!(env, Arity::Atleast(1), "begin", |args, _| {
         Ok(args.iter().last().unwrap().clone())
     });
@@ -394,9 +405,9 @@ pub fn new_env() -> Result<Environment, LispError> {
 
     primitive!(env, Arity::Exact(3), "set-pixel!", |args, app| {
         if type_match!(
-            args, 0 => LispExpr::Number(LispNumber::Integer(_)),
-            1 => LispExpr::Number(LispNumber::Integer(_)),
-            2 => LispExpr::BoolLit(_)
+        args, 0 => LispExpr::Number(LispNumber::Integer(_)),
+        1 => LispExpr::Number(LispNumber::Integer(_)),
+        2 => LispExpr::BoolLit(_)
         ) {
             let x = args[0].unwrap_number();
             let y = args[1].unwrap_number();
@@ -418,6 +429,27 @@ pub fn new_env() -> Result<Environment, LispError> {
         }
     });
 
+    primitive!(env, Arity::Exact(2), "get-pixel", |args, app| {
+        if type_match!(
+            args,
+            0 => LispExpr::Number(LispNumber::Integer(_)),
+            1 => LispExpr::Number(LispNumber::Integer(_))
+        ) {
+            let x = args[0].unwrap_number();
+            let y = args[1].unwrap_number();
+            let get_loc: MapPoint = (x.unwrap_integer(), y.unwrap_integer())
+                .try_into()
+                .map_err(|_| -> LispError { EvalError::InvalidCoordinates((x, y)).into() })?;
+            if !app.pixmap.contains(get_loc) {
+                Err(EvalError::InvalidCoordinates((x, y)).into())
+            } else {
+                Ok(LispExpr::BoolLit(app.pixmap.get(get_loc)))
+            }
+        } else {
+            Err(EvalError::TypeMismatch.into())
+        }
+    });
+
     primitive!(env, Arity::Exact(0), "commit", |_, app| {
         app.commit_operation();
         Ok(LispExpr::Unit)