Fix compound assignment type casts

Author: joaotgouveia

cpp2rust/converter/converter.cpp

@@ -2274,7 +2274,8 @@ void Converter::ConvertBinaryOperator(clang::BinaryOperator *expr) {
   if (auto *cmpd_assign_op =
           llvm::dyn_cast<clang::CompoundAssignOperator>(expr);
       expr->isCompoundAssignmentOp() &&
-      lhs_type != cmpd_assign_op->getComputationResultType()) {
+      GetUnsafeTypeAsString(lhs_type) !=
+          GetUnsafeTypeAsString(cmpd_assign_op->getComputationResultType())) {
     auto computation_result_type = cmpd_assign_op->getComputationResultType();
     if (IsUnsignedArithOp(cmpd_assign_op)) {
       Convert(lhs);

cpp2rust/converter/models/converter_refcount.cpp

@@ -109,6 +109,14 @@ ConverterRefCount::PushUnboxedIfSimple::PushUnboxedIfSimple(
                                        : ConversionKind::FullRefCount);
 }
 
+std::string
+ConverterRefCount::GetSafeTypeAsString(clang::QualType qual_type) const {
+  std::string type_as_string;
+  ConverterRefCount converter(type_as_string, ctx_);
+  converter.Convert(qual_type);
+  return std::string(Trim(type_as_string));
+}
+
 std::string ConverterRefCount::BoxType(std::string &&str) const {
   switch (getConversionKind()) {
   case ConversionKind::Unboxed:
@@ -1362,7 +1370,8 @@ void ConverterRefCount::ConvertBinaryOperator(clang::BinaryOperator *expr) {
   std::string_view opcode_as_string = expr->getOpcodeStr();
 
   if (auto *assign = llvm::dyn_cast<clang::CompoundAssignOperator>(expr);
-      assign && lhs_type != assign->getComputationResultType()) {
+      assign && GetSafeTypeAsString(lhs_type) !=
+                    GetSafeTypeAsString(assign->getComputationResultType())) {
     auto computation_result_type = assign->getComputationResultType();
     StrCat(keyword::kLet, "rhs_0", token::kAssign);
     if (IsUnsignedArithOp(assign)) {

cpp2rust/converter/models/converter_refcount.h

@@ -242,6 +242,8 @@ class ConverterRefCount final : public Converter {
   std::string ConvertPtrType(clang::QualType type);
   std::string ConvertPointeeType(clang::QualType ptr_type) override;
 
+  std::string GetSafeTypeAsString(clang::QualType qual_type) const;
+
   /// The kind of conversion that should be performed.
   enum class ConversionKind : uint8_t {
     Unboxed,

tests/unit/out/refcount/auto.rs

@@ -19,8 +19,7 @@ fn main_0() -> i32 {
     let sum: Value<i32> = Rc::new(RefCell::new(0));
     'loop_: for mut elem in v.as_pointer() as Ptr<i32> {
         let elem: Value<i32> = Rc::new(RefCell::new(elem.read().clone()));
-        let rhs_0 = (((*sum.borrow()) as i32) + (*elem.borrow())) as i32;
-        (*sum.borrow_mut()) = rhs_0;
+        (*sum.borrow_mut()) += (*elem.borrow());
     }
     return (*sum.borrow());
 }

tests/unit/out/refcount/compound_assign_ref.rs

@@ -12,8 +12,10 @@ pub fn main() {
 fn main_0() -> i32 {
     let v: Value<Vec<i32>> = Rc::new(RefCell::new(Vec::new()));
     (*v.borrow_mut()).push(10);
-    let rhs_0 = ((((v.as_pointer() as Ptr<i32>).read()) as i32) + 5) as i32;
-    (v.as_pointer() as Ptr<i32>).write(rhs_0);
+    {
+        let _ptr = (v.as_pointer() as Ptr<i32>).clone();
+        _ptr.write(_ptr.read() + 5)
+    };
     assert!((((v.as_pointer() as Ptr<i32>).read()) == 15));
     return ((v.as_pointer() as Ptr<i32>).read());
 }

tests/unit/out/refcount/foreach_mut.rs

@@ -24,8 +24,10 @@ fn main_0() -> i32 {
         (*sum.borrow_mut()) += (*x.borrow());
     }
     'loop_: for mut x in v1.as_pointer() as Ptr<i32> {
-        let rhs_0 = (((x.read()) as i32) + 10) as i32;
-        x.write(rhs_0);
+        {
+            let _ptr = x.clone();
+            _ptr.write(_ptr.read() + 10)
+        };
     }
     'loop_: for mut x in v1.as_pointer() as Ptr<i32> {
         let __rhs = (x.read());
@@ -49,8 +51,10 @@ fn main_0() -> i32 {
     }
     'loop_: for mut p in v2.as_pointer() as Ptr<Ptr<i32>> {
         let p: Value<Ptr<i32>> = Rc::new(RefCell::new(p.read().clone()));
-        let rhs_0 = ((((*p.borrow()).read()) as i32) + 5) as i32;
-        (*p.borrow()).write(rhs_0);
+        {
+            let _ptr = (*p.borrow()).clone();
+            _ptr.write(_ptr.read() + 5)
+        };
     }
     'loop_: for mut p in v2.as_pointer() as Ptr<Ptr<i32>> {
         let p: Value<Ptr<i32>> = Rc::new(RefCell::new(p.read().clone()));

tests/unit/out/refcount/unique_ptr_deref_ops.rs

@@ -11,12 +11,9 @@ pub fn main() {
 }
 fn main_0() -> i32 {
     let p: Value<Option<Value<i32>>> = Rc::new(RefCell::new(Some(Rc::new(RefCell::new(10)))));
-    let rhs_0 = (((*(*p.borrow()).as_ref().unwrap().borrow()) as i32) + 5) as i32;
-    (*(*p.borrow_mut()).as_ref().unwrap().borrow_mut()) = rhs_0;
-    let rhs_0 = (((*(*p.borrow()).as_ref().unwrap().borrow()) as i32) - 3) as i32;
-    (*(*p.borrow_mut()).as_ref().unwrap().borrow_mut()) = rhs_0;
-    let rhs_0 = (((*(*p.borrow()).as_ref().unwrap().borrow()) as i32) * 2) as i32;
-    (*(*p.borrow_mut()).as_ref().unwrap().borrow_mut()) = rhs_0;
+    (*(*p.borrow_mut()).as_ref().unwrap().borrow_mut()) += 5;
+    (*(*p.borrow_mut()).as_ref().unwrap().borrow_mut()) -= 3;
+    (*(*p.borrow_mut()).as_ref().unwrap().borrow_mut()) *= 2;
     let q: Value<Option<Value<i32>>> = Rc::new(RefCell::new(Some(Rc::new(RefCell::new(1)))));
     let sum: Value<i32> = Rc::new(RefCell::new(
         ((*(*p.borrow()).as_ref().unwrap().borrow()) + (*(*q.borrow()).as_ref().unwrap().borrow())),

tests/unit/out/unsafe/auto.rs

@@ -21,7 +21,7 @@ unsafe fn main_0() -> i32 {
     let mut sum: i32 = 0;
     'loop_: for elem in 0..(v.len()) {
         let mut elem = v[elem].clone();
-        sum = ((sum as i32) + elem) as i32;
+        sum += elem;
     }
     return sum;
 }

tests/unit/out/unsafe/compound_assign_ref.rs

@@ -14,7 +14,7 @@ pub fn main() {
 unsafe fn main_0() -> i32 {
     let mut v: Vec<i32> = Vec::new();
     v.push(10);
-    ((v).first_mut().unwrap()) = ((((v).first_mut().unwrap()) as i32) + 5) as i32;
+    (*((v).first_mut().unwrap())) += 5;
     assert!(((*((v).first_mut().unwrap())) == (15)));
     return (*((v).first_mut().unwrap()));
 }

tests/unit/out/unsafe/foreach_mut.rs

@@ -27,7 +27,7 @@ unsafe fn main_0() -> i32 {
     }
     'loop_: for x in 0..(v1.len()) {
         let mut x = v1.as_mut_ptr().add(x);
-        (*x) = (((*x) as i32) + 10) as i32;
+        (*x) += 10;
     }
     'loop_: for x in 0..(v1.len()) {
         let mut x = v1.as_ptr().add(x);
@@ -47,7 +47,7 @@ unsafe fn main_0() -> i32 {
     }
     'loop_: for p in 0..(v2.len()) {
         let mut p = v2[p].clone();
-        (*p) = (((*p) as i32) + 5) as i32;
+        (*p) += 5;
     }
     'loop_: for p in 0..(v2.len()) {
         let mut p = v2[p].clone();