src/io/ZConsReader.cpp

#include "io/ZConsReader.hpp"
#include "utils/simError.h"

ZConsReader::ZConsReader(SimInfo* info)
                   :istream(NULL){

  GenericData* data;
  StringData* filename; 
  this->info = info;

 //retrieve output filename of z force
  data = info->getProperty(ZCONSFILENAME_ID);
  if(!data) {

      
    sprintf( painCave.errMsg,
               "ZConsReader error: If you use an ZConstraint\n"
               " , you must set output filename of z-force.\n");
    painCave.isFatal = 1;
    simError();  

  }
  else{

    filename = dynamic_cast<StringData*>(data);
    
    if(!filename){

      sprintf( painCave.errMsg,
                 "ZConsReader error: Can not get property from SimInfo\n");
      painCave.isFatal = 1;
      simError();  
        
    }
    else{
      zconsFileName = filename->getData();
    }
    
  }

  istream = new ifstream(zconsFileName.c_str());

  if (!istream){
    cerr << "open " << filename << "error" << endl;
    exit(1);
  }
  
  readHeader();  
}

ZConsReader::ZConsReader(const string& filename){
  istream = new ifstream(zconsFileName.c_str());

  if (!istream){
    cerr << "open " << filename << "error" << endl;
    exit(1);
  }
  
  readHeader();  
}

ZConsReader::~ZConsReader(){
  istream->close();
}

int ZConsReader::getNumZMol(){
  return nZMol;
}

vector<int> ZConsReader::getZConsIndex(){
  return index;
}

vector<double> ZConsReader::getZConsPos(){
  return zconsPos;
}

//double ZConsReader::getKRatio(){
//  return kRatio;
//}
    
double ZConsReader::getCurTime(){
  return curTime;
}

vector<double> ZConsReader::getCurZPos(){
  return curZPos;
}

vector<double> ZConsReader::getCurFZ(){
  return curFZ;
}

void ZConsReader::readHeader(){
  const int MAXBUFFERSIZE = 2000;
  char line[MAXBUFFERSIZE];
  int zmolIndex;
  float zmolPos;
  int sscanfCount;
  
  istream->getline(line, MAXBUFFERSIZE);

  cout << line << endl;
  //skip the comment lines
  while(line[0] == '#')
      istream->getline(line, MAXBUFFERSIZE);

  sscanfCount = sscanf(line, "%d", &nZMol);

  if (sscanfCount != 1){
    cerr << "ZConsReader Error : reading file error" << endl;
    exit(1);
  }
  
  for(int i = 0 ; i < nZMol; i++){

    istream->getline(line, MAXBUFFERSIZE);
      
    sscanfCount = sscanf(line, "%d\t%f", &zmolIndex, &zmolPos);
    if (sscanfCount != 2){
      cerr << "ZConsReader Error : reading file error" << endl;
      exit(1);    
    }
    
    index.push_back(zmolIndex);
    zconsPos.push_back(zmolPos);
  }

  curZPos.resize(nZMol);
  curFZ.resize(nZMol);
}

void ZConsReader::readNextFrame(){
  const int MAXBUFFERSIZE = 2000;
  char line[MAXBUFFERSIZE];  
  int tempNZMol;
  int sscanfCount;
  int tempIndex;
  float tempCurTime;
  float tempFZ;
  float tempCurZPos;
  float tempZconsPos;
  
  istream->getline(line, MAXBUFFERSIZE);
  sscanfCount = sscanf(line, "%f", &tempCurTime);
  if (sscanfCount != 1){
    cerr << "ZConsReader Error : reading file error" << endl;
    exit(1);
  }
  curTime = tempCurTime;
  
  istream->getline(line, MAXBUFFERSIZE);
  sscanfCount = sscanf(line, "%d", &tempNZMol);
  if (sscanfCount != 1){
    cerr << "ZConsReader Error : reading file error" << endl;
    exit(1);
  }
  
  if (tempNZMol != nZMol){
    cerr << "ZConsReader Error: reading file error" << endl;
    exit(1);
  }

  for(int i = 0; i < nZMol; i++){
    istream->getline(line, MAXBUFFERSIZE);
    sscanfCount = sscanf(line, "%d\t%f\t%f\t%f", &tempIndex, &tempFZ, &tempCurZPos,&tempZconsPos);
    if (sscanfCount != 4){
      cerr << "ZConsReader Error : reading file error" << endl;
      exit(1);
    }

    index[i] = tempIndex;
    curFZ[i] = tempFZ;
    curZPos[i]= tempCurZPos;
    zconsPos[i] = tempZconsPos;
  }

}

bool ZConsReader::hasNextFrame(){
  return istream->peek() != EOF ? true : false;
}

Revision:	3
Committed:	Fri Sep 24 16:27:58 2004 UTC (20 years, 7 months ago) by tim
File size:	3787 byte(s)
Log Message:	change the #include in source files
#	User	Rev	Content
1	tim	3	#include "io/ZConsReader.hpp"
2			#include "utils/simError.h"
3	gezelter	2
4			ZConsReader::ZConsReader(SimInfo* info)
5			:istream(NULL){
6
7			GenericData* data;
8			StringData* filename;
9			this->info = info;
10
11			//retrieve output filename of z force
12			data = info->getProperty(ZCONSFILENAME_ID);
13			if(!data) {
14
15
16			sprintf( painCave.errMsg,
17			"ZConsReader error: If you use an ZConstraint\n"
18			" , you must set output filename of z-force.\n");
19			painCave.isFatal = 1;
20			simError();
21
22			}
23			else{
24
25			filename = dynamic_cast<StringData*>(data);
26
27			if(!filename){
28
29			sprintf( painCave.errMsg,
30			"ZConsReader error: Can not get property from SimInfo\n");
31			painCave.isFatal = 1;
32			simError();
33
34			}
35			else{
36			zconsFileName = filename->getData();
37			}
38
39			}
40
41			istream = new ifstream(zconsFileName.c_str());
42
43			if (!istream){
44			cerr << "open " << filename << "error" << endl;
45			exit(1);
46			}
47
48			readHeader();
49			}
50
51			ZConsReader::ZConsReader(const string& filename){
52			istream = new ifstream(zconsFileName.c_str());
53
54			if (!istream){
55			cerr << "open " << filename << "error" << endl;
56			exit(1);
57			}
58
59			readHeader();
60			}
61
62			ZConsReader::~ZConsReader(){
63			istream->close();
64			}
65
66			int ZConsReader::getNumZMol(){
67			return nZMol;
68			}
69
70			vector<int> ZConsReader::getZConsIndex(){
71			return index;
72			}
73
74			vector<double> ZConsReader::getZConsPos(){
75			return zconsPos;
76			}
77
78			//double ZConsReader::getKRatio(){
79			// return kRatio;
80			//}
81
82			double ZConsReader::getCurTime(){
83			return curTime;
84			}
85
86			vector<double> ZConsReader::getCurZPos(){
87			return curZPos;
88			}
89
90			vector<double> ZConsReader::getCurFZ(){
91			return curFZ;
92			}
93
94			void ZConsReader::readHeader(){
95			const int MAXBUFFERSIZE = 2000;
96			char line[MAXBUFFERSIZE];
97			int zmolIndex;
98			float zmolPos;
99			int sscanfCount;
100
101			istream->getline(line, MAXBUFFERSIZE);
102
103			cout << line << endl;
104			//skip the comment lines
105			while(line[0] == '#')
106			istream->getline(line, MAXBUFFERSIZE);
107
108			sscanfCount = sscanf(line, "%d", &nZMol);
109
110			if (sscanfCount != 1){
111			cerr << "ZConsReader Error : reading file error" << endl;
112			exit(1);
113			}
114
115			for(int i = 0 ; i < nZMol; i++){
116
117			istream->getline(line, MAXBUFFERSIZE);
118
119			sscanfCount = sscanf(line, "%d\t%f", &zmolIndex, &zmolPos);
120			if (sscanfCount != 2){
121			cerr << "ZConsReader Error : reading file error" << endl;
122			exit(1);
123			}
124
125			index.push_back(zmolIndex);
126			zconsPos.push_back(zmolPos);
127			}
128
129			curZPos.resize(nZMol);
130			curFZ.resize(nZMol);
131			}
132
133			void ZConsReader::readNextFrame(){
134			const int MAXBUFFERSIZE = 2000;
135			char line[MAXBUFFERSIZE];
136			int tempNZMol;
137			int sscanfCount;
138			int tempIndex;
139			float tempCurTime;
140			float tempFZ;
141			float tempCurZPos;
142			float tempZconsPos;
143
144			istream->getline(line, MAXBUFFERSIZE);
145			sscanfCount = sscanf(line, "%f", &tempCurTime);
146			if (sscanfCount != 1){
147			cerr << "ZConsReader Error : reading file error" << endl;
148			exit(1);
149			}
150			curTime = tempCurTime;
151
152			istream->getline(line, MAXBUFFERSIZE);
153			sscanfCount = sscanf(line, "%d", &tempNZMol);
154			if (sscanfCount != 1){
155			cerr << "ZConsReader Error : reading file error" << endl;
156			exit(1);
157			}
158
159			if (tempNZMol != nZMol){
160			cerr << "ZConsReader Error: reading file error" << endl;
161			exit(1);
162			}
163
164			for(int i = 0; i < nZMol; i++){
165			istream->getline(line, MAXBUFFERSIZE);
166			sscanfCount = sscanf(line, "%d\t%f\t%f\t%f", &tempIndex, &tempFZ, &tempCurZPos,&tempZconsPos);
167			if (sscanfCount != 4){
168			cerr << "ZConsReader Error : reading file error" << endl;
169			exit(1);
170			}
171
172			index[i] = tempIndex;
173			curFZ[i] = tempFZ;
174			curZPos[i]= tempCurZPos;
175			zconsPos[i] = tempZconsPos;
176			}
177
178			}
179
180			bool ZConsReader::hasNextFrame(){
181			return istream->peek() != EOF ? true : false;
182			}
183