LogEnergy.cpp

/*
* 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 "LogEnergy.h"
#include <ddMd/potentials/pair/PairPotential.h>
#ifdef SIMP_BOND
#include <ddMd/potentials/bond/BondPotential.h>
#endif
#ifdef SIMP_ANGLE
#include <ddMd/potentials/angle/AnglePotential.h>
#endif
#ifdef SIMP_DIHEDRAL
#include <ddMd/potentials/dihedral/DihedralPotential.h>
#endif
#include <util/format/Int.h>
#include <util/format/Dbl.h>
#include <util/mpi/MpiLoader.h>
#include <util/misc/ioUtil.h>

#include <sstream>

namespace DdMd
{

   using namespace Util;

   /*
   * Constructor.
   */
   LogEnergy::LogEnergy(Simulation& simulation) 
    : Analyzer(simulation),
      nSample_(0),
      isInitialized_(false)
   {  setClassName("LogEnergy"); }

   /*
   * Read interval and outputFileName. 
   */
   void LogEnergy::readParameters(std::istream& in) 
   {
      readInterval(in);
      isInitialized_ = true;
   }

   /*
   * Load internal state from an archive.
   */
   void LogEnergy::loadParameters(Serializable::IArchive &ar)
   {
      loadInterval(ar);
      MpiLoader<Serializable::IArchive> loader(*this, ar);
      loader.load(nSample_);
      isInitialized_ = true;
   }

   /*
   * Save internal state to an archive.
   */
   void LogEnergy::save(Serializable::OArchive &ar)
   {
      saveInterval(ar);
      ar << nSample_;
   }

   /*
   * Reset nSample.
   */
   void LogEnergy::clear() 
   {  nSample_ = 0;  }

   /*
   * Dump configuration to file
   */
   void LogEnergy::sample(long iStep) 
   {
      if (isAtInterval(iStep))  {
         Simulation& sim = simulation();
         sim.computeKineticEnergy();
         sim.computePotentialEnergies();
         if (sim.domain().isMaster()) {
            double kinetic = sim.kineticEnergy();
            Log::file() << Int(iStep, 10)
                        << Dbl(kinetic, 15);
            double potential = 0.0;
            double pair = sim.pairPotential().energy();
            potential += pair;
            Log::file() << Dbl(pair, 15);
            #ifdef SIMP_BOND
            if (sim.nBondType()) {
               double bond = sim.bondPotential().energy();
               potential += bond;
               Log::file() << Dbl(bond, 15);
            }
            #endif
            #ifdef SIMP_ANGLE
            if (sim.nAngleType()) {
               double angle = sim.anglePotential().energy();
               potential += angle;
               Log::file() << Dbl(angle, 15);
            }
            #endif
            #ifdef SIMP_DIHEDRAL
            if (sim.nDihedralType()) {
               double dihedral = sim.dihedralPotential().energy();
               potential += dihedral;
               Log::file() << Dbl(dihedral, 15);
            }
            #endif
            Log::file() << Dbl(kinetic + potential, 20)
                        << std::endl;
         }
         ++nSample_;
      }
   }

}