Get wk_value index from loop indices

src/trapping/cfrfme.cpp

@@ -395,22 +395,15 @@ void frfme(string data_file, string output_path) {
 	  for (int j80 = jlmf; j80 <= jlml; j80++) {
 	    dcomplex *vec_w = new dcomplex[nkv * nkv]();
 	    dcomplex **w = new dcomplex*[nkv];
-	    // dcomplex *wk_local = new dcomplex[nlmmt]();
 	    for (int wi = 0; wi < nkv; wi++) w[wi] = vec_w + wi * nkv;
 	    dcomplex wk_value;
-	    int wk_index = 0;
-	    int nkvs = nkv * nkv;
 	    for (int jy50 = 0; jy50 < nkv; jy50++) {
 	      for (int jx50 = 0; jx50 < nkv; jx50++) {
-		// for (int wi = 0; wi < nlmmt; wi++) wk_local[wi] = tt1->wk[wk_index++];
-		// w[jx50][jy50] = wk_local[j80 - 1];
-		wk_value = tt1->wk[wk_index + j80 - 1];
-		wk_index += nlmmt;
+		wk_value = tt1->wk[jy50 * nkv * nlmmt + jx50 * nlmmt + j80 - 1];
 		w[jx50][jy50] = wk_value;
 	      } // jx50
 	    } // jy50 loop
 	    int ixyz = 0;
-	    for (int wj = 0; wj < nrvc; wj++) tfrfme->wsum[j80 - 1][wj] = cc0;
 	    for (int iz75 = 0; iz75 < nzv; iz75++) {
 	      double z = _zv[iz75] + frsh;
 	      for (int iy70 = 0; iy70 < nyv; iy70++) {
@@ -442,7 +435,6 @@ void frfme(string data_file, string output_path) {
 	    } // iz75 loop
 	    delete[] vec_w;
 	    delete[] w;
-	    // delete[] wk_local;
 	  } // j80 loop
 #ifdef USE_NVTX
 	  nvtxRangePop();