OOPSE/libmdtools/ZConsReader.cpp

#include "ZConsReader.hpp"
#include "simError.h"

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

  GenericData* data;
  StringData* filename; 

  assert(info != NULL);

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