Homec4science
Diffusion lammps (master) Browse

lammps/src/CNEURONET/sym_fun_type_container.hh
1ebe65076d3bmaster

Last Change
Raw File

sym_fun_type_container.hh
View Options

#include <map>
#include <vector>
#include <tuple>
#ifndef SYM_FUN_TYPE_CONTAINER_H
#define SYM_FUN_TYPE_CONTAINER_H
using uint = unsigned int;
using TypeId = uint;
using SymFunId = uint;
using par_tup = std::tuple<double*, uint, uint>;
class SingleTypeContainer{
public:
inline void insert(double* param, uint me_id, uint other_id) {
this->params.push_back(param);
this->me_index.push_back(me_id);
this->you_index.push_back(other_id);
}
class iterator {
public:
iterator(const SingleTypeContainer & cont, bool end = false)
:index{0}, cont(cont){if (end) index = cont.params.size();}
iterator & operator++() {this->index++; return *this}
inline par_tup operator*(){
return std::forward_as_tuple(this->cont.params[this->index],
this->cont.me_index[this->index],
this->cont.you_index[this->index]);
}
protected:
uint index;
const SingleTypeContainer & cont;
};
iterator begin() {return iterator(*this);}
iterator end() const {return iterator(*this, true);}
protected:
std::vector<double*> params;
std::vector<uint> me_index;
std::vector<uint> you_index;
};
/* ---------------------------------------------------------------------- */
using sym_param_map = std::map<SymFunId, SingleTypeContainer>;
class SymFunTypeContainer {
public:
inline void insert(TypeId type_i, TypeId type_j, SymFunId sym, double* param);
inline sym_param_map & get_sym_params(TypeId type);
protected:
std::map<TypeId, sym_param_map > values;
std::map<TypeId, uint> counters;
};
/* ---------------------------------------------------------------------- */
inline void SymFunTypeContainer::insert(TypeId type_i, TypeId type_j,
SymFunId sym, double* param) {
auto i_cont& = this->values[type_i][sym];
auto j_cont& = this->values[type_j][sym];
auto & i_count = this->counters[type_i];
auto & j_count = this->counters[type_j];
this->i_cont.insert(param, i_count, j_count);
if (type_j != type_j) {
this->j_cont.insert(param, j_count, i_count);
j_count++;
}
c_count++;
}
/* ---------------------------------------------------------------------- */
inline sym_param_map & SymFunTypeContainer::get_sym_params(TypeId type) {
return this->values[type];
}
#endif /* SYM_FUN_TYPE_CONTAINER_H */
c4science · Help