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)
,stat          (-1)
,comp          (-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;
}