Extension.hxx

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// CVS: $Revision: 2099 $, $Date: 2006-04-17 17:48:43 +0100 (Mon, 17 Apr 2006) $, $Author: rsc $
//-----------------------------------------------------------------------------
#ifndef EXTENSION_H
#define EXTENSION_H

#include "Constants.h"
#include "DataBuilder.hxx"
#include "InsertData.hxx"
#include "Logger.h"
#include "TableInfo.h"
//-----------------------------------------------------------------------------
// One row per extension HDU

template<typename DataType>
class Extension : public DataBuilder<DataType>
{
public:
  Extension(const TableInfo& aInfo)
    : mInfo(aInfo)
    { DataBuilder<DataType>::builderName = "Extension"; }

  virtual ~Extension() {}

  TableData<DataType>* build(FitsFile& fpix,
                             FitsFile& fcat,
                             int fileNo)
  {
    Logger log;
    // Number of rows is number extension HDUs
    int numRows = fpix.getNumHdus() - 1;
    TableData<DataType>* data;

    // This must match the number of extensions in the
    // (not empty) catalog file
    bool useEmptyCat = fcat.getFileName().find(DefaultValues::emptycatname)
                          != string::npos;

    if (numRows != fcat.getNumHdus()-1 and useEmptyCat == false)
    {
      //throw DataBuilderException("Pixel and catalog file mismatch!");
      log.addSysInfo("in " + fpix.getFileName() + "," + fcat.getFileName() +
                     " (pix and cat mismatch)");
      data = new TableData<DataType>(0, 0);
    }
    else
    {
      int numCols = mInfo.getNumAttributes();
      data = new TableData<DataType>(numCols, numRows);

      // Insert values from primary HDU if any
      for (int rowNo = 0; rowNo < numRows; ++rowNo)
      {
        insertFitsPrimaryHeader(rowNo, fpix, mInfo, *data);
      }

      // Loop over all extension HDUs
      for (int rowNo = 0; rowNo < numRows; ++rowNo)
      {
        int hduNo = rowNo + 2;
        fpix.movabsHdu(hduNo);
        insertFitsExtensionHeader(rowNo, fpix, mInfo, *data);

        if (useEmptyCat == false)
        {
          fcat.movabsHdu(hduNo);
          insertFitsExtensionHeader(rowNo, fcat, mInfo, *data);
        }
      }
      insertExtNum(fpix, mInfo, *data);
    }
    return data;
  }

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