X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Ftypes%2Fnum_exp.c;fp=src%2Ftypes%2Fnum_exp.c;h=91ff887311979e625942ada7fc16828cc8fe3797;hp=0577acf88160d4beabd9e31a5758c2708ed61ad5;hb=ba906e039ffe9e57842ce5135aa43efa00b8a4c6;hpb=764f3f36f003401f6334ced63356451fb5c6c642

diff --git a/src/types/num_exp.c b/src/types/num_exp.c
index 0577acf..91ff887 100644
--- a/src/types/num_exp.c
+++ b/src/types/num_exp.c
@@ -32,6 +32,19 @@
 #include "call.h"
 #include "calc.h"
 
+struct psi_num_exp *psi_num_exp_init_ternary(token_t op,
+		struct psi_num_exp *cond, struct psi_num_exp *truthy,
+		struct psi_num_exp *falsy)
+{
+	struct psi_num_exp *exp = calloc(1, sizeof(*exp));
+
+	exp->op = op;
+	exp->data.t.cond = cond;
+	exp->data.t.truthy = truthy;
+	exp->data.t.falsy = falsy;
+
+	return exp;
+}
 
 struct psi_num_exp *psi_num_exp_init_binary(token_t op,
 		struct psi_num_exp *lhs, struct psi_num_exp *rhs)
@@ -129,6 +142,12 @@ struct psi_num_exp *psi_num_exp_copy(struct psi_num_exp *exp)
 		cpy->data.b.rhs = psi_num_exp_copy(exp->data.b.rhs);
 		break;
 
+	case PSI_T_IIF:
+		cpy->data.t.cond = psi_num_exp_copy(exp->data.t.cond);
+		cpy->data.t.truthy = psi_num_exp_copy(exp->data.t.truthy);
+		cpy->data.t.falsy = psi_num_exp_copy(exp->data.t.falsy);
+		break;
+
 	default:
 		assert(0);
 	}
@@ -187,6 +206,12 @@ void psi_num_exp_free(struct psi_num_exp **c_ptr)
 			psi_num_exp_free(&c->data.b.rhs);
 			break;
 
+		case PSI_T_IIF:
+			psi_num_exp_free(&c->data.t.cond);
+			psi_num_exp_free(&c->data.t.truthy);
+			psi_num_exp_free(&c->data.t.falsy);
+			break;
+
 		default:
 			assert(0);
 		}
@@ -253,6 +278,9 @@ static inline const char *psi_num_exp_op_tok(token_t op)
 	case PSI_T_LCHEVR:
 		return "<";
 
+	case PSI_T_IIF:
+		return "?";
+
 	default:
 		assert(0);
 	}
@@ -303,6 +331,14 @@ void psi_num_exp_dump(int fd, struct psi_num_exp *exp)
 		psi_num_exp_dump(fd, exp->data.b.rhs);
 		break;
 
+	case PSI_T_IIF:
+		psi_num_exp_dump(fd, exp->data.t.cond);
+		dprintf(fd, " ? ");
+		psi_num_exp_dump(fd, exp->data.t.truthy);
+		dprintf(fd, " : ");
+		psi_num_exp_dump(fd, exp->data.t.falsy);
+		break;
+
 	default:
 		assert(0);
 	}
@@ -349,6 +385,7 @@ bool psi_num_exp_validate(struct psi_data *data, struct psi_num_exp *exp,
 
 		case PSI_T_CAST:
 		case PSI_T_LPAREN:
+		case PSI_T_IIF:
 			break;
 
 		case PSI_T_PIPE:
@@ -425,6 +462,12 @@ bool psi_num_exp_validate(struct psi_data *data, struct psi_num_exp *exp,
 	case PSI_T_MODULO:
 		return psi_num_exp_validate(data, exp->data.b.lhs, impl, cb_decl, current_let, current_set, current_enum)
 				&& psi_num_exp_validate(data, exp->data.b.rhs, impl, cb_decl, current_let, current_set, current_enum);
+
+	case PSI_T_IIF:
+		return psi_num_exp_validate(data, exp->data.t.cond, impl, cb_decl, current_let, current_set, current_enum)
+				&& psi_num_exp_validate(data, exp->data.t.truthy, impl, cb_decl, current_let, current_set, current_enum)
+				&& psi_num_exp_validate(data, exp->data.t.falsy, impl, cb_decl, current_let, current_set, current_enum);
+
 	default:
 		assert(0);
 	}
@@ -535,6 +578,20 @@ static void psi_num_exp_reduce(struct psi_num_exp *exp, struct psi_plist **outpu
 		psi_num_exp_reduce(exp->data.u, &output, &input, frame, defs);
 		break;
 
+	case PSI_T_IIF:
+		{
+			impl_val cond_val = {0};
+			token_t cond_typ = psi_num_exp_exec(exp->data.t.cond, &cond_val, frame, defs);
+
+			psi_calc_bool_not(cond_typ, &cond_val, 0, NULL, &cond_val);
+			if (cond_val.u8) {
+				psi_num_exp_reduce(exp->data.t.falsy, &output, &input, frame, defs);
+			} else {
+				psi_num_exp_reduce(exp->data.t.truthy, &output, &input, frame, defs);
+			}
+		}
+		break;
+
 	default:
 		psi_num_exp_reduce(exp->data.b.lhs, &output, &input, frame, defs);
 		while (psi_plist_top(input, &entry)) {