% given idx_sum, which is ordered by col by col
% output row_id & col_id
function [row_id, col_id] = row_col_idx(nely, idx_sum)
    row_id = mod(idx_sum-1, nely) + 1;
    col_id = floor((idx_sum-1)/nely) + 1;
end