getdp_s.cpp

/*********************************************************************\
 * Program: getdp_s                                                  *
 * Purpose: extract slice data info from the GENESIS output result   *  
 * Copyright (C) 2012 Tong Zhang                                     *
 *                                                                   *
 * This program is free software: you can distribute it and/or modify*
 * it under the terms of GNU General Public License as published by  *
 * the Free Software Foundation, either version 3 of the License or  *
 * (at your option) any later version.                               *
 *                                                                   *
 * This program is distributed in the hope that it will be useful,   *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of    *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the      *
 * GNU General Public License for more details.                      *
 *                                                                   *
 * You should have received a copy of the GNU General Public License *
 * along with this program. If not see <http://www.gnu.org/licenses/>.*
\*********************************************************************/

/*********************************************************************\
This program is written for parsing output file (TDP)

***Get the given slices z-value

Usage: getdp file1 file2 slice_order
	file1: output file (TDP)
	file2: write into the extrated data
	slice_order: slice record order

Author: Tong ZHANG
e-mail: tzhang@sinap.ac.cn
Created Time: 10:05, Dec. 9th, 2011
\*********************************************************************/

#include "getdp_fun.h"
#include <iostream>

using namespace std;

int main(int argc, char **argv)
{

	if (argc != 4) // if the number of parameters is not enough, then give errors, exit
	{
		if (argc >= 2) cout << "Not enough parameters!\n\n";
		cout << "Usage: " << argv[0] << " file1 file2 s-pos\n";
		cout << "\t" << "file1: "     << "\t" << "TDP output filename, open to read\n";
		cout << "\t" << "file2: "     << "\t" << "data filename, open to write\n";
		cout << "\t" << "s-pos:"      << "\t" << "s-pos in unit of [m]\n\n";
		exit(1);
	}

	string file1name = argv[1]; // tdp filename to read
	string file2name = argv[2]; // filename for writting data

	ifstream file1;
	ofstream file2;

	double s_pos = atof(argv[3]); 	  // s-pos in [m]
	string keystr = "seperation";     // define the keyword string, to find its value
	double slice_sepe = findKeywordValue(file1name, keystr);
	int s_order = s_pos/slice_sepe+1; // transform s_pos into s-record order

	file1.open(file1name.c_str(), ifstream::in);
	file2.open(file2name.c_str(), ofstream::out);

	if (!file1 || !file2) 			// if any one of files cannot open, return error, exit
	{
		cout << "Open file error!\n\n";
		exit(1);
	}

// locate the slice [s-order]
	int count = 0; // read line by line, increase count value when encounter line starts with "*", untile count = s-order
	while (!file1.eof()) // do not stop reading untile the end of file1 
	{
		if (count==s_order)break;
		char buf[MAX_CHARS_PER_LINE];
		file1.getline(buf, MAX_CHARS_PER_LINE);
		char* token[MAX_TOKENS_PER_LINE] = {0};
		token[0] = strtok(buf, DELIMITER.c_str());
		if(token[0] && token[0][0]=='*')count++;
	}

	//pass next 5 useless lines
	string line;
	for (int i =1; i<=5; i++)
	{
		getline(file1, line);
		//file2 << line << "\n";
	}

	//extract the [s-order]th slice, 'result v.s. z'
	getline(file1, line);
	while (!line.empty())
	{
		file2 << line << "\n";
		getline(file1, line);
	}

	//close opened files
	file1.close();
	file2.close();

	return 0;
}