fabber_core/setup.cc at master · physimals/fabber_core · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/* setup.cc - FABBER default configuration setup class declarations.

 Mike Jackson, The University of Edinburgh & Michael Chappell, FMRIB Analysis Group & IBME QuBIc
 group

 Copyright (C) 2015 University of Oxford  */

#include "setup.h"

#include "inference.h"
#include "inference_vb.h"
#ifndef NO_NLLS
#include "inference_nlls.h"
#endif

#include "noisemodel.h"
#include "noisemodel_ar.h"
#include "noisemodel_white.h"

#include "fwdmodel.h"
#include "fwdmodel_linear.h"
#include "fwdmodel_poly.h"

#include "convergence.h"

void FabberSetup::SetupDefaultInferenceTechniques()
{
    InferenceTechniqueFactory *factory = InferenceTechniqueFactory::GetInstance();
    factory->Add("vb", &Vb::NewInstance);
    factory->Add("spatialvb", &Vb::NewInstance);
#ifndef NO_NLLS
    factory->Add("nlls", &NLLSInferenceTechnique::NewInstance);
#endif
}

void FabberSetup::SetupDefaultNoiseModels()
{
    NoiseModelFactory *factory = NoiseModelFactory::GetInstance();
    factory->Add("ar", &Ar1cNoiseModel::NewInstance);
    factory->Add("white", &WhiteNoiseModel::NewInstance);
}

void FabberSetup::SetupDefaultFwdModels()
{
    FwdModelFactory *factory = FwdModelFactory::GetInstance();
    factory->Add("linear", &LinearFwdModel::NewInstance);
    factory->Add("poly", &PolynomialFwdModel::NewInstance);
}

void FabberSetup::SetupDefaultConvergenceDetectors()
{
    ConvergenceDetectorFactory *factory = ConvergenceDetectorFactory::GetInstance();
    factory->Add("maxits", &CountingConvergenceDetector::NewInstance);
    factory->Add("pointzeroone", &FchangeConvergenceDetector::NewInstance);
    factory->Add("freduce", &FreduceConvergenceDetector::NewInstance);
    factory->Add("trialmode", &TrialModeConvergenceDetector::NewInstance);
    factory->Add("lm", &LMConvergenceDetector::NewInstance);
}

void FabberSetup::SetupDefaults()
{
    FabberSetup::SetupDefaultInferenceTechniques();
    FabberSetup::SetupDefaultNoiseModels();
    FabberSetup::SetupDefaultFwdModels();
    FabberSetup::SetupDefaultConvergenceDetectors();
}

void FabberSetup::Destroy()
{
    FwdModelFactory::Destroy();
    NoiseModelFactory::Destroy();
    InferenceTechniqueFactory::Destroy();
}