DataDumper.hxx

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// CVS: $Revision: 7115 $, $Date: 2010-06-21 11:49:57 +0100 (Mon, 21 Jun 2010) $, $Author: EckhardSutorius $
//-----------------------------------------------------------------------------
template <typename DataType>
void DataDumper::dumpAsCsv(TableData<DataType>& aData, std::ostream& aOut)
{
  for (unsigned rowNo = 0; rowNo < aData.qNumRows(); ++rowNo)
  {
    // If this is the only occurence of this value, or there are no
    // constraints, dump the csv output to the stream. Otherwise do nothing.
    if (isUnique(aData, rowNo))
    {
      for (unsigned colNo = 0; colNo < aData.qNumCols() - 1; ++colNo)
      {
        formatOut(aData.at(colNo, rowNo), mInfo.getFormat(colNo), aOut);
        aOut << ",";
      }
      formatOut(aData.at(aData.qNumCols()-1, rowNo),
                mInfo.getFormat(aData.qNumCols()-1),
                aOut);
      aOut << std::endl;
    }
  }
  // @@TODO: Make keyCount it's own class so it auto-clears on destruction
  mKeyCount.clear();
}
//-----------------------------------------------------------------------------
template <typename DataType>
void DataDumper::addDefaultRow(TableData<DataType>& aData, FitsFile& fcat)
{
  unsigned mNumCols = mInfo.getNumAttributes();
  unsigned mNumRows = aData.qNumRows();
  unsigned mNumExts = aData.qNumExts();
  int rowNum = 0;
  unsigned outRow = -1;
  int row1 = 0;
  
  TableData<Numeric>* tmpData = new TableData<Numeric>(mNumCols, mNumRows);
  tmpData -> setNumExts(mNumExts);
  
  for (int hduNum = 2; hduNum <= fcat.getNumHdus(); ++hduNum)
  {
    fcat.movabsHdu(hduNum);
    int hduRowNum = fcat.getNumRows();
    if (hduRowNum > 0)
    {
      outRow += 1;
      for (unsigned col = 0; col < mNumCols; ++col)
      {
        if (mInfo.getName(col) == ExpectNames::multiframeID ||
            mInfo.getName(col) == ExpectNames::extNum ||
            mInfo.getName(col) == ExpectNames::filterID ||
            mInfo.getName(col) == ExpectNames::cuEventID)
        {
          Numeric numval;
          aData.value(col, rowNum, numval);
          tmpData->assign(col, outRow, numval);
        }
        else 
        {
          string format = mInfo.getFormat(col);
          if (format == "bigint")
          {
            tmpData->assign(col, outRow, DefaultValues::bigintdef);
          }
          else if (format == "tinyint")
          {
            tmpData->assign(col, outRow, DefaultValues::tinyintdef);
          }
          else if (format == "int")
          {
            tmpData->assign(col, outRow, DefaultValues::intdef);
          }
          else if (format == "smallint")
          {
            tmpData->assign(col, outRow, DefaultValues::smallintdef);
          }
          else if (format == "real" || format == "float")
          {
            tmpData->assign(col, outRow, DefaultValues::realdef);
          }
          else if (format.find("varchar") != string::npos)
          {
            unsigned varcharLen = 0;
            int posa = format.find("(") + 1;
            int posb = format.rfind(")");
            varcharLen = atoi(format.substr(posa, posb - posa).c_str());
            if (varcharLen == 1)
            {
              tmpData->assign(col, outRow, DefaultValues::varchar1def);
            } else if (varcharLen == 2) {
              tmpData->assign(col, outRow, DefaultValues::varchar2def);
            } else if (varcharLen == 3) {
              tmpData->assign(col, outRow, DefaultValues::varchar3def);
            } else if (varcharLen > 3) {
              tmpData->assign(col, outRow, DefaultValues::varchar4def);
            }
          }
          else if (format == "datetime")
          {
            tmpData->assign(col, outRow, DefaultValues::datetimedef);
          }
          string defValue = mInfo.getDefault(col);
          if (defValue.length() > 0)
          {
            tmpData->assign(col, outRow, defValue);
          }
        }
      }
      int row2 = row1 + hduRowNum;
      if (row2 != row1)
      {
        for (int rowNo = row1; rowNo < row2; ++rowNo)
        {
          outRow += 1;
          for (unsigned col = 0; col < mNumCols; ++col)
          {
            Numeric numval;
            if (aData.isDefined(col, rowNo))
            {
              aData.value(col, rowNo, numval);
              tmpData->assign(col, outRow, numval);
            } else {
              tmpData->setundef(col, outRow);
            }
          }
        }
        row1 = row2;
      }
    }
    rowNum += hduRowNum;
  }
  row1 = 0;
  for (int hduNum = 2; hduNum <= fcat.getNumHdus(); ++hduNum)
  {
    fcat.movabsHdu(hduNum);
    int hduRowNum = fcat.getNumRows();
    int row2 = row1 + hduRowNum + 1;
    if (row2 != row1)
    {
      for (int rowNo = row1; rowNo < row2; ++rowNo)
      {
        for (unsigned col = 0; col < mNumCols; ++col)
        {
          if (tmpData->isDefined(col, rowNo))
          {
            Numeric numval;
            tmpData->value(col, rowNo, numval);
            aData.assign(col, rowNo, numval);
          }
        }
      }
      row1 = row2;
    }
  }
}
//-----------------------------------------------------------------------------
// @@TODO: Should this be a method of TableData?
template <typename DataType>
bool DataDumper::isUnique(TableData<DataType>& aData, unsigned aRow)
{
  using namespace std;
  // The key is the concatenation of values associted with the
  // constraining attributes. Only a unique occurence of this key
  // will be output
  string key;
  for (list<int>::iterator colConstItr = mColConstList.begin();
                           colConstItr != mColConstList.end();
                           ++colConstItr)
  {
    string val;
    aData.value(*colConstItr, aRow, val);
    key += val;
  }

  // Increment the number of occurences of this value
  mKeyCount[key] += 1;

  return (mKeyCount[key] == 1 ||  mColConstList.empty());
}
//-----------------------------------------------------------------------------

---