cacheresource.h

// -*-c++-*-
#ifndef STORMM_CACHERESOURCE_H
#define STORMM_CACHERESOURCE_H
 
#include "copyright.h"
#include "Accelerator/hybrid.h"
#include "DataTypes/common_types.h"
 
namespace stormm {
namespace energy {
 
using card::Hybrid;
using card::HybridTargetLevel;

template <typename T> struct CacheResourceKit {

  CacheResourceKit(int max_blocks_in, int max_atoms_in, llint* xcrd_in, llint* ycrd_in,
                   llint* zcrd_in, llint* xvel_in, llint* yvel_in, llint* zvel_in,
                   int* xcrd_ovrf_in, int* ycrd_ovrf_in, int* zcrd_ovrf_in, int* xvel_ovrf_in,
                   int* yvel_ovrf_in, int* zvel_ovrf_in, int* xfrc_ovrf_in, int* yfrc_ovrf_in,
                   int* zfrc_ovrf_in, T* charges_in, int* lj_idx_in);

  CacheResourceKit(const CacheResourceKit &original) = default;
  CacheResourceKit(CacheResourceKit &&original) = default;
  
  const int max_blocks;  
  const int max_atoms;   
  llint* xcrd;           
  llint* ycrd;           
  llint* zcrd;           
  llint* xvel;           
  llint* yvel;           
  llint* zvel;           
  int* xcrd_ovrf;        
  int* ycrd_ovrf;        
  int* zcrd_ovrf;        
  int* xvel_ovrf;        
  int* yvel_ovrf;        
  int* zvel_ovrf;        
  int* xfrc_ovrf;        
  int* yfrc_ovrf;        
  int* zfrc_ovrf;        
  T* charges;            
  int* lj_idx;           
};

class CacheResource {
public:

  CacheResource(int block_limit_in, int atom_limit_in);
  CacheResource(int block_limit_in, int2 atom_limits_in);

  CacheResource(const CacheResource &original);
  CacheResource(CacheResource &&original);

  CacheResource& operator=(const CacheResource &other);
  CacheResource& operator=(CacheResource &&other);

  CacheResourceKit<double> dpData(HybridTargetLevel tier = HybridTargetLevel::HOST);

  CacheResourceKit<float> spData(HybridTargetLevel tier = HybridTargetLevel::HOST);
 
private:
  int block_limit;                    
  int atom_limit;                     
  Hybrid<llint> x_coordinates;        
  Hybrid<llint> y_coordinates;        
  Hybrid<llint> z_coordinates;        
  Hybrid<llint> x_velocities;         
  Hybrid<llint> y_velocities;         
  Hybrid<llint> z_velocities;         
  Hybrid<int> x_coordinate_overflow;  
  Hybrid<int> y_coordinate_overflow;  
  Hybrid<int> z_coordinate_overflow;  
  Hybrid<int> x_velocity_overflow;    
  Hybrid<int> y_velocity_overflow;    
  Hybrid<int> z_velocity_overflow;    
  Hybrid<int> x_force_overflow;       
  Hybrid<int> y_force_overflow;       
  Hybrid<int> z_force_overflow;       
  Hybrid<double> charges;             
  Hybrid<float> sp_charges;           
  Hybrid<int> lennard_jones_indices;  
  Hybrid<int> int_data;               
  Hybrid<llint> llint_data;           

  void repairPointers();
};
 
} // namespace energy
} // namespace stormm
 
#include "cacheresource.tpp"
 
#endif