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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
// ************************************************************************************************
//
// BornAgain: simulate and fit reflection and scattering
//
//! @file Device/Mask/MaskStack.cpp
//! @brief Implements class MaskStack.
//!
//! @homepage http://www.bornagainproject.org
//! @license GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
// ************************************************************************************************
#include "Device/Mask/MaskStack.h"
#include "Base/Axis/Frame.h"
#include "Base/Axis/Scale.h"
#include "Device/Data/Datafield.h"
#include "Device/Mask/IShape2D.h"
struct ShapeWrapper {
ShapeWrapper(IShape2D* shape_, bool doMask_);
ShapeWrapper(const ShapeWrapper&) = delete;
~ShapeWrapper();
ShapeWrapper* clone() const;
IShape2D* shape; // cloneable
bool doMask;
};
ShapeWrapper::ShapeWrapper(IShape2D* shape_, bool doMask_)
: shape(shape_)
, doMask(doMask_)
{
}
ShapeWrapper::~ShapeWrapper()
{
delete shape;
}
ShapeWrapper* ShapeWrapper::clone() const
{
return new ShapeWrapper(shape->clone(), doMask);
}
MaskStack::MaskStack() = default;
MaskStack::~MaskStack() = default;
MaskStack* MaskStack::clone() const
{
auto* result = new MaskStack;
for (const ShapeWrapper* p : m_stack)
result->m_stack.push_back(p->clone());
return result;
}
void MaskStack::pushMask(const IShape2D& shape, bool mask_value)
{
m_stack.push_back(new ShapeWrapper(shape.clone(), mask_value));
}
bool MaskStack::isMasked(size_t i_flat, const Frame& frame) const
{
size_t nx = frame.xAxis().size();
size_t ix = i_flat % nx;
size_t iy = i_flat / nx;
for (int k = m_stack.size() - 1; k >= 0; --k) {
const ShapeWrapper* const pat = m_stack[k];
Bin1D binx = frame.xAxis().bin(ix);
Bin1D biny = frame.yAxis().bin(iy);
if (pat->shape->contains(binx, biny))
return pat->doMask;
}
return false;
}
bool MaskStack::hasMasks() const
{
return !m_stack.empty();
}
size_t MaskStack::numberOfMasks() const
{
return m_stack.size();
}
std::pair<IShape2D*, bool> MaskStack::patternAt(size_t iMask) const
{
const ShapeWrapper* pat = m_stack.at(iMask);
return {pat->shape, pat->doMask};
}
|
