Merge pull request #7 from themaplelab/wala-feature/FloatLiterals

Support for Float literals

Author: karimhamdanali

include/swift/WALASupport/InstrKindInfoGetter.h

@@ -43,6 +43,7 @@ class InstrKindInfoGetter {
   jobject handleIntegerLiteralInst();
   jobject handleStringLiteralInst();
   jobject handleConstStringLiteralInst();
+  jobject handleFloatLiteralInst();
   jobject handleProjectBoxInst();
   jobject handleDebugValueInst();
   jobject handleFunctionRefInst();

include/swift/WALASupport/WALAWalker.h

@@ -55,6 +55,8 @@ class WALAIntegration {
   void print(jobject obj);
 
   jobject makePosition(int, int, int, int);
+
+  jobject makeBigDecimal(const char *, int);
 
   WALAIntegration(JNIEnv *, Exceptions &, const char *);
 };

lib/WALASupport/InstrKindInfoGetter.cpp

@@ -235,6 +235,40 @@ jobject InstrKindInfoGetter::handleIntegerLiteralInst() {
   return Node;
 }
 
+jobject InstrKindInfoGetter::handleFloatLiteralInst() {
+  if (outs != NULL) {
+    *outs << "<< FloatLiteralInst >>" << "\n";
+  }
+  jobject node = nullptr;
+  FloatLiteralInst* castInst = cast<FloatLiteralInst>(instr);
+  APFloat value = castInst->getValue();
+
+  if (&value.getSemantics() == &APFloat::IEEEsingle()) {
+    // To Float
+    node = (*wala)->makeConstant(value.convertToFloat());
+  }
+  else if (&value.getSemantics() == &APFloat::IEEEdouble()) {
+    // To Double
+    node = (*wala)->makeConstant(value.convertToDouble());
+  }
+  else if (value.isFinite()) {
+    // To BigDecimal
+    SmallVector<char, 128> buf;
+    value.toString(buf);
+    jobject bigDecimal = (*wala).makeBigDecimal(buf.data(), buf.size());
+    node = (*wala)->makeConstant(bigDecimal);
+  } 
+  else {
+    // Infinity or NaN, convert to double 
+    // as BigDecimal constructor cannot accept strings of these
+    bool APFLosesInfo;
+    value.convert(APFloat::IEEEdouble(), APFloat::rmNearestTiesToEven, &APFLosesInfo);
+    node = (*wala)->makeConstant(value.convertToDouble());
+  }
+  nodeMap->insert(std::make_pair(castInst, node));
+  return node;
+}
+
 jobject InstrKindInfoGetter::handleStringLiteralInst() {
   // ValueKind indentifier
   if (outs != NULL) {
@@ -840,7 +874,7 @@ SILInstructionKind InstrKindInfoGetter::get() {
     }
 
     case SILInstructionKind::FloatLiteralInst: {
-      *outs << "<< FloatLiteralInst >>" << "\n";
+      node = handleFloatLiteralInst();
       break;
     }
 

lib/WALASupport/WALAWalker.cpp

@@ -54,6 +54,21 @@ jobject WALAIntegration::makePosition(int fl, int fc, int ll, int lc) {
 	return result;
 }
 
+jobject WALAIntegration::makeBigDecimal(const char *strData, int strLen) {
+	char *safeData = strndup(strData, strLen);
+	jobject val = java_env->NewStringUTF(safeData);
+	delete safeData;
+	jclass bigDecimalCls = java_env->FindClass("java/math/BigDecimal");
+	THROW_ANY_EXCEPTION(cpp_ex);
+	jmethodID bigDecimalInit = java_env->GetMethodID(bigDecimalCls, 
+		"<init>", "(Ljava/lang/String;)V");
+	THROW_ANY_EXCEPTION(cpp_ex);
+	jobject bigDecimal = java_env->NewObject(bigDecimalCls, bigDecimalInit, val);
+	THROW_ANY_EXCEPTION(cpp_ex);
+	java_env->DeleteLocalRef(val);
+	return bigDecimal;
+}
+
 void WALAIntegration::print(jobject obj) {
 	print_object(java_env, obj);
 	THROW_ANY_EXCEPTION(cpp_ex);