|
1 | 1 | import os
|
2 | 2 | from importlib.util import find_spec
|
| 3 | +from typing import Any |
3 | 4 |
|
4 | 5 | from sed import SedProcessor
|
5 | 6 | from sed.dataset import dataset
|
|
12 | 13 | data_path = dataset.dir
|
13 | 14 |
|
14 | 15 |
|
15 |
| -config_override = { |
| 16 | +config_override: dict[str, Any] = { |
16 | 17 | "core": {
|
17 | 18 | "paths": {
|
18 | 19 | "raw": data_path,
|
|
51 | 52 | system_config=config_file,
|
52 | 53 | collect_metadata=False,
|
53 | 54 | )
|
| 55 | + |
| 56 | +dataset.get("Photon_peak", root_dir="./tutorial") |
| 57 | +data_path = dataset.dir |
| 58 | + |
| 59 | +config_override = { |
| 60 | + "core": { |
| 61 | + "paths": { |
| 62 | + "raw": data_path, |
| 63 | + "processed": data_path + "/processed/", |
| 64 | + }, |
| 65 | + }, |
| 66 | + "dataframe": { |
| 67 | + "ubid_offset": 0, |
| 68 | + "channels": { |
| 69 | + "timeStamp": { |
| 70 | + "index_key": "/zraw/TIMINGINFO/TIME1.BUNCH_FIRST_INDEX.1/dGroup/index", |
| 71 | + "dataset_key": "/zraw/TIMINGINFO/TIME1.BUNCH_FIRST_INDEX.1/dGroup/time", |
| 72 | + }, |
| 73 | + "pulseId": { |
| 74 | + "index_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/index", |
| 75 | + "dataset_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/value", |
| 76 | + }, |
| 77 | + "dldPosX": { |
| 78 | + "index_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/index", |
| 79 | + "dataset_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/value", |
| 80 | + }, |
| 81 | + "dldPosY": { |
| 82 | + "index_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/index", |
| 83 | + "dataset_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/value", |
| 84 | + }, |
| 85 | + "dldTimeSteps": { |
| 86 | + "index_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/index", |
| 87 | + "dataset_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/value", |
| 88 | + }, |
| 89 | + "dldAux": { |
| 90 | + "index_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/index", |
| 91 | + "dataset_key": "/zraw/FLASH.FEL/HEXTOF.DAQ/DLD1/dGroup/value", |
| 92 | + }, |
| 93 | + "bam": { |
| 94 | + "index_key": "/zraw/FLASH.SDIAG/BAM.DAQ/4DBC3.HIGH_CHARGE_ARRIVAL_TIME/dGroup/index", |
| 95 | + "dataset_key": "/zraw/FLASH.SDIAG/BAM.DAQ/4DBC3.HIGH_CHARGE_ARRIVAL_TIME/dGroup/value", |
| 96 | + }, |
| 97 | + "delayStage": { |
| 98 | + "index_key": "/zraw/FLASH.SYNC/LASER.LOCK.EXP/FLASH1.MOD1.PG.OSC/FMC0.MD22.1.ENCODER_POSITION.RD/dGroup/index", |
| 99 | + "dataset_key": "/zraw/FLASH.SYNC/LASER.LOCK.EXP/FLASH1.MOD1.PG.OSC/FMC0.MD22.1.ENCODER_POSITION.RD/dGroup/value", |
| 100 | + }, |
| 101 | + "opticalDiode": { |
| 102 | + "format": "per_train", |
| 103 | + "index_key": "/uncategorised/FLASH.LASER/FLACPUPGLASER1.PULSEENERGY/PG2_incoupl/PULSEENERGY.MEAN/index", |
| 104 | + "dataset_key": "/uncategorised/FLASH.LASER/FLACPUPGLASER1.PULSEENERGY/PG2_incoupl/PULSEENERGY.MEAN/value", |
| 105 | + }, |
| 106 | + }, |
| 107 | + }, |
| 108 | +} |
| 109 | + |
| 110 | +runs = ["40887"] |
| 111 | +for run in runs: |
| 112 | + sp = SedProcessor( |
| 113 | + runs=run, |
| 114 | + config=config_override, |
| 115 | + system_config=config_file, |
| 116 | + collect_metadata=False, |
| 117 | + ) |
0 commit comments