Catalog.hxx

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// CVS: $Revision: 4825 $, $Date: 2008-06-05 15:32:48 +0100 (Thu, 05 Jun 2008) $, $Author: EckhardSutorius $
//-----------------------------------------------------------------------------
#ifndef CATALOG_H
#define CATALOG_H

#include "DataBuilder.hxx"
#include "InsertData.hxx"
#include "TableInfo.h"
//-----------------------------------------------------------------------------
template<typename DataType>
class Catalog : public DataBuilder<DataType>
{
public:
  Catalog(const TableInfo& aInfo)
    : mInfo(aInfo)
    { DataBuilder<DataType>::builderName = "Catalog"; }

  virtual ~Catalog() {}

  TableData<DataType>* build(FitsFile& fpix,
                             FitsFile& fcat,
                             int fileNo)
  {
    // Count number of sources; include line for each extension's default row
    int numSources = 0;
    //int mNumExts = fcat.getNumHdus() - 1;
    unsigned mValidExts = 0;

    for (int hduNo = 2; hduNo <= fcat.getNumHdus(); ++hduNo)
    {
      fcat.movabsHdu(hduNo);
      int hduRowNum = fcat.getNumRows();
      numSources += hduRowNum;
      if (mInfo.isDetection() and hduRowNum > 0)
      {
        numSources += 1;
        mValidExts += 1;
      }
    }
    
    // Creat new data table
    int numCols = mInfo.getNumAttributes();
    TableData<DataType>* data = new TableData<DataType>(numCols, numSources);
    data -> setNumExts(mValidExts);

    // Load the sources in the table
    int firstRow = 0;
    int row1 = 0;
    for (int hduNo = 2; hduNo <= fcat.getNumHdus(); ++hduNo)
    {
      fcat.movabsHdu(hduNo);
      // read extension header keywords (set in schema with --/K)
      int row2 = row1 + fcat.getNumRows();
      if (row2 != row1)
      {
        for (int rowNo = row1; rowNo < row2; ++rowNo)
        {
          insertFitsExtensionHeader(rowNo, fcat, mInfo, *data);
        }
        // read catalogue data
        insertFitsColumns(firstRow, fcat, mInfo, *data);
        firstRow += fcat.getNumRows();
        row1 = row2;
      }
    }

    insertExtNum(fcat, mInfo, *data);

    return data;
  }

private:
  TableInfo mInfo;
};
//-----------------------------------------------------------------------------
#endif
//-----------------------------------------------------------------------------
// Change log:
//
// 27-May-2004,  IAB: Original version.