From a7158b10aaf5b31471183556c437d3ec43c98dbc Mon Sep 17 00:00:00 2001
From: Abdul Zreika <azre6702@uni.sydney.edu.au>
Date: Thu, 22 Oct 2020 16:23:36 +1100
Subject: [PATCH] Removed uses of old get type in ast2ram main.

---
 src/ast2ram/AstToRamTranslator.cpp | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/src/ast2ram/AstToRamTranslator.cpp b/src/ast2ram/AstToRamTranslator.cpp
index 67d5df12d35..6cfca6a54bc 100644
--- a/src/ast2ram/AstToRamTranslator.cpp
+++ b/src/ast2ram/AstToRamTranslator.cpp
@@ -243,9 +243,8 @@ Own<ram::Expression> AstToRamTranslator::translateValue(const ast::Argument* arg
         }
 
         Own<ram::Expression> visitNumericConstant(const ast::NumericConstant& c) override {
-            assert(c.getType().has_value() && "At this points all constants should have type.");
-
-            switch (*c.getType()) {
+            assert(!polyAnalysis->hasInvalidType(&c) && "constant should have valid type");
+            switch (polyAnalysis->getOverloadedType(&c)) {
                 case ast::NumericConstant::Type::Int:
                     return mk<ram::SignedConstant>(RamSignedFromString(c.getConstant(), nullptr, 0));
                 case ast::NumericConstant::Type::Uint:
@@ -400,8 +399,8 @@ RamDomain AstToRamTranslator::getConstantRamRepresentation(const ast::Constant&
     } else if (isA<ast::NilConstant>(&constant)) {
         return 0;
     } else if (auto* numConstant = dynamic_cast<const ast::NumericConstant*>(&constant)) {
-        assert(numConstant->getType().has_value());
-        switch (*numConstant->getType()) {
+        assert(!polyAnalysis->hasInvalidType(numConstant) && "constant should have valid type");
+        switch (polyAnalysis->getOverloadedType(numConstant)) {
             case ast::NumericConstant::Type::Int:
                 return RamSignedFromString(numConstant->getConstant(), nullptr, 0);
             case ast::NumericConstant::Type::Uint:
@@ -417,7 +416,7 @@ Own<ram::Expression> AstToRamTranslator::translateConstant(ast::Constant const&
     auto const rawConstant = getConstantRamRepresentation(c);
 
     if (auto* const c_num = dynamic_cast<const ast::NumericConstant*>(&c)) {
-        switch (*c_num->getType()) {
+        switch (polyAnalysis->getOverloadedType(c_num)) {
             case ast::NumericConstant::Type::Int: return mk<ram::SignedConstant>(rawConstant);
             case ast::NumericConstant::Type::Uint: return mk<ram::UnsignedConstant>(rawConstant);
             case ast::NumericConstant::Type::Float: return mk<ram::FloatConstant>(rawConstant);