int main_ecalib(int argc, char* argv[])
{
long calsize[3] = { 24, 24, 24 };
int maps = 2;
bool one = false;
bool calcen = false;
bool print_svals = false;
struct ecalib_conf conf = ecalib_defaults;
const struct opt_s opts[] = {
OPT_FLOAT('t', &conf.threshold, "threshold", "This determined the size of the null-space."),
OPT_FLOAT('c', &conf.crop, "crop_value", "Crop the sensitivities if the eigenvalue is smaller than {crop_value}."),
OPT_VEC3('k', &conf.kdims, "ksize", "kernel size"),
OPT_VEC3('K', &conf.kdims, "", "()"),
OPT_VEC3('r', &calsize, "cal_size", "Limits the size of the calibration region."),
OPT_VEC3('R', &calsize, "", "()"),
OPT_INT('m', &maps, "maps", "Number of maps to compute."),
OPT_SET('S', &conf.softcrop, "create maps with smooth transitions (Soft-SENSE)."),
OPT_SET('W', &conf.weighting, "soft-weighting of the singular vectors."),
OPT_SET('I', &conf.intensity, "intensity correction"),
OPT_SET('1', &one, "perform only first part of the calibration"),
OPT_CLEAR('P', &conf.rotphase, "Do not rotate the phase with respect to the first principal component"),
OPT_CLEAR('O', &conf.orthiter, "()"),
OPT_FLOAT('b', &conf.perturb, "", "()"),
OPT_SET('V', &print_svals, "()"),
OPT_SET('C', &calcen, "()"),
OPT_SET('g', &conf.usegpu, "()"),
OPT_FLOAT('p', &conf.percentsv, "", "()"),
OPT_INT('n', &conf.numsv, "", "()"),
OPT_FLOAT('v', &conf.var, "variance", "Variance of noise in data."),
OPT_SET('a', &conf.automate, "Automatically pick thresholds."),
OPT_INT('d', &debug_level, "level", "Debug level"),
};
cmdline(&argc, argv, 2, 3, usage_str, help_str, ARRAY_SIZE(opts), opts);
if (-1. != conf.percentsv)
conf.threshold = -1.;
if (-1 != conf.numsv)
conf.threshold = -1.;
if (conf.automate) {
conf.crop = -1.;
conf.weighting = true;
}
if (conf.weighting) {
conf.numsv = -1.;
conf.threshold = 0;
conf.percentsv = -1.;
conf.orthiter = false;
}
int N = DIMS;
long ksp_dims[N];
complex float* in_data = load_cfl(argv[1], N, ksp_dims);
// assert((kdims[0] < calsize_ro) && (kdims[1] < calsize_ro) && (kdims[2] < calsize_ro));
// assert((ksp_dims[0] == 1) || (calsize_ro < ksp_dims[0]));
if (1 != ksp_dims[MAPS_DIM])
error("MAPS dimension is not of size one.\n");
long cal_dims[N];
complex float* cal_data = NULL;
if (!calcen) {
#ifdef USE_CC_EXTRACT_CALIB
cal_data = cc_extract_calib(cal_dims, calsize, ksp_dims, in_data);
#else
cal_data = extract_calib(cal_dims, calsize, ksp_dims, in_data, false);
#endif
} else {
for (int i = 0; i < 3; i++)
cal_dims[i] = (calsize[i] < ksp_dims[i]) ? calsize[i] : ksp_dims[i];
for (int i = 3; i < N; i++)
cal_dims[i] = ksp_dims[i];
cal_data = md_alloc(5, cal_dims, CFL_SIZE);
md_resize_center(5, cal_dims, cal_data, ksp_dims, in_data, CFL_SIZE);
}
for (int i = 0; i < 3; i++)
if (1 == ksp_dims[i])
conf.kdims[i] = 1;
long channels = cal_dims[3];
unsigned int K = conf.kdims[0] * conf.kdims[1] * conf.kdims[2] * channels;
float svals[K];
for (unsigned int i = 0; i < 3; i++)
if ((1 == cal_dims[i]) && (1 != ksp_dims[i]))
error("Calibration region not found!\n");
// To reproduce old results turn off rotation of phase.
// conf.rotphase = false;
// FIXME: we should scale the data
(conf.usegpu ? num_init_gpu : num_init)();
if ((conf.var < 0) && (conf.weighting || (conf.crop < 0)))
conf.var = estvar_calreg(conf.kdims, cal_dims, cal_data);
if (one) {
#if 0
long maps = out_dims[4];
assert(caldims[3] == out_dims[3]);
assert(maps <= channels);
#endif
long cov_dims[4];
calone_dims(&conf, cov_dims, channels);
complex float* imgcov = md_alloc(4, cov_dims, CFL_SIZE);
calone(&conf, cov_dims, imgcov, K, svals, cal_dims, cal_data);
complex float* out = create_cfl(argv[2], 4, cov_dims);
md_copy(4, cov_dims, out, imgcov, CFL_SIZE);
unmap_cfl(4, cov_dims, out);
// caltwo(crthr, out_dims, out_data, emaps, cov_dims, imgcov, NULL, NULL);
md_free(imgcov);
} else {
long out_dims[N];
long map_dims[N];
for (int i = 0; i < N; i++) {
out_dims[i] = 1;
map_dims[i] = 1;
if ((i < 3) && (1 < conf.kdims[i])) {
out_dims[i] = ksp_dims[i];
map_dims[i] = ksp_dims[i];
}
}
assert(maps <= ksp_dims[COIL_DIM]);
out_dims[COIL_DIM] = ksp_dims[COIL_DIM];
out_dims[MAPS_DIM] = maps;
map_dims[COIL_DIM] = 1;
map_dims[MAPS_DIM] = maps;
const char* emaps_file = NULL;
if (4 == argc)
emaps_file = argv[3];
complex float* out_data = create_cfl(argv[2], N, out_dims);
complex float* emaps = (emaps_file ? create_cfl : anon_cfl)(emaps_file, N, map_dims);
calib(&conf, out_dims, out_data, emaps, K, svals, cal_dims, cal_data);
unmap_cfl(N, out_dims, out_data);
unmap_cfl(N, map_dims, emaps);
}
if (print_svals) {
for (unsigned int i = 0; i < K; i++)
printf("SVALS %d %f\n", i, svals[i]);
}
printf("Done.\n");
unmap_cfl(N, ksp_dims, in_data);
md_free(cal_data);
return 0;
}
int main_nufft(int argc, char* argv[])
{
bool adjoint = false;
bool inverse = false;
bool use_gpu = false;
bool precond = false;
bool dft = false;
struct nufft_conf_s conf = nufft_conf_defaults;
struct iter_conjgrad_conf cgconf = iter_conjgrad_defaults;
long coilim_vec[3] = { 0 };
float lambda = 0.;
const struct opt_s opts[] = {
OPT_SET('a', &adjoint, "adjoint"),
OPT_SET('i', &inverse, "inverse"),
OPT_VEC3('d', &coilim_vec, "x:y:z", "dimensions"),
OPT_VEC3('D', &coilim_vec, "", "()"),
OPT_SET('t', &conf.toeplitz, "Toeplitz embedding for inverse NUFFT"),
OPT_SET('c', &precond, "Preconditioning for inverse NUFFT"),
OPT_FLOAT('l', &lambda, "lambda", "l2 regularization"),
OPT_UINT('m', &cgconf.maxiter, "", "()"),
OPT_SET('s', &dft, "DFT"),
};
cmdline(&argc, argv, 3, 3, usage_str, help_str, ARRAY_SIZE(opts), opts);
long coilim_dims[DIMS] = { 0 };
md_copy_dims(3, coilim_dims, coilim_vec);
// Read trajectory
long traj_dims[DIMS];
complex float* traj = load_cfl(argv[1], DIMS, traj_dims);
assert(3 == traj_dims[0]);
num_init();
if (inverse || adjoint) {
long ksp_dims[DIMS];
const complex float* ksp = load_cfl(argv[2], DIMS, ksp_dims);
assert(1 == ksp_dims[0]);
assert(md_check_compat(DIMS, ~(PHS1_FLAG|PHS2_FLAG), ksp_dims, traj_dims));
md_copy_dims(DIMS - 3, coilim_dims + 3, ksp_dims + 3);
if (0 == md_calc_size(DIMS, coilim_dims)) {
estimate_im_dims(DIMS, coilim_dims, traj_dims, traj);
debug_printf(DP_INFO, "Est. image size: %ld %ld %ld\n", coilim_dims[0], coilim_dims[1], coilim_dims[2]);
}
complex float* img = create_cfl(argv[3], DIMS, coilim_dims);
md_clear(DIMS, coilim_dims, img, CFL_SIZE);
const struct linop_s* nufft_op;
if (!dft)
nufft_op = nufft_create(DIMS, ksp_dims, coilim_dims, traj_dims, traj, NULL, conf, use_gpu);
else
nufft_op = nudft_create(DIMS, FFT_FLAGS, ksp_dims, coilim_dims, traj_dims, traj);
if (inverse) {
const struct operator_s* precond_op = NULL;
if (conf.toeplitz && precond)
precond_op = nufft_precond_create(nufft_op);
lsqr(DIMS, &(struct lsqr_conf){ lambda }, iter_conjgrad, CAST_UP(&cgconf),
nufft_op, NULL, coilim_dims, img, ksp_dims, ksp, precond_op);
if (conf.toeplitz && precond)
operator_free(precond_op);
} else {
