ddMd/chemistry/Mask.h

#ifndef DDMD_MASK_H
#define DDMD_MASK_H

/*
* Simpatico - Simulation Package for Polymeric and Molecular Liquids
*
* Copyright 2010 - 2017, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/

#include <util/global.h>

namespace DdMd
{

   class Atom;

   class Mask 
   {

   public:

      /*
      * Maximum number of masked atoms per parent atom.
      */
      static const int Capacity  = 4;

      Mask();

      void clear();

      void append(int id);

      bool isMasked(int id) const;

      int operator [] (int i) const;

      int size() const;

   private:

      int atomIds_[Capacity];      

      int  size_;

   }; 

   // Inline methods

   /*
   * Check if an Atom is masked.
   */
   inline bool Mask::isMasked(int id) const
   {
      for (int i=0; i < size_ ; ++i) {
         if (atomIds_[i] == id) return true;
      }
      return false;
   }

   /*
   * Return the number of masked atoms for this parent Atom.
   */
   inline int Mask::size() const
   { return size_; }

   /*
   * Return value of atom index.
   */
   inline int Mask::operator[] (int i) const
   {
      assert(i >= 0);
      assert(i <  size_);
      return atomIds_[i]; 
   }

} 
#endif