TpcSpaceChargeMatrixContainerv1.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file TpcSpaceChargeMatrixContainerv1.h

#ifndef TPCCALIB_TPCSPACECHARGEMATRIXCONTAINERV1_H
#define TPCCALIB_TPCSPACECHARGEMATRIXCONTAINERV1_H

#include "TpcSpaceChargeMatrixContainer.h"

#include <array>

class TpcSpaceChargeMatrixContainerv1 : public TpcSpaceChargeMatrixContainer
{
  public:

  TpcSpaceChargeMatrixContainerv1();

  ~TpcSpaceChargeMatrixContainerv1() override = default;



  void identify(std::ostream &os = std::cout) const override;

  void get_grid_dimensions( int& phibins, int& rbins, int& zbins ) const override;

  int get_grid_size() const override;

  int get_cell_index( int iphibin, int irbin, int izbin ) const override;

  int get_entries( int cell_index ) const override;

  float get_lhs( int cell_index, int i, int j ) const override;

  float get_rhs( int cell_index, int i ) const override;




  void Reset() override;


  void set_grid_dimensions( int phibins, int rbins, int zbins ) override;

  void add_to_entries( int cell_index ) override
  { add_to_entries( cell_index, 1 ); }

  void add_to_entries( int cell_index, int value ) override;

  void add_to_lhs( int cell_index, int i, int j, float value ) override;

  void add_to_rhs( int cell_index, int i, float value ) override;

  bool add( const TpcSpaceChargeMatrixContainer& other ) override;


  private:

  bool bound_check( int cell_index ) const;

  bool bound_check( int cell_index, int i ) const;

  bool bound_check( int cell_index, int i, int j ) const;

  int get_flat_index( int i, int j ) const;


  int m_phibins = 36;
  int m_rbins = 16;
  int m_zbins = 80;

  static constexpr int m_ncoord = 3;


  using matrix_t = std::array<float, m_ncoord*m_ncoord>;
  using column_t = std::array<float, m_ncoord>;

  std::vector<matrix_t> m_lhs;

  std::vector<column_t> m_rhs;

  std::vector<int> m_entries;

  ClassDefOverride(TpcSpaceChargeMatrixContainerv1, 1)

};

#endif