BinNormalisationPETFromComponents.h

Go to the documentation of this file.

//
//
/*
    Copyright (C) 2022, University College London
    This file is part of STIR.

    SPDX-License-Identifier: Apache-2.0

    See STIR/LICENSE.txt for details
*/

#ifndef __stir_recon_buildblock_BinNormalisationPETFromComponents_H__
#define __stir_recon_buildblock_BinNormalisationPETFromComponents_H__

#include "stir/recon_buildblock/BinNormalisation.h"
#include "stir/ML_norm.h"
#include "stir/ProjData.h"
#include "stir/shared_ptr.h"

START_NAMESPACE_STIR

class BinNormalisationPETFromComponents : public BinNormalisation
{
private:
  using base_type = BinNormalisation;

public:
  // need to have this here as it's a pure virtual in BinNormalisation.
  // it is normally implemented via RegisteredParsingObject, but
  // we currently do no derive from that.
  std::string get_registered_name() const override { return this->registered_name; }


  BinNormalisationPETFromComponents();


  bool is_trivial() const override;


  Succeeded set_up(const shared_ptr<const ExamInfo>& exam_info_sptr, const shared_ptr<const ProjDataInfo>&) override;

  // import all apply/undo methods from base-class (we'll override some below)
  using base_type::apply;
  using base_type::undo;


  void apply(RelatedViewgrams<float>& viewgrams) const override;


  void undo(RelatedViewgrams<float>& viewgrams) const override;

  float get_bin_efficiency(const Bin& bin) const override;

#if 0
  virtual shared_ptr<ProjData> get_norm_proj_data_sptr() const;
#endif


  void
  allocate(shared_ptr<const ProjDataInfo>, bool do_eff, bool do_geo, bool do_block = false, bool do_symmetry_per_block = false);

  DetectorEfficiencies& crystal_efficiencies()
  {
    return efficiencies;
  }
  GeoData3D& geometric_factors()
  {
    return geo_data;
  }
  BlockData3D& block_factors()
  {
    return block_data;
  }


  void set_defaults() override;

protected:
  DetectorEfficiencies efficiencies;
  GeoData3D geo_data;
  BlockData3D block_data;


  bool has_crystal_efficiencies() const;
  bool has_geometric_factors() const;
  bool has_block_factors() const;

private:
  static const char* const registered_name;
  shared_ptr<ProjDataInMemory> invnorm_proj_data_sptr;
  bool _already_allocated;
  bool _is_trivial;
  void create_proj_data();
};

END_NAMESPACE_STIR

#endif