OPAL.h

Go to the documentation of this file.

#pragma once

#include "ImageDatabase.h"
#include "OPALSettings.h"
#include "SSD.h"
#include "MaxVoteLabelEstimator.h"
#include "DummyLabelEstimator.h"

#include <chrono>
#include <random>
#include <functional>

class OPAL {
public:
  // Fixed types.
  using ImgPixelType = double;
  using SegPixelType = int;
  using DatabaseType = ImageDatabase<ImgPixelType, SegPixelType>;
  using ImgType      = DatabaseType::ImgType;
  using SegType      = DatabaseType::SegType;
  using SSDType      = SSD<DatabaseType>;

  OPAL(const OPALSettings &settings, const DatabaseType &database);

  void ConstrainedInitialization();

  void BuildSegmentation();

  SegType GetOutput() const {
    return OutputSegmentation;
  }

  void Run();


  const Image<int>& getFieldX() const { return FieldX; }

  const Image<int>& getFieldY() const { return FieldY; }

  const Image<size_t>& getFieldT() const { return FieldT; }

  const Image<SSDType>& getSSDMap() const { return SSDMap; }


private:
  const OPALSettings &Sets;

  const DatabaseType &Database;

  // Output displacement fields.
  Image<int>     FieldX; 
  Image<int>     FieldY; 
  Image<size_t>  FieldT; 
  Image<SSDType> SSDMap; 

  // Size of the images in the database. Just to replace call to
  // Database.getImageHeight(), Database.getImageWidth().
  size_t ImageHeight; 
  size_t ImageWidth;  

  ImgType InputImage;

  SegType OutputSegmentation;


  using RandomGeneratorType = std::mt19937;

  RandomGeneratorType randGen;


  using FinalLabelEstimator      = DummyLabelEstimator<SegPixelType>;
  using CandidateLabelsContainer = FinalLabelEstimator::CandidateContainer;
  FinalLabelEstimator finalLabelEstimator;

private:
  void EvenPropagation(size_t iteration);

  void OddPropagation(size_t iteration);

  int PropagateRightDown(size_t x, size_t y);

  int PropagateLeftUp(size_t x, size_t y);

  void SaveCurrentFields(const std::string &fileName) const;

  SSDType SSDAt(size_t i, size_t j) const;

  void UpdateSSDMap();

  void GetCandidateLabelsForPixel(size_t i, size_t j,
                                  CandidateLabelsContainer &result) const;
};