TPipeline.h

Go to the documentation of this file.

#ifndef _TPIPELINE_H_
#define _TPIPELINE_H_

#include <string>
#include <vector>

#include "TDirectory.h"

#include "StoppableThread.h"
#include "ThreadsafeQueue.h"
#include "TRawEvent.h"
#include "TRawEventSource.h"
#include "TUnpackedEvent.h"

class TDataLoop;
class TUnpackLoop;
class TRootInputLoop;
class THistogramLoop;
class TWriteLoop;
class TTerminalLoop;

class TPipeline {
public:
  TPipeline();
  virtual ~TPipeline();

  bool CanStart(bool print_reason = false);

  void Start();
  void Pause();
  void Resume();
  void Stop();
  void Join();
  std::string Status();
  void Write();


  TDirectory* GetDirectory();
  void ClearHistograms();
  void ReplaceRawDataFile(std::string filename);

  void SetHistogramLibrary(std::string filename);
  std::string GetHistogramLibrary() const;

  void SetReplaceVariable(const char* name, double value);
  void RemoveVariable(const char* name);
  TList* GetVariables();

  bool InLearningPhase();

  void AddRawDataFile(std::string filename);

  void SetOutputRootFile(std::string filename);

  void SetIsOnline(bool is_online);

  void SetTimeOrdering(bool time_order);

  void AddInputRootFile(std::string filename);

  void SetInputRing(std::string ringname);

  bool IsFinished();

private:
  TPipeline(const TPipeline&) { }
  TPipeline& operator=(const TPipeline&) { }

  bool AllQueuesEmpty();

  int Initialize();

  void SetupRawReadLoop();
  void SetupRootReadLoop();
  void SetupOutputFile();
  void SetupHistogramLoop();
  void SetupOutputLoop();
  TRawEventSource* OpenSingleFile(const std::string& filename);

  TFile* output_file;
  TDirectory* output_directory;

  // Output of the TDataLoop.  Unused if reading from a root file.
  ThreadsafeQueue<TRawEvent> raw_event_queue;

  // Output of the TUnpackLoop, or the TRootInputLoop, depending on settings
  ThreadsafeQueue<TUnpackedEvent*> unpacked_event_queue;

  // Output of the THistogramLoop
  ThreadsafeQueue<TUnpackedEvent*> post_histogram_queue;

  std::vector<StoppableThread*> pipeline;
  TDataLoop* data_loop;
  TUnpackLoop* unpack_loop;
  TRootInputLoop* root_input_loop;
  THistogramLoop* histogram_loop;
  TWriteLoop* write_loop;
  TTerminalLoop* terminal_loop;

  // The setup parameters
  std::vector<std::string> input_raw_files;
  std::vector<std::string> input_root_files;
  std::string input_ring;

  std::string histogram_library;
  std::string output_root_file;
  bool is_online;

  bool time_order;
  int time_order_depth;

  ClassDef(TPipeline, 0);
};

#endif /* _TPIPELINE_H_ */