TCMakerDBSCANAlgorithm.cpp

Go to the documentation of this file.

 
#include "triggeralgs/dbscan/TCMakerDBSCANAlgorithm.hpp"
 
#include "TRACE/trace.h"
#define TRACE_NAME "TCMakerDBSCANAlgorithm"
 
namespace triggeralgs {
 
void
TCMakerDBSCANAlgorithm::set_new_tc(const TriggerActivity& input_ta)
{
  m_current_tc = TriggerCandidate();
  m_current_tc.inputs.push_back(input_ta);
  m_current_tp_count = input_ta.inputs.size();
  return;
}
 
void
TCMakerDBSCANAlgorithm::configure(const nlohmann::json& config)
{
  TriggerCandidateMaker::configure(config);
 
  if (config.contains("max_tp_count"))
    m_max_tp_count = config["max_tp_count"];
  return;
}
 
void
TCMakerDBSCANAlgorithm::set_tc_attributes()
{
  auto& first_ta = m_current_tc.inputs.front();
  auto& last_ta = m_current_tc.inputs.back();
 
  m_current_tc.time_start = first_ta.time_start;
  m_current_tc.time_end = last_ta.time_end;
  m_current_tc.time_candidate = last_ta.time_start; // Since this is the TA that closed the TC.
 
  m_current_tc.detid = first_ta.detid;
  m_current_tc.algorithm = TriggerCandidate::Algorithm::kDBSCAN;
  m_current_tc.type = m_tc_type_out;
  return;
}
 
void
TCMakerDBSCANAlgorithm::process(const TriggerActivity& input_ta, std::vector<TriggerCandidate>& output_tcs)
{
  // Start a new TC if not already going.
  if (m_current_tc.inputs.empty()) {
    set_new_tc(input_ta);
    return;
  }
 
  // Check to close the TC based on TP contents.
  if (input_ta.inputs.size() + m_current_tp_count > m_max_tp_count) {
    set_tc_attributes();
    output_tcs.push_back(m_current_tc);
    set_new_tc(input_ta);
    return;
  }
 
  // Append the new TA and increase the TP count.
  m_current_tc.inputs.push_back(input_ta);
  m_current_tp_count += input_ta.inputs.size();
  return;
}
 
// Register algo in TC Factory.
REGISTER_TRIGGER_CANDIDATE_MAKER(TRACE_NAME, TCMakerDBSCANAlgorithm)
 
} // namespace triggeralgs