BatchQueue.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2007 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/Util/genutil/BatchQueue.h
 
#ifndef __BatchQueue_h
#define __BatchQueue_h
 
#include "ThreadContext.h"
#include "Functor.h"
 
#include <vector>

class BatchQueue {
 
public:

  BatchQueue (unsigned nthread = 0);

  ~BatchQueue ();

  void resize (unsigned nthread);

  class Job;

  void submit (Job*);

  template<class Class, typename Method>
  void submit (Class* instance, Method method);

  template<class Class, typename Method, typename Argument>
  void submit (Class* instance, Method method, Argument argument);

  template<class Class, typename Method, typename Arg1, typename Arg2>
  void submit (Class* instance, Method method, Arg1 a1, Arg2 a2);

  void wait ();
 
protected:

  void add (Job*);

  void remove (Job*);

  ThreadContext* context;

  std::vector<Job*> active;
 
};
 
class BatchQueue::Job {
 
public:
 
  virtual ~Job () {}

  void run ();

  virtual void execute () = 0;
 
protected:
 
  friend class BatchQueue;

  BatchQueue* queue;
 
};
 
template<class Class, typename Method>
void BatchQueue::submit (Class* instance, Method method)
{
  class Job0 : public Job {
    Functor< void() > functor;
    void execute () { functor (); }
  public:
    Job0 (Class* instance, Method method) : functor (instance, method) {}
  };
  
  submit (new Job0( instance, method ));
}
 
template<class Class, typename Method, typename Argument>
void BatchQueue::submit (Class* instance, Method method, Argument arg)
{
  class Job1 : public Job {
    Functor< void(Argument) > functor;
    Argument argument;
    void execute () { functor (argument); }
  public:
    Job1 (Class* instance, Method method, Argument arg)
      : functor (instance, method) { argument = arg; }
  };
  
  submit (new Job1( instance, method, arg ));
}
 
template<class Class, typename Method, typename Arg1, typename Arg2>
void BatchQueue::submit (Class* instance, Method method, Arg1 a1, Arg2 a2)
{
  class Job2 : public Job {
    Functor< void(Arg1, Arg2) > functor;
    Arg1 arg1;
    Arg2 arg2;
    void execute () { functor (arg1, arg2); }
  public:
    Job2 (Class* instance, Method method, Arg1 a1, Arg2 a2)
      : functor (instance, method) { arg1 = a1; arg2 = a2; }
  };
  
  submit (new Job2( instance, method, a1, a2 ));
}
 
#endif // !defined(__BatchQueue_h)