123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490 |
- /**
- * @file utils/Trace.hpp
- * @author See the AUTHORS file
- */
- /*
- * Copyright (C) 2012-2017 ULCO http://www.univ-littoral.fr
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the 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/>.
- */
- #ifndef ARTIS_UTILS_TRACE
- #define ARTIS_UTILS_TRACE
- #include <algorithm>
- #include <iterator>
- #include <memory>
- #include <mutex>
- #include <sstream>
- #include <string>
- #include <vector>
- #include <map>
- #include <artis/utils/DateTime.hpp>
- #include <artis/kernel/Any.hpp>
- #if WIN32
- #include <iso646.h>
- #endif
- //fwd decl for friend classes in TraceElement
- namespace artis { namespace kernel {
- template < typename T, typename U, typename V, typename W >
- class AbstractCoupledModel;
- template < typename T, typename U, typename V >
- class AbstractAtomicModel;
- template < typename Time, typename Parameters >
- class AbstractModel;
- template < typename T, typename U, typename V >
- class Internals;
- template < typename T, typename U, typename V >
- class Externals;
- }}
- namespace artis { namespace utils {
- enum TraceType { NONE = 0, CHECK = 1, CONSTRUCT, SUBMODEL_ADD, INTERNAL_DECL,
- EXTERNAL_DECL, INTERNAL_LINK, INIT, START, BEFORE_COMPUTE,
- COMPUTE, PUT, AFTER_COMPUTE, DESTRUCT, KERNEL};
- static const std::vector <std::string> TraceTypesStr = {
- "none", "check", "construct", "submodel_add", "internal_decl",
- "external_decl", "internal_link", "init", "start", "before_compute",
- "compute", "put", "after_compute", "destruct", "kernel"};
- class KernelInfo
- {
- public:
- static std::map < std::string, int > elt_dictionary;
- static std::vector < std::string > elt_names;
- static unsigned int term(const std::string& v) {
- if(elt_dictionary.find(v) == elt_dictionary.end()) {
- elt_dictionary[v] = elt_names.size();
- elt_names.push_back(v);
- }
- return elt_dictionary[v];
- }
- static const std::string & term(unsigned int i) {return elt_names[i];}
- KernelInfo() :
- _internal_var(false), _empty(true)
- {
- set_var("");
- set_value("");
- set_tgt_model("");
- set_tgt_internal_var("");
- }
- KernelInfo(const std::string& var, bool internal_var,
- const std::string& value) :
- _internal_var(internal_var), _empty(false)
- {
- set_var(var);
- set_value(value);
- set_tgt_model("");
- set_tgt_internal_var("");
- }
- KernelInfo(const std::string& var, const std::string& tgt_model,
- const std::string& tgt_internal_var) :
- _internal_var(true), _empty(false)
- {
- set_var(var);
- set_value("");
- set_tgt_model(tgt_model);
- set_tgt_internal_var(tgt_internal_var);
- }
- KernelInfo(const std::string& var, bool internal_var) :
- _internal_var(internal_var), _empty(false)
- {
- set_var(var);
- set_value("");
- set_tgt_model("");
- set_tgt_internal_var("");
- }
- KernelInfo(const std::string& tgt_model) :
- _internal_var(false), _empty(false)
- {
- set_var("");
- set_value("");
- set_tgt_model(tgt_model);
- set_tgt_internal_var("");
- }
- void set_var(const std::string& v)
- { _var = KernelInfo::term(v); }
- void set_value(const std::string& v)
- { _value = KernelInfo::term(v); }
- void set_tgt_model(const std::string& v)
- { _tgt_model = KernelInfo::term(v); }
- void set_tgt_internal_var(const std::string& v)
- { _tgt_internal_var = KernelInfo::term(v); }
- const std::string& var() const
- { return KernelInfo::term(_var); }
- const std::string& value() const
- { return KernelInfo::term(_value); }
- const std::string& tgt_model() const
- { return KernelInfo::term(_tgt_model); }
- const std::string& tgt_internal_var() const
- { return KernelInfo::term(_tgt_internal_var); }
- unsigned int var_idx() const
- { return _var; }
- unsigned int tgt_internal_var_idx() const
- { return _tgt_internal_var; }
- unsigned int tgt_model_idx() const
- { return _tgt_model; }
- bool is_internal_var() const
- { return _internal_var; }
- bool empty() const
- { return _empty; }
- std::string to_string() const
- {
- std::ostringstream ss;
- if( not KernelInfo::term(_var).empty() ){
- ss << ":";
- if (not _internal_var)
- ss << "*";
- ss << KernelInfo::term(_var);
- }
- if( not KernelInfo::term(_value).empty()) {
- ss << "=" << KernelInfo::term(_value);
- }
- if(not KernelInfo::term(_tgt_model).empty()) {
- ss << " -> " << KernelInfo::term(_tgt_model);
- if(not KernelInfo::term(_tgt_internal_var).empty()) {
- ss << ":" << KernelInfo::term(_tgt_internal_var);
- }
- }
- return ss.str();
- }
- private:
- unsigned int _var;
- unsigned int _value;
- unsigned int _tgt_model;
- unsigned int _tgt_internal_var;
- bool _internal_var;
- bool _empty;
- };
- template < class Time >
- class TraceElement
- {
- template < typename T, typename U, typename V, typename W >
- friend class artis::kernel::AbstractCoupledModel;
- template < typename T, typename U, typename V >
- friend class artis::kernel::AbstractAtomicModel;
- template < typename T, typename P >
- friend class artis::kernel::AbstractModel;
- template < typename T, typename U, typename V >
- friend class artis::kernel::Internals;
- template < typename T, typename U, typename V >
- friend class artis::kernel::Externals;
- private:
- TraceElement(bool from_kernel, const std::string& model_name,
- typename Time::type time, TraceType type) :
- _time(time), _type(type),
- _from_kernel(from_kernel)
- {
- _model_name = KernelInfo::term(model_name);
- }
- public:
- TraceElement() : _time(Time::null), _type(NONE),
- _from_kernel(false)
- { }
- TraceElement(const std::string& model_name, typename Time::type time,
- TraceType type) :
- _time(time), _type(type),
- _from_kernel(false)
- {
- _model_name = KernelInfo::term(model_name);
- }
- virtual ~TraceElement()
- { }
- const std::string& get_comment() const
- { return _comment; }
- const std::string& get_model_name() const
- { return KernelInfo::term(_model_name); }
- unsigned int get_model_name_idx() const
- { return _model_name; }
- typename Time::type get_time() const
- { return _time; }
- TraceType get_type() const
- { return _type; }
- const KernelInfo& get_kernel_info() const
- { return _kernel_info; }
- void set_comment(const std::string& comment)
- { _comment = comment; }
- bool from_kernel() const
- { return _from_kernel; }
- void set_kernel_info(const KernelInfo& info)
- { _kernel_info = info; }
- std::string to_string(artis::utils::DateFormat date_format =
- artis::utils::DATE_FORMAT_EXTENDED) const
- {
- std::ostringstream ss;
- ss << (from_kernel() ? "KERNEL" : "TRACE ");
- if (get_time() != Time::null) {
- ss << "(" << utils::DateTime::toJulianDayFmt(get_time(),
- date_format)
- << ")";
- }
- ss << ": ";
- ss << "<" + TraceTypesStr[get_type()] + ">";
- ss << " " << get_model_name();
- if (not get_kernel_info().empty()) {
- ss << get_kernel_info().to_string();
- }
- if (not get_comment().empty()) {
- ss << " => " << get_comment();
- }
- return ss.str();
- }
- private:
- unsigned int _model_name;
- typename Time::type _time;
- TraceType _type;
- std::string _comment;
- KernelInfo _kernel_info;
- bool _from_kernel;
- };
- template < class Time >
- class TraceElements : public std::vector < TraceElement < Time > >
- {
- public:
- TraceElements()
- { }
- virtual ~TraceElements()
- { }
- TraceElements filter_model_name(
- const std::string& model_name) const
- {
- TraceElements < Time > result;
- std::copy_if(
- TraceElements < Time >::begin(),
- TraceElements < Time >::end(), std::back_inserter(result),
- [model_name](TraceElement < Time > const & x) {
- if (not x.get_kernel_info().empty()) {
- if (x.get_kernel_info().tgt_model() == model_name) {
- return true;
- }
- }
- return x.get_model_name() == model_name;
- });
- return result;
- }
- TraceElements filter_time(typename Time::type time) const
- {
- TraceElements result;
- std::copy_if(TraceElements < Time >::begin(),
- TraceElements < Time >::end(), std::back_inserter(result),
- [time](TraceElement < Time > const & x)
- { return x.get_time() == time; });
- return result;
- }
- TraceElements filter_type(TraceType type) const
- {
- TraceElements result;
- std::copy_if(TraceElements < Time >::begin(),
- TraceElements < Time >::end(), std::back_inserter(result),
- [type](TraceElement < Time > const & x)
- { return x.get_type() == type; });
- return result;
- }
- TraceElements filter_variable(const std::string& var_name) const
- {
- TraceElements result;
- std::copy_if(TraceElements < Time >::begin(),
- TraceElements < Time >::end(), std::back_inserter(result),
- [var_name](TraceElement < Time > const & x)
- { if(x.get_kernel_info().empty())
- return false;
- return ((x.get_kernel_info().var() == var_name)
- or (x.get_kernel_info().tgt_internal_var() == var_name));
- });
- return result;
- }
- std::string to_string(artis::utils::DateFormat date_format =
- artis::utils::DATE_FORMAT_EXTENDED) const
- {
- std::ostringstream ss;
- for (typename TraceElements < Time >::const_iterator it =
- TraceElements < Time >::begin();
- it != TraceElements < Time >::end(); ++it) {
- ss << it->to_string(date_format) << std::endl;
- }
- return ss.str();
- }
- };
- template < class Time >
- class Trace
- {
- public:
- virtual ~Trace()
- { }
- static Trace& trace()
- {
- std::call_once(_flag, [] ()
- { _instance.reset(new Trace()); }
- );
- return *_instance;
- }
- void clear()
- { _trace.clear(); }
- const TraceElements < Time >& elements() const
- { return _trace; }
- void flush()
- {
- if (_sstream) {
- _element.set_comment(_sstream->str());
- delete _sstream;
- _sstream = 0;
- }
- _trace.push_back(_element);
- }
- std::mutex& mutex()
- { return _mutex; }
- void set_element(const TraceElement < Time >& element)
- { _element = element; }
- void set_kernel_info(const KernelInfo& info)
- { _element.set_kernel_info(info); }
- std::ostringstream& sstream()
- {
- if (_sstream == 0) {
- _sstream = new std::ostringstream();
- }
- return *_sstream;
- }
- private:
- Trace()
- { _sstream = 0; }
- static std::shared_ptr < Trace < Time > > _instance;
- static std::once_flag _flag;
- TraceElements < Time > _trace;
- TraceElement < Time > _element;
- std::ostringstream* _sstream;
- std::mutex _mutex;
- };
- } }
- template < class Time >
- artis::utils::Trace < Time >& operator<<(
- artis::utils::Trace < Time >& trace,
- const artis::utils::TraceElement < Time >& e)
- {
- std::lock_guard < std::mutex > lock(trace.mutex());
- trace.set_element(e);
- return trace;
- }
- template < class Time >
- artis::utils::Trace < Time >& operator<<(
- artis::utils::Trace < Time >& trace,
- const artis::utils::KernelInfo& info)
- {
- std::lock_guard < std::mutex > lock(trace.mutex());
- trace.set_kernel_info(info);
- return trace;
- }
- template < class Time >
- artis::utils::Trace < Time >& operator<<(
- artis::utils::Trace < Time >& trace,
- const std::string& str)
- {
- std::lock_guard < std::mutex > lock(trace.mutex());
- trace.sstream() << str;
- return trace;
- }
- template < class Time >
- artis::utils::Trace < Time >& operator<<(
- artis::utils::Trace < Time >& trace,
- typename Time::type t)
- {
- std::lock_guard < std::mutex > lock(trace.mutex());
- trace.sstream() << t;
- return trace;
- }
- template < class Time >
- std::shared_ptr < artis::utils::Trace < Time > >
- artis::utils::Trace < Time >::_instance;
- template < class Time >
- std::once_flag artis::utils::Trace < Time >::_flag;
- #endif
|