alpar@119: /* -*- C++ -*-
alpar@119:  *
alpar@119:  * This file is a part of LEMON, a generic C++ optimization library
alpar@119:  *
alpar@119:  * Copyright (C) 2003-2008
alpar@119:  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
alpar@119:  * (Egervary Research Group on Combinatorial Optimization, EGRES).
alpar@119:  *
alpar@119:  * Permission to use, modify and distribute this software is granted
alpar@119:  * provided that this copyright notice appears in all copies. For
alpar@119:  * precise terms see the accompanying LICENSE file.
alpar@119:  *
alpar@119:  * This software is provided "AS IS" with no warranty of any kind,
alpar@119:  * express or implied, and with no claim as to its suitability for any
alpar@119:  * purpose.
alpar@119:  *
alpar@119:  */
alpar@119: 
alpar@119: #ifndef LEMON_COUNTER_H
alpar@119: #define LEMON_COUNTER_H
alpar@119: 
alpar@119: #include <string>
alpar@119: #include <iostream>
alpar@119: 
alpar@119: ///\ingroup timecount
alpar@119: ///\file
alpar@119: ///\brief Tools for counting steps and events
alpar@119: 
alpar@119: namespace lemon 
alpar@119: {
alpar@119: 
alpar@119:   template<class P> class _SubNoCounter;
alpar@119: 
alpar@119:   template<class P>
alpar@119:   class _SubCounter 
alpar@119:   {
alpar@119:     P &_parent;
alpar@119:     std::string _title;
alpar@119:     std::ostream &_os;
alpar@119:     int count;
alpar@119:   public:
alpar@119: 
alpar@119:     typedef _SubCounter<_SubCounter<P> > SubCounter;
alpar@119:     typedef _SubNoCounter<_SubCounter<P> > SubNoCounter;
alpar@119: 
alpar@119:     _SubCounter(P &parent)
alpar@119:       : _parent(parent), _title(), _os(std::cerr), count(0) {}
alpar@119:     _SubCounter(P &parent,std::string title,std::ostream &os=std::cerr)
alpar@119:       : _parent(parent), _title(title), _os(os), count(0) {}
alpar@119:     _SubCounter(P &parent,const char *title,std::ostream &os=std::cerr)
alpar@119:       : _parent(parent), _title(title), _os(os), count(0) {}
alpar@119:     ~_SubCounter() { 
alpar@119:       _os << _title << count <<std::endl;
alpar@119:       _parent+=count;
alpar@119:     }
alpar@119:     _SubCounter &operator++() { count++; return *this;}
alpar@119:     int operator++(int) { return count++; }
alpar@119:     _SubCounter &operator--() { count--; return *this;}
alpar@119:     int operator--(int) { return count--; }
alpar@119:     _SubCounter &operator+=(int c) { count+=c; return *this;}
alpar@119:     _SubCounter &operator-=(int c) { count-=c; return *this;}
alpar@119:     void reset(int c=0) {count=c;}
alpar@119:     operator int() {return count;}
alpar@119:   };
alpar@119: 
alpar@119:   template<class P>
alpar@119:   class _SubNoCounter 
alpar@119:   {
alpar@119:     P &_parent;
alpar@119:   public:
alpar@119:     typedef _SubNoCounter<_SubNoCounter<P> > SubCounter;
alpar@119:     typedef _SubNoCounter<_SubNoCounter<P> > SubNoCounter;
alpar@119:   
alpar@119:     _SubNoCounter(P &parent) :_parent(parent) {}
alpar@119:     _SubNoCounter(P &parent,std::string,std::ostream &) 
alpar@119:       :_parent(parent) {}
alpar@119:     _SubNoCounter(P &parent,std::string) 
alpar@119:       :_parent(parent) {}
alpar@119:     _SubNoCounter(P &parent,const char *,std::ostream &)
alpar@119:       :_parent(parent) {}
alpar@119:     _SubNoCounter(P &parent,const char *)
alpar@119:       :_parent(parent) {}
alpar@119:     ~_SubNoCounter() {}
alpar@119:     _SubNoCounter &operator++() { ++_parent; return *this;}
alpar@119:     int operator++(int) { _parent++; return 0;}
alpar@119:     _SubNoCounter &operator--() { --_parent; return *this;}
alpar@119:     int operator--(int) { _parent--; return 0;}
alpar@119:     _SubNoCounter &operator+=(int c) { _parent+=c; return *this;}
alpar@119:     _SubNoCounter &operator-=(int c) { _parent-=c; return *this;}
alpar@119:     void reset(int) {}
alpar@119:     void reset() {}
alpar@119:     operator int() {return 0;}
alpar@119:   };
alpar@119: 
alpar@119: 
alpar@119:   /// \addtogroup timecount
alpar@119:   /// @{
alpar@119: 
alpar@119:   ///A counter class
alpar@119: 
alpar@119:   ///This class makes it easier to count certain events. You can increment
alpar@119:   ///or decrement the counter using operator++ and operator--.
alpar@119:   ///A report is automatically printed on destruction.
alpar@119:   ///\todo More doc
alpar@119:   class Counter 
alpar@119:   {
alpar@119:     std::string _title;
alpar@119:     std::ostream &_os;
alpar@119:     int count;
alpar@119:   public:
alpar@119:     ///\e
alpar@119: 
alpar@119:     ///\todo document please.
alpar@119:     ///
alpar@119:     typedef _SubCounter<Counter> SubCounter;
alpar@119:     ///\e
alpar@119: 
alpar@119:     ///\todo document please.
alpar@119:     ///
alpar@119:     typedef _SubNoCounter<Counter> SubNoCounter;
alpar@119: 
alpar@119:     ///\e
alpar@119:     Counter() : _title(), _os(std::cerr), count(0) {}
alpar@119:     ///\e
alpar@119:     Counter(std::string title,std::ostream &os=std::cerr) 
alpar@119:       : _title(title), _os(os), count(0) {}
alpar@119:     ///\e
alpar@119:     Counter(const char *title,std::ostream &os=std::cerr)
alpar@119:       : _title(title), _os(os), count(0) {}
alpar@119:     ///Destructor. Prints the given title and the value of the counter.
alpar@119:     ~Counter() {
alpar@119:       _os << _title << count <<std::endl;
alpar@119:     }
alpar@119:     ///\e
alpar@119:     Counter &operator++() { count++; return *this;}
alpar@119:     ///\e
alpar@119:     int operator++(int) { return count++;}
alpar@119:     ///\e
alpar@119:     Counter &operator--() { count--; return *this;}
alpar@119:     ///\e
alpar@119:     int operator--(int) { return count--;}
alpar@119:     ///\e
alpar@119:     Counter &operator+=(int c) { count+=c; return *this;}
alpar@119:     ///\e
alpar@119:     Counter &operator-=(int c) { count-=c; return *this;}
alpar@119:     ///\e
alpar@119:     void reset(int c=0) {count=c;}
alpar@119:     ///\e
alpar@119:     operator int() {return count;}
alpar@119:   };
alpar@119: 
alpar@119:   ///'Do nothing' version of \ref Counter
alpar@119: 
alpar@119:   ///'Do nothing' version of \ref Counter.
alpar@119:   ///\sa Counter
alpar@119:   class NoCounter
alpar@119:   {
alpar@119:   public:
alpar@119:     typedef _SubNoCounter<NoCounter> SubCounter;
alpar@119:     typedef _SubNoCounter<NoCounter> SubNoCounter;
alpar@119: 
alpar@119:     NoCounter() {}
alpar@119:     NoCounter(std::string,std::ostream &) {}
alpar@119:     NoCounter(const char *,std::ostream &) {}
alpar@119:     NoCounter(std::string) {}
alpar@119:     NoCounter(const char *) {}
alpar@119:     NoCounter &operator++() { return *this; }
alpar@119:     int operator++(int) { return 0; }
alpar@119:     NoCounter &operator--() { return *this; }
alpar@119:     int operator--(int) { return 0; }
alpar@119:     NoCounter &operator+=(int) { return *this;}
alpar@119:     NoCounter &operator-=(int) { return *this;}
alpar@119:     void reset(int) {}
alpar@119:     void reset() {}
alpar@119:     operator int() {return 0;}
alpar@119:   };
alpar@119: 
alpar@119:   ///@}
alpar@119: }
alpar@119: 
alpar@119: #endif