CaSearchWindowMap.h

Go to the documentation of this file.

/* Copyright (C) 2022 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 <boost/serialization/array.hpp>
#include <boost/serialization/string.hpp>
 
#include <algorithm>
#include <cmath>
#include <tuple>
#include <vector>
 
namespace cbm::algo::ca
{
  class SearchWindowMap {
   public:
    struct SearchWindow {
      float xMin;
      float xMax;
      float yMin;
      float yMax;
 
     public:
      std::string ToString() const;
    };

    SearchWindowMap() = default;

    ~SearchWindowMap() = default;

    void SetRange(float xMin, float xMax, int xNbins, float yMin, float yMax, int yNbins)
    {
      fXmin   = xMin;
      fXmax   = xMax;
      fXnBins = xNbins;
      fYmin   = yMin;
      fYmax   = yMax;
      fYnBins = yNbins;
      fNbins  = xNbins * yNbins;
      fBins.clear();
      fBins.resize(fNbins);
    }

    void Clear()
    {
      for (auto& bin : fBins) {
        bin.dx = 0.f;
        bin.dy = 0.f;
      }
    }

    void Update(float x, float y, float dx, float dy, float ms)
    {
      auto& bin = fBins[GetBin(x, y)];
      bin.dx    = std::max(bin.dx, std::fabs(dx));
      bin.dy    = std::max(bin.dy, std::fabs(dy));
      bin.ms    = std::max(bin.ms, std::fabs(ms));
    }

    int GetBin(float x, float y) const
    {
      int binX = (x - fXmin) / (fXmax - fXmin) * fXnBins;
      int binY = (y - fYmin) / (fYmax - fYmin) * fYnBins;
      binX     = std::clamp(binX, 0, fXnBins - 1);
      binY     = std::clamp(binY, 0, fYnBins - 1);
      return binX * fYnBins + binY;
    }

    SearchWindow GetSearchWindow(float x, float y) const
    {
      auto& b = fBins[GetBin(x, y)];
      return {x - b.dx, x + b.dx, y - b.dy, y + b.dy};
    };
 
    std::tuple<SearchWindow, float> GetSearchWindowAndMs(float x, float y) const
    {
      auto& b = fBins[GetBin(x, y)];
      return {{x - b.dx, x + b.dx, y - b.dy, y + b.dy}, b.ms};
    };

    std::string ToString() const;
 
    // TODO: make the members private
   public:
    struct BinData {
      float dx{0.f};
      float dy{0.f};
      float ms{0.f};

      friend class boost::serialization::access;
      template<class Archive>
      void serialize(Archive& ar, const unsigned int)
      {
        ar& dx;
        ar& dy;
        ar& ms;
      }
    };
 
    float fXmin{0.f};              
    float fXmax{0.f};              
    int fXnBins{0};                
    float fYmin{0.f};              
    float fYmax{0.f};              
    int fYnBins{0};                
    int fNbins{0};                 
    std::vector<BinData> fBins{};  

    friend class boost::serialization::access;
    template<class Archive>
    void serialize(Archive& ar, const unsigned int)
    {
      ar& fXmin;
      ar& fXmax;
      ar& fXnBins;
      ar& fYmin;
      ar& fYmax;
      ar& fYnBins;
      ar& fNbins;
      ar& fBins;
    }
  };
}  // namespace cbm::algo::ca