95 lines
1.6 KiB
C++
95 lines
1.6 KiB
C++
#include <cmath>
|
|
#include "lib/Function.hpp"
|
|
#include "lib/Loader.hpp"
|
|
#include "lib/Value.hpp"
|
|
|
|
extern "C" void num_arith(jk::Loader& loader)
|
|
{
|
|
loader.declare("+", [](auto args){
|
|
int res = 0;
|
|
|
|
for (auto arg: args)
|
|
{
|
|
res += arg->as_int();
|
|
}
|
|
|
|
return jk::Value::make_int(res);
|
|
});
|
|
|
|
loader.declare("-", [](auto args){
|
|
int res = args[0]->as_int();
|
|
|
|
for (size_t i=1; i<args.size(); i++)
|
|
{
|
|
res -= args[i]->as_int();
|
|
}
|
|
|
|
return jk::Value::make_int(res);
|
|
});
|
|
|
|
loader.declare("*", [](auto args){
|
|
int res = 1;
|
|
|
|
for (auto arg: args)
|
|
{
|
|
res *= arg->as_int();
|
|
}
|
|
|
|
return jk::Value::make_int(res);
|
|
});
|
|
|
|
loader.declare("/", [](auto args){
|
|
int res = args[0]->as_int();
|
|
|
|
for (size_t i=1; i<args.size(); i++)
|
|
{
|
|
res /= args[i]->as_int();
|
|
}
|
|
|
|
return jk::Value::make_int(res);
|
|
});
|
|
|
|
loader.declare("^", [](auto args){
|
|
int res = args[0]->as_int();
|
|
|
|
for (size_t i=1; i<args.size(); i++)
|
|
{
|
|
res = std::pow(res, args[i]->as_int());
|
|
}
|
|
|
|
return jk::Value::make_int(res);
|
|
});
|
|
|
|
loader.declare("%", [](auto args){
|
|
int res = args[0]->as_int();
|
|
|
|
for (size_t i=1; i<args.size(); i++)
|
|
{
|
|
res %= args[i]->as_int();
|
|
}
|
|
|
|
return jk::Value::make_int(res);
|
|
});
|
|
}
|
|
|
|
extern "C" void lib(jk::Loader& loader)
|
|
{
|
|
num_arith(loader);
|
|
loader.declare("dump", [](auto args){
|
|
std::string sep;
|
|
|
|
for (auto arg: args)
|
|
{
|
|
std::cout << sep << arg->string();
|
|
sep = " ";
|
|
}
|
|
|
|
if (args.empty() == false)
|
|
{
|
|
std::cout << std::endl;
|
|
}
|
|
|
|
return jk::Value::make_nil();
|
|
});
|
|
}
|