CaMcMatch.h

Go to the documentation of this file.

/* Copyright (C) 2022 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */

 
#ifndef CaMcMatch_h
#define CaMcMatch_h 1
 
#include "CaDefs.h"
 
#include <boost/functional/hash.hpp>
 
#include <string>
 
namespace cbm::algo::ca
{
  struct McMatch {
 
   public:
    McMatch() = default;

    McMatch(const McMatch&) = default;

    McMatch(McMatch&&) = default;

    McMatch& operator=(const McMatch&) = default;

    McMatch& operator=(McMatch&&) = default;

    ~McMatch() = default;

    void AddLink(int linkIndex, float linkWeight)
    {
      if (fNofLinks < 5 && linkWeight > 0.f && linkIndex >= 0) {
        fLinks[fNofLinks]       = linkIndex;
        fLinkWeights[fNofLinks] = linkWeight;
        if (fBestLinkIndex < 0 || linkWeight > fLinkWeights[fBestLinkIndex]) {
          fBestLinkIndex = fNofLinks;
        }
        ++fNofLinks;
      }
    }

    int GetNofLinks() const { return fNofLinks; }

    int GetBestLinkIndex() const { return fBestLinkIndex; }

    int GetLink(int i) const
    {
      if (i >= 0 && i < fNofLinks) {
        return fLinks[i];
      }
      return -1;
    }

    float GetLinkWeight(int i) const
    {
      if (i >= 0 && i < fNofLinks) {
        return fLinkWeights[i];
      }
      return 0.f;
    }
 
    bool IsEmpty() const { return fNofLinks == 0; }
 
    void Clear()
    {
      fNofLinks      = 0;
      fBestLinkIndex = -1;
      fLinks.fill(-1);
      fLinkWeights.fill(0.f);
    }
 
    bool operator==(const McMatch& other) const
    {
      for (int i = 0; i < fNofLinks; ++i) {
        if (fLinkWeights[i] <= 0.f) {
          continue;
        }
        for (int j = 0; j < other.fNofLinks; ++j) {
          if (other.fLinkWeights[j] <= 0.f) {
            continue;
          }
          if (fLinks[i] == other.fLinks[j]) {
            return true;
          }
        }
      }
      return false;
    }
 
    bool operator!=(const McMatch& other) const { return !(*this == other); }
 
    std::string ToString() const
    {
      std::ostringstream out;
      out << "McMatch: NofLinks = " << fNofLinks << ", BestLinkIndex = " << fBestLinkIndex << '\n';
      for (int i = 0; i < fNofLinks; ++i) {
        out << "  Link[" << i << "] = " << fLinks[i] << ", Weight = " << fLinkWeights[i] << '\n';
      }
      return out.str();
    }
 
   private:
    int fNofLinks                     = 0;                     
    std::array<int, 5> fLinks         = {-1, -1, -1, -1, -1};  
    std::array<float, 5> fLinkWeights = {0, 0, 0, 0, 0};       
    int fBestLinkIndex                = -1;                    
  };
}  // namespace cbm::algo::ca
 
#endif  // CaMcMatch_h