source: trunk/src/Utils/Section.hh @ 526

// -----------------------------------------------------------------------
// Section.hh: Definition of the Section class, used to store
//             definitions of subsections of a FITS file.
// -----------------------------------------------------------------------
// 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 SECTION_H
#define SECTION_H

#include <string>
#include <vector>

namespace duchamp
{

  /**
   * A class to store information on subsections of a cube. This keeps
   * the subsection itself as a string, plus the parsed information in
   * the format of vectors of starting positions and lengths of each
   * dimension.
   */

  class Section
  {
  public:
    Section();
    Section(std::string &s){subsection=s;};
    Section(const Section& s);
    Section& operator= (const Section& s);
    virtual ~Section(){};

    /** Convert the subsection string into the lists of numerical values. */
    int parse(std::vector<long> dimAxes);
    int parse(long *dim, int size);

    /** Test whether a given voxel (x,y,z) lies within the subsection */
    bool isInside(int x, int y, int z){
      return (  ( (x>=starts[0])&&(x<starts[0]+dims[0])  ) &&
                ( (y>=starts[1])&&(y<starts[1]+dims[1])  ) &&
                ( (z>=starts[2])&&(z<starts[2]+dims[2])  )   );
    }

    /** Save the subsection string */
    void setSection(std::string s){subsection=s;};
    /** Return the subsection string */
    std::string getSection(){return subsection;};

    /** Save a single dimension's subsection string */
    void setSection(int i, std::string s){sections[i] = s;};
    /** Return a particular dimension's subsection string */
    std::string getSection(int i){if(i>=numSections) return "*"; else return sections[i];};

    /** Return a particular starting value */
    int getStart(int i){if(i>=numSections) return 0; else return starts[i];};
    /** Set a particular starting value */
    void setStart(int i, int val){starts[i]=val;};

    /** Return a particular dimension length */
    int getDim(int i){if(i>=numSections) return 1; else return dims[i];};
    /** Set a particular dimension length */
    void setDim(int i, int val){dims[i]=val;};

    /** Return a particular ending value */
    int getEnd(int i){if(i>=numSections) return 0; else return starts[i]+dims[i]-1;};
    /** Set a particular ending value, using the current starting value */
    void setEnd(int i, int val){dims[i]=val-starts[i]+1;};

    /** Return the full list of start values */
    std::vector<int> getStartList(){return starts;};
    /** Return the full list of dimension lengths */
    std::vector<int> getDimList(){return dims;};

  private:
    std::string subsection;   ///< The string indicating the subsection, of the format [a:b,c:d,e:f] etc.
    std::vector<std::string> sections; // The individual section strings for each dimension
    int numSections;          ///< The number of sections in the string.
    std::vector<int> starts;  ///< The list of starting values (a,c,e)
    std::vector<int> dims;    ///< The list of lengths of each dimension (b-a+1,d-c+1,f-e+1)
  };

  std::string nullSection(int ndim);

}

#endif //SECTION_H