OutputFileShare.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2014 by Paul Demorest
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __OutputFileShare_h
#define __OutputFileShare_h
 
#include "Error.h"
#include "MJD.h"
#include "dsp/HasInput.h"
#include "dsp/Operation.h"
#include "dsp/BitSeries.h"
#include "dsp/OutputFile.h"
 
class ThreadContext;
 
namespace dsp {
 
 
  class OutputFileShare : public Reference::Able { public:
    
    OutputFileShare (unsigned contributors);
    
    virtual ~OutputFileShare ();
 
    OutputFile* get_output_file () const { return output_file; }
 
    void set_output_file (OutputFile* f) { output_file = f; }
 
    void set_context (ThreadContext* t) { context = t; }
 
    ThreadContext* get_context () const { return context; }
 
    class Submit;
 
    Submit* new_Submit (unsigned contributor);
 
    MJD get_next_time () const { return next_time; }
 
    void set_next_time (MJD time) { next_time=time; }
 
    void signal_ready (unsigned contributor, MJD start_time);
 
    void signal_done (unsigned contributor);
 
  protected:
 
    unsigned contributors;
 
    Reference::To<OutputFile> output_file;
 
    ThreadContext* context;
 
    MJD next_time;
 
    std::vector<MJD> start_times;
 
    bool first;
 
    unsigned nready;
 
  };
 
  class OutputFileShare::Submit : public OutputFile
  {
  public:
 
    Submit (OutputFileShare* parent, unsigned contributor);
 
    void operation ();
 
    std::string get_extension () const 
    { 
      return parent->output_file->get_extension();
    };
 
    void write_header () { parent->output_file->write_header(); }
 
  protected:
 
    Reference::To<OutputFileShare> parent;
    unsigned contributor;
 
  };
 
}
 
#endif // !defined(__OutputFileShare_h)