source: trunk/FACT++/src/mcp.cc@ 11574

Last change on this file since 11574 was 11574, checked in by tbretz, 13 years ago
Added the possibility to setup DIM_HOST_NODE with the --host program option.
File size: 18.4 KB
Line 
1#include "Dim.h"
2#include "Event.h"
3#include "Shell.h"
4#include "StateMachineDim.h"
5#include "Connection.h"
6#include "Configuration.h"
7#include "Console.h"
8#include "Converter.h"
9#include "DimServiceInfoList.h"
10
11#include "tools.h"
12
13#include "LocalControl.h"
14
15#include "HeadersFTM.h"
16#include "HeadersFAD.h"
17
18
19namespace ba = boost::asio;
20namespace bs = boost::system;
21namespace dummy = ba::placeholders;
22
23using namespace std;
24
25// ------------------------------------------------------------------------
26
27#include "DimDescriptionService.h"
28
29// ------------------------------------------------------------------------
30
31class StateMachineMCP : public StateMachineDim, public DimInfoHandler
32{
33 /*
34 int Wrap(boost::function<void()> f)
35 {
36 f();
37 return T::GetCurrentState();
38 }
39
40 boost::function<int(const EventImp &)> Wrapper(boost::function<void()> func)
41 {
42 return bind(&StateMachineMCP::Wrap, this, func);
43 }*/
44
45private:
46 enum states_t
47 {
48 kStateDimNetworkNA = 1,
49 kStateDisconnected,
50 kStateConnecting,
51 kStateConnected,
52 kStateIdle,
53 kStateReadyForDataTaking,
54 kStateConfiguring1,
55 kStateConfiguring2,
56 kStateConfiguring3,
57 kStateConfigured,
58 };
59
60 DimServiceInfoList fNetwork;
61
62 pair<Time, int> fStatusDim;
63 pair<Time, int> fStatusFTM;
64 pair<Time, int> fStatusFAD;
65 pair<Time, int> fStatusLog;
66
67 DimStampedInfo fDim;
68 DimStampedInfo fFTM;
69 DimStampedInfo fFAD;
70 DimStampedInfo fLog;
71
72 pair<Time, int> GetNewState(DimStampedInfo &info) const
73 {
74 const bool disconnected = info.getSize()==0;
75
76 // Make sure getTimestamp is called _before_ getTimestampMillisecs
77 const int tsec = info.getTimestamp();
78 const int tms = info.getTimestampMillisecs();
79
80 return make_pair(Time(tsec, tms*1000),
81 disconnected ? -2 : info.getQuality());
82 }
83
84 void infoHandler()
85 {
86 DimInfo *curr = getInfo(); // get current DimInfo address
87 if (!curr)
88 return;
89
90 if (curr==&fFTM)
91 {
92 fStatusFTM = GetNewState(fFTM);
93 return;
94 }
95
96 if (curr==&fFAD)
97 {
98 fStatusFAD = GetNewState(fFAD);
99 return;
100 }
101
102 if (curr==&fLog)
103 {
104 fStatusLog = GetNewState(fLog);
105 return;
106 }
107
108 if (curr==&fDim)
109 {
110 fStatusDim = GetNewState(fDim);
111 fStatusDim.second = curr->getSize()==4 ? curr->getInt() : 0;
112 return;
113 }
114
115 }
116
117 bool CheckEventSize(size_t has, const char *name, size_t size)
118 {
119 if (has==size)
120 return true;
121
122 ostringstream msg;
123 msg << name << " - Received event has " << has << " bytes, but expected " << size << ".";
124 Fatal(msg);
125 return false;
126 }
127
128 int SetVerbosity(const EventImp &)
129 {
130 /*
131 if (!CheckEventSize(evt.GetSize(), "SetVerbosity", 1))
132 return T::kSM_FatalError;
133
134 fFSC.SetVerbose(evt.GetBool());
135
136 */
137
138 return GetCurrentState();
139 }
140
141
142 void PrintState(const pair<Time,int> &state, const char *server)
143 {
144 const State rc = fNetwork.GetState(server, state.second);
145
146 Out() << state.first.GetAsStr("%H:%M:%S.%f").substr(0, 12) << " - ";
147 Out() << kBold << server << ": ";
148 Out() << rc.name << "[" << rc.index << "]";
149 Out() << kReset << " - " << kBlue << rc.comment << endl;
150 }
151
152 int Print()
153 {
154 Out() << fStatusDim.first.GetAsStr("%H:%M:%S.%f").substr(0, 12) << " - ";
155 Out() << kBold << "DIM_DNS: ";
156 if (fStatusDim.second==0)
157 Out() << "Offline" << endl;
158 else
159 Out() << "V" << fStatusDim.second/100 << 'r' << fStatusDim.second%100 << endl;
160
161 PrintState(fStatusFTM, "FTM_CONTROL");
162 PrintState(fStatusFAD, "FAD_CONTROL");
163 PrintState(fStatusLog, "DATA_LOGGER");
164
165 return GetCurrentState();
166 }
167
168 int GetReady()
169 {
170
171 return GetCurrentState();
172 }
173
174 int StopRun(const EventImp &)
175 {
176 if (fStatusFTM.second==FTM::kTakingData)
177 Dim::SendCommand("FTM_CONTROL/STOP_RUN");
178
179 // FIXME: Do step 2 only when FTM is stopped
180 if (fStatusFAD.second==FAD::kConnected)
181 {
182 Dim::SendCommand("FAD_CONTROL/ENABLE_TRIGGER_LINE", bool(false));
183 Dim::SendCommand("FAD_CONTROL/ENABLE_CONTINOUS_TRIGGER", bool(false));
184 }
185
186 return GetCurrentState();
187 }
188
189 int Reset(const EventImp &)
190 {
191 fRunType = "";
192 return kStateIdle;
193 /*
194 // FIMXE: Handle error states!
195 if (fStatusLog.second>=20)//kSM_NightlyOpen
196 Dim::SendCommand("DATA_LOGGER/STOP");
197
198 if (fStatusLog.second==0)
199 Dim::SendCommand("DATA_LOGGER/WAIT_FOR_RUN_NUMBER");
200
201 if (fStatusFAD.second==FAD::kConnected)
202 {
203 Dim::SendCommand("FAD_CONTROL/ENABLE_TRIGGER_LINE", bool(false));
204 Dim::SendCommand("FAD_CONTROL/ENABLE_CONTINOUS_TRIGGER", bool(false));
205 }
206
207 if (fStatusFTM.second==FTM::kTakingData)
208 Dim::SendCommand("FTM_CONTROL/STOP");
209
210 return GetCurrentState(); */
211 }
212
213 uint64_t fMaxTime;
214 uint64_t fNumEvents;
215 string fRunType;
216
217 int StartRun(const EventImp &evt)
218 {
219 fMaxTime = evt.Get<int64_t>();
220 fNumEvents = evt.Get<int64_t>(8);
221 fRunType = evt.Ptr<char>(16);
222
223 ostringstream str;
224 str << "Starting configuration '" << fRunType << "' for new run";
225 if (fNumEvents>0 || fMaxTime>0)
226 str << " [";
227 if (fNumEvents>0)
228 str << fNumEvents << " events";
229 if (fNumEvents>0 && fMaxTime>0)
230 str << " / ";
231 if (fMaxTime>0)
232 str << fMaxTime << "s";
233 if (fNumEvents>0 || fMaxTime>0)
234 str << "]";
235 Message(str);
236
237 return kStateConfiguring1;
238 }
239
240 void ConfigureFAD()
241 {
242 struct Value
243 {
244 uint64_t time;
245 uint64_t nevts;
246 char type[];
247 };
248
249 const size_t len = sizeof(Value)+fRunType.length()+1;
250
251 char *buf = new char[len];
252
253 Value *val = reinterpret_cast<Value*>(buf);
254
255 val->time = fMaxTime;
256 val->nevts = fNumEvents;
257
258 strcpy(val->type, fRunType.c_str());
259
260 Dim::SendCommand("FAD_CONTROL/CONFIGURE", buf, len);
261
262 delete buf;
263 }
264
265 int Execute()
266 {
267 // Dispatch (execute) at most one handler from the queue. In contrary
268 // to run_one(), it doesn't wait until a handler is available
269 // which can be dispatched, so poll_one() might return with 0
270 // handlers dispatched. The handlers are always dispatched/executed
271 // synchronously, i.e. within the call to poll_one()
272 //poll_one();
273
274 if (fStatusDim.second==0)
275 return kStateDimNetworkNA;
276
277 if (fStatusFTM.second >= FTM::kConnected &&
278 fStatusFAD.second >= FAD::kConnected &&
279 fStatusLog.second >= kSM_Ready)
280 {
281 if (GetCurrentState()==kStateConfiguring1)
282 {
283 if (fStatusLog.second!=30/*kSM_WaitForRun*/)
284 Dim::SendCommand("DATA_LOGGER/WAIT_FOR_RUN_NUMBER");
285 Dim::SendCommand("FTM_CONTROL/CONFIGURE", fRunType);
286 return kStateConfiguring2;
287 }
288
289 if (GetCurrentState()==kStateConfiguring2)
290 {
291 if ((fStatusFTM.second != FTM::kConfiguring2 &&
292 fStatusFTM.second != FTM::kConfigured) ||
293 fStatusLog.second != 30/*kSM_WaitForRun*/)
294 return GetCurrentState();
295
296 Message("Starting FAD");
297 ConfigureFAD();
298 return kStateConfiguring3;
299 }
300
301 if (GetCurrentState()==kStateConfiguring3)
302 {
303 if (fStatusFTM.second != FTM::kConfigured ||
304 fStatusFAD.second != FAD::kConfigured)
305 return GetCurrentState();
306
307 Message("Starting Trigger (FTM)");
308 Dim::SendCommand("FTM_CONTROL/START_RUN");
309 return kStateConfigured;
310 }
311
312 if (GetCurrentState()==kStateConfigured)
313 return GetCurrentState();
314
315 return kStateIdle;
316 }
317
318 /*
319 if (fStatusFTM.second >= FTM::kConnected &&
320 fStatusFAD.second >= FAD::kConnected &&
321 fStatusLog.second >= kSM_Ready)
322 return kStateIdle;
323 */
324 if (fStatusFTM.second >-2 &&
325 fStatusFAD.second >-2 &&
326 fStatusLog.second >-2)
327 return kStateConnected;
328
329 if (fStatusFTM.second >-2 ||
330 fStatusFAD.second >-2 ||
331 fStatusLog.second >-2)
332 return kStateConnecting;
333
334 return kStateDisconnected;
335 }
336
337public:
338 StateMachineMCP(ostream &out=cout) : StateMachineDim(out, "MCP"),
339 fStatusDim(make_pair(Time(), -2)),
340 fStatusFTM(make_pair(Time(), -2)),
341 fStatusFAD(make_pair(Time(), -2)),
342 fStatusLog(make_pair(Time(), -2)),
343 fDim("DIS_DNS/VERSION_NUMBER", (void*)NULL, 0, this),
344 fFTM("FTM_CONTROL/STATE", (void*)NULL, 0, this),
345 fFAD("FAD_CONTROL/STATE", (void*)NULL, 0, this),
346 fLog("DATA_LOGGER/STATE", (void*)NULL, 0, this)
347 {
348 // ba::io_service::work is a kind of keep_alive for the loop.
349 // It prevents the io_service to go to stopped state, which
350 // would prevent any consecutive calls to run()
351 // or poll() to do nothing. reset() could also revoke to the
352 // previous state but this might introduce some overhead of
353 // deletion and creation of threads and more.
354
355 // State names
356 AddStateName(kStateDimNetworkNA, "DimNetworkNotAvailable",
357 ".");
358
359 AddStateName(kStateDisconnected, "Disconnected",
360 ".");
361
362 AddStateName(kStateConnecting, "Connecting",
363 ".");
364
365 AddStateName(kStateConnected, "Connected",
366 ".");
367
368 AddStateName(kStateIdle, "Idle",
369 ".");
370
371 AddStateName(kStateReadyForDataTaking, "ReadyForDataTaking",
372 ".");
373
374 AddStateName(kStateConfiguring1, "Configuring1",
375 ".");
376
377 AddStateName(kStateConfiguring2, "Configuring2",
378 ".");
379
380 AddStateName(kStateConfiguring3, "Configuring3",
381 ".");
382
383 AddStateName(kStateConfigured, "Configured",
384 ".");
385
386
387 AddEvent("START", "X:2;C")//, kStateIdle)
388 (bind(&StateMachineMCP::StartRun, this, placeholders::_1))
389 ("");
390
391 AddEvent("STOP")
392 (bind(&StateMachineMCP::StopRun, this, placeholders::_1))
393 ("");
394
395 AddEvent("RESET", kStateConfiguring1, kStateConfiguring2, kStateConfigured)
396 (bind(&StateMachineMCP::Reset, this, placeholders::_1))
397 ("");
398
399 // Verbosity commands
400 AddEvent("SET_VERBOSE", "B:1")
401 (bind(&StateMachineMCP::SetVerbosity, this, placeholders::_1))
402 ("set verbosity state"
403 "|verbosity[bool]:disable or enable verbosity for received data (yes/no), except dynamic data");
404
405 AddEvent("PRINT")
406 (bind(&StateMachineMCP::Print, this))
407 ("");
408 }
409
410 int EvalOptions(Configuration &)
411 {
412 //SetEndpoint(conf.Get<string>("addr"));
413
414 //fFSC.SetVerbose(!conf.Get<bool>("quiet"));
415
416 return -1;
417 }
418};
419
420// ------------------------------------------------------------------------
421
422#include "Main.h"
423
424/*
425void RunThread(StateMachineImp *io_service)
426{
427 // This is necessary so that the StateMachien Thread can signal the
428 // Readline to exit
429 io_service->Run();
430 Readline::Stop();
431}
432*/
433/*
434template<class S, class T>
435int RunDim(Configuration &conf)
436{
437 WindowLog wout;
438
439 ReadlineColor::PrintBootMsg(wout, conf.GetName(), false);
440
441
442 if (conf.Has("log"))
443 if (!wout.OpenLogFile(conf.Get<string>("log")))
444 wout << kRed << "ERROR - Couldn't open log-file " << conf.Get<string>("log") << ": " << strerror(errno) << endl;
445
446 // Start io_service.Run to use the StateMachineImp::Run() loop
447 // Start io_service.run to only use the commandHandler command detaching
448 StateMachineMCP<S, T> io_service(wout);
449 if (!io_service.EvalConfiguration(conf))
450 return -1;
451
452 io_service.Run();
453
454 return 0;
455}
456*/
457
458template<class T>
459int RunShell(Configuration &conf)
460{
461 return Main<T, StateMachineMCP>(conf);
462/*
463 static T shell(conf.GetName().c_str(), conf.Get<int>("console")!=1);
464
465 WindowLog &win = shell.GetStreamIn();
466 WindowLog &wout = shell.GetStreamOut();
467
468 if (conf.Has("log"))
469 if (!wout.OpenLogFile(conf.Get<string>("log")))
470 win << kRed << "ERROR - Couldn't open log-file " << conf.Get<string>("log") << ": " << strerror(errno) << endl;
471
472 StateMachineMCP<S, R> io_service(wout);
473 if (!io_service.EvalConfiguration(conf))
474 return -1;
475
476 shell.SetReceiver(io_service);
477
478 boost::thread t(bind(RunThread, &io_service));
479 // boost::thread t(bind(&StateMachineMCP<S>::Run, &io_service));
480
481 if (conf.Has("cmd"))
482 {
483 const vector<string> v = conf.Get<vector<string>>("cmd");
484 for (vector<string>::const_iterator it=v.begin(); it!=v.end(); it++)
485 shell.ProcessLine(*it);
486 }
487
488 if (conf.Has("exec"))
489 {
490 const vector<string> v = conf.Get<vector<string>>("exec");
491 for (vector<string>::const_iterator it=v.begin(); it!=v.end(); it++)
492 shell.Execute(*it);
493 }
494
495 if (conf.Get<bool>("quit"))
496 shell.Stop();
497
498 shell.Run(); // Run the shell
499 io_service.Stop(); // Signal Loop-thread to stop
500 // io_service.Close(); // Obsolete, done by the destructor
501
502 // Wait until the StateMachine has finished its thread
503 // before returning and destroying the dim objects which might
504 // still be in use.
505 t.join();
506
507 return 0;
508*/
509}
510
511void SetupConfiguration(Configuration &conf)
512{
513 const string n = conf.GetName()+".log";
514
515 po::options_description config("Program options");
516 config.add_options()
517 ("dns", var<string>("localhost"), "Dim nameserver (overwites DIM_DNS_NODE environment variable)")
518 ("host", var<string>(""), "Address with which the Dim nameserver can connect to this host (overwites DIM_HOST_NODE environment variable)")
519 ("log,l", var<string>(n), "Write log-file")
520// ("no-dim,d", po_bool(), "Disable dim services")
521 ("console,c", var<int>(), "Use console (0=shell, 1=simple buffered, X=simple unbuffered)")
522 ("cmd", vars<string>(), "Execute one or more commands at startup")
523 ("exec,e", vars<string>(), "Execute one or more scrips at startup")
524 ("quit,q", po_switch(), "Quit after startup");
525 ;
526/*
527 po::options_description control("FTM control options");
528 control.add_options()
529 ("addr,a", var<string>("localhost:5000"), "Network address of FTM")
530 ("quiet,q", po_bool(), "Disable printing contents of all received messages (except dynamic data) in clear text.")
531 ;
532*/
533 conf.AddEnv("dns", "DIM_DNS_NODE");
534 conf.AddEnv("host", "DIM_HOST_NODE");
535
536 conf.AddOptions(config);
537// conf.AddOptions(control);
538}
539
540/*
541 Extract usage clause(s) [if any] for SYNOPSIS.
542 Translators: "Usage" and "or" here are patterns (regular expressions) which
543 are used to match the usage synopsis in program output. An example from cp
544 (GNU coreutils) which contains both strings:
545 Usage: cp [OPTION]... [-T] SOURCE DEST
546 or: cp [OPTION]... SOURCE... DIRECTORY
547 or: cp [OPTION]... -t DIRECTORY SOURCE...
548 */
549void PrintUsage()
550{
551 cout <<
552 "The ftmctrl controls the FSC (FACT Slow Control) board.\n"
553 "\n"
554 "The default is that the program is started without user intercation. "
555 "All actions are supposed to arrive as DimCommands. Using the -c "
556 "option, a local shell can be initialized. With h or help a short "
557 "help message about the usuage can be brought to the screen.\n"
558 "\n"
559 "Usage: fscctrl [-c type] [OPTIONS]\n"
560 " or: fscctrl [OPTIONS]\n";
561 cout << endl;
562}
563
564void PrintHelp()
565{
566 /* Additional help text which is printed after the configuration
567 options goes here */
568
569 /*
570 cout << "bla bla bla" << endl << endl;
571 cout << endl;
572 cout << "Environment:" << endl;
573 cout << "environment" << endl;
574 cout << endl;
575 cout << "Examples:" << endl;
576 cout << "test exam" << endl;
577 cout << endl;
578 cout << "Files:" << endl;
579 cout << "files" << endl;
580 cout << endl;
581 */
582}
583
584int main(int argc, const char* argv[])
585{
586 Configuration conf(argv[0]);
587 conf.SetPrintUsage(PrintUsage);
588 SetupConfiguration(conf);
589
590 po::variables_map vm;
591 try
592 {
593 vm = conf.Parse(argc, argv);
594 }
595#if BOOST_VERSION > 104000
596 catch (po::multiple_occurrences &e)
597 {
598 cerr << "Program options invalid due to: " << e.what() << " of '" << e.get_option_name() << "'." << endl;
599 return -1;
600 }
601#endif
602 catch (exception& e)
603 {
604 cerr << "Program options invalid due to: " << e.what() << endl;
605 return -1;
606 }
607
608 if (conf.HasVersion() || conf.HasPrint())
609 return -1;
610
611 if (conf.HasHelp())
612 {
613 PrintHelp();
614 return -1;
615 }
616
617 Dim::Setup(conf.Get<string>("dns"), conf.Get<string>("host"));
618
619 //try
620 {
621 // No console access at all
622 if (!conf.Has("console"))
623 {
624// if (conf.Get<bool>("no-dim"))
625// return RunShell<LocalStream, StateMachine, ConnectionFSC>(conf);
626// else
627 return RunShell<LocalStream>(conf);
628 }
629 // Cosole access w/ and w/o Dim
630/* if (conf.Get<bool>("no-dim"))
631 {
632 if (conf.Get<int>("console")==0)
633 return RunShell<LocalShell, StateMachine, ConnectionFSC>(conf);
634 else
635 return RunShell<LocalConsole, StateMachine, ConnectionFSC>(conf);
636 }
637 else
638*/ {
639 if (conf.Get<int>("console")==0)
640 return RunShell<LocalShell>(conf);
641 else
642 return RunShell<LocalConsole>(conf);
643 }
644 }
645 /*catch (std::exception& e)
646 {
647 cerr << "Exception: " << e.what() << endl;
648 return -1;
649 }*/
650
651 return 0;
652}
Note: See TracBrowser for help on using the repository browser.