NamTracer.h

// Copyright 2008 Michael Marsh, University of Maryland.
//
// This file is part of pydtn.
//
// pydtn 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.
//
// pydtn 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 pydtn.  If not, see <http://www.gnu.org/licenses/>.
//
// The views and conclusions contained in the software and documentation
// are those of the authors and should not be interpreted as representing
// official policies, either expressed or implied, of the University
// of Maryland.
//
// pydtn extends and embeds the Python interpreter, which is
// Copyright 2001-2006 Python Software Foundation, All Rights Reserved,
// and is released under the PSF License Agreement.
//
// RANLUX random number generation uses the Boost library,
// Copyright 1994-2006 by various authors (details in individual files),
// which is released under the Boost Software License, Version 1.0.

#ifndef __NAM_TRACE_NAM_TRACER_H__
#define __NAM_TRACE_NAM_TRACER_H__

#include "config.h"

#include "pydtn/Tracer.h"
#include "simlpy/Entity.h"
#include "simlpy/Event.h"
#include "dtn/ByteString.h"
#include <map>
#include <set>
#include <ostream>


class NamTracer;

class NamTraceEvent : public Event
{
   public :
      NamTraceEvent( Entity* source,
                     NamTracer* destination,
                     const Time& t ) :
         Event( source, t ),
         m_destination( destination )
      {}

      virtual ~NamTraceEvent() {}

      bool handle();

   private :
      NamTracer* m_destination;
};

class NamTracer : public Tracer, public Entity
{
   public :
      typedef std::map< DTN::ByteString , int >  NodeIDMap;
      typedef std::set< const WrapNode* > NodeSet;
      typedef std::set< const WrapLink* > LinkSet;

      NamTracer( std::ostream& s );
      virtual ~NamTracer();

      void setQueueState( bool q ) { m_queueState = q; }

      void node( const WrapNode& wn );

      void link( const WrapLink& wl );

      void enqueue( const DTN::Bundle& b, const DTN::Node* n );

      void dequeue( const DTN::Bundle& b, const DTN::Node* n );

      void send( const DTN::Bundle& b, const DTN::Node* n );

      void receive( const DTN::Bundle& b, const DTN::Node* n );

      void drop( const DTN::Bundle& b,
                 const DTN::DropCause& c,
                 const DTN::Node* n );

      Entity* create() const { return 0; }
      void configure( const ArgList& args ) {}
      void emit( const ArgList& args ) {}
      bool handler( NamTraceEvent& event );
      std::string identifier() const { return "namtrace"; }

   protected :
      void traceNode( const WrapNode& wn, bool initial = false );

      void traceLink( const WrapLink& wl, bool initial = false );

      void traceBundle( const DTN::Bundle& b );

      int nodeID( const WrapNode& wn );

      int nodeID( const DTN::ByteString& addr );

      DTN::ByteString vQueue( const WrapNode& wn );

      DTN::ByteString vQueue( const DTN::ByteString& addr );

      double time( const Time& t );

      std::string stringify( const DTN::ByteString& b ) const;

   private :
      std::ostream&  m_stream;
      int            m_lastNode;
      NodeIDMap      m_nodes;
      bool           m_queueState;
      bool           m_tracing;
      bool           m_notified;
      NodeSet        m_nodeSet;
      LinkSet        m_linkSet;
};


#endif // __NAM_TRACE_NAM_TRACER_H__

---