FrameID.hxx

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// CVS: $Revision: 4785 $, $Date: 2008-05-20 14:19:36 +0100 (Tue, 20 May 2008) $, $Author: RossCollins $
//-----------------------------------------------------------------------------
#ifndef FRAMEID_H
#define FRAMEID_H

#include <string>

#include "DataMethod.hxx"
#include "DataOps.h"
#include "FitsFile.h"
#include "Logger.h"
//-----------------------------------------------------------------------------
template<typename DataType>
class DirectFrameID : public DataMethod<DataType>
{
public:
  DirectFrameID(const FitsFile& aFile)
  {
    DataMethod<DataType>::methodName = "DirectFrameID";
    mFrameID = DataOps::makeFrameID(aFile);
  }

  ~DirectFrameID() { }

  void setSource(IntMap& tbls, IntMap& cols, StringMap units) { }

  void setTarget(IntMap& cols, StringMap units) { colNo = cols[0]; }

  void doit(TableData<DataType>& data, int row1, int row2)
  {
    for (int rowNo = row1; rowNo < row2; ++rowNo)
    {
      data.assign(colNo, rowNo, mFrameID);      
    }
  }


  // No data from raw data included
  void doit2(TableData<DataType>& inpData, TableData<DataType>& data, int row1, int row2) { }

private:
  int colNo;
  std::string mFrameID;
};
//-----------------------------------------------------------------------------
template<typename DataType>
class IndirectFrameID : public DataMethod<DataType>
{
public:
  IndirectFrameID(const std::string& aFilePath)
    : mFilePath(aFilePath)
    { DataMethod<DataType>::methodName = "IndirectFrameID"; }

  virtual ~IndirectFrameID() { }

  void setSource(IntMap& tbls, IntMap& cols, StringMap units)
  {
    fileColno = cols[0];
    if (fileColno < 0)
    {
      throw DataMethodException("IndirectFrameID method. Couldn't source");
    }
  }

  void setTarget(IntMap& cols, StringMap units) { fidColno = cols[0]; }

  void doit(TableData<DataType>& data, int row1, int row2)
  {
    Logger log;

    for (int rowNo = row1; rowNo < row2; ++rowNo)
    {
      std::string fileName;
      data.value(fileColno, rowNo, fileName);

      if (fileName.length() > 0)
      {
        int posb = fileName.rfind(".fit")+4;
        int posa = 0;
        if (fileName.rfind(" ") != std::string::npos)
        {
          posa = (int) fileName.rfind(" ",posb)+1;
        }

        fileName = mFilePath + fileName.substr(posa,posb-posa);

        log.addMessage("Calculate frame ID from this file name: '"
                        + fileName + "'");
      }
      else
      {
        log.addMessage("No file name available.");
      }

      // Try and open a FITS file for this file name and calculate its
      // frame ID
      try
      {
        FitsFile fitsFile(fileName);
        std::string frameID = DataOps::makeFrameID(fitsFile);
        fitsFile.closeFile();

        // Catch files with mfID == 0 and undefine it
        if (frameID == "0")
        {
          data.setundef(fidColno, rowNo);
        }
        else
        {
          data.assign(fidColno, rowNo, frameID);
        }
      }
      catch (FitsIOException)
      {
        // Catch the case if there is a problem opening the FITS file.
        // Here we log a warning and not try to assign a value for this row.
        // If specified in the schema, a default value will be assigned later.
        data.setundef(fidColno, rowNo);
        log.addWarning("Could not open FITS file object for " + fileName);
      }
    }
  }

  // No data from raw data included
  void doit2(TableData<DataType>& inpData, TableData<DataType>& data, int row1, int row2) { }

private:
    int fidColno;
    int fileColno;
    std::string mFilePath;
};
//-----------------------------------------------------------------------------
#endif
//-----------------------------------------------------------------------------
// Change log:
//
// 27-May-2004,  IAB: Original version.
// 30-May-2005, ETWS: Added this header; made FrameID undefined if file
//                    is not available
// 31-Mar-2006, ETWS: Undefine FrameID if mFrameID = 0
//  7-Apr-2008, ETWS: Upgraded to use new detection table layout.