CaDataManager.h

Go to the documentation of this file.

/* Copyright (C) 2022-2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov, Sergei Zharko [committer] */
 
 
#pragma once  // include this header only once per compilation unit
 
#include "CaDefs.h"
#include "CaInputData.h"
 
namespace cbm::algo::ca
{
  class alignas(constants::misc::Alignment) DataManager {
   public:
    DataManager() = default;
 
    ~DataManager() = default;
 
    DataManager(const DataManager& other) = delete;
 
    DataManager(DataManager&& other) = delete;
 
    DataManager& operator=(const DataManager& other) = delete;
 
    DataManager& operator=(DataManager&& other) = delete;
 
    int GetNofHits() { return fInputData.fvHits.size(); }
 
    void ReadInputData(const std::string& fileName);
 
    void ReserveNhits(HitIndex_t nHits) { fInputData.fvHits.reserve(nHits); }
 
    void ResetInputData(HitIndex_t nHits = 0) noexcept;
 
    void PushBackHit(const Hit& hit, int64_t streamId)
    {
      if (fInputData.fvStreamStartIndices.size() == 0 || fLastStreamId != streamId) {  // new data stream
        fLastStreamId = streamId;
        fInputData.fvStreamStartIndices.push_back(fInputData.fvHits.size());
        // for a case.. it is fixed later in InitData()
        fInputData.fvStreamStopIndices.push_back(fInputData.fvHits.size());
      }
      fInputData.fvHits.push_back(hit);
    }
 
    void PushBackHit(const Hit& hit) { fInputData.fvHits.push_back(hit); }
 
    void SetNhitKeys(int nKeys) { fInputData.fNhitKeys = nKeys; }
 
    InputData&& TakeInputData();
 
    void WriteInputData(const std::string& fileName) const;
 
 
   private:
    void InitData();
 
    template<int Level>
    bool CheckInputData() const;
 
 
    // ***************************
    // ** Member variables list **
    // ***************************
    InputData fInputData{};     
    int64_t fLastStreamId{-1};  
  };
 
 
  // *************************************
  // ** Inline functions implementation **
  // *************************************
 
  // -------------------------------------------------------------------------------------------------------------------
  //
 
  // TODO: Complete this function
  template<int Level>
  inline bool DataManager::CheckInputData() const
  {
    if constexpr (Level == 0) {
      return true;
    }                                // Level = 0 -> do nothing
    else if constexpr (Level > 0) {  // Level = 1 and higher
      // ----- Check if the hits container is not empty ----------------------------------------------------------------
      if (fInputData.fvHits.size() == 0) {
        LOG(warn) << "DataManager [check input]: Sample contains empty hits, tracking will not be executed";
        return false;
      }
 
      // ----- Checks if the number of hit keys is valid ---------------------------------------------------------------
      if (fInputData.fNhitKeys < 1) {
        LOG(error) << "DataManager [check input]: Incorrect number of keys passed (" << fInputData.fNhitKeys
                   << "), tracking will not be executed";
        return false;
      }
 
      // ----- Checks the indexes of first and last hits in stations
      // TODO: Add one of the two following checks for fvStartHitIn
 
      if constexpr (Level > 1) {  // Level = 2 and higher
        // ----- Checks for hits sorting -------------------------------------------------------------------------------
        // TODO...
        if constexpr (Level > 2) {  // Level = 3 and higher
          // ----- Checks for consistency of the particular hit --------------------------------------------------------
          // TODO...
        }
      }
      return true;
    }
    return true;
  }
}  // namespace cbm::algo::ca