36 |
|
* [1] Meineke, et al., J. Comp. Chem. 26, 252-271 (2005). |
37 |
|
* [2] Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006). |
38 |
|
* [3] Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008). |
39 |
< |
* [4] Vardeman & Gezelter, in progress (2009). |
39 |
> |
* [4] Kuang & Gezelter, J. Chem. Phys. 133, 164101 (2010). |
40 |
> |
* [5] Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011). |
41 |
|
*/ |
42 |
|
|
43 |
|
/** |
47 |
|
#ifndef IO_RESTREADER_HPP |
48 |
|
#define IO_RESTREADER_HPP |
49 |
|
|
50 |
< |
#include "io/DumpReader.hpp" |
50 |
> |
#include <cstdio> |
51 |
> |
#include <string> |
52 |
> |
#include "brains/SimInfo.hpp" |
53 |
> |
#include "primitives/StuntDouble.hpp" |
54 |
|
|
55 |
|
namespace OpenMD { |
56 |
|
|
57 |
|
/** |
58 |
|
* @class RestReader RestReader.hpp "io/RestReader.hpp" |
59 |
|
*/ |
60 |
< |
class RestReader : public DumpReader { |
60 |
> |
class RestReader { |
61 |
|
public: |
62 |
< |
RestReader(SimInfo* info, const std::string &filename) : DumpReader(info, filename){} |
63 |
< |
void readReferenceStructure(); |
62 |
> |
RestReader(SimInfo* info, const std::string &filename, |
63 |
> |
std::vector<int> stuntDoubleIndex) |
64 |
> |
: info_(info), filename_(filename){ |
65 |
> |
stuntDoubleIndex_ = stuntDoubleIndex; |
66 |
> |
#ifdef IS_MPI |
67 |
> |
|
68 |
> |
if (worldRank == 0) { |
69 |
> |
#endif |
70 |
> |
|
71 |
> |
inFile_ = new std::ifstream(filename_.c_str(), |
72 |
> |
ifstream::in | ifstream::binary); |
73 |
> |
|
74 |
> |
if (inFile_->fail()) { |
75 |
> |
sprintf(painCave.errMsg, |
76 |
> |
"RestReader: Cannot open file: %s\n", |
77 |
> |
filename_.c_str()); |
78 |
> |
painCave.isFatal = 1; |
79 |
> |
simError(); |
80 |
> |
} |
81 |
> |
|
82 |
> |
#ifdef IS_MPI |
83 |
> |
|
84 |
> |
} |
85 |
> |
|
86 |
> |
strcpy(checkPointMsg, "Reference file opened for reading successfully."); |
87 |
> |
errorCheckPoint(); |
88 |
> |
|
89 |
> |
#endif |
90 |
> |
|
91 |
> |
return; |
92 |
> |
} |
93 |
> |
|
94 |
> |
void readReferenceStructure(void); |
95 |
> |
|
96 |
> |
~RestReader(){ |
97 |
> |
#ifdef IS_MPI |
98 |
> |
if (worldRank == 0){ |
99 |
> |
#endif |
100 |
> |
|
101 |
> |
delete inFile_; |
102 |
> |
|
103 |
> |
#ifdef IS_MPI |
104 |
> |
} |
105 |
> |
|
106 |
> |
strcpy(checkPointMsg, "Reference file closed successfully."); |
107 |
> |
errorCheckPoint(); |
108 |
> |
#endif |
109 |
> |
} |
110 |
> |
|
111 |
|
protected: |
112 |
< |
virtual void parseDumpLine(const std::string&); |
113 |
< |
virtual void readFrameProperties(std::istream& inputStream); |
112 |
> |
void parseDumpLine(const std::string& line); |
113 |
> |
void readStuntDoubles(std::istream& inpuStream); |
114 |
> |
void readFrameProperties(std::istream& inputStream); |
115 |
> |
void scanFile(void); |
116 |
> |
void readSet(void); |
117 |
|
private: |
118 |
+ |
|
119 |
+ |
SimInfo* info_; |
120 |
+ |
|
121 |
|
std::vector<Vector3d> all_pos_; |
122 |
+ |
std::vector<int> stuntDoubleIndex_; |
123 |
+ |
|
124 |
+ |
std::istream* inFile_; |
125 |
+ |
std::string filename_; |
126 |
+ |
|
127 |
+ |
std::streampos framePos_; |
128 |
+ |
|
129 |
+ |
const static int bufferSize = 4096; |
130 |
+ |
char buffer[bufferSize]; |
131 |
|
}; |
132 |
|
|
133 |
|
} //end namespace OpenMD |