In the middle of adding a general binary writer

2025-06-17 15:27:06 +01:00 · 2016-10-04 11:24:08 +01:00
parent cfbc1a26b8
commit d9b5fbd374
4 changed files with 163 additions and 55 deletions
--- a/lib/qcd/hmc/BinaryCheckpointer.h
+++ b/lib/qcd/hmc/BinaryCheckpointer.h
@ -0,0 +1,106 @@
+    /*************************************************************************************
+
+    Grid physics library, www.github.com/paboyle/Grid 
+
+    Source file: ./lib/qcd/hmc/NerscCheckpointer.h
+
+    Copyright (C) 2015
+
+Author: paboyle <paboyle@ph.ed.ac.uk>
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License along
+    with this program; if not, write to the Free Software Foundation, Inc.,
+    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+    See the full license in the file "LICENSE" in the top level distribution directory
+    *************************************************************************************/
+    /*  END LEGAL */
+#ifndef NERSC_CHECKPOINTER
+#define NERSC_CHECKPOINTER
+
+#include <string>
+#include <iostream>
+#include <sstream>
+
+
+namespace Grid{
+  namespace QCD{
+    
+    template<class fobj,class sobj>
+    struct BinarySimpleUnmunger{
+      void operator() (sobj &in,fobj &out,uint32_t &csum){
+        fobj = sobj;
+        csum =0;
+    };
+
+
+    // Only for gauge fields
+    template<class Impl> 
+    class BinaryHmcCheckpointer : public HmcObservable<typename Impl::Field> {
+    private:
+      std::string configStem;
+      std::string rngStem;
+      int SaveInterval;
+    public:
+      INHERIT_FIELD_TYPES(Impl); // The Field is a Lattice object
+
+      typedef typename Field::vector_object vobj;
+      typedef typename vobj::scalar_object sobj;
+ 
+
+      BinaryHmcCheckpointer(std::string cf, std::string rn,int savemodulo) {
+        configStem  = cf;
+        rngStem     = rn;
+        SaveInterval= savemodulo;
+      };
+
+      void TrajectoryComplete(int traj, Field &U, GridSerialRNG &sRNG,
+                              GridParallelRNG &pRNG) {
+        if ((traj % SaveInterval) == 0) {
+          std::string rng;
+          {
+            std::ostringstream os;
+            os << rngStem << "." << traj;
+            rng = os.str();
+          }
+          std::string config;
+          {
+            std::ostringstream os;
+            os << configStem << "." << traj;
+            config = os.str();
+          }
+
+          int prec = getPrecision<Field>::value; //get precision of oject
+          std::string floating_point = "IEEE64BIG"; //default precision
+          if (prec == 1)
+            floating_point = "IEEE32BIG"
+
+          BinarySimpleUnmunger<sobj, sobj> munge;
+          BinaryIO::writeRNGSerial(sRNG, pRNG, rng, 0);
+          BinaryIO::writeObjectParallel<vobj, sobj>(U, config, munge, 0 ,floating_point);
+        }
+      };
+
+      void CheckpointRestore(int traj, GaugeField &U, GridSerialRNG &sRNG, GridParallelRNG & pRNG ){
+
+  std::string rng;   { std::ostringstream os; os << rngStem     <<"."<< traj; rng = os.str(); }
+  std::string config;{ std::ostringstream os; os << configStem  <<"."<< traj; config = os.str();}
+
+  NerscField header;
+  NerscIO::readRNGState(sRNG,pRNG,header,rng);
+  NerscIO::readConfiguration(U,header,config);
+      };
+
+    };
+}}
+#endif