sndPlaneTools.cxx

Go to the documentation of this file.

#include "sndPlaneTools.h"
 
#include <vector>
 
#include "TClonesArray.h"
#include "Scifi.h"
#include "MuFilter.h"
#include "sndConfiguration.h"
#include "sndScifiPlane.h"
#include "sndUSPlane.h"
#include "sndScifiHit.h"
#include "MuFilterHit.h"
 
std::vector<snd::analysis_tools::ScifiPlane> snd::analysis_tools::FillScifi(const snd::Configuration &configuration, TClonesArray *sf_hits, Scifi *scifi_geometry)
{
 
  std::vector<snd::analysis_tools::ScifiPlane> scifi_planes;
  int n_sf_hits{sf_hits->GetEntries()};
 
  const int max_station = configuration.scifi_n_stations;
  std::vector<std::vector<sndScifiHit*>> stations_hits(max_station);
 
  for (int i{0}; i < n_sf_hits; ++i) {
      auto hit = static_cast<sndScifiHit*>(sf_hits->At(i));
      int station_id = hit->GetStation()-1;
 
      if (station_id > -1 && station_id < max_station) {
          stations_hits[station_id].push_back(hit);
      }
      else throw std::runtime_error{"Invalid SciFi station"};
  }
  for (int st{0}; st < max_station; ++st) {
          scifi_planes.emplace_back(snd::analysis_tools::ScifiPlane(stations_hits[st], configuration, scifi_geometry, st+1));
  }
  return scifi_planes;
}
 
 
std::vector<snd::analysis_tools::USPlane> snd::analysis_tools::FillUS(const snd::Configuration &configuration, TClonesArray *mufi_hits, MuFilter *mufilter_geometry)
{
 
  std::vector<snd::analysis_tools::USPlane> us_planes;
  int n_mufi_hits{mufi_hits->GetEntries()};
 
  const int n_station = configuration.us_n_stations;
  std::vector<std::vector<MuFilterHit*>> plane_hits(n_station);
 
  for (int i{0}; i < n_mufi_hits; ++i) {
    auto hit = static_cast<MuFilterHit*>(mufi_hits->At(i));
    if (hit->GetSystem()!=2) continue;
    int station_id = hit->GetPlane();
    if (station_id > -1 && station_id < n_station) {
          plane_hits[station_id].push_back(hit);
      }
      else throw std::runtime_error{"Invalid US plane"};
  }
  for (int st{0}; st < n_station; ++st) {
          us_planes.emplace_back(snd::analysis_tools::USPlane(plane_hits[st], configuration, mufilter_geometry, st+1));
  }
  return us_planes;
}