mech_mapping.cpp

Go to the documentation of this file.

/*
# =============================================================================
# Copyright (c) 2016 - 2021 Blue Brain Project/EPFL
#
# See top-level LICENSE file for details.
# =============================================================================
*/
 
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <map>
 
#include "coreneuron/mechanism/mech_mapping.hpp"
#include "coreneuron/mechanism/mechanism.hpp"
#include "coreneuron/permute/data_layout.hpp"
 
namespace coreneuron {
using Offset = size_t;
using MechId = int;
using VariableName = const char*;
 
struct cmp_str {
    bool operator()(char const* a, char const* b) const {
        return std::strcmp(a, b) < 0;
    }
};
 
/*
 * Structure that map variable names of mechanisms to their value's location (offset) in memory
 */
using MechNamesMapping = std::map<MechId, std::map<VariableName, Offset, cmp_str>>;
static MechNamesMapping mechNamesMapping;
 
static void set_an_offset(int mech_id, const char* variable_name, int offset) {
    mechNamesMapping[mech_id][variable_name] = offset;
}
 
double* get_var_location_from_var_name(int mech_id,
                                       const char* variable_name,
                                       Memb_list* ml,
                                       int node_index) {
    if (mechNamesMapping.find(mech_id) == mechNamesMapping.end()) {
        std::cerr << "ERROR : no variable name mapping exist for mechanism id: " << mech_id
                  << std::endl;
        abort();
    }
    if (mechNamesMapping.at(mech_id).find(variable_name) == mechNamesMapping.at(mech_id).end()) {
        std::cerr << "ERROR : no value associtated to variable name: " << variable_name
                  << std::endl;
        abort();
    }
    int variable_rank = mechNamesMapping.at(mech_id).at(variable_name);
    int ix = get_data_index(node_index, variable_rank, mech_id, ml);
    return &(ml->data[ix]);
}
 
void register_all_variables_offsets(int mech_id, SerializedNames variable_names) {
    int idx = 0;
    int nb_parsed_variables = 0;
    int current_categorie = 1;
    while (current_categorie < NB_MECH_VAR_CATEGORIES) {
        if (variable_names[idx]) {
            set_an_offset(mech_id, variable_names[idx], nb_parsed_variables);
            nb_parsed_variables++;
        } else {
            current_categorie++;
        }
        idx++;
    }
    idx++;
}
 
}  // namespace coreneuron