323):
324 """Simple script to create an OKS configuration file for all
325 ReadoutApplications defined in a readout map.
326
327 The file will automatically include the relevant schema files and
328 any other OKS files you specify.
329
330 Example:
331 generate_readoutOKS -i hosts \
332 -i appmodel/connections.data.xml -i appmodel/moduleconfs \
333 config/np04readoutmap.data.xml readoutApps.data.xml
334
335 Will load hosts, connections and moduleconfs data files as well as
336 the readoutmap (config/np04readoutmap.data.xml) and write the
337 generated apps to readoutApps.data.xml.
338
339 generate_readoutOKS --session --segment \
340 -i appmodel/fsm -i hosts \
341 -i appmodel/connections.data.xml -i appmodel/moduleconfs \
342 config/np04readoutmap.data.xml np04readout-session.data.xml
343
344 Will do the same but in addition it will generate a containing
345 Segment for the apps and a containing Session for the Segment.
346
347 NB: Currently FSM generation is not implemented so you must include
348 an fsm file in order to generate a Segment
349
350 """
351
352 if not readoutmap.endswith(".data.xml"):
353 readoutmap = readoutmap + ".data.xml"
354
355 print(f"Readout map file {readoutmap}")
356
357 includefiles = [
358 "schema/confmodel/dunedaq.schema.xml",
359 "schema/appmodel/application.schema.xml",
360 "schema/appmodel/trigger.schema.xml",
361 "schema/appmodel/fdmodules.schema.xml",
362 "schema/appmodel/wiec.schema.xml",
363 readoutmap,
364 ]
365
366 searchdirs = [path for path in os.environ["DUNEDAQ_DB_PATH"].split(":")]
367 searchdirs.append(os.path.dirname(oksfile))
368 for inc in include:
369
370 match = False
371 inc = inc.removesuffix(".xml")
372 if inc.endswith(".data"):
373 sub_dirs = ["config", "data"]
374 elif inc.endswith(".schema"):
375 sub_dirs = ["schema"]
376 else:
377 sub_dirs = ["*"]
378 inc = inc + "*"
379 for path in searchdirs:
380
381 matches = glob.glob(f"{inc}.xml", root_dir=path)
382 if len(matches) == 0:
383 for search_dir in sub_dirs:
384
385 matches = glob.glob(f"{search_dir}/{inc}.xml", root_dir=path)
386 for filename in matches:
387 if filename not in includefiles:
388 print(f"Adding {filename} to include list")
389 includefiles.append(filename)
390 else:
391 print(f"{filename} already in include list")
392 match = True
393 break
394 if match:
395 break
396 if match:
397 break
398 else:
399 for filename in matches:
400 if filename not in includefiles:
401 print(f"Adding {filename} to include list")
402 includefiles.append(filename)
403 else:
404 print(f"{filename} already in include list")
405 match = True
406 break
407
408 if not match:
409 print(f"Error could not find include file for {inc}")
410 return
411
414 if not oksfile.endswith(".data.xml"):
415 oksfile = oksfile + ".data.xml"
416 print(f"Creating OKS database file {oksfile}")
417 db.create_db(oksfile, includefiles)
418 db.set_active(oksfile)
419
420 detector_connections = db.get_dals(class_name="DetectorToDaqConnection")
421 daqapp_control = db.get_dal(class_name="Service", uid="daqapp_control")
422 rccontroller_control = db.get_dal(class_name="Service", uid="rccontroller_control")
423
424 try:
425 rule = db.get_dal(
426 class_name="NetworkConnectionRule", uid="data-req-readout-net-rule"
427 )
428 except:
429 print(
430 'Expected NetworkConnectionRule "data-req-readout-net-rule" not found in input databases!'
431 )
432 else:
433 netrules = [rule]
434
435 for rule in ["tpset-net-rule", "ts-net-rule", "ta-net-rule"]:
436 netrules.append(db.get_dal(class_name="NetworkConnectionRule", uid=rule))
437
438 try:
439 rule = db.get_dal(
440 class_name="QueueConnectionRule", uid="fd-dlh-data-requests-queue-rule"
441 )
442 except:
443 print(
444 'Expected QueueConnectionRule "fd-dlh-data-requests-queue-rule" not found in input databases!'
445 )
446 else:
447 qrules = [rule]
448 for rule in [
449 "fa-queue-rule",
450 "tp-queue-rule",
451 ]:
452 qrules.append(db.get_dal(class_name="QueueConnectionRule", uid=rule))
453
454 hosts = []
455 if len(hosts_to_use) == 0:
456 for vhost in db.get_dals(class_name="VirtualHost"):
457 if vhost.id == "vlocalhost":
458 hosts.append(vhost.id)
459 if "vlocalhost" not in hosts:
460 cpus = dal.ProcessingResource("cpus", cpu_cores=[0, 1, 2, 3])
461 db.update_dal(cpus)
462 phdal = dal.PhysicalHost("localhost", contains=[cpus])
463 db.update_dal(phdal)
464 host = dal.VirtualHost("vlocalhost", runs_on=phdal, uses=[cpus])
465 db.update_dal(host)
466 hosts.append("vlocalhost")
467 else:
468 for vhost in db.get_dals(class_name="VirtualHost"):
469 if vhost.id in hosts_to_use:
470 hosts.append(vhost.id)
471 assert len(hosts) > 0
472
473 rohw = dal.RoHwConfig(f"rohw-{detector_connections[0].id}")
474 db.update_dal(rohw)
475
476 opmon_conf = db.get_dal(class_name="OpMonConf", uid="slow-all-monitoring")
477
478 appnum = 0
479 nicrec = None
480 flxcard = None
481 wm_conf = None
482 hermes_conf = None
483 ruapps = []
484 for connection in detector_connections:
485
486 det_id = 0
487 for resource in connection.contains:
488 if "ResourceSetAND" in resource.oksTypes():
489 for stream in resource.contains:
490 det_id = stream.contains[0].geo_id.detector_id
491 break
492 break
493
494 if det_id == 0:
495 raise Exception(f"Unable to determine detector ID from Hardware Map!")
496
497 tphandler = db.get_dal(class_name="DataHandlerConf", uid="def-tp-handler")
498
499 if det_id == 2:
500 if "DAPHNEStream" in emulated_file_name:
501 linkhandler = db.get_dal(
502 class_name="DataHandlerConf", uid="def-pds-stream-link-handler"
503 )
504 det_q = db.get_dal(
505 class_name="QueueConnectionRule", uid="pds-stream-raw-data-rule"
506 )
507 else:
508 linkhandler = db.get_dal(
509 class_name="DataHandlerConf", uid="def-pds-link-handler"
510 )
511 det_q = db.get_dal(
512 class_name="QueueConnectionRule", uid="pds-raw-data-rule"
513 )
514
515 elif det_id == 3:
516 linkhandler = db.get_dal(
517 class_name="DataHandlerConf", uid="def-link-handler"
518 )
519 det_q = db.get_dal(
520 class_name="QueueConnectionRule", uid="wib-eth-raw-data-rule"
521 )
522 elif det_id == 11:
523 linkhandler = db.get_dal(
524 class_name="DataHandlerConf", uid="def-tde-link-handler"
525 )
526 det_q = db.get_dal(
527 class_name="QueueConnectionRule", uid="tde-raw-data-rule"
528 )
529
530 hostnum = appnum % len(hosts)
531
532 host = db.get_dal(class_name="VirtualHost", uid=hosts[hostnum])
533
534
535 for resource in connection.contains:
536 if "DetDataReceiver" in resource.oksTypes():
537 receiver = resource
538 break
539
540 if type(receiver).__name__ == "FakeDataReceiver":
541 if nicrec == None:
542 try:
543 stream_emu = db.get_dal(
544 class_name="StreamEmulationParameters", uid="stream-emu"
545 )
546 stream_emu.data_file_name = resolve_asset_file(emulated_file_name)
547 db.update_dal(stream_emu)
548 except:
549 stream_emu = dal.StreamEmulationParameters(
550 "stream-emu",
551 data_file_name=resolve_asset_file(emulated_file_name),
552 input_file_size_limit=5777280,
553 set_t0=True,
554 random_population_size=100000,
555 frame_error_rate_hz=0,
556 generate_periodic_adc_pattern=True,
557 TP_rate_per_channel=1,
558 )
559 db.update_dal(stream_emu)
560
561 print("Generating fake DataReaderConf")
562 nicrec = dal.DPDKReaderConf(
563 f"nicrcvr-fake-gen",
564 template_for="FDFakeReaderModule",
565 emulation_mode=1,
566 emulation_conf=stream_emu,
567 )
568 db.update_dal(nicrec)
569 datareader = nicrec
570 elif type(receiver).__name__ == "DPDKReceiver":
571 if nicrec == None:
572 print("Generating DPDKReaderConf")
573 nicrec = dal.DPDKReaderConf(
574 f"nicrcvr-dpdk-gen", template_for="DPDKReaderModule"
575 )
576 db.update_dal(nicrec)
577 if wm_conf == None:
578 try:
579 wm_conf = db.get_dal("WIBModuleConf", "def-wib-conf")
580 except:
581 print(
582 'Expected WIBModuleConf "def-wib-conf" not found in input databases!'
583 )
584 if hermes_conf == None:
585 try:
586 hermes_conf = db.get_dal("HermesModuleConf", "def-hermes-conf")
587 except:
588 print(
589 'Expected HermesModuleConf "def-hermes-conf" not found in input databases!'
590 )
591
592 datareader = nicrec
593
594 wiec_app = dal.WIECApplication(
595 f"wiec-{connection.id}",
596 application_name="daq_application",
597 runs_on=host,
598 contains=[connection],
599 wib_module_conf=wm_conf,
600 hermes_module_conf=hermes_conf,
601 exposes_service=[daqapp_control],
602 )
603 db.update_dal(wiec_app)
604
605 elif type(receiver).__name__ == "FelixInterface":
606 if flxcard == None:
607 print("Generating Felix DataReaderConf")
608 flxcard = dal.DataReaderConf(
609 f"flxConf-1", template_for="FelixReaderModule"
610 )
611 db.update_dal(flxcard)
612 datareader = flxcard
613 else:
614 print(
615 f"ReadoutGroup contains unknown interface type {type(receiver).__name__}"
616 )
617 continue
618
619 db.commit()
620
621
622 dataRequests = db.get_dal(class_name="Service", uid="dataRequests")
623 timeSyncs = db.get_dal(class_name="Service", uid="timeSyncs")
624 triggerActivities = db.get_dal(class_name="Service", uid="triggerActivities")
625 triggerPrimitives = db.get_dal(class_name="Service", uid="triggerPrimitives")
626
627
628 readout_start = db.get_dal(class_name="ActionPlan", uid="readout-start")
629 readout_stop = db.get_dal(class_name="ActionPlan", uid="readout-stop")
630
631 ru = dal.ReadoutApplication(
632 f"ru-{connection.id}",
633 application_name="daq_application",
634 runs_on=host,
635 contains=[connection],
636 network_rules=netrules,
637 queue_rules=qrules + [det_q],
638 link_handler=linkhandler,
639 data_reader=datareader,
640 opmon_conf=opmon_conf,
641 tp_generation_enabled=tpg_enabled,
642 ta_generation_enabled=tpg_enabled,
643 uses=rohw,
644 exposes_service=[daqapp_control, dataRequests, timeSyncs],
645 action_plans=[readout_start, readout_stop],
646 )
647 if tpg_enabled:
648 ru.tp_handler = tphandler
649 tp_sources = []
650 tpbaseid = (appnum * 3) + 100
651
652
653 for plane in range(1 if det_id not in [3, 10, 11] else 3):
654 s_id = tpbaseid + plane
655 tps_dal = dal.SourceIDConf(
656 f"tp-srcid-{s_id}", sid=s_id, subsystem="Trigger"
657 )
658 db.update_dal(tps_dal)
659 tp_sources.append(tps_dal)
660 ru.tp_source_ids = tp_sources
661 ru.exposes_service += [triggerActivities, triggerPrimitives]
662 appnum = appnum + 1
663 print(f"{ru=}")
664 db.update_dal(ru)
665 db.commit()
666 ruapps.append(ru)
667 if appnum == 0:
668 print(f"No ReadoutApplications generated\n")
669 return
670
671 db.commit()
672
673 if generate_segment:
674
675 fsm = db.get_dal(class_name="FSMconfiguration", uid="FSMconfiguration_noAction")
676 controller = dal.RCApplication(
677 "ru-controller",
678 application_name="drunc-controller",
679 runs_on=host,
680 fsm=fsm,
681 opmon_conf=opmon_conf,
682 exposes_service=[rccontroller_control],
683 )
684 db.update_dal(controller)
685 db.commit()
686
687 seg = dal.Segment(f"ru-segment", controller=controller, applications=ruapps)
688 db.update_dal(seg)
689 db.commit()
690
691 db.commit()
692 return
693
694