void
GenericReconCartesianGrappaGadget::perform_calib(IsmrmrdReconBit &recon_bit, ReconObjType &recon_obj, size_t e) {
size_t RO = recon_bit.data_.data_.get_size(0);
size_t E1 = recon_bit.data_.data_.get_size(1);
size_t E2 = recon_bit.data_.data_.get_size(2);
hoNDArray<std::complex<float> > &src = recon_obj.ref_calib_;
hoNDArray<std::complex<float> > &dst = recon_obj.ref_calib_dst_;
size_t ref_RO = src.get_size(0);
size_t ref_E1 = src.get_size(1);
size_t ref_E2 = src.get_size(2);
size_t srcCHA = src.get_size(3);
size_t ref_N = src.get_size(4);
size_t ref_S = src.get_size(5);
size_t ref_SLC = src.get_size(6);
size_t dstCHA = dst.get_size(3);
recon_obj.unmixing_coeff_.create(RO, E1, E2, srcCHA, ref_N, ref_S, ref_SLC);
recon_obj.gfactor_.create(RO, E1, E2, 1, ref_N, ref_S, ref_SLC);
Gadgetron::clear(recon_obj.unmixing_coeff_);
Gadgetron::clear(recon_obj.gfactor_);
if (acceFactorE1_[e] <= 1 && acceFactorE2_[e] <= 1) {
Gadgetron::conjugate(recon_obj.coil_map_, recon_obj.unmixing_coeff_);
} else {
// allocate buffer for kernels
size_t kRO = grappa_kSize_RO.value();
size_t kNE1 = grappa_kSize_E1.value();
size_t kNE2 = grappa_kSize_E2.value();
size_t convKRO(1), convKE1(1), convKE2(1);
bool fitItself = this->downstream_coil_compression.value();
if (E2 > 1) {
std::vector<int> kE1, oE1;
std::vector<int> kE2, oE2;
grappa3d_kerPattern(kE1, oE1, kE2, oE2, convKRO, convKE1, convKE2, (size_t) acceFactorE1_[e],
(size_t) acceFactorE2_[e], kRO, kNE1, kNE2, fitItself);
} else {
std::vector<int> kE1, oE1;
Gadgetron::grappa2d_kerPattern(kE1, oE1, convKRO, convKE1, (size_t) acceFactorE1_[e], kRO, kNE1,
fitItself);
recon_obj.kernelIm_.create(RO, E1, 1, srcCHA, dstCHA, ref_N, ref_S, ref_SLC);
}
recon_obj.kernel_.create(convKRO, convKE1, convKE2, srcCHA, dstCHA, ref_N, ref_S, ref_SLC);
Gadgetron::clear(recon_obj.kernel_);
Gadgetron::clear(recon_obj.kernelIm_);
long long num = ref_N * ref_S * ref_SLC;
long long ii;
// only allow this for loop openmp if num>1 and 2D recon
#pragma omp parallel for default(none) private(ii) shared(src, dst, recon_obj, e, num, ref_N, ref_S, ref_RO, ref_E1, ref_E2, RO, E1, E2, dstCHA, srcCHA, convKRO, convKE1, convKE2, kRO, kNE1, kNE2, fitItself) if(num>1)
for (ii = 0; ii < num; ii++) {
size_t slc = ii / (ref_N * ref_S);
size_t s = (ii - slc * ref_N * ref_S) / (ref_N);
size_t n = ii - slc * ref_N * ref_S - s * ref_N;
std::stringstream os;
os << "n" << n << "_s" << s << "_slc" << slc << "_encoding_" << e;
std::string suffix = os.str();
std::complex<float> *pSrc = &(src(0, 0, 0, 0, n, s, slc));
hoNDArray<std::complex<float> > ref_src(ref_RO, ref_E1, ref_E2, srcCHA, pSrc);
std::complex<float> *pDst = &(dst(0, 0, 0, 0, n, s, slc));
hoNDArray<std::complex<float> > ref_dst(ref_RO, ref_E1, ref_E2, dstCHA, pDst);
// -----------------------------------
if (E2 > 1) {
hoNDArray<std::complex<float> > ker(convKRO, convKE1, convKE2, srcCHA, dstCHA,
&(recon_obj.kernel_(0, 0, 0, 0, 0, n, s, slc)));
if (fitItself)
{
Gadgetron::grappa3d_calib_convolution_kernel(ref_src, ref_dst, (size_t)acceFactorE1_[e],
(size_t)acceFactorE2_[e], grappa_reg_lamda.value(),
grappa_calib_over_determine_ratio.value(), kRO, kNE1,
kNE2, ker);
}
else
{
Gadgetron::grappa3d_calib_convolution_kernel(ref_src, ref_src, (size_t)acceFactorE1_[e],
(size_t)acceFactorE2_[e], grappa_reg_lamda.value(),
grappa_calib_over_determine_ratio.value(), kRO, kNE1,
kNE2, ker);
}
//if (!debug_folder_full_path_.empty())
//{
// gt_exporter_.export_array_complex(ker, debug_folder_full_path_ + "convKer3D_" + suffix);
//}
hoNDArray<std::complex<float> > coilMap(RO, E1, E2, dstCHA,
&(recon_obj.coil_map_(0, 0, 0, 0, n, s, slc)));
hoNDArray<std::complex<float> > unmixC(RO, E1, E2, srcCHA,
&(recon_obj.unmixing_coeff_(0, 0, 0, 0, n, s, slc)));
hoNDArray<float> gFactor(RO, E1, E2, 1, &(recon_obj.gfactor_(0, 0, 0, 0, n, s, slc)));
Gadgetron::grappa3d_unmixing_coeff(ker, coilMap, (size_t) acceFactorE1_[e],
(size_t) acceFactorE2_[e], unmixC, gFactor);
//if (!debug_folder_full_path_.empty())
//{
// gt_exporter_.export_array_complex(unmixC, debug_folder_full_path_ + "unmixC_3D_" + suffix);
//}
//if (!debug_folder_full_path_.empty())
//{
// gt_exporter_.export_array(gFactor, debug_folder_full_path_ + "gFactor_3D_" + suffix);
//}
} else {
hoNDArray<std::complex<float> > acsSrc(ref_RO, ref_E1, srcCHA,
const_cast< std::complex<float> *>(ref_src.begin()));
hoNDArray<std::complex<float> > acsDst(ref_RO, ref_E1, dstCHA,
const_cast< std::complex<float> *>(ref_dst.begin()));
hoNDArray<std::complex<float> > convKer(convKRO, convKE1, srcCHA, dstCHA,
&(recon_obj.kernel_(0, 0, 0, 0, 0, n, s, slc)));
hoNDArray<std::complex<float> > kIm(RO, E1, srcCHA, dstCHA,
&(recon_obj.kernelIm_(0, 0, 0, 0, 0, n, s, slc)));
if (fitItself)
{
Gadgetron::grappa2d_calib_convolution_kernel(acsSrc, acsDst, (size_t)acceFactorE1_[e],
grappa_reg_lamda.value(), kRO, kNE1, convKer);
}
else
{
Gadgetron::grappa2d_calib_convolution_kernel(acsSrc, acsSrc, (size_t)acceFactorE1_[e],
grappa_reg_lamda.value(), kRO, kNE1, convKer);
}
Gadgetron::grappa2d_image_domain_kernel(convKer, RO, E1, kIm);
/*if (!debug_folder_full_path_.empty())
{
gt_exporter_.export_array_complex(convKer, debug_folder_full_path_ + "convKer_" + suffix);
}
if (!debug_folder_full_path_.empty())
{
gt_exporter_.export_array_complex(kIm, debug_folder_full_path_ + "kIm_" + suffix);
}*/
hoNDArray<std::complex<float> > coilMap(RO, E1, dstCHA,
&(recon_obj.coil_map_(0, 0, 0, 0, n, s, slc)));
hoNDArray<std::complex<float> > unmixC(RO, E1, srcCHA,
&(recon_obj.unmixing_coeff_(0, 0, 0, 0, n, s, slc)));
hoNDArray<float> gFactor;
Gadgetron::grappa2d_unmixing_coeff(kIm, coilMap, (size_t) acceFactorE1_[e], unmixC, gFactor);
memcpy(&(recon_obj.gfactor_(0, 0, 0, 0, n, s, slc)), gFactor.begin(),
gFactor.get_number_of_bytes());
/*if (!debug_folder_full_path_.empty())
{
gt_exporter_.export_array_complex(unmixC, debug_folder_full_path_ + "unmixC_" + suffix);
}
if (!debug_folder_full_path_.empty())
{
gt_exporter_.export_array(gFactor, debug_folder_full_path_ + "gFactor_" + suffix);
}*/
}
// -----------------------------------
}
}
}
void MultiChannelCartesianGrappaReconGadget::perform_calib(IsmrmrdReconBit& recon_bit, ReconObjType& recon_obj, size_t e)
{
try
{
size_t RO = recon_bit.data_.data_.get_size(0);
size_t E1 = recon_bit.data_.data_.get_size(1);
size_t E2 = recon_bit.data_.data_.get_size(2);
hoNDArray< std::complex<float> >& src = recon_obj.ref_calib_;
hoNDArray< std::complex<float> >& dst = recon_obj.ref_calib_;
size_t ref_RO = src.get_size(0);
size_t ref_E1 = src.get_size(1);
size_t ref_E2 = src.get_size(2);
size_t srcCHA = src.get_size(3);
size_t ref_N = src.get_size(4);
size_t ref_S = src.get_size(5);
size_t ref_SLC = src.get_size(6);
size_t dstCHA = dst.get_size(3);
recon_obj.unmixing_coeff_.create(RO, E1, E2, srcCHA, ref_N, ref_S, ref_SLC);
recon_obj.gfactor_.create(RO, E1, E2, 1, ref_N, ref_S, ref_SLC);
Gadgetron::clear(recon_obj.unmixing_coeff_);
Gadgetron::clear(recon_obj.gfactor_);
if (acceFactorE1_[e] <= 1 && acceFactorE2_[e] <= 1)
{
Gadgetron::conjugate(recon_obj.coil_map_, recon_obj.unmixing_coeff_);
}
else
{
// allocate buffer for kernels
size_t kRO = grappa_kSize_RO.value();
size_t kNE1 = grappa_kSize_E1.value();
size_t kNE2 = grappa_kSize_E2.value();
size_t convKRO(1), convKE1(1), convKE2(1);
if (E2 > 1)
{
std::vector<int> kE1, oE1;
std::vector<int> kE2, oE2;
bool fitItself = true;
grappa3d_kerPattern(kE1, oE1, kE2, oE2, convKRO, convKE1, convKE2, (size_t)acceFactorE1_[e], (size_t)acceFactorE2_[e], kRO, kNE1, kNE2, fitItself);
}
else
{
std::vector<int> kE1, oE1;
bool fitItself = true;
Gadgetron::grappa2d_kerPattern(kE1, oE1, convKRO, convKE1, (size_t)acceFactorE1_[e], kRO, kNE1, fitItself);
recon_obj.kernelIm_.create(RO, E1, 1, srcCHA, dstCHA, ref_N, ref_S, ref_SLC);
}
recon_obj.kernel_.create(convKRO, convKE1, convKE2, srcCHA, dstCHA, ref_N, ref_S, ref_SLC);
Gadgetron::clear(recon_obj.kernel_);
Gadgetron::clear(recon_obj.kernelIm_);
long long num = ref_N*ref_S*ref_SLC;
long long ii;
#pragma omp parallel for default(none) private(ii) shared(src, dst, recon_obj, e, num, ref_N, ref_S, ref_RO, ref_E1, ref_E2, RO, E1, E2, dstCHA, srcCHA, convKRO, convKE1, convKE2, kRO, kNE1, kNE2) if(num>1)
for (ii = 0; ii < num; ii++)
{
size_t slc = ii / (ref_N*ref_S);
size_t s = (ii - slc*ref_N*ref_S) / (ref_N);
size_t n = ii - slc*ref_N*ref_S - s*ref_N;
std::stringstream os;
os << "n" << n << "_s" << s << "_slc" << slc << "_encoding_" << e;
std::string suffix = os.str();
std::complex<float>* pSrc = &(src(0, 0, 0, 0, n, s, slc));
hoNDArray< std::complex<float> > ref_src(ref_RO, ref_E1, ref_E2, srcCHA, pSrc);
std::complex<float>* pDst = &(dst(0, 0, 0, 0, n, s, slc));
hoNDArray< std::complex<float> > ref_dst(ref_RO, ref_E1, ref_E2, dstCHA, pDst);
// -----------------------------------
if (E2 > 1)
{
hoNDArray< std::complex<float> > ker(convKRO, convKE1, convKE2, srcCHA, dstCHA, &(recon_obj.kernel_(0, 0, 0, 0, 0, n, s, slc)));
Gadgetron::grappa3d_calib_convolution_kernel(ref_src, ref_dst, (size_t)acceFactorE1_[e], (size_t)acceFactorE2_[e], grappa_reg_lamda.value(), grappa_calib_over_determine_ratio.value(), kRO, kNE1, kNE2, ker);
hoNDArray< std::complex<float> > coilMap(RO, E1, E2, dstCHA, &(recon_obj.coil_map_(0, 0, 0, 0, n, s, slc)));
hoNDArray< std::complex<float> > unmixC(RO, E1, E2, srcCHA, &(recon_obj.unmixing_coeff_(0, 0, 0, 0, n, s, slc)));
hoNDArray<float> gFactor(RO, E1, E2, 1, &(recon_obj.gfactor_(0, 0, 0, 0, n, s, slc)));
Gadgetron::grappa3d_unmixing_coeff(ker, coilMap, (size_t)acceFactorE1_[e], (size_t)acceFactorE2_[e], unmixC, gFactor);
}
else
{
hoNDArray< std::complex<float> > acsSrc(ref_RO, ref_E1, srcCHA, const_cast< std::complex<float>*>(ref_src.begin()));
hoNDArray< std::complex<float> > acsDst(ref_RO, ref_E1, dstCHA, const_cast< std::complex<float>*>(ref_dst.begin()));
hoNDArray< std::complex<float> > convKer(convKRO, convKE1, srcCHA, dstCHA, &(recon_obj.kernel_(0, 0, 0, 0, 0, n, s, slc)));
hoNDArray< std::complex<float> > kIm(RO, E1, srcCHA, dstCHA, &(recon_obj.kernelIm_(0, 0, 0, 0, 0, n, s, slc)));
Gadgetron::grappa2d_calib_convolution_kernel(acsSrc, acsDst, (size_t)acceFactorE1_[e], grappa_reg_lamda.value(), kRO, kNE1, convKer);
Gadgetron::grappa2d_image_domain_kernel(convKer, RO, E1, kIm);
hoNDArray< std::complex<float> > coilMap(RO, E1, dstCHA, &(recon_obj.coil_map_(0, 0, 0, 0, n, s, slc)));
hoNDArray< std::complex<float> > unmixC(RO, E1, srcCHA, &(recon_obj.unmixing_coeff_(0, 0, 0, 0, n, s, slc)));
hoNDArray<float> gFactor;
Gadgetron::grappa2d_unmixing_coeff(kIm, coilMap, (size_t)acceFactorE1_[e], unmixC, gFactor);
memcpy(&(recon_obj.gfactor_(0, 0, 0, 0, n, s, slc)), gFactor.begin(), gFactor.get_number_of_bytes());
}
// -----------------------------------
}
}
}
catch (...)
{
GADGET_THROW("Errors happened in MultiChannelCartesianGrappaReconGadget::perform_calib(...) ... ");
}
}
