mirror of
https://github.com/nasa/trick.git
synced 2024-12-20 21:53:10 +00:00
91 lines
2.9 KiB
C++
91 lines
2.9 KiB
C++
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <stdio.h>
|
|
#include <stdarg.h>
|
|
#include <time.h>
|
|
#include <math.h>
|
|
#include <unistd.h>
|
|
|
|
#include "trick/MessagePublisher.hh"
|
|
#include "trick/message_proto.h"
|
|
#include "trick/exec_proto.h"
|
|
|
|
#define MAX_MSG_HEADER_SIZE 256
|
|
|
|
Trick::MessagePublisher * the_message_publisher ;
|
|
|
|
Trick::MessagePublisher::MessagePublisher() {
|
|
|
|
sim_name = " " ;
|
|
the_message_publisher = this ;
|
|
|
|
tics_per_sec = 1000000 ;
|
|
set_print_format() ;
|
|
|
|
}
|
|
|
|
Trick::MessagePublisher::~MessagePublisher() {
|
|
the_message_publisher = NULL;
|
|
}
|
|
|
|
void Trick::MessagePublisher::set_print_format() {
|
|
num_digits = (int)round(log10((double)tics_per_sec)) ;
|
|
snprintf(print_format, sizeof(print_format), "|L %%3d|%%s|%%s|%%s|T %%d|%%lld.%%0%dlld| ", num_digits) ;
|
|
}
|
|
|
|
int Trick::MessagePublisher::init() {
|
|
tics_per_sec = exec_get_time_tic_value() ;
|
|
set_print_format() ;
|
|
return 0 ;
|
|
}
|
|
|
|
int Trick::MessagePublisher::publish(int level , std::string message) {
|
|
|
|
/** @par Design Details: */
|
|
std::list<Trick::MessageSubscriber *>::iterator p ;
|
|
|
|
char date_buf[MAX_MSG_HEADER_SIZE];
|
|
char header_buf[MAX_MSG_HEADER_SIZE];
|
|
char hostname[64];
|
|
time_t date ;
|
|
std::string header ;
|
|
long long tics = exec_get_time_tics() ;
|
|
|
|
/** @li Create message header with level, date, host, sim name, process id, sim time. */
|
|
date = time(NULL) ;
|
|
strftime(date_buf, (size_t) 20, "%Y/%m/%d,%H:%M:%S", localtime(&date));
|
|
(void) gethostname(hostname, (size_t) 48);
|
|
snprintf(header_buf, sizeof(header_buf), print_format , level, date_buf, hostname,
|
|
sim_name.c_str(), exec_get_process_id(), tics/tics_per_sec ,
|
|
(long long)((double)(tics % tics_per_sec) * (double)(pow(10 , num_digits)/tics_per_sec)) ) ;
|
|
header = header_buf ;
|
|
|
|
/** @li Go through all its subscribers and send a message update to the subscriber that is enabled. */
|
|
if ( ! subscribers.empty() ) {
|
|
for ( p = subscribers.begin() ; p != subscribers.end() ; p++ ) {
|
|
if ( (*p)->enabled ) {
|
|
(*p)->update(level , header , message) ;
|
|
}
|
|
}
|
|
} else {
|
|
// If there are no subscribers, that probably means things have not been inited yet... just print message only
|
|
|
|
// Building the final string in a temporary stream ensures an automic call to cout, which prevents
|
|
// multithreaded sims from interleaving header and message elements.
|
|
std::ostringstream oss;
|
|
oss << header << message ;
|
|
std::cout << oss.str() << std::flush ; } return(0) ;
|
|
|
|
}
|
|
|
|
Trick::MessageSubscriber * Trick::MessagePublisher::getSubscriber( std::string sub_name ) {
|
|
std::list<Trick::MessageSubscriber *>::iterator lit ;
|
|
for ( lit = subscribers.begin() ; lit != subscribers.end() ; lit++ ) {
|
|
if ( ! (*lit)->name.compare(sub_name) ) {
|
|
return *lit ;
|
|
}
|
|
}
|
|
return NULL ;
|
|
}
|