void Deconvolution::performSimpleClean(size_t currentChannelIndex, bool& reachedMajorThreshold, size_t majorIterationNr, PolarizationEnum polarization)
{
deconvolution::SingleImageSet
residualImage(_imgWidth*_imgHeight, *_imageAllocator),
modelImage(_imgWidth*_imgHeight, *_imageAllocator);
ImageBufferAllocator::Ptr psfImage;
_imageAllocator->Allocate(_imgWidth*_imgHeight, psfImage);
_residualImages->Load(residualImage.Data(), polarization, currentChannelIndex, false);
_modelImages->Load(modelImage.Data(), polarization, currentChannelIndex, false);
_psfImages->Load(psfImage.data(), _psfPolarization, currentChannelIndex, false);
std::vector<double*> psfs(1, psfImage.data());
TypedDeconvolutionAlgorithm<deconvolution::SingleImageSet>& tAlgorithm =
static_cast<TypedDeconvolutionAlgorithm<deconvolution::SingleImageSet>&>(*_cleanAlgorithm);
tAlgorithm.ExecuteMajorIteration(residualImage, modelImage, psfs, _imgWidth, _imgHeight, reachedMajorThreshold);
_modelImages->Store(modelImage.Data(), polarization, currentChannelIndex, false);
_residualImages->Store(residualImage.Data(), polarization, currentChannelIndex, false);
}
void Deconvolution::performJoinedPolClean(size_t currentChannelIndex, bool& reachedMajorThreshold, size_t majorIterationNr)
{
typename JoinedClean<deconvolution::PolarizedImageSet<PolCount>>::ImageSet
modelSet(_imgWidth*_imgHeight, *_imageAllocator),
residualSet(_imgWidth*_imgHeight, *_imageAllocator);
ImageBufferAllocator::Ptr psfImage;
_imageAllocator->Allocate(_imgWidth*_imgHeight, psfImage);
_psfImages->Load(psfImage.data(), _psfPolarization, currentChannelIndex, false);
modelSet.Load(*_modelImages, _polarizations, currentChannelIndex);
residualSet.Load(*_residualImages, _polarizations, currentChannelIndex);
std::vector<double*> psfImages(1, psfImage.data());
static_cast<TypedDeconvolutionAlgorithm<deconvolution::PolarizedImageSet<PolCount>>&>(*_cleanAlgorithm).ExecuteMajorIteration(residualSet, modelSet, psfImages, _imgWidth, _imgHeight, reachedMajorThreshold);
psfImage.reset();
modelSet.Store(*_modelImages, _polarizations, currentChannelIndex);
residualSet.Store(*_residualImages, _polarizations, currentChannelIndex);
}
void MoreSane::ExecuteMajorIteration(double* dataImage, double* modelImage, const double* psfImage, size_t width, size_t height)
{
if(_iterationNumber!=0)
{
Logger::Info << "Convolving model with psf...\n";
ImageBufferAllocator::Ptr preparedPsf;
_allocator->Allocate(width*height, preparedPsf);
FFTConvolver::PrepareKernel(preparedPsf.data(), psfImage, width, height);
FFTConvolver::ConvolveSameSize(_fftwManager, modelImage, preparedPsf.data(), width, height);
Logger::Info << "Adding model back to residual...\n";
for(size_t i=0; i!=width*height; ++i)
dataImage[i] += modelImage[i];
}
std::ostringstream outputStr;
outputStr << _prefixName << "-tmp-moresaneoutput" << _iterationNumber;
const std::string
dirtyName(_prefixName + "-tmp-moresaneinput-dirty.fits"),
psfName(_prefixName + "-tmp-moresaneinput-psf.fits"),
maskName(_prefixName + "-tmp-moresaneinput-mask.fits"),
outputName(outputStr.str());
FitsWriter writer;
writer.SetImageDimensions(width, height);
if(this->_cleanMask != 0)
writer.WriteMask(maskName, _cleanMask);
writer.Write(dirtyName, dataImage);
writer.Write(psfName, psfImage);
std::ostringstream commandLine;
commandLine
<< "time python \"" << _moresaneLocation << "\" ";
if(!_allowNegativeComponents)
commandLine << "-ep ";
if(this->_cleanMask != 0)
commandLine << "-m \"" << maskName + "\" ";
if(!_moresaneArguments.empty())
commandLine << _moresaneArguments<< ' ';
commandLine << "\"" << dirtyName << "\" \"" << psfName << "\" \"" << outputName << '\"';
if(!_moresaneSigmaLevels.empty()) {
commandLine << " -sl " << _moresaneSigmaLevels[std::min(_iterationNumber,_moresaneSigmaLevels.size()-1)] << " ";
}
// TODO should use Application::Run().
Logger::Info << "Running: " << commandLine.str() << '\n';
int pid = vfork();
switch (pid) {
case -1: // Error
throw std::runtime_error("Could not vfork() new process for executing MoreSane");
case 0: // Child
execl("/bin/sh", "sh", "-c", commandLine.str().c_str(), NULL);
_exit(127);
}
// Wait for process to terminate
int pStatus;
do {
int pidReturn;
do {
pidReturn = waitpid(pid, &pStatus, 0);
} while (pidReturn == -1 && errno == EINTR);
} while(!WIFEXITED(pStatus) && !WIFSIGNALED(pStatus));
if(WIFEXITED(pStatus))
{
// all good
// const int exitStatus = WEXITSTATUS(pStatus);
} else {
throw std::runtime_error("MoreSane returned an error");
}
FitsReader modelReader(outputName+"_model.fits");
modelReader.Read(modelImage);
FitsReader residualReader(outputName+"_residual.fits");
residualReader.Read(dataImage);
unlink(dirtyName.c_str());
unlink(psfName.c_str());
unlink(maskName.c_str());
unlink((outputName+"_model.fits").c_str());
unlink((outputName+"_residual.fits").c_str());
}