start with the implementation of NeXus IDF V2.

2026-02-04 19:48:38 +01:00
parent ceb9673330
commit 43cb8b2460
2 changed files with 153 additions and 192 deletions
--- a/src/include/PRunDataHandler.h
+++ b/src/include/PRunDataHandler.h
@@ -685,6 +685,159 @@ Bool_t PRunDataHandler::ReadNexusFileIdf1(T& nxs_file)
 template <typename T>
 Bool_t PRunDataHandler::ReadNexusFileIdf2(T& nxs_file)
 {
+  PRawRunData    runData;
+  PRawRunDataSet dataSet;
+  TString str;
+  std::string sstr;
+  Int_t ival;
+  Double_t dval, factor;
+  bool ok;
+
+  // get header information
+
+  // get/set laboratory
+  sstr = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/instrument/source/name"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/instrument/source/name").GetData()[0];
+  runData.SetLaboratory(sstr);
+
+  // get/set beamline
+  sstr = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/instrument/name"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/instrument/name").GetData()[0];
+  runData.SetBeamline(sstr);
+  runData.SetBeamline(sstr);
+
+  // get/set muon source
+  sstr = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/instrument/source/type"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/instrument/source/type").GetData()[0];
+  runData.SetMuonSource(sstr);
+
+  // get/set muon species
+  sstr = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/instrument/source/probe"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/instrument/source/probe").GetData()[0];
+  runData.SetMuonSpecies(sstr);
+
+  // get/set run title
+  sstr = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/title"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/title").GetData()[0];
+  runData.SetRunTitle(sstr);
+
+  // get/set run number
+  ival = -1;
+  if (nxs_file->HasDataset("/raw_data_1/run_number"))
+    ival = nxs_file->template GetDataset<std::string>("/raw_data_1/run_number").GetData()[0];
+  runData.SetRunNumber(ival);
+
+  // get/set temperature
+  dval = PMUSR_UNDEFINED;
+  sstr = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/sample/temperature")) {
+    auto tmp_ds = nxs_file->template GetDataset<float>("/raw_data_1/sample/temperature");
+    dval = tmp_ds.GetData()[0];
+    if (tmp_ds.HasAttribute("units"))
+      sstr = std::any_cast<std::string>(tmp_ds.GetAttribute("units"));
+    if (sstr == "Celcius")
+      dval += 273.16;
+  }
+  runData.SetTemperature(0, dval, 0.0);
+
+         // get/set field
+  dval = PMUSR_UNDEFINED;
+  sstr = "n/a";
+  factor = 1.0;
+  if (nxs_file->HasDataset("/raw_data_1/sample/magnetic_field")) {
+    auto mag_ds = nxs_file->template GetDataset<float>("/raw_data_1/sample/magnetic_field");
+    dval = mag_ds.GetData()[0];
+    if (mag_ds.HasAttribute("units"))
+      sstr = std::any_cast<std::string>(mag_ds.GetAttribute("units"));
+    if (sstr == "Tesla")
+      factor = 1.0e4;
+  }
+  runData.SetField(dval*factor);
+
+  // get/set implantation energy
+  runData.SetEnergy(PMUSR_UNDEFINED);
+
+  // get/set implantation energy
+  runData.SetEnergy(PMUSR_UNDEFINED);
+
+  // get/set moderator HV
+  runData.SetTransport(PMUSR_UNDEFINED);
+
+  // get/set RA HV's (LEM specific)
+  for (UInt_t i=0; i<4; i++)
+    runData.SetRingAnode(i, PMUSR_UNDEFINED);
+
+  // get/set setup
+  sstr  = "n/a";
+  runData.SetSetup(str);
+
+  // get/set sample
+  sstr = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/sample/name"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/sample/name").GetData()[0];
+  runData.SetSample(sstr);
+
+  // get/set orientation
+  runData.SetOrientation("n/a");
+
+         // get/set time resolution (ns)
+  dval = PMUSR_UNDEFINED;
+  sstr = "n/a";
+  factor = 1.0;
+  if (nxs_file->HasDataset("/raw_data_1/instrument/detector_1/resolution")) {
+    auto res_ds = nxs_file->template GetDataset<int>("/raw_data_1/instrument/detector_1/resolution");
+    dval = res_ds.GetData()[0];
+    if (res_ds.HasAttribute("units"))
+      sstr = std::any_cast<std::string>(res_ds.GetAttribute("units"));
+    if ((sstr == "picoseconds") || (sstr == "pico.seconds"))
+      factor = 1.0e-3; // ps -> ns
+  }
+  runData.SetTimeResolution(dval*factor);
+
+  // get/set start/stop time
+  sstr = "n/a";
+  TString date{"n/a"}, time{"n/a"};
+  if (nxs_file->HasDataset("/raw_data_1/start_time"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/start_time").GetData()[0];
+  str = sstr;
+  SplitTimeDate(str, time, date, ok);
+  if (ok) {
+    runData.SetStartTime(time);
+    runData.SetStartDate(date);
+  }
+
+  sstr = "n/a";
+  date = "n/a";
+  time = "n/a";
+  if (nxs_file->HasDataset("/raw_data_1/end_time"))
+    sstr = nxs_file->template GetDataset<std::string>("/raw_data_1/end_time").GetData()[0];
+  str = sstr;
+  SplitTimeDate(str, time, date, ok);
+  if (ok) {
+    runData.SetStopTime(time);
+    runData.SetStopDate(date);
+  }
+
+  // get/set deadtime relevant parameters
+  if (nxs_file->HasDataset("/raw_data_1/instrument/detector1/dead_time")) {
+    std::vector<float> dt;
+    dt = nxs_file->template GetDataset<float>("/raw_data_1/instrument/detector1/dead_time").GetData();
+    runData.SetDeadTimeParam(dt);
+  }
+  if (nxs_file->HasDataset("/raw_data_1/good_frames")) {
+    ival = nxs_file->template GetDataset<float>("/raw_data_1/good_frames").GetData()[0];
+    runData.SetNumberOfGoodFrames(ival);
+  }
+
+  // data with its metadata
+
+  // *** STILL MISSING ***
+
  return true;
 }