MoCoRecon.h

// ====================================================================================================   
//                                                                                                        
//   SuPReMo: Surrogate Parameterised Respiratory Motion Model                                            
//            An implementation of the generalised motion modelling and image registration framework      
//                                                                                                        
//   Copyright (c) University College London (UCL). All rights reserved.                                  
//                                                                                                        
//   This software is distributed WITHOUT ANY WARRANTY; without even                                      
//   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR                                  
//   PURPOSE.                                                                                             
//                                                                                                        
//   See LICENSE.txt in the top level directory for details.                                              
//                                                                                                        
// ====================================================================================================   
 
 
 
#pragma once
 
//----------
// includes 
//----------
#include <memory>
#include <vector>
#include "nifti1_io.h"
#include "SupremoUtils.h"
 
 
 
//----------------------
// forward declarations
//----------------------
class CorrespondenceModel;
class ImageAcquisition;
 
 
 
 
class MoCoRecon
{
public:
  typedef float PrecisionType;  
  typedef std::vector< std::vector<PrecisionType> > SurrogateSignalType;
 
  MoCoRecon();
 
  virtual ~MoCoRecon();
 
  void SetCorrespondenceModel( const std::shared_ptr<CorrespondenceModel>& correspondenceModelIn );
  
  void SetImageAcquisition( const std::shared_ptr<ImageAcquisition>& imageAcquisitionIn );
 
  void SetSurrogateSignals( const SurrogateSignalType & surrSignalsIn );
  
  void SetDynamicImages( const std::vector<nifti_image*>& dynamicImagesIn );
  
  void SetReconstructionGeometryImage( nifti_image* reconstructionGeometryImageIn );
  
  virtual void Update() = 0;
 
  nifti_image* GetReconstructedImage();
 
 
  bool GetRepeatedUpdateChangesMCR() { return this->repeatedUpdateChangesMCRImage; };
 
  void CopyReconstructedImageContentsToImage( nifti_image* destinationImage );
 
protected:
  void AllocateReconstructedImage();
 
  void ClearReconstructedImage();
 
  std::shared_ptr<CorrespondenceModel> correspondenceModel;  
  std::shared_ptr<ImageAcquisition> imageAcquisition;        
  SurrogateSignalType surrogateSignals;                      
  std::vector<nifti_image*> dynamicImages;                   
  size_t numberOfDynamicImages;                              
  nifti_image* reconstructionGeometryImage;                  
  nifti_image* reconstructedImage;                           
  PrecisionType paddingValue;                                
  bool repeatedUpdateChangesMCRImage;                        
};