Implement double_to_scientific()

And use it for ast.h and asr.h
rebcabin · Jun 10, 2022 · 60cd916 · 60cd916
1 parent 7b2e083
commit 60cd916
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/grammar/asdl_cpp.py b/grammar/asdl_cpp.py
@@ -644,7 +644,7 @@ def visitField(self, field, cons, last):
                     self.emit('s.append(std::to_string(x.m_%s));' % field.name, 2)
             elif field.type == "float" and not field.seq and not field.opt:
                 self.emit('s.append("\\n" + indtd + "%s" + "%s=");' % (arr, field.name), 2)
-                self.emit('s.append(std::to_string(x.m_%s));' % field.name, 2)
+                self.emit('s.append(double_to_scientific(x.m_%s));' % field.name, 2)
             elif field.type == "bool" and not field.seq and not field.opt:
                 self.emit('s.append("\\n" + indtd + "%s" + "%s=");' % (arr, field.name), 2)
                 self.emit("if (x.m_%s) {" % field.name, 2)
@@ -1178,7 +1178,7 @@ def visitField(self, field, cons):
                 else:
                     self.emit('s.append(std::to_string(x.m_%s));' % field.name, 2)
             elif field.type == "float" and not field.seq and not field.opt:
-                self.emit('s.append(std::to_string(x.m_%s));' % field.name, 2)
+                self.emit('s.append(double_to_scientific(x.m_%s));' % field.name, 2)
             elif field.type == "bool" and not field.seq and not field.opt:
                 self.emit("if (x.m_%s) {" % field.name, 2)
                 self.emit(    's.append(".true.");', 3)

diff --git a/src/libasr/containers.h b/src/libasr/containers.h
@@ -180,6 +180,12 @@ struct Str {
 static_assert(std::is_standard_layout<Str>::value);
 static_assert(std::is_trivial<Str>::value);
 
+static inline std::string double_to_scientific(double x) {
+    std::string buf(30, '\0');
+    std::sprintf(&buf[0], "%e", x);
+    return buf;
+}
+
 } // namespace LFortran