fwd_thread_arg.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "fwd_thread_arg.h"
#include <mne/mne_source_space.h>
#include "fwd_coil_set.h"
#include "fwd_bem_model.h"
#include "fwd_comp_data.h"
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace Eigen;
using namespace FWDLIB;
using namespace MNELIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
FwdThreadArg::FwdThreadArg()
: res(nullptr)
, res_grad(nullptr)
, off(0)
, field_pot(nullptr)
, vec_field_pot(nullptr)
, field_pot_grad(nullptr)
, coils_els(nullptr)
, client(nullptr)
, s(nullptr)
, fixed_ori(false)
, comp(-1)
, stat(-1)
{
}
 
//=============================================================================================================
 
FwdThreadArg::~FwdThreadArg()
{
    if (client_free)
        client_free();
}
 
//=============================================================================================================
 
FwdThreadArg::UPtr FwdThreadArg::create_eeg_multi_thread_duplicate(FwdThreadArg& one, bool bem_model)
{
    auto res = std::make_unique<FwdThreadArg>();
 
    *res = one;
    res->client_free = nullptr; /* Don't copy the source's deleter */
    if (bem_model) {
        auto bem = std::make_shared<FwdBemModel>();
        *bem = *static_cast<FwdBemModel*>(res->client);
        bem->v0.resize(0);
        res->client = bem.get();
        res->client_free = [bem]() {
        }; /* shared_ptr releases FwdBemModel on destruction */
    }
    return res;
}
 
//=============================================================================================================
 
FwdThreadArg::UPtr FwdThreadArg::create_meg_multi_thread_duplicate(FwdThreadArg& one, bool bem_model)
{
    auto res = std::make_unique<FwdThreadArg>();
    FwdCompData* orig = static_cast<FwdCompData*>(one.client);
 
    *res = one;
    res->client_free = nullptr; /* Don't copy the source's deleter */
 
    auto comp = std::make_shared<FwdCompData>();
    *comp = *orig;
    comp->comp_coils = nullptr; /* Non-owning: shared with original, prevent ~FwdCompData from deleting */
    comp->set = nullptr;        /* Will be replaced below; prevent dtor from deleting orig's copy */
    comp->work.resize(0);
    comp->vec_work.resize(0, 0);
 
    std::shared_ptr<MNECTFCompDataSet> set_guard(
        orig->set ? new MNECTFCompDataSet(*(orig->set)) : nullptr);
    comp->set = set_guard.get();
 
    res->client = comp.get();
 
    if (bem_model) {
        auto bem = std::make_shared<FwdBemModel>();
        *bem = *static_cast<FwdBemModel*>(comp->client);
        bem->v0.resize(0);
        comp->client = bem.get();
        /* shared_ptrs release their objects when client_free is destroyed */
        res->client_free = [comp, set_guard, bem]() {
            comp->set = nullptr; /* Prevent ~FwdCompData double-free; set_guard owns it */
        };
    } else {
        res->client_free = [comp, set_guard]() {
            comp->set = nullptr; /* Prevent ~FwdCompData double-free; set_guard owns it */
        };
    }
    return res;
}