BlockDescriptor.h

#ifndef PSCF_BLOCK_DESCRIPTOR_H
#define PSCF_BLOCK_DESCRIPTOR_H
 
/*
* PSCF - Polymer Self-Consistent Field Theory
*
* Copyright 2016 - 2022, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include "PolymerType.h"
#include <util/containers/Pair.h>
 
#include <iostream>
 
namespace Pscf
{ 
 
   using namespace Util;
 
   class BlockDescriptor
   {
   public:
 
      BlockDescriptor();
 
      virtual ~BlockDescriptor();
 
      template <class Archive>
      void serialize(Archive& ar, unsigned int versionId);
 
 
    
      void setId(int id);
  
      void setVertexIds(int vertexAId, int vertexBId);
  
      void setMonomerId(int monomerId);
  
      virtual void setLength(double length);
  
      void setPolymerType(PolymerType::Enum type);
  
 
    
      int id() const;
  
      int monomerId() const;
  
      const Pair<int>& vertexIds() const;
  
      int vertexId(int i) const;
  
      double length() const;
 
      PolymerType::Enum polymerType() const;
 
 
   private:
  
      int id_;
 
      int monomerId_;
 
      double length_;
 
      Pair<int> vertexIds_;
 
      PolymerType::Enum polymerType_;
 
      friend 
      std::istream& operator >> (std::istream& in, BlockDescriptor &block);
 
      friend 
      std::ostream& operator << (std::ostream& out, 
                                 const BlockDescriptor &block);
 
   };
 
   std::istream& operator >> (std::istream& in, BlockDescriptor &block);
 
   std::ostream& operator << (std::ostream& out, const BlockDescriptor &block);
 
   // Inline member functions
 
   /*
   * Get the id of this block.
   */ 
   inline int BlockDescriptor::id() const
   {  return id_; }
 
   /*
   * Get the monomer type id.
   */ 
   inline int BlockDescriptor::monomerId() const
   {  return monomerId_; }
 
   /*
   * Get the pair of associated vertex ids.
   */ 
   inline const Pair<int>& BlockDescriptor::vertexIds() const
   {  return vertexIds_; }
 
   /*
   * Get id of an associated vertex.
   */ 
   inline int BlockDescriptor::vertexId(int i) const
   {  return vertexIds_[i]; }
 
   /*
   * Get the length (number of monomers) in this block.
   */
   inline double BlockDescriptor::length() const
   {  return length_; }
    
   /*
   * Get the polymer type (branched or linear).
   */
   inline PolymerType::Enum BlockDescriptor::polymerType() const
   {  return polymerType_; }
    
   /*
   * Serialize to/from an archive.
   */
   template <class Archive>
   void BlockDescriptor::serialize(Archive& ar, unsigned int)
   {
      ar & id_;
      ar & monomerId_;
      ar & vertexIds_;
      ar & length_;
   }
    
} 
#endif