// -----------------------------------------------------------------------
// Object3D.hh: Definition of Object3D and ChanMap, classes that hold
//              pixel information for a three-dimensional object.
// -----------------------------------------------------------------------
// Copyright (C) 2006, Matthew Whiting, ATNF
//
// This program is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by the
// Free Software Foundation; either version 2 of the License, or (at your
// option) any later version.
//
// Duchamp is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
// for more details.
//
// You should have received a copy of the GNU General Public License
// along with Duchamp; if not, write to the Free Software Foundation,
// Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA
//
// Correspondence concerning Duchamp may be directed to:
//    Internet email: Matthew.Whiting [at] atnf.csiro.au
//    Postal address: Dr. Matthew Whiting
//                    Australia Telescope National Facility, CSIRO
//                    PO Box 76
//                    Epping NSW 1710
//                    AUSTRALIA
// -----------------------------------------------------------------------
#ifndef OBJECT3D_H
#define OBJECT3D_H

#include <duchamp/PixelMap/Voxel.hh>
#include <duchamp/PixelMap/Scan.hh>
#include <duchamp/PixelMap/Object2D.hh>
#include <vector>
#include <algorithm>
#include <iostream>

namespace PixelInfo
{

  /**
   * A class to store a channel+Object2D map.
   * This represents a 2-dimensional set of pixels that has an
   * associated channel number. Sets of these will form a three
   * dimensional object.
   */

  class ChanMap
  {
  public:
    ChanMap();
    ChanMap(long z){itsZ=z;};
    ChanMap(long z, Object2D obj){itsZ=z; itsObject=obj;};
    ChanMap(const ChanMap& m);
    ChanMap& operator= (const ChanMap& m);
    virtual ~ChanMap(){};

    /** Define the ChanMap using a channel number and Object2D. */
    void     define(long z, Object2D obj){itsZ=z; itsObject=obj;};

    /** Return the value of the channel number. */
    long     getZ(){return itsZ;};

    /** Set the value of the channel number. */
    void     setZ(long l){itsZ=l;};

    /** Return the Object2D set of scans. */
    Object2D getObject(){return itsObject;};

    /** Return the i-th scan of the Object2D .*/
    Scan     getScan(int i){return itsObject.scanlist[i];};

    /** The number of scans in the Object2D set. */
    long     getNumScan(){return itsObject.scanlist.size();};

    void     addOffsets(long xoff, long yoff, long zoff){
      /**
       * Add constant offsets to each of the coordinates.
       */
      itsZ += zoff;
      itsObject.addOffsets(xoff,yoff);
    };

    friend bool operator< (ChanMap lhs, ChanMap rhs){
      /** The less-than operator: only acting on the channel number. */
      return (lhs.itsZ<rhs.itsZ);
    };

    friend ChanMap operator+ (ChanMap lhs, ChanMap rhs){
      /**
       *  Add two ChanMaps together.
       *  If they are the same channel, add the Objects, otherwise
       *  return a blank ChanMap.
       */

      ChanMap newmap;
      if(lhs.itsZ==rhs.itsZ){
	newmap.itsZ = lhs.itsZ;
	newmap.itsObject = lhs.itsObject + rhs.itsObject;
      }
      return newmap;
    };

    friend class Object3D;

  private:
    long     itsZ;      ///< The channel number.
    Object2D itsObject; ///< The set of scans of object pixels.

  };


  //=====================================

  /**
   * A set of pixels in 3D.
   * This stores the pixels in a list of ChanMap objects -- ie. a set
   * of Object2Ds, each with a different channel number.
   * Also recorded are the average x-, y- and z-values (via their
   * sums), as well as their extrema.
   */

  class Object3D
  {
  public:
    Object3D();
    Object3D(const Object3D& o);
    Object3D& operator= (const Object3D& o);  
    virtual ~Object3D(){};

    /** Is a 3-D voxel in the Object? */
    bool     isInObject(long x, long y, long z);
    /** Is a 3-D voxel in the Object? */
    bool     isInObject(Voxel v){return this->isInObject(v.getX(),v.getY(),v.getZ());};
 
    /** Add a single 3-D voxel to the Object. */
    void     addPixel(long x, long y, long z);
    /** Add a single 3-D voxel to the Object. */
    void     addPixel(Voxel v){this->addPixel(v.getX(),v.getY(),v.getZ());};
    /** Add a full channel map to the Object. */
    void     addChannel(ChanMap channel);
    /** Add a full channel map to the Object. */
    void     addChannel(long z, Object2D obj){
      ChanMap channel(z,obj);
      this->addChannel(channel);
    }

    /** Sort the list of channel maps by their channel number. */
    void     order(){
      //     for(int i=0;i<maplist.size();i++) maplist[i].itsObject.order();
      std::stable_sort(maplist.begin(),maplist.end());
    };

    /** Calculate the averages and extrema of the three coordinates. */
    void     calcParams();
    /** Return the average x-value.*/
    float    getXcentre(){return xSum/float(numVox);};
    /** Return the average y-value.*/
    float    getYcentre(){return ySum/float(numVox);};
    /** Return the average z-value.*/
    float    getZcentre(){return zSum/float(numVox);};
    /** Return the minimum x-value.*/
    long     getXmin(){return xmin;}; 
    /** Return the minimum y-value.*/
    long     getYmin(){return ymin;};
    /** Return the minimum z-value.*/
    long     getZmin(){return zmin;};
    /** Return the maximum x-value.*/
    long     getXmax(){return xmax;};
    /** Return the maximum y-value.*/
    long     getYmax(){return ymax;};
    /** Return the maximum z-value.*/
    long     getZmax(){return zmax;};

    /** Return the number of distinct voxels in the Object. */
    long     getSize(){return numVox;};

    /** Return the number of distinct channels in the Object. */
    long     getNumDistinctZ();

    /** Return the number of channels in the Object. */
    long     getNumChanMap(){return this->maplist.size();};

    /** Return the number of spatial pixels -- ie. the number of
	distinct (x,y) sets in the Object. */
    long     getSpatialSize();
 
    /** Get the pixNum-th voxel */
    Voxel    getPixel(int pixNum);

    /** Return a vector set of all voxels in the Object. */
    std::vector<Voxel> getPixelSet();

    /** Get the i-th channel map. */
    ChanMap  getChanMap(int i){return this->maplist[i];};

    /** Get the channel number of the i-th channel map. */
    long     getZ(int i){return this->maplist[i].itsZ;};

    /** Return an Object2D showing the spatial (x,y) distribution of
	voxels in the Object */
    Object2D getSpatialMap();

    void     addOffsets(long xoff, long yoff, long zoff){
      /**
       *  Add constant offsets to each of the dimensions, changing the
       *  parameters at the same time.
       */
      for(int i=0;i<maplist.size();i++) maplist[i].addOffsets(xoff,yoff,zoff);
      xSum += xoff*numVox;
      xmin += xoff; xmax += xoff;
      ySum += yoff*numVox;
      ymin += yoff; ymax += yoff;
      zSum += zoff*numVox;
      zmin += zoff; zmax += zoff;
    };

    /** Output operator for the Object3D. */
    friend std::ostream& operator<< ( std::ostream& theStream, Object3D& obj);

    /** Add two Object3Ds. 
	Overlapping channels are combined using addChannel(). */
    friend Object3D operator+ (Object3D lhs, Object3D rhs){
      Object3D output = lhs;
      for(int m=0;m<rhs.maplist.size();m++) output.addChannel(rhs.maplist[m]);
      return output;
    }

  private:
    std::vector<ChanMap> maplist;   ///< The list of channel maps
    long                 numVox;    ///< How many voxels in the Object?
    float                xSum;      ///< Sum of the x-values
    float                ySum;      ///< Sum of the y-values
    float                zSum;      ///< Sum of the z-values
    long                 xmin,xmax; ///< min and max x-values of object
    long                 ymin,ymax; ///< min and max y-values of object
    long                 zmin,zmax; ///< min and max z-values of object
  }; 

}

#endif //OBJECT3D_H