CbmBbaConstraint.h

Go to the documentation of this file.

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

 
#pragma once
 
#include <set>
 
namespace cbm::bba
{
  class Constraint {
   public:
    enum ConstraintType
    {
      kShiftX,  
      kShiftY,  
      kShiftZ   
    };

    Constraint() = default;

    ~Constraint() = default;

    void Reset()
    {
      fBodies.clear();
      fConstraints.clear();
    }

    void AddBody(int bodyIndex) { fBodies.insert(bodyIndex); }

    void AddConstraint(ConstraintType type) { fConstraints.insert(type); }

    const std::set<int>& GetBodies() const { return fBodies; }

    const std::set<ConstraintType>& GetConstraints() const { return fConstraints; }
 
   private:
    std::set<int> fBodies;                  
    std::set<ConstraintType> fConstraints;  
 
    ClassDefNV(Constraint, 0);
  };
}  // namespace cbm::bba