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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#include <iostream>
#include <fstream>
#include "itkStdStreamLogOutput.h"
#include "itkThreadLogger.h"
#include "itkMultiThreaderBase.h"
#include "itkLogTester.h"
struct ThreadDataStruct
{
itk::LoggerBase * logger;
};
using ThreadDataVec = std::vector<ThreadDataStruct>;
ITK_THREAD_RETURN_FUNCTION_CALL_CONVENTION
ThreadedGenerateLogMessages(void * arg)
{
const auto * workUnitInfo = static_cast<itk::MultiThreaderBase::WorkUnitInfo *>(arg);
if (workUnitInfo)
{
const unsigned int workUnitID = workUnitInfo->WorkUnitID;
std::string threadPrefix;
{
std::ostringstream msg;
msg << "<Thread " << workUnitID << "> ";
threadPrefix = msg.str();
}
const ThreadDataVec * dataVec = static_cast<ThreadDataVec *>(workUnitInfo->UserData);
if (dataVec)
{
const ThreadDataStruct threadData = (*dataVec)[workUnitID];
{
std::ostringstream msg;
msg << threadPrefix << "unpacked arg\n";
threadData.logger->Write(itk::LoggerBase::PriorityLevelEnum::INFO, msg.str());
threadData.logger->Flush();
msg.str("");
msg << threadPrefix << "Done logging\n";
threadData.logger->Write(itk::LoggerBase::PriorityLevelEnum::INFO, msg.str());
// std::cout << msg.str() << std::endl;
}
// do stuff
}
else
{
std::cerr << "ERROR: UserData was not of type ThreadDataVec*" << std::endl;
return ITK_THREAD_RETURN_DEFAULT_VALUE;
}
}
else
{
std::cerr << "ERROR: arg was not of type itk::MultiThreaderBase::WorkUnitInfo*" << std::endl;
return ITK_THREAD_RETURN_DEFAULT_VALUE;
}
return ITK_THREAD_RETURN_DEFAULT_VALUE;
}
ThreadDataVec
create_threaded_data(int num_threads, itk::LoggerBase * logger)
{
ThreadDataVec threadData;
for (int ii = 0; ii < num_threads; ++ii)
{
threadData.push_back(ThreadDataStruct());
threadData[ii].logger = logger;
}
return threadData;
}
int
itkThreadLoggerTest(int argc, char * argv[])
{
try
{
if (argc < 2)
{
std::cout << "Usage: " << itkNameOfTestExecutableMacro(argv) << " logFilename [num threads, default = 10]"
<< std::endl;
return EXIT_FAILURE;
}
int numWorkUnits = 10;
if (argc > 2)
{
numWorkUnits = std::stoi(argv[2]);
}
// Create an ITK StdStreamLogOutputs
itk::StdStreamLogOutput::Pointer coutput = itk::StdStreamLogOutput::New();
itk::StdStreamLogOutput::Pointer foutput = itk::StdStreamLogOutput::New();
coutput->SetStream(std::cout);
std::ofstream fout(argv[1]);
foutput->SetStream(fout);
// Create an ITK ThreadLogger
itk::ThreadLogger::Pointer logger = itk::ThreadLogger::New();
std::cout << "Testing itk::ThreadLogger" << std::endl;
// Setting the logger
logger->SetName("org.itk.threadLogger");
logger->SetPriorityLevel(itk::LoggerBase::PriorityLevelEnum::INFO);
logger->SetLevelForFlushing(itk::LoggerBase::PriorityLevelEnum::CRITICAL);
std::cout << " Adding console and file stream LogOutputs" << std::endl;
logger->AddLogOutput(coutput);
logger->AddLogOutput(foutput);
// Printing the logger's member variables
std::cout << " Name: " << logger->GetName() << std::endl;
std::cout << " Priority Level: " << logger->GetPriorityLevel() << std::endl;
std::cout << " Level For Flushing: " << logger->GetLevelForFlushing() << std::endl;
// Print logger itself
std::cout << logger << std::endl;
// Logging by the itkLogMacro from a class with itk::ThreadLogger
itk::Testing::LogTester tester;
tester.SetLogger(logger);
tester.log();
// Logging by the itkLogMacroStatic from a class with itk::ThreadLogger
itk::Testing::LogTester::logStatic(&tester);
std::cout << " The printed order of 'Messages ##' below might not be predictable because of multi-threaded logging"
<< std::endl;
std::cout << " But the logged messages will be in order." << std::endl;
std::cout << " Each line is an atom for synchronization." << std::endl;
// Writing by the logger
logger->Write(itk::LoggerBase::PriorityLevelEnum::DEBUG, "This is the DEBUG message.\n");
std::cout << " Message #1" << std::endl;
logger->Write(itk::LoggerBase::PriorityLevelEnum::INFO, "This is the INFO message.\n");
logger->Write(itk::LoggerBase::PriorityLevelEnum::WARNING, "This is the WARNING message.\n");
std::cout << " Message #2" << std::endl;
logger->Write(itk::LoggerBase::PriorityLevelEnum::CRITICAL, "This is the CRITICAL message.\n");
logger->Write(itk::LoggerBase::PriorityLevelEnum::FATAL, "This is the FATAL message.\n");
logger->Write(itk::LoggerBase::PriorityLevelEnum::MUSTFLUSH, "This is the MUSTFLUSH message.\n");
std::cout << " Message #3" << std::endl;
logger->Flush();
std::cout << " Flushing by the ThreadLogger is synchronized." << std::endl;
std::cout << "Beginning multi-threaded portion of test." << std::endl;
ThreadDataVec threadData = create_threaded_data(numWorkUnits, logger);
itk::MultiThreaderBase::Pointer threader = itk::MultiThreaderBase::New();
itk::MultiThreaderBase::SetGlobalMaximumNumberOfThreads(numWorkUnits + 10);
threader->SetNumberOfWorkUnits(numWorkUnits);
threader->SetSingleMethod(ThreadedGenerateLogMessages, &threadData);
threader->SingleMethodExecute();
logger->Flush();
std::cout << "Ended multi-threaded portion of test." << std::endl;
}
catch (...)
{
std::cerr << "Exception caught!" << std::endl;
return EXIT_FAILURE;
}
std::cout << "[PASSED]" << std::endl;
return EXIT_SUCCESS;
}
|
