#ifndef REGIONSET_HEADER
#define REGIONSET_HEADER
//////////////////////////////////////////////////////////////
//! docentry="Image.Image Processing.Region.Data Structures"
//! userlevel=Normal
//! example="exSeg.cc"
//! file="amma/Image/IPStream/IPRegion/RegionSet.hh"
//! rcsid="$Id: RegionSet.hh,v 1.12 2000/12/14 09:55:20 ees1cg Exp $"
//! lib=IPRegion
//! author="Roshano Roberts"
//! date="12/06/98"

#include "amma/RCHandle.hh"
#include "amma/Num/NumLabel.hh"
#include "amma/Num/NumDataSet.hh"
#include "amma/Region/RegionSetB.hh"
#include "amma/DList.hh"
#include "amma/Image.hh"
#include "amma/ImageRec.hh"

// ---------------------------------------------------------------------------
// **********  RegionSetC  ***************************************************
// ---------------------------------------------------------------------------
//: Class for representing the segmentation
     
template<class StatT>
class RegionSetC : public RCHandleC<RegionSetBodyC<StatT > > {
public:
  RegionSetC()
    : RCHandleC<RegionSetBodyC<StatT> > (*new RegionSetBodyC<StatT>())
  {}
  //: Default constructor

  RegionSetC(const ImageC<NumLabelC> & map, const NumDataSetC<NumLabelC,StatT> & stats)
    : RCHandleC<RegionSetBodyC<StatT> > (*new RegionSetBodyC<StatT>(map, stats))
  {}
  //: Constructor from the segmentation map and the associated statistics for each region
  
  inline ImageC<NumLabelC> Image() const;
  //: Returns the segmentation map
  
  inline NumDataSetC<NumLabelC,StatT> Stats() const;
  //: Returns the statistics associated to each region in the segmentation map

  inline NumDataSetC<NumLabelC,StatT> & Stats();
  //: Returns the statistics associated to each region in the segmentation map

  inline ByteImageC ByteImage() const;
  //: Returns the segmentation map in the form of a ByteImageC

  inline IntImageC IntImage() const;
  //: Returns the segmentation map in the form of a IntImageC

  inline RGBImageC RandomImage() const;
  //: Returns the segmentation map in the form of a colour random image

  inline YUVImageC RandomTaintImage(ByteGreyValueT max=100) const;
  //: Returns the segmentation map in the form of a colour random image. The Y channel is left blank (e.g., for displaying the original data). The labels in the U and V channels are in the range 0 to 'max'.

  inline void SetMap(const IntImageC & map);
  //: The segmentation map for this is changed to map; UpdateStats should be performed after this operation

  template<class ClassT, class PValueT> void UpdateStats(const ImageC<PValueT> & img);
  //: Calculates the statistics of the regions in the segmentation map from scratch

  inline void MakeConnectedComponents();
  //: Connected component labelling is performed on the segmentation map; Each new component is given the same statistic as the coreesponding previous region


  inline void MakeConnectedComponents(const ByteImageC & roi);
  //: Connected component labelling is performed on the segmentation map; Each new component is given the same statistic as the coreesponding previous region. The whole operation is performed on the region of interest "roi". The region of interest should be labelled 0 and all other pixels in the image should be labelled 1. In the output segmentation map, the pixels outside the region of interest are undefined.

protected:
  RegionSetC(RegionSetBodyC<StatT> &bod)
    : RCHandleC<RegionSetBodyC<StatT> >(bod)
  {}
  //: Constructor from the body class
  
};


template<class StatT>
inline ImageC<NumLabelC> RegionSetC<StatT>::Image() const { return Body().Image(); }

template<class StatT>
inline NumDataSetC<NumLabelC,StatT> RegionSetC<StatT>::Stats() const { return Body().Stats(); }

template<class StatT>
inline NumDataSetC<NumLabelC,StatT> & RegionSetC<StatT>::Stats() { return Body().Stats(); }

template<class StatT>
inline ByteImageC RegionSetC<StatT>::ByteImage() const { return Body().ByteImage(); }

template<class StatT>
inline IntImageC RegionSetC<StatT>::IntImage() const { return Body().IntImage(); }


template<class StatT>
inline RGBImageC RegionSetC<StatT>::RandomImage() const { return Body().RandomImage(); }

template<class StatT>
inline YUVImageC RegionSetC<StatT>::RandomTaintImage(ByteGreyValueT max) const { return Body().RandomTaintImage(max); }


template<class StatT>
inline void RegionSetC<StatT>::SetMap(const IntImageC & map) {
  Body().SetMap(map);
}

template<class StatT>
template<class ClassT, class PValueT>
void RegionSetC<StatT>::UpdateStats(const ImageC<PValueT> & img) {
  Body().UpdateStats<ClassT, PValueT>(img);
}

template<class StatT>
inline void RegionSetC<StatT>::MakeConnectedComponents() {
  Body().MakeConnectedComponents();
}

template<class StatT>
inline void RegionSetC<StatT>::MakeConnectedComponents(const ByteImageC & roi) {
  Body().MakeConnectedComponents(roi);
}

#endif