// -----------------------------------------------------------------------
// columns.hh: Define the column class that formats Duchamp output.
// -----------------------------------------------------------------------
// 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 COLUMNS_H
#define COLUMNS_H

#include <iostream>
#include <iomanip>
#include <string>
#include <vector>

namespace duchamp
{

  class Detection;
  class FitsHeader;

  /**
   * A namespace controlling the formatting of columns of output for Duchamp.
   */

  namespace Column
  {
    // First, define some basic quantities.
    // The default values for the columns, and default values for
    //  the precision of different types of columns.

    /** Total number of columns being considered.*/
    const int numColumns=34;    
    /** Enumerated column titles */
    enum COLNAME {NUM=0, NAME, X, Y, Z,
		  RA, DEC, RAJD, DECJD, VEL, 
		  WRA, WDEC, WVEL,
		  FINT, FTOT, FPEAK, SNRPEAK,
		  X1, X2, Y1, Y2, Z1, Z2, NPIX, FLAG,
		  XAV, YAV, ZAV, XCENT, YCENT, ZCENT, 
		  XPEAK, YPEAK, ZPEAK,
		  UNKNOWN};  // the last is for unclassified columns

    /** Guide for which columns are used for the results file */
    const bool isFile[numColumns+1]=
      {true,true,true,true,true,
       true,true,false,false,true,
       true,true,true,
       true,true,true,true,
       true,true,true,true,true,true,true,true,
       true,true,true,true,true,true,
       true,true,true,
       false};

    /** Guide for which columns are used for the results printed to screen */
    const bool isScreen[numColumns+1]=
      {true,true,true,true,true,
       true,true,false,false,true,
       true,true,true,
       true,true,true,true,
       true,true,true,true,true,true,true,true,
       false,false,false,false,false,false,
       false,false,false,
       false};

    /** Guide for which columns are used for the log file */
    const bool isLog[numColumns+1]=
      {true,false,true,true,true,
       false,false,false,false,false,
       false,false,false,
       false,true,true,true,
       true,true,true,true,true,true,true,false,
       false,false,false,false,false,false,
       false,false,false,
       false};

     /** Guide for which columns are used for the VOTable file */
    const bool isVOTable[numColumns+1]=
      {true,true,false,false,false,
       false,false,true,true,true,
       true,true,true,
       true,true,true,true,
       false,false,false,false,false,false,false,true,
       true,true,true,true,true,true,
       true,true,true,
       false};

    /** Total number of columns used in logfile (no WCS ones). */
    const int numColumnsLog=14; 
    /** Enumerated column titles for logfile*/
    enum LOGNAME {lNUM=0, lX, lY, lZ,
		  lFTOT, lFPEAK, lSNRPEAK,
		  lX1, lX2, lY1, lY2, lZ1, lZ2, lNPIX}; 

    /** Number of types of precision. */
    const int numPrec=6;        
    /** Enumerated precision categories */
    enum PrecType {prFLUX=3, prVEL=3, prXYZ=1, prPOS=6, prWPOS=2, prSNR=2}; 

    /** Default widths of all columns.*/
    const int defaultWidth[numColumns+1]=
      {5, 8, 6, 6, 6,
       13, 13, 11, 11, 7, 9, 9, 7,
       10, 10, 9, 7,
       4, 4, 4, 4, 4, 4, 6, 5,
       6, 6, 6, 7, 7, 7, 7, 7, 7,
       1};

    /** Default precisions for all columns.*/
    const int defaultPrec[numColumns+1]=
      {0, 0, prXYZ, prXYZ, prXYZ,
       0, 0, prPOS, prPOS, prVEL, 
       prWPOS, prWPOS, prVEL,
       prFLUX, prFLUX, prFLUX, 
       prSNR, 0, 0, 0, 0, 0, 0, 0, 0,
       prXYZ, prXYZ, prXYZ, 
       prXYZ, prXYZ, prXYZ, 
       prXYZ, prXYZ, prXYZ,
       0}; 

    /** Default Titles of all columns. */
    const std::string defaultName[numColumns+1]=
      {"Obj#","Name","X","Y","Z",
       "RA","DEC","RAJD","DECJD","VEL",
       "w_RA","w_DEC","w_VEL",
       "F_int","F_tot","F_peak", "S/Nmax",
       "X1","X2","Y1","Y2","Z1","Z2","Npix","Flag",
       "X_av", "Y_av", "Z_av", "X_cent", "Y_cent", "Z_cent",
       "X_peak", "Y_peak", "Z_peak",
       " "};

    /** Default units of all columns. */
    const std::string defaultUnits[numColumns+1]=
      {"","","","","",
       "","","[deg]","[deg]","",
       "[arcmin]","[arcmin]","",
       "","","", "",
       "","","","","","","[pix]","",
       "","","","","","","","","",
       ""}; 


    /** Class to store information about a single column.
     * This contains information about a given column -- its width, the 
     *  precision associated with it, its title and the units.
     * Plus the usual array of accessor functions and so on.
     */
    class Col{
    public:
      Col();          ///< Basic constructor
      Col(int num, int prec=-1);   ///< Specific constructor
      Col(const Col& c);///< Assignment constructor
      Col& operator=(const Col& c); ///< Copy constructor
      virtual ~Col(); ///< Default destructor;
      //--------------
      // basic accessor functions
      //
      int    getWidth(){return width;};         
      void   setWidth(int i){width=i;};    
      int    getPrecision(){return precision;};     
      void   setPrecision(int i){precision=i;};
      std::string getName(){return name;};          
      void   setName(std::string s){name=s;};  
      std::string getUnits(){return units;};         
      void   setUnits(std::string s){units=s;}; 
      COLNAME getType(){return type;};
      void   setType(COLNAME coltype){type=coltype;};

      //--------------
      // other functions
      //
      /** Make the column one space wider. */
      void   widen(){width++;};

      /** Increase the precision by one, widening the column if necessary. */
      void   upPrec(){precision++; width++;};

      //--------------
      // Outputting functions -- all in columns.cc
      //
      /** write the title of the column to the stream */
      void   printTitle(std::ostream &stream);

      /** write the units of the column to the stream */
      void   printUnits(std::ostream &stream);

      /** write dashes the full width of the column to the stream */
      void   printDash (std::ostream &stream);

      /** write blanks the full width of the column to the stream */
      void   printBlank(std::ostream &stream);

      /** Print a given value in a column with correct width & precision. */
      template <class T> void printEntry(std::ostream &stream, T value);

      /** Decides whether the column is used for a given table type */
      bool   doCol(std::string type, bool flagFint=true);


    private:
      int width;          ///< How wide is the column (in ascii spaces)
      int precision;      ///< What precision should be used to print
                          ///   the values? (If 0, the setprecision command
                          ///   is not used.)
      std::string name;   ///< The title of the column
      std::string units;  ///< The units that the values in the column
                          ///   are expressed in.
      COLNAME type;       ///< The type of the column
    };

  
    /** Returns a vector of Col for results file output.*/
    std::vector<Column::Col> getFullColSet(std::vector<Detection> &objectList, 
					   FitsHeader &head);

    /** Returns a vector of Col for logfile output.*/
    std::vector<Column::Col> getLogColSet(std::vector<Detection> &objectList, 
					  FitsHeader &head);

  }

}

#endif