vtkParallelTimer.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
  Module:    vtkParallelTimer.h
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
//
#ifndef vtkParallelTimer_h
#define vtkParallelTimer_h
 
#define vtkParallelTimerDEBUG -1
 
#include "vtkObject.h"
#include "vtkRenderingParallelLICModule.h" // for export
 
//BTX
#include <vector> // for vector
#include <string> // for string
#include <sstream> // for sstream
#if vtkParallelTimerDEBUG > 0
#include <iostream> // for cerr
#endif
//ETX
 
class vtkParallelTimerBuffer;
 
class VTKRENDERINGPARALLELLIC_EXPORT vtkParallelTimer : public vtkObject
{
public:
  static vtkParallelTimer *New();
  vtkTypeMacro(vtkParallelTimer,vtkObject);
  void PrintSelf(ostream& os, vtkIndent indent);
 
 
  class LogHeaderType
    {
    public:
      template<typename T> LogHeaderType &operator<<(const T& s);
    };
 
 
  class LogBodyType
    {
    public:
      template<typename T> LogBodyType &operator<<(const T& s);
    };
 
 
  vtkSetMacro(WriterRank,int);
  vtkGetMacro(WriterRank,int);
 
 
  vtkSetStringMacro(FileName);
  vtkGetStringMacro(FileName);
  //BTX
  void SetFileName(const std::string &fileName)
    { this->SetFileName(fileName.c_str()); }
  //ETX
 
 
  void StartEvent(const char *event);
  void StartEvent(int rank, const char *event);
  void EndEvent(const char *event);
  void EndEvent(int rank, const char *event);
  void EndEventSynch(const char *event);
  void EndEventSynch(int rank, const char *event);
 
  //BTX
 
  template<typename T>
  vtkParallelTimer &operator<<(const T& s);
 
 
  vtkParallelTimer::LogHeaderType GetHeader()
    { return vtkParallelTimer::LogHeaderType(); }
 
 
  vtkParallelTimer::LogBodyType GetBody()
    { return vtkParallelTimer::LogBodyType(); }
  //ETX
 
  void Clear();
 
  void Update();
 
  int Write();
 
  static vtkParallelTimer *GetGlobalInstance();
 
  static void DeleteGlobalInstance();
 
 
  vtkSetMacro(WriteOnClose, int);
  vtkGetMacro(WriteOnClose, int);
 
 
  vtkSetMacro(GlobalLevel, int);
  vtkGetMacro(GlobalLevel, int);
 
protected:
  vtkParallelTimer();
  virtual ~vtkParallelTimer();
 
private:
  vtkParallelTimer(const vtkParallelTimer&); // Not implemented
  void operator=(const vtkParallelTimer&); // Not implemented
 
//BTX
 
  class VTKRENDERINGPARALLELLIC_EXPORT vtkParallelTimerDestructor
    {
    public:
      vtkParallelTimerDestructor() : Log(0) {}
      ~vtkParallelTimerDestructor();
 
      void SetLog(vtkParallelTimer *log){ this->Log = log; }
 
    private:
      vtkParallelTimer *Log;
    };
//ETX
 
private:
  int GlobalLevel;
  int Initialized;
  int WorldRank;
  int WriterRank;
  char *FileName;
  int WriteOnClose;
  std::vector<double> StartTime;
  #if vtkParallelTimerDEBUG < 0
  std::vector<std::string> EventId;
  #endif
 
  vtkParallelTimerBuffer *Log;
 
  static vtkParallelTimer *GlobalInstance;
  static vtkParallelTimerDestructor GlobalInstanceDestructor;
 
  std::ostringstream HeaderBuffer;
 
  friend class LogHeaderType;
  friend class LogBodyType;
};
 
//BTX
//-----------------------------------------------------------------------------
template<typename T>
vtkParallelTimer &vtkParallelTimer::operator<<(const T& s)
{
  if (this->WorldRank == this->WriterRank)
    {
    this->HeaderBuffer << s;
    #if vtkParallelTimerDEBUG > 0
    std::cerr << s;
    #endif
    }
  return *this;
}
 
//-----------------------------------------------------------------------------
template<typename T>
vtkParallelTimer::LogHeaderType &vtkParallelTimer::LogHeaderType::operator<<(const T& s)
{
  vtkParallelTimer *log = vtkParallelTimer::GetGlobalInstance();
 
  if (log->WorldRank == log->WriterRank)
    {
    log->HeaderBuffer << s;
    #if vtkParallelTimerDEBUG > 0
    std::cerr << s;
    #endif
    }
 
  return *this;
}
 
//-----------------------------------------------------------------------------
template<typename T>
vtkParallelTimer::LogBodyType &vtkParallelTimer::LogBodyType::operator<<(const T& s)
{
  vtkParallelTimer *log = vtkParallelTimer::GetGlobalInstance();
 
  *(log->Log) <<  s;
  #if vtkParallelTimerDEBUG > 0
  std::cerr << s;
  #endif
 
  return *this;
}
//ETX
 
#endif