Rand4Hits.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_FASTCALOGPU_RAND4HITS_H
#define ISF_FASTCALOGPU_RAND4HITS_H
 
#include <vector>
#include <random>
 
#include "GpuGeneral_structs.h"
 
class Rand4Hits {
public:
  Rand4Hits() = default;
  ~Rand4Hits();
 
  float* rand_ptr( int nhits ) {
    if ( over_alloc( nhits ) ) {
      rd_regen();
      return m_rand_ptr;
    } else {
      float* f_ptr = &( m_rand_ptr[3 * m_current_hits] );
      return f_ptr;
    }
  };
  float*       rand_ptr_base() { return m_rand_ptr; }
  void         set_rand_ptr( float* ptr ) { m_rand_ptr = ptr; };
  void         set_t_a_hits( int nhits ) { m_total_a_hits = nhits; };
  void         set_c_hits( int nhits ) { m_current_hits = nhits; };
  unsigned int get_c_hits() { return m_current_hits; };
  unsigned int get_t_a_hits() { return m_total_a_hits; };
 
  void create_gen( unsigned long long seed, size_t numhits, bool useCPU = false );
 
  void allocate_simulation( long long maxhits, unsigned short maxbins, unsigned short maxhitct, unsigned long n_cells );
 
  float*  get_cells_energy() { return m_cells_energy; };
  Cell_E* get_cell_e() { return m_cell_e; };
  Cell_E* get_cell_e_h() { return m_cell_e_h; };
 
  int* get_ct() { return m_ct; };
 
  unsigned long* get_hitcells() { return m_hitcells; };
  int*           get_hitcells_ct() { return m_hitcells_ct; };
 
  void rd_regen();
 
  void add_a_hits( int nhits ) {
    if ( over_alloc( nhits ) )
      m_current_hits = nhits;
    else
      m_current_hits += nhits;
  };
  bool over_alloc( int nhits ) {
    return m_current_hits + nhits > m_total_a_hits;
  }; // return true if hits over spill, need regenerat rand..
 
private:
  float* genCPU( size_t num );
  void   createCPUGen( unsigned long long seed );
  void   destroyCPUGen();
 
  float*       m_rand_ptr{nullptr};
  unsigned int m_total_a_hits{0};
  unsigned int m_current_hits{0};
  void*        m_gen{nullptr};
  bool         m_useCPU{false};
 
  // patch in some GPU pointers for cudaMalloc
  float*  m_cells_energy {0};
  Cell_E* m_cell_e {0};
  int*    m_ct {0};
 
  // host side ;
  unsigned long* m_hitcells{nullptr};
  int*           m_hitcells_ct{nullptr};
  Cell_E*        m_cell_e_h{nullptr};
 
  std::vector<float> m_rnd_cpu;
 
};
 
#endif