r1d/sweep/SweepParameter.h

#ifndef R1D_SWEEP_PARAMETER_H
#define R1D_SWEEP_PARAMETER_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 <util/containers/DArray.h>
#include <util/containers/GArray.h>
#include <iostream>
#include <string>
#include <pscf/sweep/ParameterType.h>
 
using namespace Util;
 
namespace Pscf {
namespace R1d {
 
   class System;
 
   class SweepParameter
   {
 
   public:
 
      SweepParameter();
 
      SweepParameter(System& system);
 
      void setSystem(System& system)
      {  systemPtr_ = &system; }
 
      void setParameterTypesArray(GArray<ParameterType>& array)
      {  parameterTypesPtr_ = &array; }
 
      ParameterType& parameterType() const;
 
      int parameterTypeId() const
      {  return parameterTypeId_; }
 
      bool isSpecialized() const
      {  return (parameterTypeId_ != -1); }
 
      void getInitial();
 
      void update(double newVal);
 
      std::string type() const;
 
      void writeParamType(std::ostream& out) const;
 
      int id(int i) const
      {  return id_[i];}
 
      double current()
      {  return get_(); }
 
      double initial() const
      {  return initial_; }
 
      double change() const
      {  return change_; }
 
      template <class Archive>
      void serialize(Archive ar, const unsigned int version);
 
   private:
 
      enum ParamType { Block, Chi, Kuhn, Phi_Polymer, Phi_Solvent,
                       Mu_Polymer, Mu_Solvent, Solvent, Cell_Param, 
                       Special, Null};
 
      ParamType type_;
 
      int nId_;
 
      DArray<int> id_;
 
      double initial_;
 
      double change_;
 
      System* systemPtr_;
 
      GArray<ParameterType>* parameterTypesPtr_;
 
      int parameterTypeId_;
 
      void readParamType(std::istream& in);
 
      double get_();
 
      void set_(double newVal);
 
   // friends:
 
      friend
      std::istream& operator >> (std::istream&, SweepParameter&);
 
      friend
      std::ostream&
      operator << (std::ostream&, SweepParameter const &);
 
   };
 
   template <class Archive>
   void SweepParameter::serialize(Archive ar, const unsigned int version)
   {
      serializeEnum(ar, type_, version);
      ar & nId_;
      for (int i = 0; i < nId_; ++i) {
         ar & id_[i];
      }
      ar & initial_;
      ar & change_;
   }
 
}
}
#endif