static void prox_lineq_apply(const operator_data_t* _data, float mu, complex float* dst, const complex float* src)
{
UNUSED(mu);
struct prox_lineq_data* pdata = CAST_DOWN(prox_lineq_data, _data);
const struct linop_s* op = pdata->op;
linop_normal(op, linop_domain(op)->N, linop_domain(op)->dims, pdata->tmp, src);
md_zsub(linop_domain(op)->N, linop_domain(op)->dims, dst, src, pdata->tmp);
md_zadd(linop_domain(op)->N, linop_domain(op)->dims, dst, dst, pdata->adj);
}
void data_consistency(const long dims[DIMS], complex float* dst, const complex float* pattern, const complex float* kspace1, const complex float* kspace2)
{
assert(1 == dims[MAPS_DIM]);
long strs[DIMS];
long dims1[DIMS];
long strs1[DIMS];
md_select_dims(DIMS, ~COIL_FLAG, dims1, dims);
md_calc_strides(DIMS, strs1, dims1, CFL_SIZE);
md_calc_strides(DIMS, strs, dims, CFL_SIZE);
complex float* tmp = md_alloc_sameplace(DIMS, dims, CFL_SIZE, dst);
md_zmul2(DIMS, dims, strs, tmp, strs, kspace2, strs1, pattern);
md_zsub(DIMS, dims, tmp, kspace2, tmp);
md_zfmac2(DIMS, dims, strs, tmp, strs, kspace1, strs1, pattern);
md_copy(DIMS, dims, dst, tmp, CFL_SIZE);
md_free(tmp);
}
// [RING] Caclucate intersection points
static void calc_intersections(unsigned int Nint, unsigned int N, unsigned int no_intersec_sp, float dist[2][Nint], long idx[2][Nint], const float angles[N], const long kc_dims[DIMS], const complex float* kc, const int ROI)
{
long spoke_dims[DIMS];
md_copy_dims(DIMS, spoke_dims, kc_dims);
spoke_dims[PHS2_DIM] = 1;
complex float* spoke_i = md_alloc(DIMS, spoke_dims, CFL_SIZE);
complex float* spoke_j = md_alloc(DIMS, spoke_dims, CFL_SIZE);
long pos_i[DIMS] = { 0 };
long pos_j[DIMS] = { 0 };
long coilPixel_dims[DIMS];
md_copy_dims(DIMS, coilPixel_dims, spoke_dims);
coilPixel_dims[PHS1_DIM] = 1;
complex float* coilPixel_l = md_alloc(DIMS, coilPixel_dims, CFL_SIZE);
complex float* coilPixel_m = md_alloc(DIMS, coilPixel_dims, CFL_SIZE);
complex float* diff = md_alloc(DIMS, coilPixel_dims, CFL_SIZE);
long diff_rss_dims[DIMS];
md_copy_dims(DIMS, diff_rss_dims, coilPixel_dims);
diff_rss_dims[COIL_DIM] = 1;
diff_rss_dims[PHS1_DIM] = 1;
complex float* diff_rss = md_alloc(DIMS, diff_rss_dims, CFL_SIZE);
long pos_l[DIMS] = { 0 };
long pos_m[DIMS] = { 0 };
// Array to store indices of spokes that build an angle close to pi/2 with the current spoke
long intersec_sp[no_intersec_sp];
for (unsigned int i = 0; i < no_intersec_sp; i++)
intersec_sp[i] = -1;
// Boundaries for spoke comparison
int myROI = ROI;
myROI += (ROI % 2 == 0) ? 1 : 0; // make odd
int low = 0;
int high = myROI - 1;
int count = 0;
// Intersection determination
for (unsigned int i = 0; i < N; i++) {
pos_i[PHS2_DIM] = i;
md_slice(DIMS, PHS2_FLAG, pos_i, kc_dims, spoke_i, kc, CFL_SIZE);
find_intersec_sp(no_intersec_sp, intersec_sp, i, N, angles);
for (unsigned int j = 0; j < no_intersec_sp; j++) {
pos_j[PHS2_DIM] = intersec_sp[j];
md_slice(DIMS, PHS2_FLAG, pos_j, kc_dims, spoke_j, kc, CFL_SIZE);
idx[0][i * no_intersec_sp + j] = i;
idx[1][i * no_intersec_sp + j] = intersec_sp[j];
// Elementwise rss comparisson
float rss = FLT_MAX;
for (int l = low; l <= high; l++) {
pos_l[PHS1_DIM] = l;
md_copy_block(DIMS, pos_l, coilPixel_dims, coilPixel_l, spoke_dims, spoke_i, CFL_SIZE);
for (int m = low; m <= high; m++) {
pos_m[PHS1_DIM] = m;
md_copy_block(DIMS, pos_m, coilPixel_dims, coilPixel_m, spoke_dims, spoke_j, CFL_SIZE);
md_zsub(DIMS, coilPixel_dims, diff, coilPixel_l, coilPixel_m);
md_zrss(DIMS, coilPixel_dims, PHS1_FLAG|COIL_FLAG, diff_rss, diff);
if (cabsf(diff_rss[0]) < rss) { // New minimum found
rss = cabsf(diff_rss[0]);
dist[0][i * no_intersec_sp + j] = (l + 1/2 - ROI/2);
dist[1][i * no_intersec_sp + j] = (m + 1/2 - ROI/2);
}
}
}
count++;
}
}
#ifdef RING_PAPER
// Print projection angles and corresponding offsets to files
const char* idx_out = "projangle.txt";
FILE* fp = fopen(idx_out, "w");
const char* d_out = "offset.txt";
FILE* fp1 = fopen(d_out, "w");
for (unsigned int i = 0; i < N; i++) {
for (unsigned int j = 0; j < no_intersec_sp; j++) {
fprintf(fp, "%f \t %f\n", angles[idx[0][i * no_intersec_sp + j]], angles[idx[1][i * no_intersec_sp + j]]);
fprintf(fp1, "%f \t %f\n", dist[0][i * no_intersec_sp + j], dist[1][i * no_intersec_sp + j]);
}
}
fclose(fp);
fclose(fp1);
#endif
md_free(spoke_i);
md_free(spoke_j);
md_free(coilPixel_l);
md_free(coilPixel_m);
md_free(diff);
md_free(diff_rss);
}
