]> git.treefish.org Git - phys/latlib.git/blob - configcache.cpp
...
[phys/latlib.git] / configcache.cpp
1 #include "configcache.h"
2
3 #include <stdlib.h>
4 #include <iostream>
5 #include <time.h>
6 #include <dirent.h>
7
8 configcache::configcache(const string& cacheid, const int& nequi, const int& nskip, const string& datadir, char **configmem, const int& configMemSize, const int& cachemode){
9   NEQUI = nequi;
10   NSKIP = nskip;
11   DATADIR = datadir;
12   CACHEID = cacheid;
13
14   configMem = (char*)malloc(configMemSize);
15   tmpConfig = (char*)malloc(configMemSize);
16
17   *configmem = configMem;
18   configSize = configMemSize;
19
20   outBuffer = NULL;
21   inBuffer = NULL;
22
23   MODE = cachemode;
24
25   refetchDataFiles = false;
26 }
27
28 string configcache::getFileId(const bool& shortid)
29 {
30   stringstream fileid;
31
32   if(!shortid) fileid << CACHEID << "_" << NEQUI << "_" << NSKIP;
33   for(int ipara=0; ipara<Paras.size(); ipara++)
34     fileid << "_" << Paras[ipara].id << Paras[ipara].val;
35
36   return fileid.str();
37 }
38
39 void configcache::fetchDataFiles()
40 {
41   struct dirent *de=NULL;
42   DIR *d=NULL;
43   
44   d=opendir(DATADIR.c_str());
45   if(d != NULL){
46     while(de = readdir(d)){
47       string filename = de->d_name;
48       if(isValidInFile(filename)) 
49         {
50           inFiles.push_back(filename);
51         }
52     }
53   }
54 }
55
56 bool configcache::isValidInFile(const string& infile)
57 {
58   char *inchar, *inParts;
59   string truncIn, truncOut;
60
61   if( infile.size() < 4 ) return false;
62
63   if( infile.substr(infile.size()-3) != "dat" ) return false;
64
65   inchar = new char [infile.size()+1];
66   strcpy (inchar, infile.c_str());
67   
68   inParts = strtok( inchar, "_" );
69   for(int iPart=0; inParts!=NULL; iPart++)
70     {
71       if( iPart>3 ) { truncIn += "_"; truncIn += inParts; }
72
73       switch(iPart)
74         {
75         case 1: if(inParts != CACHEID) return false; break;
76         case 2: if(atoi(inParts) < NEQUI) return false; break;
77         case 3: if(atoi(inParts) < NSKIP) return false; break;
78         }
79       inParts = strtok( NULL, "_");
80     }
81   truncIn = truncIn.substr(0, truncIn.size()-4);
82
83   delete[] inchar;
84
85   if( truncIn.find( getFileId(true) + "_" ) == string::npos ) return false;
86
87   return true;
88 }
89
90 bool configcache::readConfig()
91 {
92   if(DATADIR == "" || MODE == CACHE_MODE_DISABLED) return false;
93
94   if(refetchDataFiles){
95     refetchDataFiles = false;
96     fetchDataFiles();
97   }
98
99   while(true)
100     {
101       if( (!inFile.is_open()) && inFiles.size() == 0 ) return false;
102
103       while( (!inFile.is_open()) && inFiles.size() > 0 )
104         {
105           inFile.open( (DATADIR + "/" + inFiles.back()).c_str(), std::ios::binary );
106           inFiles.pop_back();
107
108           if( !inFile.is_open() ) continue;
109
110           inBuffer = new boost::iostreams::filtering_istreambuf;
111           inBuffer->push( boost::iostreams::bzip2_decompressor() );
112           inBuffer->push(inFile);
113         }
114
115       if( inFile.is_open() ) 
116         {
117           if( boost::iostreams::read(*inBuffer, tmpConfig, configSize) == configSize )
118             {
119               memcpy(configMem, tmpConfig, configSize);
120               return true;
121             }
122           else inFile.close();
123         }
124     }
125 }
126
127 void configcache::writeConfig()
128 {
129   if( DATADIR == "" || MODE < 2 ) return;
130
131   if(!outFile.is_open()){
132     time_t secstamp = time(NULL);
133
134     outFileName.str("");
135     outFileName << DATADIR << "/" << secstamp << "_" << getFileId() << "_.dat.tmp";    
136     outFile.open( outFileName.str().c_str(), std::ios::binary );
137
138     outBuffer = new boost::iostreams::filtering_ostreambuf;
139     outBuffer->push(boost::iostreams::bzip2_compressor());
140     outBuffer->push(outFile);
141   }
142
143   boost::iostreams::write(*outBuffer, configMem, configSize);
144 }
145
146 void configcache::addPara(const string& parid, const double& val){
147   parameter newPara;
148   newPara.id = parid;
149   newPara.val = val;
150   Paras.push_back(newPara);
151 }
152
153 int configcache::getParIndex(const string& parid){
154   for(int ipara=0; ipara<Paras.size(); ipara++)
155     if(Paras[ipara].id == parid) return ipara;
156 }
157
158 void configcache::setPara(const string& parid, const double& value){
159   Paras[getParIndex(parid)].val = value;
160   finishOutFile();
161   if(inBuffer != NULL) { delete inBuffer; inBuffer=NULL; } 
162   inFile.close();
163   inFiles.clear();
164
165   refetchDataFiles = true;
166 }
167
168 configcache::~configcache()
169 {
170   finishOutFile();
171   delete inBuffer;
172   inBuffer = NULL;
173 }
174
175 void configcache::finishOutFile()
176 {
177   if( outBuffer != NULL )
178     {
179       delete outBuffer;
180       outBuffer = NULL;
181     }
182
183   if( outFile.is_open() )
184     {
185       outFile.close();
186       rename( outFileName.str().c_str(), outFileName.str().substr(0, outFileName.str().size()-4).c_str() );
187     }
188 }