mirror of
https://github.com/nasa/trick.git
synced 2024-12-22 06:27:49 +00:00
128 lines
3.9 KiB
C++
128 lines
3.9 KiB
C++
#include <sstream>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <limits>
|
|
|
|
#include "trick/MonteVarFile.hh"
|
|
#include "trick/message_proto.h"
|
|
#include "trick/exec_proto.h"
|
|
|
|
Trick::MonteVarFile::MonteVarFile(std::string in_name, std::string in_file_name, unsigned int in_column, std::string in_unit) : input_file_stream(NULL) {
|
|
this->name = in_name;
|
|
this->column = in_column;
|
|
this->unit = in_unit;
|
|
|
|
set_file_name(in_file_name);
|
|
}
|
|
|
|
Trick::MonteVarFile::~MonteVarFile() {
|
|
delete input_file_stream;
|
|
}
|
|
|
|
// Composite the various properties of this MonteVarFile.
|
|
std::string Trick::MonteVarFile::describe_variable() {
|
|
std::stringstream ss;
|
|
|
|
ss << "#NAME:\t\t" << this->name << "\n"
|
|
<< "#TYPE:\t\tFILE\n"
|
|
<< "#UNIT:\t\t" << this->unit << "\n"
|
|
<< "#FILE:\t\t" << this->file_name << "\n"
|
|
<< "#COLUMN:\t" << this->column << "\n";
|
|
|
|
return ss.str();
|
|
}
|
|
|
|
std::string Trick::MonteVarFile::get_next_value() {
|
|
|
|
// Reopen the file if not open.
|
|
if(!input_file_stream->is_open()) {
|
|
input_file_stream->open(this->file_name.c_str(), std::ifstream::in);
|
|
|
|
// If the stream position has been set, re-open file to this position.
|
|
if(this->stream_position != NULL) {
|
|
input_file_stream->seekg(this->stream_position);
|
|
}
|
|
}
|
|
|
|
if (input_file_stream->good()) {
|
|
std::string line;
|
|
// Skip the comments and empty lines in the data file.
|
|
do {
|
|
std::getline(*input_file_stream, line);
|
|
|
|
if(input_file_stream->eof()) {
|
|
input_file_stream->close();
|
|
return "EOF";
|
|
}
|
|
}
|
|
while(line[0] == '#' || line[0] == '\0');
|
|
|
|
// Store the current stream position and close the file.
|
|
this->stream_position = input_file_stream->tellg();
|
|
input_file_stream->close();
|
|
|
|
// Count the number of columns in the input file.
|
|
char *token;
|
|
unsigned int ntokens = 0;
|
|
char* temp_str = strdup(line.c_str());
|
|
token = strtok(temp_str, " \t");
|
|
while (token != NULL) {
|
|
token = strtok(NULL, " \t");
|
|
ntokens++;
|
|
}
|
|
|
|
// Verify the input column number is valid.
|
|
if ((column == 0) || (column > ntokens)) {
|
|
char string[100];
|
|
sprintf(string, "Trick:MonteVarFile An invalid column number %d, valid column numbers are 1 - %d", column, ntokens);
|
|
exec_terminate_with_return(-1, __FILE__, __LINE__, string);
|
|
}
|
|
|
|
// Get the next value.
|
|
temp_str = strdup(line.c_str());
|
|
token = strtok(temp_str, " \t");
|
|
|
|
for(unsigned int i = 1; i < column; i++) {
|
|
// Iterate through each token in the temp_str.
|
|
if(token != NULL)
|
|
token = strtok(NULL, " \t");
|
|
}
|
|
|
|
// Return the value as a string.
|
|
this->value = token;
|
|
std::stringstream ss;
|
|
|
|
if(unit.empty())
|
|
ss << this->name << " = " << token;
|
|
else
|
|
ss << this->name << " = " << "trick.attach_units(\"" << unit << "\", " << token << ")";
|
|
|
|
return ss.str();
|
|
}
|
|
char string[100];
|
|
sprintf(string, "Trick:MonteVarFile the input file \"%s\" is not open for reading", file_name.c_str());
|
|
exec_terminate_with_return(-1, __FILE__, __LINE__, string);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void Trick::MonteVarFile::set_file_name(std::string in_file_name) {
|
|
delete input_file_stream;
|
|
|
|
input_file_stream = new std::ifstream(in_file_name.c_str(), std::ifstream::in);
|
|
if (input_file_stream->fail()) {
|
|
std::stringstream string_stream;
|
|
|
|
string_stream << "Error: " << strerror(errno) << std::endl
|
|
<< " Trick:MonteVarFile input file \"" << in_file_name << "\" failed to open";
|
|
|
|
exec_terminate_with_return(-1, __FILE__, __LINE__, string_stream.str().c_str());
|
|
}
|
|
this->file_name = in_file_name;
|
|
}
|
|
|
|
void Trick::MonteVarFile::set_column(unsigned int in_column) {
|
|
this->column = in_column;
|
|
}
|