NonseparableSpatiallyVaryingFilters.h

Go to the documentation of this file.

//
//
/*
    Copyright (C) 2000- 2007, Hammersmith Imanet
    SPDX-License-Identifier: Apache-2.0
    See STIR/LICENSE.txt for details
*/

#ifndef __stir_NonseparableSpatiallyVaryingFilters_H__
#define __stir_NonseparableSpatiallyVaryingFilters_H__

#include "stir_experimental/ModifiedInverseAverigingArrayFilter.h"
#include "stir/DiscretisedDensity.h"
#include "stir/DataProcessor.h"
#include "stir/RegisteredParsingObject.h"
#include "stir/VectorWithOffset.h"
#include "stir/ProjData.h"
#include "stir_experimental/ArrayFilter2DUsingConvolution.h"
#include "stir_experimental/ArrayFilter3DUsingConvolution.h"

#include "stir/shared_ptr.h"

START_NAMESPACE_STIR

// TODO!! remove define

#define num_dimensions 3

template <typename elemT>
class NonseparableSpatiallyVaryingFilters
    : public RegisteredParsingObject<NonseparableSpatiallyVaryingFilters<elemT>,
                                     DataProcessor<DiscretisedDensity<num_dimensions, elemT>>,
                                     DataProcessor<DiscretisedDensity<num_dimensions, elemT>>>
{
private:
  typedef RegisteredParsingObject<NonseparableSpatiallyVaryingFilters<elemT>,
                                  DataProcessor<DiscretisedDensity<num_dimensions, elemT>>,
                                  DataProcessor<DiscretisedDensity<num_dimensions, elemT>>>
      base_type;

public:
  static const char* const registered_name;

  NonseparableSpatiallyVaryingFilters();

  NonseparableSpatiallyVaryingFilters(string proj_data_filename,
                                      string attenuation_proj_data_filename,
                                      const Array<2, float>& filter_coefficients,
                                      shared_ptr<ProjData> proj_data_ptr,
                                      shared_ptr<ProjData> attenuation_proj_data_ptr,
                                      DiscretisedDensity<3, float>* initial_image,
                                      DiscretisedDensity<3, float>* sensitivity_image,
                                      DiscretisedDensity<3, float>* precomputed_coefficients_image,
                                      DiscretisedDensity<3, float>* normalised_bck_image,
                                      int mask_size,
                                      int num_dim);

private:
  int mask_size;
  Array<2, float> filter_coefficients_for_parsing;
  mutable Array<2, float> filter_coefficients;

  shared_ptr<ProjData> proj_data_ptr;
  string proj_data_filename;

  shared_ptr<ProjData> attenuation_proj_data_ptr;
  string attenuation_proj_data_filename;

  DiscretisedDensity<3, float>* initial_image;
  string initial_image_filename;

  DiscretisedDensity<3, float>* sensitivity_image;
  string sensitivity_image_filename;

  DiscretisedDensity<3, float>* precomputed_coefficients_image;
  string precomputed_coefficients_filename;

  DiscretisedDensity<3, float>* normalised_bck_image;
  string normalised_bck_filename;
  int num_dim;
  int number_of_coefficients_before_padding;

  Succeeded virtual_set_up(const DiscretisedDensity<num_dimensions, elemT>& density);
  // new
  void virtual_apply(DiscretisedDensity<num_dimensions, elemT>& out_density,
                     const DiscretisedDensity<num_dimensions, elemT>& in_density) const;
  void virtual_apply(DiscretisedDensity<num_dimensions, elemT>& density) const;
  void precalculate_filter_coefficients_2D(
      VectorWithOffset<VectorWithOffset<VectorWithOffset<shared_ptr<ArrayFilter2DUsingConvolution<float>>>>>&
          all_filter_coefficients,
      DiscretisedDensity<3, elemT>* in_density) const;

  virtual void set_defaults();
  virtual void initialise_keymap();

  virtual bool post_processing();
};

#undef num_dimensions

END_NAMESPACE_STIR

#endif