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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/***********************************************/
/**
* @file loopMatrix.h
*
* @brief Loop over rows of a matrix.
*
* @author Torsten Mayer-Guerr
* @author Sebastian Strasser
* @date 2017-01-27
*
*/
/***********************************************/
#ifndef __GROOPS_LOOPMATRIX__
#define __GROOPS_LOOPMATRIX__
// Latex documentation
#ifdef DOCSTRING_Loop
static const char *docstringLoopMatrix = R"(
\subsection{Matrix}
Loop over rows of a matrix. To define the loop variables the standard
data variables of the matrix are available, see~\reference{dataVariables}{general.parser:dataVariables}.
)";
#endif
/***********************************************/
#include "base/import.h"
#include "parser/dataVariables.h"
#include "files/fileMatrix.h"
#include "classes/loop/loop.h"
/***** CLASS ***********************************/
/** @brief Loop over rows of a matrix.
* @ingroup loopGroup
* @see Loop */
class LoopMatrix : public Loop
{
Matrix A;
std::vector<ExpressionVariablePtr> variables;
std::string nameIndex, nameCount;
VariableList varListMatrix;
public:
LoopMatrix(Config &config);
UInt count() const override {return A.rows();}
Bool iteration(VariableList &varList) override;
};
/***********************************************/
/***** Inlines *********************************/
/***********************************************/
inline LoopMatrix::LoopMatrix(Config &config)
{
try
{
FileName fileName;
ExpressionVariablePtr startRow, countRows;
Bool transpose;
readConfig(config, "inputfile", fileName, Config::MUSTSET, "", "");
readConfig(config, "transpose", transpose, Config::DEFAULT, "0", "effectively loop over columns");
readConfig(config, "startRow", startRow, Config::DEFAULT, "0", "start at this row (variable: rows)");
readConfig(config, "countRows", countRows, Config::DEFAULT, "rows", "use this many rows (variable: rows)");
readConfig(config, "variableLoop", variables, Config::MUSTSET, "loopNumber=data0", "define a variable by name = expression (input columns are named data0, data1, ...)");
readConfig(config, "variableLoopIndex", nameIndex, Config::OPTIONAL, "", "variable with index of current iteration (starts with zero)");
readConfig(config, "variableLoopCount", nameCount, Config::OPTIONAL, "", "variable with total number of iterations");
readConfigCondition(config);
if(isCreateSchema(config)) return;
readFileMatrix(fileName, A);
if(transpose)
A = A.trans();
varListMatrix.setVariable("rows", A.rows());
A = A.row(static_cast<UInt>(startRow->evaluate(varListMatrix)), static_cast<UInt>(countRows->evaluate(varListMatrix)));
for(auto &variable : variables)
variable->parseVariableName(); // get real variable names, otherwise all named 'variableLoop'
addDataVariables(A, varListMatrix);
}
catch(std::exception &e)
{
GROOPS_RETHROW(e)
}
}
/***********************************************/
inline Bool LoopMatrix::iteration(VariableList &varList)
{
if(index() >= count())
return FALSE;
evaluateDataVariables(A, index(), varListMatrix);
for(auto &variable : variables)
varList.setVariable(variable->name(), variable->evaluate(varListMatrix));
if(!nameIndex.empty()) varList.setVariable(nameIndex, index());
if(!nameCount.empty()) varList.setVariable(nameCount, count());
return checkCondition(varList);
}
/***********************************************/
#endif
|
