CaTripletSearchWindowMap.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 "CaSearchWindowMap.h"
 
#include <boost/serialization/array.hpp>
#include <boost/serialization/string.hpp>
 
#include <cmath>
 
namespace cbm::algo::ca
{
  class TripletSearchWindowMap {
   public:
    TripletSearchWindowMap() = default;

    ~TripletSearchWindowMap() = default;

    void Init(float targetX, float targetY, float targetZ, float refZ, float xMin, float xMax, int nBinsX, float yMin,
              float yMax, int nBinsY)
    {
      fTargetX = targetX;
      fTargetY = targetY;
      fTargetZ = targetZ;
      fRefZ    = refZ;
      fMap.SetRange(xMin, xMax, nBinsX, yMin, yMax, nBinsY);
    }

    std::tuple<SearchWindowMap::SearchWindow, float> GetSearchWindowAndMs(  //
      float x1, float y1, float z1, float dx1, float dy1,                   //
      float x2, float y2, float z2, float dx2, float dy2                    //
    ) const
    {
      // hit :  x, y, z
      // track : straight line at (x1, y1, z1)
      // tx = (x2 - x1) / (z2 - z1)
      // ty = (y2 - y1) / (z2 - z1)
      // intersection with the plane at zRef
      // c     = (fRefZ - z1) / (z2 - z1);
      // xe = x1 + tx * (zRef-z1) = x1 + (x2 - x1) * c = x1 * (1. - c) + x2 * c
      // ye = y1 + ty * (zRef-z1) = y1 + (y2 - y1) * c = y1 * (1. - c) + y2 * c
      // dxe^2 = dx1^2 * (1. - c)^2 + dx2^2 * c^2
      // dye^2 = dy1^2 * (1. - c)^2 + dy2^2 * c^2
 
      float c           = (fRefZ - z1) / (z2 - z1);
      float cm1         = (1.f - c);
      float xe          = x1 + (x2 - x1) * c;
      float ye          = y1 + (y2 - y1) * c;
      float dxe         = std::sqrt(dx1 * dx1 * cm1 * cm1 + dx2 * dx2 * c * c);
      float dye         = std::sqrt(dy1 * dy1 * cm1 * cm1 + dy2 * dy2 * c * c);
      auto [window, ms] = fMap.GetSearchWindowAndMs(xe, ye);
      window.xMin -= dxe;
      window.xMax += dxe;
      window.yMin -= dye;
      window.yMax += dye;
      return {window, ms};
    };
 
    SearchWindowMap& GetMap() { return fMap; }

    std::string ToString() const;
 
    // TODO: make the members private
   public:
    SearchWindowMap fMap;  
    float fTargetX{0.f};   
    float fTargetY{0.f};   
    float fTargetZ{0.f};   
    float fRefZ{0.f};      

    friend class boost::serialization::access;
    template<class Archive>
    void serialize(Archive& ar, const unsigned int)
    {
      ar& fMap;
      ar& fTargetX;
      ar& fTargetY;
      ar& fTargetZ;
      ar& fRefZ;
    }
  };
}  // namespace cbm::algo::ca