Handle flonum for if, while, do, !, ?:, || and &&

codegen.c

@@ -150,6 +150,17 @@ static void store(Type *ty) {
 }
 
 static void cmp_zero(Type *ty) {
+  switch (ty->kind) {
+  case TY_FLOAT:
+    println("  xorps %%xmm1, %%xmm1");
+    println("  ucomiss %%xmm1, %%xmm0");
+    return;
+  case TY_DOUBLE:
+    println("  xorpd %%xmm1, %%xmm1");
+    println("  ucomisd %%xmm1, %%xmm0");
+    return;
+  }
+
   if (is_integer(ty) && ty->size <= 4)
     println("  cmp $0, %%eax");
   else
@@ -337,7 +348,7 @@ static void gen_expr(Node *node) {
   case ND_COND: {
     int c = count();
     gen_expr(node->cond);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->cond->ty);
     println("  je .L.else.%d", c);
     gen_expr(node->then);
     println("  jmp .L.end.%d", c);
@@ -348,7 +359,7 @@ static void gen_expr(Node *node) {
   }
   case ND_NOT:
     gen_expr(node->lhs);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->lhs->ty);
     println("  sete %%al");
     println("  movzx %%al, %%rax");
     return;
@@ -359,10 +370,10 @@ static void gen_expr(Node *node) {
   case ND_LOGAND: {
     int c = count();
     gen_expr(node->lhs);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->lhs->ty);
     println("  je .L.false.%d", c);
     gen_expr(node->rhs);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->rhs->ty);
     println("  je .L.false.%d", c);
     println("  mov $1, %%rax");
     println("  jmp .L.end.%d", c);
@@ -374,10 +385,10 @@ static void gen_expr(Node *node) {
   case ND_LOGOR: {
     int c = count();
     gen_expr(node->lhs);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->lhs->ty);
     println("  jne .L.true.%d", c);
     gen_expr(node->rhs);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->rhs->ty);
     println("  jne .L.true.%d", c);
     println("  mov $0, %%rax");
     println("  jmp .L.end.%d", c);
@@ -579,7 +590,7 @@ static void gen_stmt(Node *node) {
   case ND_IF: {
     int c = count();
     gen_expr(node->cond);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->cond->ty);
     println("  je  .L.else.%d", c);
     gen_stmt(node->then);
     println("  jmp .L.end.%d", c);
@@ -596,7 +607,7 @@ static void gen_stmt(Node *node) {
     println(".L.begin.%d:", c);
     if (node->cond) {
       gen_expr(node->cond);
-      println("  cmp $0, %%rax");
+      cmp_zero(node->cond->ty);
       println("  je %s", node->brk_label);
     }
     gen_stmt(node->then);
@@ -613,7 +624,7 @@ static void gen_stmt(Node *node) {
     gen_stmt(node->then);
     println("%s:", node->cont_label);
     gen_expr(node->cond);
-    println("  cmp $0, %%rax");
+    cmp_zero(node->cond->ty);
     println("  jne .L.begin.%d", c);
     println("%s:", node->brk_label);
     return;

test/control.c

@@ -68,6 +68,21 @@ int main() {
   ASSERT(7, ({ int i=0; int j=0; do { j++; } while (i++ < 6); j; }));
   ASSERT(4, ({ int i=0; int j=0; int k=0; do { if (++j > 3) break; continue; k++; } while (1); j; }));
 
+  ASSERT(0, 0.0 && 0.0);
+  ASSERT(0, 0.0 && 0.1);
+  ASSERT(0, 0.3 && 0.0);
+  ASSERT(1, 0.3 && 0.5);
+  ASSERT(0, 0.0 || 0.0);
+  ASSERT(1, 0.0 || 0.1);
+  ASSERT(1, 0.3 || 0.0);
+  ASSERT(1, 0.3 || 0.5);
+  ASSERT(5, ({ int x; if (0.0) x=3; else x=5; x; }));
+  ASSERT(3, ({ int x; if (0.1) x=3; else x=5; x; }));
+  ASSERT(5, ({ int x=5; if (0.0) x=3; x; }));
+  ASSERT(3, ({ int x=5; if (0.1) x=3; x; }));
+  ASSERT(10, ({ double i=10.0; int j=0; for (; i; i--, j++); j; }));
+  ASSERT(10, ({ double i=10.0; int j=0; do j++; while(--i); j; }));
+
   printf("OK\n");
   return 0;
 }

test/float.c

@@ -80,6 +80,14 @@ int main() {
   ASSERT(0, 0.0/0.0 > 0);
   ASSERT(0, 0.0/0.0 >= 0);
 
+  ASSERT(0, !3.);
+  ASSERT(1, !0.);
+  ASSERT(0, !3.f);
+  ASSERT(1, !0.f);
+
+  ASSERT(5, 0.0 ? 3 : 5);
+  ASSERT(3, 1.2 ? 3 : 5);
+
   printf("OK\n");
   return 0;
 }