tmp

Author: ubaidsk

src/libasr/codegen/asr_to_c_cpp.h

@@ -1422,28 +1422,28 @@ PyMODINIT_FUNC PyInit_lpython_module_)" + fn_name + R"((void) {
         int n_args, bool check_for_bounds, bool is_unbounded_pointer_to_data) {
         std::string prod = "1";
         std::string idx = "0";
-        std::reverse(m_args.begin(), m_args.end());
-        std::reverse(diminfo.begin(), diminfo.end());
-        if (!is_unbounded_pointer_to_data) {
-            for (int i = 0; i < (int)diminfo.size() - 1; i += 2) {
-                std::swap(diminfo[i], diminfo[i + 1]);
+        if (is_unbounded_pointer_to_data) {
+            for (int r = 0; r < n_args; r++) {
+                std::string curr_llvm_idx = m_args[r];
+                std::string lval = diminfo[r];
+                curr_llvm_idx = "(" + curr_llvm_idx + " - " + lval + ")";
+                if( check_for_bounds ) {
+                    // check_single_element(curr_llvm_idx, arr); TODO: To be implemented
+                }
+                idx = "(" + idx + " + " + "(" + curr_llvm_idx + ")" + ")";
             }
+            return idx;
         }
-        for( int r = 0, r1 = 0; r < n_args; r++ ) {
+        for( int r = n_args - 1, r1 = 2 * n_args - 1; r >= 0; r--, r1 -= 2) {
             std::string curr_llvm_idx = m_args[r];
-            std::string lval = diminfo[r1];
+            std::string lval = diminfo[r1 - 1];
             curr_llvm_idx = "(" + curr_llvm_idx + " - " + lval + ")";
             if( check_for_bounds ) {
                 // check_single_element(curr_llvm_idx, arr); TODO: To be implemented
             }
             idx = "(" + idx + " + " + "(" + prod + " * " + curr_llvm_idx + ")" + ")";
-            if (is_unbounded_pointer_to_data) {
-                r1 += 1;
-            } else {
-                std::string dim_size = diminfo[r1 + 1];
-                r1 += 2;
-                prod = "(" + prod + " * " + dim_size + ")";
-            }
+            std::string dim_size = diminfo[r1];
+            prod = "(" + prod + " * " + dim_size + ")";
         }
         return idx;
     }
@@ -1496,24 +1496,22 @@ PyMODINIT_FUNC PyInit_lpython_module_)" + fn_name + R"((void) {
             update_target_desc += indent + target_desc + "->offset = 0;\n"; // offset not available yet
 
             std::string target_dim_des_array = target_desc + "->dims";
-            int j = 0;
-            for( int i = 0; i < value_rank; i++ ) {
-                std::string stride = "1";
-                for (int r = (i * 2) + 1; r < (int)diminfo.size(); r += 2) {
-                    stride = "(" + stride + "*" + diminfo[r] + ")";
-                }
+            int j = target_rank - 1;
+            int r = (int)diminfo.size() - 1;
+            std::string stride = "1";
+            for( int i = value_rank - 1; i >= 0; i-- ) {
                 if( ds[i] != "" ) {
                     std::string dim_length = "(((" + ubs[i] + " - " + lbs[i] + ")" + "/" + ds[i] + ") + 1)";
                     std::string target_dim_des = target_dim_des_array + "[" + std::to_string(j) + "]";
                     update_target_desc += indent + target_dim_des + ".stride = " + stride + ";\n";
                     update_target_desc += indent + target_dim_des + ".lower_bound = 1;\n";
                     update_target_desc += indent + target_dim_des + ".length = " + dim_length + ";\n";
-                    j++;
+                    j--;
                 }
-                // stride = "(" + stride + "*" + diminfo[r] + ")";
-                // r += 2;
+                stride = "(" + stride + "*" + diminfo[r] + ")";
+                r -= 2;
             }
-            LCOMPILERS_ASSERT(j == target_rank);
+            LCOMPILERS_ASSERT(j == -1);
             update_target_desc += indent + target_desc + "->n_dims = " + std::to_string(target_rank) + ";\n";
             src = update_target_desc;
     }