shm_impl.h

Go to the documentation of this file.

#ifndef SHM_IMPL_H
#define SHM_IMPL_H
 
#include <boost/type_traits.hpp>
#include <boost/mpl/find.hpp>
#include <boost/mpl/size.hpp>
#include <boost/mpl/push_back.hpp>
#include <iostream>
#include <cassert>
 
#include "scalars.h"
#include "celltreenode.h"
#include "shm_array.h"
 
#include "archives_config.h"
 
namespace vistle {
 
typedef boost::mpl::push_back<Scalars, CelltreeNode<sizeof(Index), 1>>::type VectorTypes1;
typedef boost::mpl::push_back<VectorTypes1, CelltreeNode<sizeof(Index), 2>>::type VectorTypes2;
typedef boost::mpl::push_back<VectorTypes2, CelltreeNode<sizeof(Index), 3>>::type VectorTypes;
 
template<typename T, class allocator>
int shm_array<T, allocator>::typeId()
{
    const size_t pos = boost::mpl::find<VectorTypes, T>::type::pos::value;
    static_assert(pos < boost::mpl::size<VectorTypes>::value, "Scalar type not found");
    return pos;
}
 
template<typename T>
const ShmVector<T> Shm::getArrayFromName(const std::string &name) const
{
    typedef vistle::shm_array<T, typename vistle::shm<T>::allocator> array;
 
    if (name.empty()) {
        return vistle::shm_array_ref<array>();
    }
 
    auto arr = vistle::shm<array>::find_and_ref(name);
 
    if (!arr) {
        return vistle::shm_array_ref<array>();
    }
    assert(arr->shmtype == ShmData::ARRAY);
    if (array::typeId() != arr->type()) {
        arr->unref();
        return vistle::shm_array_ref<array>();
    }
 
    auto ref = vistle::shm_array_ref<array>(name, arr);
    arr->unref();
    return ref;
}
 
template<class Archive>
void shm_name_t::save(Archive &ar) const
{
    std::string n(name.data());
    //std::cerr << "SHM_NAME_T save: '" << n << "'" << std::endl;
    ar &V_NAME(ar, "shm_name_t", n);
}
 
template<class Archive>
void shm_name_t::load(Archive &ar)
{
    std::string n;
    ar &V_NAME(ar, "shm_name_t", n);
    auto end = n.find('\0');
    if (end != std::string::npos) {
        n = n.substr(0, end);
    }
    if (n.size() < name.size()) {
        std::copy(n.begin(), n.end(), name.data());
        name[n.size()] = '\0';
    } else {
        std::cerr << "shm_name_t: name too long: " << n << " (" << n.size() << " chars)" << std::endl;
        memset(name.data(), 0, name.size());
        assert(n.size() < name.size());
    }
    //std::cerr << "SHM_NAME_T load: '" << name.data() << "'" << std::endl;
}
 
#ifdef NO_SHMEM
#else
template<typename T>
typename managed_shm::segment_manager::template construct_proxy<T>::type shm<T>::construct(const std::string &name)
{
    return Shm::the().shm().construct<T>(name.c_str());
}
#endif
 
} // namespace vistle
 
#endif