From 37f7140bb834124d6aab127bd358a96d8fbf71dc Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Sun, 29 Mar 2020 15:13:08 +0800
Subject: [PATCH] support numerical constants

---
 Cargo.lock  |  2 ++
 Cargo.toml  |  2 ++
 src/main.rs | 28 +++++++++++++++++++++++-----
 3 files changed, 27 insertions(+), 5 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 9ddf54e..4d9ab34 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -380,6 +380,8 @@ name = "nac3"
 version = "0.1.0"
 dependencies = [
  "inkwell",
+ "num-bigint",
+ "num-traits",
  "rustpython-parser",
 ]
 
diff --git a/Cargo.toml b/Cargo.toml
index 77cb88e..9566924 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -5,5 +5,7 @@ authors = ["Sebastien Bourdeauducq <sb@m-labs.hk>"]
 edition = "2018"
 
 [dependencies]
+num-bigint = "0.2"
+num-traits = "0.2"
 inkwell = { git = "https://github.com/TheDan64/inkwell", branch = "llvm8-0" }
 rustpython-parser = { git = "https://github.com/RustPython/RustPython", branch = "master" }
diff --git a/src/main.rs b/src/main.rs
index e03e88d..3115f7c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,15 @@
+extern crate num_bigint;
 extern crate inkwell;
 extern crate rustpython_parser;
 
+use std::error::Error;
+use std::fmt;
+use std::path::Path;
+use std::collections::HashMap;
+use std::fs;
+
+use num_traits::cast::ToPrimitive;
+
 use rustpython_parser::{ast, parser};
 
 use inkwell::OptimizationLevel;
@@ -12,11 +21,6 @@ use inkwell::types;
 use inkwell::types::BasicType;
 use inkwell::values;
 
-use std::error::Error;
-use std::fmt;
-use std::path::Path;
-use std::collections::HashMap;
-use std::fs;
 
 #[derive(Debug)]
 enum CompileErrorKind {
@@ -176,6 +180,20 @@ impl<'ctx> CodeGen<'ctx> {
         self.set_source_location(expression.location);
 
         match &expression.node {
+            ast::ExpressionType::Number { value: ast::Number::Integer { value } } => {
+                let mut bits = value.bits();
+                if value.sign() == num_bigint::Sign::Minus {
+                    bits += 1;
+                }
+                match bits {
+                     0..=32 => Ok(self.context.i32_type().const_int(value.to_i32().unwrap() as _, true).into()),
+                    33..=64 => Ok(self.context.i64_type().const_int(value.to_i64().unwrap() as _, true).into()),
+                          _ => Err(self.compile_error(CompileErrorKind::Unsupported("integers larger than 64 bits")))
+                }
+            },
+            ast::ExpressionType::Number { value: ast::Number::Float { value } } => {
+                Ok(self.context.f64_type().const_float(*value).into())
+            }
             ast::ExpressionType::Identifier { name } => {
                 match self.namespace.get(name) {
                     Some(value) => Ok(*value),