ProgressMediator.h

Go to the documentation of this file.

 
/*
 * ProgressMediator.h
 *
 *  Created on: Sep 07, 2018
 *      Author: Alejandro Alvarez Ayllon
 */
 
#ifndef _SEMAIN_PROGRESSLISTENER_H
#define _SEMAIN_PROGRESSLISTENER_H
 
#include "SEFramework/Source/SourceGroupInterface.h"
#include "SEFramework/Pipeline/Segmentation.h"
#include "SEUtils/Observable.h"
#include <atomic>
#include <mutex>
 
namespace SourceXtractor {
 
struct ProgressInfo {
  std::string m_label;
  int m_done, m_total;
 
  ProgressInfo(const std::string &name, int done, int total): m_label(name), m_done(done), m_total(total) {}
};
 
typedef Observable<std::list<ProgressInfo>> ProgressObservable;
typedef Observable<bool> DoneObservable;
 
class ProgressMediator: public ProgressObservable, public DoneObservable {
public:
  typedef Observer<SegmentationProgress> segmentation_observer_t;
  typedef Observer<SourceInterface>      source_observer_t;
  typedef Observer<SourceGroupInterface> group_observer_t;
 
  ~ProgressMediator() = default;
 
  ProgressMediator();
 
  std::shared_ptr<segmentation_observer_t>& getSegmentationObserver(void);
 
  std::shared_ptr<source_observer_t>& getDetectionObserver(void);
 
  std::shared_ptr<group_observer_t>& getDeblendingObserver(void);
 
  std::shared_ptr<group_observer_t>& getMeasurementObserver(void);
 
  void done();
 
  void update(void);
 
private:
  SegmentationProgress m_segmentation_progress;
  std::atomic_int m_detected, m_deblended, m_measured;
 
  std::shared_ptr<segmentation_observer_t> m_segmentation_listener;
  std::shared_ptr<source_observer_t> m_detection_listener;
  std::shared_ptr<group_observer_t> m_deblending_listener, m_measurement_listener;
 
  // Mediator serializes the notifications, so the observers do not need to worry about
  // being called from multiple threads
  std::mutex m_mutex;
 
  // This is a set of internal classes that implement the Observer pattern for different stages.
  class ProgressCounter;
  class SourceCounter;
  class GroupCounter;
};
 
} // end SourceXtractor
 
#endif // _SEMAIN_PROGRESSLISTENER_H