From 592661bae5547a0c1675816595b803f902d83041 Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Fri, 3 Mar 2023 13:49:50 +0800
Subject: [PATCH] 1

---
 a8/lisp.cc | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++----
 a8/lisp.h  | 12 ++++++---
 2 files changed, 75 insertions(+), 8 deletions(-)

diff --git a/a8/lisp.cc b/a8/lisp.cc
index 055fe13..5059a23 100644
--- a/a8/lisp.cc
+++ b/a8/lisp.cc
@@ -43,6 +43,58 @@ namespace a8
             vars_[name] = std::make_shared<Value>(proc);
         }
 
+        GlobalScope::GlobalScope()
+        {
+            RegisterCProc
+                (
+                 "+",
+                 [] (const List& params) -> std::shared_ptr<Value>
+                 {
+                     Atom result;
+                     for (auto param : params) {
+                         Atom&& atom = std::any_cast<Atom>(param->value);
+                         result.val += atom.val;
+                     }
+                     return std::make_shared<Value>(result);
+                 });
+            RegisterCProc
+                (
+                 "-",
+                 [] (const List& params) -> std::shared_ptr<Value>
+                 {
+                     Atom result;
+                     for (auto param : params) {
+                         Atom&& atom = std::any_cast<Atom>(param->value);
+                         result.val -= atom.val;
+                     }
+                     return std::make_shared<Value>(result);
+                 });
+            RegisterCProc
+                (
+                 "*",
+                 [] (const List& params) -> std::shared_ptr<Value>
+                 {
+                     Atom result;
+                     for (auto param : params) {
+                         Atom&& atom = std::any_cast<Atom>(param->value);
+                         result.val *= atom.val;
+                     }
+                     return std::make_shared<Value>(result);
+                 });
+            RegisterCProc
+                (
+                 "/",
+                 [] (const List& params) -> std::shared_ptr<Value>
+                 {
+                     Atom result;
+                     for (auto param : params) {
+                         Atom&& atom = std::any_cast<Atom>(param->value);
+                         result.val /= atom.val;
+                     }
+                     return std::make_shared<Value>(result);
+                 });
+        }
+
         std::shared_ptr<Value> Expr::Compile(const std::string& script, std::shared_ptr<Scope> scope)
         {
 
@@ -67,12 +119,21 @@ namespace a8
                     {
                         List&& list = std::any_cast<List>(x->value);
                         auto leader = list.at(0);
-                        if (leader->type != ValueType::kCProc) {
-                            abort();
+                        if (leader->type == ValueType::kCProc) {
+                            CProc&& cproc = std::any_cast<CProc>(leader->value);
+                            List exps;
+                            for (auto itr = list.begin() + 1; itr != list.end(); ++itr) {
+                                exps.push_back(Eval(*itr, env));
+                            }
+                            return cproc(exps);
+                        } else {
+                            CProc&& cproc = std::any_cast<CProc>(Eval(leader, env)->value);
+                            List exps;
+                            for (auto itr = list.begin() + 1; itr != list.end(); ++itr) {
+                                exps.push_back(Eval(*itr, env));
+                            }
+                            return cproc(exps);
                         }
-                        CProc&& cproc = std::any_cast<CProc>(leader->value);
-                        List exps = List(list.begin() + 1, list.end());
-                        return cproc(exps);
                     }
                     break;
                 default:
diff --git a/a8/lisp.h b/a8/lisp.h
index 16a8f08..c462e7e 100644
--- a/a8/lisp.h
+++ b/a8/lisp.h
@@ -21,7 +21,7 @@ namespace a8
         };
         struct Value;
         typedef std::vector<std::shared_ptr<Value>> List;
-        typedef std::function<std::shared_ptr<Value>(const List)> CProc;
+        typedef std::function<std::shared_ptr<Value>(const List&)> CProc;
 
         struct Atom
         {
@@ -56,12 +56,18 @@ namespace a8
             std::shared_ptr<Scope> outer_;
         };
 
+        class GlobalScope : public Scope
+        {
+        public:
+            GlobalScope();
+        };
+
         class Expr
         {
         public:
 
-            std::shared_ptr<Value> Compile(const std::string& script, std::shared_ptr<Scope> scope);
-            std::shared_ptr<Value> Eval(std::shared_ptr<Value> x, std::shared_ptr<Scope> env);
+            static std::shared_ptr<Value> Compile(const std::string& script, std::shared_ptr<Scope> scope);
+            static std::shared_ptr<Value> Eval(std::shared_ptr<Value> x, std::shared_ptr<Scope> env);
 
         private:
         };