source: branches/pixelmap-refactor-branch/src/PixelMap/Object3D.hh @ 567

// -----------------------------------------------------------------------
// Object3D.hh: Definition of the Object3D class that holds
//              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 <map>
#include <iostream>

namespace PixelInfo
{

  /// @brief A set of pixels in 3D.  
  /// 
  /// @details 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(){};

    /// @brief Is a 3-D voxel in the Object? 
    bool     isInObject(long x, long y, long z);
    /// @brief Is a 3-D voxel in the Object? 
    bool     isInObject(Voxel v){return isInObject(v.getX(),v.getY(),v.getZ());};
 
    /// @brief Add a single 3-D voxel to the Object. 
    virtual void     addPixel(long x, long y, long z);
    /// @brief Add a single 3-D voxel to the Object. 
    virtual void     addPixel(Voxel v){addPixel(v.getX(),v.getY(),v.getZ());};
    /// @brief Add a scan to the object 
    void     addScan(Scan s, long z);
    /// @brief Add a full channel map to the Object. 
    void     addChannel(const long &z, Object2D &obj);

    /// @brief Calculate the averages and extrema of the three coordinates. 
    void     calcParams();
//     /// @brief Return the average x-value.
//     float    getXaverage(){if(numVox>0) return xSum/float(numVox); else return 0.;};
//     /// @brief Return the average y-value.
//     float    getYaverage(){if(numVox>0) return ySum/float(numVox); else return 0.;};
//     /// @brief Return the average z-value.
//     float    getZaverage(){if(numVox>0) return zSum/float(numVox); else return 0.;};
    /// @brief Return the average x-value.
    float    getXaverage();
    /// @brief Return the average y-value.
    float    getYaverage();
    /// @brief Return the average z-value.
    float    getZaverage();
    /// @brief Return the minimum x-value.
    long     getXmin(){return xmin;}; 
    /// @brief Return the minimum y-value.
    long     getYmin(){return ymin;};
    /// @brief Return the minimum z-value.
    long     getZmin(){return zmin;};
    /// @brief Return the maximum x-value.
    long     getXmax(){return xmax;};
    /// @brief Return the maximum y-value.
    long     getYmax(){return ymax;};
    /// @brief Return the maximum z-value.
    long     getZmax(){return zmax;};

    /// @brief Return the number of distinct voxels in the Object. 
    unsigned int getSize(){return numVox;};

    /// @brief Return the number of distinct channels in the Object. 
    long     getNumDistinctZ(){return chanlist.size();};

    /// @brief Return the number of channels in the Object. 
    long     getNumChanMap(){return chanlist.size();};

    /// @brief Return the number of spatial pixels -- ie. the number of distinct (x,y) sets in the Object. 
    long     getSpatialSize();

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

    /// @brief Return a vector list of the channel numbers in the Object
    std::vector<long> getChannelList();

    /// @brief Return the largest number of adjacent channels in the Object.
    int getMaxAdjacentChannels();

    /// @brief Get the channel map for channel z. 
    Object2D  getChanMap(long z);

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

    /// @brief Add constant offsets to each of the dimensions, changing the parameters at the same time. 
    virtual void     addOffsets(long xoff, long yoff, long zoff);

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

    /// @brief Add two Object3Ds. Overlapping channels are combined using addChannel(). 
    friend Object3D operator+ (Object3D lhs, Object3D rhs);

  protected:
    std::map<long,Object2D> chanlist; ///< The list of 2D 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