mirror of
https://github.com/nasa/trick.git
synced 2024-12-19 05:07:54 +00:00
118 lines
3.7 KiB
C++
118 lines
3.7 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);
|
|
buffer = new char[4096];
|
|
}
|
|
|
|
Trick::MonteVarFile::~MonteVarFile() {
|
|
delete input_file_stream;
|
|
delete buffer;
|
|
}
|
|
|
|
// 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() {
|
|
if (input_file_stream->good()) {
|
|
double file_value;
|
|
char *current_position = buffer;
|
|
do {
|
|
input_file_stream->getline(buffer, 4096);
|
|
file_value = strtold(buffer, ¤t_position);
|
|
if (input_file_stream->eof()) {
|
|
input_file_stream->close();
|
|
return "EOF";
|
|
}
|
|
} while (file_value == 0 && current_position == buffer && input_file_stream->good());
|
|
|
|
// Count the number of columns in the input file
|
|
char* token;
|
|
unsigned int ntokens = 0;
|
|
char temp_str[4096];
|
|
strcpy(temp_str, buffer) ;
|
|
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);
|
|
}
|
|
|
|
if (current_position != buffer) {
|
|
for (unsigned int i = 1; i < column; ++i) {
|
|
file_value = strtold(current_position, ¤t_position);
|
|
}
|
|
|
|
std::stringstream string_stream;
|
|
string_stream.precision(std::numeric_limits<double>::digits10);
|
|
string_stream << file_value ;
|
|
value = string_stream.str();
|
|
string_stream.str("");
|
|
if (unit.empty()) {
|
|
string_stream << name << " = " << file_value ;
|
|
}
|
|
else {
|
|
string_stream << name << " = " << "trick.attach_units(\"" << unit << "\", " << file_value
|
|
<< ")";
|
|
}
|
|
return string_stream.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;
|
|
}
|
|
|