...

[phys/latlib.git] / configcache.h

diff --git a/configcache.h b/configcache.h
index 3227233b8be515f7c095f3fb7cc95a6283841197..0af533ac694531185aed99c1caa930c8afd2438d 100644 (file)
--- a/configcache.h
+++ b/configcache.h
@@ -5,6 +5,7 @@
 #include <string>
 #include <fstream>
 #include <sstream>
 #include <string>
 #include <fstream>
 #include <sstream>

+#include <ostream>

 
 #include <boost/iostreams/filtering_streambuf.hpp>
 #include <boost/iostreams/stream.hpp>
 
 #include <boost/iostreams/filtering_streambuf.hpp>
 #include <boost/iostreams/stream.hpp>


@@ -12,6 +13,10 @@
 #include <boost/iostreams/device/array.hpp>
 #include <boost/iostreams/copy.hpp>
 
 #include <boost/iostreams/device/array.hpp>
 #include <boost/iostreams/copy.hpp>
 

+#define CACHE_MODE_DISABLED 0
+#define CACHE_MODE_READ 1
+#define CACHE_MODE_FULL 2
+

 using namespace std;
 
 struct parameter{
 using namespace std;
 
 struct parameter{


@@ -19,23 +24,43 @@ struct parameter{
   double val;
 };
 
   double val;
 };
 

+struct infiledesc {
+  string filename;
+  int nequi;
+  int nskip;
+  bool doVirtualEquilibration;
+  bool extended;
+  int readEquilibratedConfigs;
+};
+

 class configcache{
  public:
   ~configcache();
 class configcache{
  public:
   ~configcache();

-  configcache(const string& cacheid, const int& nequi, const int& nskip, const string& datadir, char **configmem, const int& configMemSize);
-  bool readConfig();
-  void writeConfig();
+  configcache(const string& cacheid, const int& nequi, const int& nskip, const string& datadir, char **configmem, const int& configMemSize, 
+             const int& cachemode=CACHE_MODE_FULL, ostream *_log=NULL);
+  void readConfig(bool& readnewconfig, int& nequileft, vector<unsigned long> *excludeFileHashes=NULL);
+  void writeConfig(int actnequi);

   void addPara(const string& parid, const double& val=0);
   void setPara(const string& parid, const double& value);
   void addPara(const string& parid, const double& val=0);
   void setPara(const string& parid, const double& value);

-  
+  void writeHeader(const string& headerid, const char *header, long unsigned int size, int actnequi);
+  void * getHeader(const string& headerid);
+  string getOutFileName() { return outFileName.str(); }
+  string getInFileName() { return DATADIR + "/" + openFileDesc.filename; }
+  static unsigned long hash(const string& str);
+  void closeInFile() { inFile.close(); }
+  int inFilesLeft() { return inFiles.size(); }
+

  private:
  private:

+  ostream* log;
+  infiledesc openFileDesc;

   void finishOutFile();
   int getParIndex(const string& parid);
   int NEQUI;
   int NSKIP;
   string DATADIR;
   string CACHEID;
   void finishOutFile();
   int getParIndex(const string& parid);
   int NEQUI;
   int NSKIP;
   string DATADIR;
   string CACHEID;

-  string getFileId(const bool& shortid=false);
+  int MODE;
+  string getFileId(int actnequi, const bool& shortid=false);

 
   ofstream outFile;
   ifstream inFile;
 
   ofstream outFile;
   ifstream inFile;


@@ -57,11 +82,29 @@ class configcache{
 
   void fetchDataFiles();
   
 
   void fetchDataFiles();
   

-  bool isValidInFile(const string& infile);
+  bool isValidInFile(const string& infile, infiledesc *filedesc);

 
 

-  vector<string> inFiles;
+  vector<infiledesc> inFiles;

 
   vector<parameter> Paras;
 
   vector<parameter> Paras;

+
+  int readDataToMem(char *tmpData, long unsigned int dataSize);
+
+  void openOutFile(int actnequi);
+
+  int readHeader();
+
+  bool headerWritten;
+
+  int readFullBlock(char *tmpData, long unsigned int dataSize);
+
+  vector< pair<unsigned long, void *> > headerStore;
+
+  void deleteHeaderStore();
+
+  bool readAllHeaders();
+
+  vector<infiledesc>::iterator getNextInfile(vector<unsigned long> *excludeFileHashes);

 };
 
 #endif
 };
 
 #endif