gmqcc/intrin.cpp

2049 lines
51 KiB
C++
Raw Normal View History

#include <string.h>
2015-01-16 01:27:17 +00:00
#include "ast.h"
#include "fold.h"
#include "parser.h"
2013-03-09 08:57:37 +00:00
2015-01-15 23:11:41 +00:00
lex_ctx_t intrin::ctx() const {
return parser_ctx(m_parser);
}
2013-10-02 17:00:34 +00:00
2015-01-20 15:43:58 +00:00
ast_function *intrin::value(ast_value **out, const char *name, qc_type vtype) {
2015-01-15 23:11:41 +00:00
ast_value *value = nullptr;
2015-01-15 20:18:33 +00:00
ast_function *func = nullptr;
2015-01-15 23:11:41 +00:00
char buffer[1024];
char stype [1024];
2013-10-02 17:00:34 +00:00
util_snprintf(buffer, sizeof(buffer), "__builtin_%s", name);
util_snprintf(stype, sizeof(stype), "<%s>", type_name[vtype]);
2015-01-30 17:59:48 +00:00
value = new ast_value(ctx(), buffer, TYPE_FUNCTION);
2015-01-24 11:25:46 +00:00
value->m_intrinsic = true;
2015-01-30 17:59:48 +00:00
value->m_next = new ast_value(ctx(), stype, vtype);
func = ast_function::make(ctx(), buffer, value);
2015-01-24 11:25:46 +00:00
value->m_flags |= AST_FLAG_ERASEABLE;
2013-10-02 17:00:34 +00:00
*out = value;
2013-10-02 17:00:34 +00:00
return func;
}
2015-01-15 23:11:41 +00:00
void intrin::reg(ast_value *const value, ast_function *const func) {
m_parser->functions.push_back(func);
2015-01-30 17:59:48 +00:00
m_parser->globals.push_back(value);
2013-10-02 17:00:34 +00:00
}
2013-03-09 08:57:37 +00:00
#define QC_POW_EPSILON 0.00001f
2013-03-09 08:57:37 +00:00
2015-01-15 23:11:41 +00:00
ast_expression *intrin::nullfunc() {
ast_value *val = nullptr;
ast_function *func = value(&val, nullptr, TYPE_VOID);
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::isfinite_() {
/*
* float isfinite(float x) {
* return !(isnan(x) || isinf(x));
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "isfinite", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
ast_call *callisnan = ast_call::make(ctx(), func_self("isnan", "isfinite"));
ast_call *callisinf = ast_call::make(ctx(), func_self("isinf", "isfinite"));
ast_block *block = new ast_block(ctx());
/* float x; */
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
/* <callisnan> = isnan(x); */
2015-01-30 17:59:48 +00:00
callisnan->m_params.push_back(x);
/* <callisinf> = isinf(x); */
2015-01-30 17:59:48 +00:00
callisinf->m_params.push_back(x);
/* return (!<callisnan> || <callisinf>); */
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
ast_unary::make(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_NOT_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_OR,
2015-01-30 17:59:48 +00:00
callisnan,
callisinf
)
)
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(block);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::isinf_() {
/*
* float isinf(float x) {
* return (x != 0.0) && (x + x == x);
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "isinf", TYPE_FLOAT);
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_AND,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_NE_F,
2015-01-30 17:59:48 +00:00
x,
m_fold->m_imm_float[0]
),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_EQ_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
x,
x
),
2015-01-30 17:59:48 +00:00
x
)
)
)
);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::isnan_() {
/*
* float isnan(float x) {
* float local;
* local = x;
*
* return (x != local);
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *arg1 = new ast_value(ctx(), "x",TYPE_FLOAT);
ast_value *local = new ast_value(ctx(), "local", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "isnan", TYPE_FLOAT);
2015-01-24 11:25:46 +00:00
body->m_locals.push_back(local);
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
local,
arg1
)
);
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_NE_F,
2015-01-30 17:59:48 +00:00
arg1,
local
)
)
);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(arg1);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::isnormal_() {
/*
* float isnormal(float x) {
* return isfinite(x);
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_call *callisfinite = ast_call::make(ctx(), func_self("isfinite", "isnormal"));
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "isnormal", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
callisfinite->m_params.push_back(x);
/* return <callisfinite> */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
callisfinite
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::signbit_() {
/*
* float signbit(float x) {
* return (x < 0);
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "signbit", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
/* return (x < 0); */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_ternary(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LT,
2015-01-30 17:59:48 +00:00
x,
m_fold->m_imm_float[0]
),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
m_fold->m_imm_float[0]
)
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::acosh_() {
/*
* float acosh(float x) {
* return log(x + sqrt((x * x) - 1));
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_call *calllog = ast_call::make(ctx(), func_self("log", "acosh"));
ast_call *callsqrt = ast_call::make(ctx(), func_self("sqrt", "acosh"));
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "acosh", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
/* <callsqrt> = sqrt((x * x) - 1); */
2015-01-24 11:25:46 +00:00
callsqrt->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_SUB_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
x,
x
),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1]
)
);
/* <calllog> = log(x + <callsqrt>); */
2015-01-24 11:25:46 +00:00
calllog->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
x,
callsqrt
)
);
/* return <calllog>; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
calllog
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::asinh_() {
/*
* float asinh(float x) {
* return log(x + sqrt((x * x) + 1));
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_call *calllog = ast_call::make(ctx(), func_self("log", "asinh"));
ast_call *callsqrt = ast_call::make(ctx(), func_self("sqrt", "asinh"));
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "asinh", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
/* <callsqrt> = sqrt((x * x) + 1); */
2015-01-24 11:25:46 +00:00
callsqrt->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
x,
x
),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1]
)
);
/* <calllog> = log(x + <callsqrt>); */
2015-01-24 11:25:46 +00:00
calllog->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
x,
callsqrt
)
);
/* return <calllog>; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
calllog
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::atanh_() {
/*
* float atanh(float x) {
* return 0.5 * log((1 + x) / (1 - x))
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_call *calllog = ast_call::make(ctx(), func_self("log", "atanh"));
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "atanh", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
/* <callog> = log((1 + x) / (1 - x)); */
2015-01-24 11:25:46 +00:00
calllog->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
x
),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_SUB_F,
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
x
)
)
);
/* return 0.5 * <calllog>; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
m_fold->constgen_float(0.5, false),
calllog
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::exp_() {
/*
* float exp(float x) {
* float sum = 1.0;
* float acc = 1.0;
* float i;
* for (i = 1; i < 200; ++i)
* sum += (acc *= x / i);
*
* return sum;
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_value *sum = new ast_value(ctx(), "sum", TYPE_FLOAT);
ast_value *acc = new ast_value(ctx(), "acc", TYPE_FLOAT);
ast_value *i = new ast_value(ctx(), "i", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "exp", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
2015-01-15 04:34:43 +00:00
2015-01-24 11:25:46 +00:00
body->m_locals.push_back(sum);
body->m_locals.push_back(acc);
body->m_locals.push_back(i);
/* sum = 1.0; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
sum,
m_fold->m_imm_float[1]
)
);
/* acc = 1.0; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
acc,
m_fold->m_imm_float[1]
)
);
/*
* for (i = 1; i < 200; ++i)
* sum += (acc *= x / i);
*/
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_loop(
2015-01-15 23:11:41 +00:00
ctx(),
/* i = 1; */
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
i,
m_fold->m_imm_float[1]
),
/* i < 200; */
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LT,
2015-01-30 17:59:48 +00:00
i,
m_fold->constgen_float(200.0f, false)
),
false,
2015-01-15 20:18:33 +00:00
nullptr,
false,
/* ++i; */
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
i,
m_fold->m_imm_float[1]
),
/* sum += (acc *= (x / i)) */
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
sum,
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
acc,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
x,
i
)
)
)
)
);
/* return sum; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
sum
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::exp2_() {
/*
* float exp2(float x) {
* return pow(2, x);
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_call *callpow = ast_call::make(ctx(), func_self("pow", "exp2"));
ast_value *arg1 = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "exp2", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(arg1);
2015-01-30 17:59:48 +00:00
callpow->m_params.push_back(m_fold->m_imm_float[3]);
callpow->m_params.push_back(arg1);
/* return <callpow> */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
callpow
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::expm1_() {
/*
* float expm1(float x) {
* return exp(x) - 1;
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_call *callexp = ast_call::make(ctx(), func_self("exp", "expm1"));
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "expm1", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(x);
/* <callexp> = exp(x); */
2015-01-30 17:59:48 +00:00
callexp->m_params.push_back(x);
/* return <callexp> - 1; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_SUB_F,
2015-01-30 17:59:48 +00:00
callexp,
m_fold->m_imm_float[1]
)
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::pow_() {
/*
*
* float pow(float base, float exp) {
* float result;
* float low;
* float high;
* float mid;
* float square;
* float accumulate;
*
* if (exp == 0.0)
2013-11-23 16:14:42 +00:00
* return 1;
* if (exp == 1.0)
* return base;
* if (exp < 0)
* return 1.0 / pow(base, -exp);
* if (exp >= 1) {
* result = pow(base, exp / 2);
* return result * result;
* }
*
* low = 0.0f;
* high = 1.0f;
* square = sqrt(base);
* accumulate = square;
* mid = high / 2.0f
*
* while (fabs(mid - exp) > QC_POW_EPSILON) {
* square = sqrt(square);
* if (mid < exp) {
* low = mid;
* accumulate *= square;
* } else {
* high = mid;
* accumulate *= (1.0f / square);
* }
* mid = (low + high) / 2;
* }
* return accumulate;
2013-04-21 10:24:55 +00:00
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
ast_function *func = value(&val, "pow", TYPE_FLOAT);
/* prepare some calls for later */
2015-01-30 17:59:48 +00:00
ast_call *callpow1 = ast_call::make(ctx(), val); /* for pow(base, -exp) */
ast_call *callpow2 = ast_call::make(ctx(), val); /* for pow(vase, exp / 2) */
ast_call *callsqrt1 = ast_call::make(ctx(), func_self("sqrt", "pow")); /* for sqrt(base) */
ast_call *callsqrt2 = ast_call::make(ctx(), func_self("sqrt", "pow")); /* for sqrt(square) */
ast_call *callfabs = ast_call::make(ctx(), func_self("fabs", "pow")); /* for fabs(mid - exp) */
/* prepare some blocks for later */
2015-01-30 17:59:48 +00:00
ast_block *expgt1 = new ast_block(ctx());
ast_block *midltexp = new ast_block(ctx());
ast_block *midltexpelse = new ast_block(ctx());
ast_block *whileblock = new ast_block(ctx());
/* float pow(float base, float exp) */
2015-01-30 17:59:48 +00:00
ast_value *base = new ast_value(ctx(), "base", TYPE_FLOAT);
ast_value *exp = new ast_value(ctx(), "exp", TYPE_FLOAT);
/* { */
2015-01-30 17:59:48 +00:00
ast_block *body = new ast_block(ctx());
/*
* float result;
* float low;
* float high;
* float square;
* float accumulate;
* float mid;
*/
2015-01-30 17:59:48 +00:00
ast_value *result = new ast_value(ctx(), "result", TYPE_FLOAT);
ast_value *low = new ast_value(ctx(), "low", TYPE_FLOAT);
ast_value *high = new ast_value(ctx(), "high", TYPE_FLOAT);
ast_value *square = new ast_value(ctx(), "square", TYPE_FLOAT);
ast_value *accumulate = new ast_value(ctx(), "accumulate", TYPE_FLOAT);
ast_value *mid = new ast_value(ctx(), "mid", TYPE_FLOAT);
2015-01-24 11:25:46 +00:00
body->m_locals.push_back(result);
body->m_locals.push_back(low);
body->m_locals.push_back(high);
body->m_locals.push_back(square);
body->m_locals.push_back(accumulate);
body->m_locals.push_back(mid);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(base);
val->m_type_params.emplace_back(exp);
/*
* if (exp == 0.0)
2013-11-23 16:14:42 +00:00
* return 1;
*/
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_EQ_F,
2015-01-30 17:59:48 +00:00
exp,
m_fold->m_imm_float[0]
),
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1]
2013-11-23 16:14:42 +00:00
),
2015-01-15 20:18:33 +00:00
nullptr
2013-11-23 16:14:42 +00:00
)
);
/*
* if (exp == 1.0)
* return base;
*/
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
2013-11-23 16:14:42 +00:00
INSTR_EQ_F,
2015-01-30 17:59:48 +00:00
exp,
m_fold->m_imm_float[1]
2013-11-23 16:14:42 +00:00
),
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
base
),
2015-01-15 20:18:33 +00:00
nullptr
)
);
/* <callpow1> = pow(base, -exp) */
2015-01-30 17:59:48 +00:00
callpow1->m_params.push_back(base);
2015-01-24 11:25:46 +00:00
callpow1->m_params.push_back(
2015-01-30 17:59:48 +00:00
ast_unary::make(
2015-01-15 23:11:41 +00:00
ctx(),
VINSTR_NEG_F,
2015-01-30 17:59:48 +00:00
exp
)
);
/*
* if (exp < 0)
* return 1.0 / <callpow1>;
*/
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LT,
2015-01-30 17:59:48 +00:00
exp,
m_fold->m_imm_float[0]
),
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
callpow1
)
),
2015-01-15 20:18:33 +00:00
nullptr
)
);
/* <callpow2> = pow(base, exp / 2) */
2015-01-30 17:59:48 +00:00
callpow2->m_params.push_back(base);
2015-01-24 11:25:46 +00:00
callpow2->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
exp,
m_fold->m_imm_float[3] /* 2.0f */
)
);
/*
* <expgt1> = {
* result = <callpow2>;
* return result * result;
* }
*/
2015-01-24 11:25:46 +00:00
expgt1->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
result,
callpow2
)
);
2015-01-24 11:25:46 +00:00
expgt1->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
result,
result
)
)
);
/*
* if (exp >= 1) {
* <expgt1>
* }
*/
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_GE,
2015-01-30 17:59:48 +00:00
exp,
m_fold->m_imm_float[1]
),
2015-01-30 17:59:48 +00:00
expgt1,
2015-01-15 20:18:33 +00:00
nullptr
)
);
/*
* <callsqrt1> = sqrt(base)
*/
2015-01-30 17:59:48 +00:00
callsqrt1->m_params.push_back(base);
/*
* low = 0.0f;
* high = 1.0f;
* square = sqrt(base);
* accumulate = square;
* mid = high / 2.0f;
*/
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
low,
m_fold->m_imm_float[0]
)
);
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
high,
m_fold->m_imm_float[1]
)
);
2013-11-25 07:11:45 +00:00
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
square,
callsqrt1
)
);
2013-11-25 07:11:45 +00:00
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
accumulate,
square
)
);
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
mid,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
high,
m_fold->m_imm_float[3] /* 2.0f */
)
)
);
/*
* <midltexp> = {
* low = mid;
* accumulate *= square;
* }
*/
2015-01-24 11:25:46 +00:00
midltexp->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
low,
mid
)
);
2015-01-24 11:25:46 +00:00
midltexp->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
accumulate,
square
)
);
/*
* <midltexpelse> = {
* high = mid;
* accumulate *= (1.0 / square);
* }
*/
2015-01-24 11:25:46 +00:00
midltexpelse->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
high,
mid
)
);
2015-01-24 11:25:46 +00:00
midltexpelse->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
accumulate,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
square
)
)
);
/*
* <callsqrt2> = sqrt(square)
*/
2015-01-30 17:59:48 +00:00
callsqrt2->m_params.push_back(square);
/*
* <whileblock> = {
* square = <callsqrt2>;
* if (mid < exp)
* <midltexp>;
* else
* <midltexpelse>;
*
* mid = (low + high) / 2;
* }
*/
2015-01-24 11:25:46 +00:00
whileblock->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
square,
callsqrt2
)
);
2015-01-24 11:25:46 +00:00
whileblock->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LT,
2015-01-30 17:59:48 +00:00
mid,
exp
),
2015-01-30 17:59:48 +00:00
midltexp,
midltexpelse
)
);
2015-01-24 11:25:46 +00:00
whileblock->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
mid,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
low,
high
),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[3] /* 2.0f */
)
)
);
/*
* <callabs> = fabs(mid - exp)
*/
2015-01-24 11:25:46 +00:00
callfabs->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_SUB_F,
2015-01-30 17:59:48 +00:00
mid,
exp
)
);
/*
* while (<callfabs> > epsilon)
* <whileblock>
*/
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_loop(
2015-01-15 23:11:41 +00:00
ctx(),
/* init */
2015-01-15 20:18:33 +00:00
nullptr,
/* pre condition */
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_GT,
2015-01-30 17:59:48 +00:00
callfabs,
m_fold->constgen_float(QC_POW_EPSILON, false)
),
/* pre not */
false,
/* post condition */
2015-01-15 20:18:33 +00:00
nullptr,
/* post not */
false,
/* increment expression */
2015-01-15 20:18:33 +00:00
nullptr,
/* code block */
2015-01-30 17:59:48 +00:00
whileblock
)
);
2013-11-23 16:15:28 +00:00
/* return accumulate */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
accumulate
)
);
/* } */
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
2013-03-09 08:57:37 +00:00
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::mod_() {
/*
2013-11-23 08:53:07 +00:00
* float mod(float a, float b) {
* float div = a / b;
* float sign = (div < 0.0f) ? -1 : 1;
* return a - b * sign * floor(sign * div);
* }
2013-04-21 10:24:55 +00:00
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_call *call = ast_call::make(ctx(), func_self("floor", "mod"));
ast_value *a = new ast_value(ctx(), "a", TYPE_FLOAT);
ast_value *b = new ast_value(ctx(), "b", TYPE_FLOAT);
ast_value *div = new ast_value(ctx(), "div", TYPE_FLOAT);
ast_value *sign = new ast_value(ctx(), "sign", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "mod", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(a);
val->m_type_params.emplace_back(b);
2013-11-23 08:53:07 +00:00
2015-01-24 11:25:46 +00:00
body->m_locals.push_back(div);
body->m_locals.push_back(sign);
2013-11-23 08:53:07 +00:00
/* div = a / b; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
2013-11-23 08:53:07 +00:00
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
div,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
2013-11-23 08:53:07 +00:00
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
a,
b
2013-11-23 08:53:07 +00:00
)
)
);
/* sign = (div < 0.0f) ? -1 : 1; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
2013-11-23 08:53:07 +00:00
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
sign,
new ast_ternary(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
2013-11-23 08:53:07 +00:00
INSTR_LT,
2015-01-30 17:59:48 +00:00
div,
m_fold->m_imm_float[0]
2013-11-23 08:53:07 +00:00
),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[2],
m_fold->m_imm_float[1]
2013-11-23 08:53:07 +00:00
)
)
);
/* floor(sign * div) */
2015-01-24 11:25:46 +00:00
call->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
2013-11-23 08:53:07 +00:00
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
sign,
div
)
);
2013-11-23 08:53:07 +00:00
/* return a - b * sign * <call> */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_SUB_F,
2015-01-30 17:59:48 +00:00
a,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
b,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
2013-11-23 08:53:07 +00:00
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
sign,
call
2013-11-23 08:53:07 +00:00
)
2013-03-09 08:57:37 +00:00
)
)
)
);
2013-03-09 08:57:37 +00:00
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::fabs_() {
/*
* float fabs(float x) {
* return x < 0 ? -x : x;
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *arg1 = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "fabs", TYPE_FLOAT);
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_ternary(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LE,
2015-01-30 17:59:48 +00:00
arg1,
m_fold->m_imm_float[0]
),
2015-01-30 17:59:48 +00:00
ast_unary::make(
2015-01-15 23:11:41 +00:00
ctx(),
VINSTR_NEG_F,
2015-01-30 17:59:48 +00:00
arg1
),
2015-01-30 17:59:48 +00:00
arg1
)
)
);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(arg1);
2015-01-15 04:34:43 +00:00
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::epsilon_() {
/*
* float epsilon(void) {
* float eps = 1.0f;
* do { eps /= 2.0f; } while ((1.0f + (eps / 2.0f)) != 1.0f);
* return eps;
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *eps = new ast_value(ctx(), "eps", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "epsilon", TYPE_FLOAT);
2015-01-24 11:25:46 +00:00
body->m_locals.push_back(eps);
/* eps = 1.0f; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
eps,
m_fold->m_imm_float[0]
)
);
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_loop(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-15 20:18:33 +00:00
nullptr,
nullptr,
false,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_NE_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
eps,
m_fold->m_imm_float[3] /* 2.0f */
)
),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1]
),
false,
2015-01-15 20:18:33 +00:00
nullptr,
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
eps,
m_fold->m_imm_float[3] /* 2.0f */
)
)
);
/* return eps; */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
eps
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::nan_() {
/*
* float nan(void) {
* float x = 0.0f;
* return x / x;
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "nan", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
ast_block *block = new ast_block(ctx());
2015-01-24 11:25:46 +00:00
block->m_locals.push_back(x);
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
x,
m_fold->m_imm_float[0]
)
);
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
x,
x
)
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(block);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::inf_() {
/*
* float inf(void) {
* float x = 1.0f;
* float y = 0.0f;
* return x / y;
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *x = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_value *y = new ast_value(ctx(), "y", TYPE_FLOAT);
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "inf", TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
ast_block *block = new ast_block(ctx());
size_t i;
2015-01-24 11:25:46 +00:00
block->m_locals.push_back(x);
block->m_locals.push_back(y);
/* to keep code size down */
for (i = 0; i <= 1; i++) {
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
((i == 0) ? x : y),
m_fold->m_imm_float[i]
)
);
}
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
x,
y
)
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(block);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::ln_() {
/*
* float log(float power, float base) {
* float whole;
* float nth
* float sign = 1.0f;
* float eps = epsilon();
*
* if (power <= 1.0f || bbase <= 1.0) {
* if (power <= 0.0f || base <= 0.0f)
* return nan();
*
* if (power < 1.0f) {
* power = 1.0f / power;
* sign *= -1.0f;
* }
*
* if (base < 1.0f) {
* sign *= -1.0f;
* base = 1.0f / base;
* }
* }
*
* float A_i = 1;
* float B_i = 0;
* float A_iminus1 = 0;
* float B_iminus1 = 1;
*
* for (;;) {
* whole = power;
* nth = 0.0f;
*
* while (whole >= base) {
* float base2 = base;
* float n2 = 1.0f;
* float newbase2 = base2 * base2;
*
* while (whole >= newbase2) {
* base2 = newbase2;
* n2 *= 2;
* newbase2 *= newbase2;
* }
*
* whole /= base2;
* nth += n2;
* }
*
* float b_iplus1 = n;
* float A_iplus1 = b_iplus1 * A_i + A_iminus1;
* float B_iplus1 = b_iplus1 * B_i + B_iminus1;
*
* A_iminus1 = A_i;
* B_iminus1 = B_i;
* A_i = A_iplus1;
* B_i = B_iplus1;
*
* if (whole <= 1.0f + eps)
* break;
*
* power = base;
* bower = whole;
* }
* return sign * A_i / B_i;
* }
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_value *power = new ast_value(ctx(), "power", TYPE_FLOAT);
ast_value *base = new ast_value(ctx(), "base",TYPE_FLOAT);
ast_value *whole= new ast_value(ctx(), "whole", TYPE_FLOAT);
ast_value *nth = new ast_value(ctx(), "nth", TYPE_FLOAT);
ast_value *sign = new ast_value(ctx(), "sign", TYPE_FLOAT);
ast_value *A_i = new ast_value(ctx(), "A_i", TYPE_FLOAT);
ast_value *B_i = new ast_value(ctx(), "B_i", TYPE_FLOAT);
ast_value *A_iminus1 = new ast_value(ctx(), "A_iminus1", TYPE_FLOAT);
ast_value *B_iminus1 = new ast_value(ctx(), "B_iminus1", TYPE_FLOAT);
ast_value *b_iplus1 = new ast_value(ctx(), "b_iplus1", TYPE_FLOAT);
ast_value *A_iplus1 = new ast_value(ctx(), "A_iplus1", TYPE_FLOAT);
ast_value *B_iplus1 = new ast_value(ctx(), "B_iplus1", TYPE_FLOAT);
ast_value *eps = new ast_value(ctx(), "eps", TYPE_FLOAT);
ast_value *base2 = new ast_value(ctx(), "base2", TYPE_FLOAT);
ast_value *n2 = new ast_value(ctx(), "n2",TYPE_FLOAT);
ast_value *newbase2 = new ast_value(ctx(), "newbase2", TYPE_FLOAT);
ast_block *block = new ast_block(ctx());
ast_block *plt1orblt1 = new ast_block(ctx()); // (power <= 1.0f || base <= 1.0f)
ast_block *plt1 = new ast_block(ctx()); // (power < 1.0f)
ast_block *blt1 = new ast_block(ctx()); // (base< 1.0f)
ast_block *forloop = new ast_block(ctx()); // for(;;)
ast_block *whileloop = new ast_block(ctx()); // while (whole >= base)
ast_block *nestwhile= new ast_block(ctx()); // while (whole >= newbase2)
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, "ln", TYPE_FLOAT);
size_t i;
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(power);
val->m_type_params.emplace_back(base);
2015-01-24 11:25:46 +00:00
block->m_locals.push_back(whole);
block->m_locals.push_back(nth);
block->m_locals.push_back(sign);
block->m_locals.push_back(eps);
block->m_locals.push_back(A_i);
block->m_locals.push_back(B_i);
block->m_locals.push_back(A_iminus1);
block->m_locals.push_back(B_iminus1);
/* sign = 1.0f; */
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
sign,
m_fold->m_imm_float[1]
)
);
/* eps = __builtin_epsilon(); */
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
eps,
ast_call::make(
2015-01-15 23:11:41 +00:00
ctx(),
func_self("__builtin_epsilon", "ln")
)
)
);
/*
* A_i = 1;
* B_i = 0;
* A_iminus1 = 0;
* B_iminus1 = 1;
*/
for (i = 0; i <= 1; i++) {
int j;
for (j = 1; j >= 0; j--) {
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
((j) ? ((i) ? B_iminus1 : A_i)
: ((i) ? A_iminus1 : B_i)),
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[j]
)
);
}
}
/*
* <plt1> = {
* power = 1.0f / power;
* sign *= -1.0f;
* }
* <blt1> = {
* base = 1.0f / base;
* sign *= -1.0f;
* }
*/
for (i = 0; i <= 1; i++) {
2015-01-24 11:25:46 +00:00
((i) ? blt1 : plt1)->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
((i) ? base : power),
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
((i) ? base : power)
)
)
);
2015-01-24 11:25:46 +00:00
plt1->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
sign,
m_fold->m_imm_float[2]
)
);
}
/*
* <plt1orblt1> = {
* if (power <= 0.0 || base <= 0.0f)
* return __builtin_nan();
* if (power < 1.0f)
* <plt1>
* if (base < 1.0f)
* <blt1>
* }
*/
2015-01-24 11:25:46 +00:00
plt1orblt1->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_OR,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LE,
2015-01-30 17:59:48 +00:00
power,
m_fold->m_imm_float[0]
),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LE,
2015-01-30 17:59:48 +00:00
base,
m_fold->m_imm_float[0]
)
),
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
ast_call::make(
2015-01-15 23:11:41 +00:00
ctx(),
func_self("__builtin_nan", "ln")
)
),
2015-01-15 20:18:33 +00:00
nullptr
)
);
for (i = 0; i <= 1; i++) {
2015-01-24 11:25:46 +00:00
plt1orblt1->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LT,
2015-01-30 17:59:48 +00:00
((i) ? base : power),
m_fold->m_imm_float[1]
),
2015-01-30 17:59:48 +00:00
((i) ? blt1 : plt1),
2015-01-15 20:18:33 +00:00
nullptr
)
);
}
2015-01-30 17:59:48 +00:00
block->m_exprs.push_back(plt1orblt1);
/* whole = power; */
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
whole,
power
)
);
/* nth = 0.0f; */
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
nth,
m_fold->m_imm_float[0]
)
);
/* base2 = base; */
2015-01-24 11:25:46 +00:00
whileloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
base2,
base
)
);
/* n2 = 1.0f; */
2015-01-24 11:25:46 +00:00
whileloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
n2,
m_fold->m_imm_float[1]
)
);
/* newbase2 = base2 * base2; */
2015-01-24 11:25:46 +00:00
whileloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
newbase2,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
base2,
base2
)
)
);
/* while loop locals */
2015-01-24 11:25:46 +00:00
whileloop->m_locals.push_back(base2);
whileloop->m_locals.push_back(n2);
whileloop->m_locals.push_back(newbase2);
/* base2 = newbase2; */
2015-01-24 11:25:46 +00:00
nestwhile->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
base2,
newbase2
)
);
/* n2 *= 2; */
2015-01-24 11:25:46 +00:00
nestwhile->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
n2,
m_fold->m_imm_float[3] /* 2.0f */
)
);
/* newbase2 *= newbase2; */
2015-01-24 11:25:46 +00:00
nestwhile->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
newbase2,
newbase2
)
);
/* while (whole >= newbase2) */
2015-01-24 11:25:46 +00:00
whileloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_loop(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-15 20:18:33 +00:00
nullptr,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_GE,
2015-01-30 17:59:48 +00:00
whole,
newbase2
),
false,
2015-01-15 20:18:33 +00:00
nullptr,
false,
2015-01-15 20:18:33 +00:00
nullptr,
2015-01-30 17:59:48 +00:00
nestwhile
)
);
/* whole /= base2; */
2015-01-24 11:25:46 +00:00
whileloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
whole,
base2
)
);
/* nth += n2; */
2015-01-24 11:25:46 +00:00
whileloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_binstore(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
nth,
n2
)
);
/* while (whole >= base) */
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_loop(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-15 20:18:33 +00:00
nullptr,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_GE,
2015-01-30 17:59:48 +00:00
whole,
base
),
false,
2015-01-15 20:18:33 +00:00
nullptr,
false,
2015-01-15 20:18:33 +00:00
nullptr,
2015-01-30 17:59:48 +00:00
whileloop
)
);
2015-01-24 11:25:46 +00:00
forloop->m_locals.push_back(b_iplus1);
forloop->m_locals.push_back(A_iplus1);
forloop->m_locals.push_back(B_iplus1);
/* b_iplus1 = nth; */
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
b_iplus1,
nth
)
);
/*
* A_iplus1 = b_iplus1 * A_i + A_iminus1;
* B_iplus1 = b_iplus1 * B_i + B_iminus1;
*/
for (i = 0; i <= 1; i++) {
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
((i) ? B_iplus1 : A_iplus1),
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
b_iplus1,
((i) ? B_i : A_i)
),
2015-01-30 17:59:48 +00:00
((i) ? B_iminus1 : A_iminus1)
)
)
);
}
/*
* A_iminus1 = A_i;
* B_iminus1 = B_i;
*/
for (i = 0; i <= 1; i++) {
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
((i) ? B_iminus1 : A_iminus1),
((i) ? B_i : A_i)
)
);
}
/*
* A_i = A_iplus1;
* B_i = B_iplus1;
*/
for (i = 0; i <= 1; i++) {
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
((i) ? B_i : A_i),
((i) ? B_iplus1 : A_iplus1)
)
);
}
/*
* if (whole <= 1.0f + eps)
* break;
*/
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_ifthen(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_LE,
2015-01-30 17:59:48 +00:00
whole,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_ADD_F,
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
eps
)
),
2015-01-30 17:59:48 +00:00
new ast_breakcont(
2015-01-15 23:11:41 +00:00
ctx(),
false,
0
),
2015-01-15 20:18:33 +00:00
nullptr
)
);
/*
* power = base;
* base = whole;
*/
for (i = 0; i <= 1; i++) {
2015-01-24 11:25:46 +00:00
forloop->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_store(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_STORE_F,
2015-01-30 17:59:48 +00:00
((i) ? base : power),
((i) ? whole : base)
)
);
}
/* add the for loop block */
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_loop(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-15 20:18:33 +00:00
nullptr,
/* for(; 1; ) ?? (can this be nullptr too?) */
2015-01-30 17:59:48 +00:00
m_fold->m_imm_float[1],
false,
2015-01-15 20:18:33 +00:00
nullptr,
false,
2015-01-15 20:18:33 +00:00
nullptr,
2015-01-30 17:59:48 +00:00
forloop
)
);
/* return sign * A_i / B_il */
2015-01-24 11:25:46 +00:00
block->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_MUL_F,
2015-01-30 17:59:48 +00:00
sign,
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
INSTR_DIV_F,
2015-01-30 17:59:48 +00:00
A_i,
B_i
)
)
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(block);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::log_variant(const char *name, float base) {
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_call *callln = ast_call::make(ctx(), func_self("__builtin_ln", name));
ast_value *arg1 = new ast_value(ctx(), "x", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, name, TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(arg1);
2015-01-30 17:59:48 +00:00
callln->m_params.push_back(arg1);
callln->m_params.push_back(m_fold->constgen_float(base, false));
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
callln
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::log_() {
return log_variant("log", 2.7182818284590452354);
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::log10_() {
return log_variant("log10", 10);
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::log2_() {
return log_variant("log2", 2);
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::logb_() {
/* FLT_RADIX == 2 for now */
2015-01-15 23:11:41 +00:00
return log_variant("log2", 2);
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::shift_variant(const char *name, size_t instr) {
/*
* float [shift] (float a, float b) {
* return floor(a [instr] pow(2, b));
*/
2015-01-15 23:11:41 +00:00
ast_value *val = nullptr;
2015-01-30 17:59:48 +00:00
ast_call *callpow = ast_call::make(ctx(), func_self("pow", name));
ast_call *callfloor = ast_call::make(ctx(), func_self("floor", name));
ast_value *a = new ast_value(ctx(), "a", TYPE_FLOAT);
ast_value *b = new ast_value(ctx(), "b", TYPE_FLOAT);
ast_block *body = new ast_block(ctx());
2015-01-15 23:11:41 +00:00
ast_function *func = value(&val, name, TYPE_FLOAT);
2015-01-30 17:59:48 +00:00
val->m_type_params.emplace_back(a);
val->m_type_params.emplace_back(b);
/* <callpow> = pow(2, b) */
2015-01-30 17:59:48 +00:00
callpow->m_params.push_back(m_fold->m_imm_float[3]);
callpow->m_params.push_back(b);
/* <callfloor> = floor(a [instr] <callpow>) */
2015-01-24 11:25:46 +00:00
callfloor->m_params.push_back(
2015-01-30 17:59:48 +00:00
new ast_binary(
2015-01-15 23:11:41 +00:00
ctx(),
instr,
2015-01-30 17:59:48 +00:00
a,
callpow
)
);
/* return <callfloor> */
2015-01-24 11:25:46 +00:00
body->m_exprs.push_back(
2015-01-30 17:59:48 +00:00
new ast_return(
2015-01-15 23:11:41 +00:00
ctx(),
2015-01-30 17:59:48 +00:00
callfloor
)
);
2015-01-24 11:25:46 +00:00
func->m_blocks.emplace_back(body);
2015-01-15 23:11:41 +00:00
reg(val, func);
2015-01-30 17:59:48 +00:00
return val;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::lshift() {
return shift_variant("lshift", INSTR_MUL_F);
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::rshift() {
return shift_variant("rshift", INSTR_DIV_F);
2013-08-14 03:03:49 +00:00
}
2015-01-15 23:11:41 +00:00
void intrin::error(const char *fmt, ...) {
va_list ap;
va_start(ap, fmt);
2015-01-15 23:11:41 +00:00
vcompile_error(ctx(), fmt, ap);
va_end(ap);
}
/* exposed */
2015-01-15 23:11:41 +00:00
ast_expression *intrin::debug_typestring() {
return (ast_expression*)0x1;
}
2015-01-15 23:11:41 +00:00
intrin::intrin(parser_t *parser)
: m_parser(parser)
2015-01-16 01:27:17 +00:00
, m_fold(&parser->m_fold)
2015-01-15 23:11:41 +00:00
{
static const intrin_func_t intrinsics[] = {
{&intrin::isfinite_, "__builtin_isfinite", "isfinite", 1},
{&intrin::isinf_, "__builtin_isinf", "isinf", 1},
{&intrin::isnan_, "__builtin_isnan", "isnan", 1},
{&intrin::isnormal_, "__builtin_isnormal", "isnormal", 1},
{&intrin::signbit_, "__builtin_signbit", "signbit", 1},
{&intrin::acosh_, "__builtin_acosh", "acosh", 1},
{&intrin::asinh_, "__builtin_asinh", "asinh", 1},
{&intrin::atanh_, "__builtin_atanh", "atanh", 1},
{&intrin::exp_, "__builtin_exp", "exp", 1},
{&intrin::exp2_, "__builtin_exp2", "exp2", 1},
{&intrin::expm1_, "__builtin_expm1", "expm1", 1},
{&intrin::mod_, "__builtin_mod", "mod", 2},
{&intrin::pow_, "__builtin_pow", "pow", 2},
{&intrin::fabs_, "__builtin_fabs", "fabs", 1},
{&intrin::log_, "__builtin_log", "log", 1},
{&intrin::log10_, "__builtin_log10", "log10", 1},
{&intrin::log2_, "__builtin_log2", "log2", 1},
{&intrin::logb_, "__builtin_logb", "logb", 1},
{&intrin::lshift, "__builtin_lshift", "", 2},
{&intrin::rshift, "__builtin_rshift", "", 2},
{&intrin::epsilon_, "__builtin_epsilon", "", 0},
{&intrin::nan_, "__builtin_nan", "", 0},
{&intrin::inf_, "__builtin_inf", "", 0},
{&intrin::ln_, "__builtin_ln", "", 2},
{&intrin::debug_typestring, "__builtin_debug_typestring", "", 0},
{&intrin::nullfunc, "#nullfunc", "", 0}
};
2015-01-15 08:01:26 +00:00
for (auto &it : intrinsics) {
2015-01-15 23:11:41 +00:00
m_intrinsics.push_back(it);
m_generated.push_back(nullptr);
2015-01-15 08:01:26 +00:00
}
}
2013-03-09 08:57:37 +00:00
2015-01-16 01:27:17 +00:00
ast_expression *intrin::do_fold(ast_value *val, ast_expression **exprs) {
2015-01-30 17:59:48 +00:00
if (!val || !val->m_name.length())
2015-01-15 20:18:33 +00:00
return nullptr;
2015-01-15 23:11:41 +00:00
static constexpr size_t kPrefixLength = 10; // "__builtin_"
for (auto &it : m_intrinsics) {
2015-01-30 17:59:48 +00:00
if (val->m_name == it.name)
2015-01-15 08:01:26 +00:00
return (vec_size(exprs) != it.args)
2015-01-15 20:18:33 +00:00
? nullptr
2015-01-30 17:59:48 +00:00
: m_fold->intrinsic(val->m_name.c_str() + kPrefixLength, exprs);
2015-01-15 08:01:26 +00:00
}
2015-01-15 20:18:33 +00:00
return nullptr;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::func_try(size_t offset, const char *compare) {
for (auto &it : m_intrinsics) {
const size_t index = &it - &m_intrinsics[0];
2015-01-15 08:01:26 +00:00
if (strcmp(*(char **)((char *)&it + offset), compare))
continue;
2015-01-15 23:11:41 +00:00
if (m_generated[index])
return m_generated[index];
return m_generated[index] = (this->*it.intrin_func_t::function)();
}
2015-01-15 08:01:26 +00:00
return nullptr;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::func_self(const char *name, const char *from) {
2015-01-15 06:35:56 +00:00
ast_expression *find;
/* try current first */
2015-01-24 11:25:46 +00:00
if ((find = parser_find_global(m_parser, name)) && ((ast_value*)find)->m_vtype == TYPE_FUNCTION)
2015-01-15 23:11:41 +00:00
for (auto &it : m_parser->functions)
2015-01-30 17:59:48 +00:00
if (reinterpret_cast<ast_value*>(find)->m_name.length() && it->m_name == reinterpret_cast<ast_value*>(find)->m_name && it->m_builtin < 0)
return find;
/* try name second */
2015-01-15 23:11:41 +00:00
if ((find = func_try(offsetof(intrin_func_t, name), name)))
return find;
/* try alias third */
2015-01-15 23:11:41 +00:00
if ((find = func_try(offsetof(intrin_func_t, alias), name)))
return find;
2013-11-23 15:58:03 +00:00
if (from) {
2015-01-15 23:11:41 +00:00
error("need function `%s', compiler depends on it for `__builtin_%s'", name, from);
return func_self("#nullfunc", nullptr);
2013-11-23 15:58:03 +00:00
}
2015-01-15 20:18:33 +00:00
return nullptr;
}
2015-01-15 23:11:41 +00:00
ast_expression *intrin::func(const char *name) {
return func_self(name, nullptr);
2013-03-09 08:57:37 +00:00
}