CListModeDataPENN.h

Go to the documentation of this file.

/*
    Copyright (C) 2020-2022 University of Pennsylvania
    This file is part of STIR.
 
    SPDX-License-Identifier: Apache-2.0
 
    See STIR/LICENSE.txt for details
*/
 
#ifndef __stir_listmode_CListModeDataPENN_H__
#define __stir_listmode_CListModeDataPENN_H__
 
#include "stir/listmode/CListModeData.h"
#include "stir/listmode/CListRecordPENN.h"
#include "stir/IO/InputStreamWithRecordsFromUPENN.h"
#include "stir/shared_ptr.h"
 
START_NAMESPACE_STIR
class CListModeDataPENN : public CListModeData
{
public:
  CListModeDataPENN(const std::string& listmode_filename_prefix);
 
  virtual std::string get_name() const;
 
  virtual shared_ptr<CListRecord> get_empty_record_sptr() const;
 
  virtual Succeeded get_next_record(CListRecord& record) const;
 
  virtual Succeeded reset();
 
  virtual SavedPosition save_get_position();
 
  virtual Succeeded set_get_position(const SavedPosition&);

  virtual inline unsigned long int get_total_number_of_events() const
  {
    shared_ptr<CListRecord> sptr(new CListRecordT(this->get_proj_data_info_sptr()->get_scanner_sptr()));
    CListRecordPENN& record = static_cast<CListRecordPENN&>(*sptr);
 
    return lm_data_sptr->get_total_number_of_events(record);
  }
 
  virtual bool has_delayeds() const { return true; }
 
  //    inline const PET::ListFileHeader* get_file_header() const
  //    {
  //        return current_lm_data_ptr->get_file_header();
  //    }
 
  inline void set_output_filename(const std::string ofname) { lm_data_sptr->create_output_file(ofname); }
 
  inline void set_event() { lm_data_sptr->set_current_record(); }
 
  void set_event(const bool& is_delay,
                 const short int& _dt,
                 const unsigned short int& _xa,
                 const unsigned short int& _xb,
                 const unsigned short int& _za,
                 const unsigned short int& _zb,
                 const unsigned short int& _ea,
                 const unsigned short int& _eb);
 
private:
  typedef CListRecordPENN CListRecordT;
  std::string listmode_filename;
  shared_ptr<InputStreamWithRecordsFromUPENN> lm_data_sptr;
 
  std::string filename;
};
 
END_NAMESPACE_STIR
 
#endif