qsescanaverager.cpp

Go to the documentation of this file.

#include "qsescanaverager.h"
#include "qsedataset.h"
#include "qsescan.h"

#include <stdio.h>

QseScanAverager::QseScanAverager
(QseDataSet* inData, QseDataSet* outData, const QString& name, QObject* parent)
  : QseEXAFSOperator(inData, outData, name, parent),
    m_Result(NULL),
    m_Totals(NULL),
    m_NScans(0),
    m_LookupIndex(0)
{
}

void QseScanAverager::exec()
{
}

void QseScanAverager::startAveraging()
{
  m_OutputData -> setHeader(m_InputData -> header());
  m_OutputData -> setFileName(m_InputData -> fileName()+"_avg");

  m_Result = m_OutputData -> appendScan();
  m_Totals = new QseScan("totals");
//   m_Totals = m_OutputData -> appendScan();
  m_NScans = 0;
}

void QseScanAverager::addScanToAverage(int i)
{
  QseScan *scan = m_InputData -> scan(i);

  if (scan) {
    if (m_NScans == 0) {
      m_Result -> setName(scan->name()+"_avg");
      m_Result -> setScanCommand(scan->scanCommand()+"_avg");
      m_Result -> setScanDateTime(QDateTime::currentDateTime());
      m_Result -> mergeHeaders(scan->header());

      m_Totals -> setName(scan->name()+"_totals");
      m_Totals -> setScanCommand(scan->scanCommand()+"_totals");
      m_Totals -> setScanDateTime(QDateTime::currentDateTime());
      m_Totals -> mergeHeaders(scan->header());

      int ncols = scan -> columnCount();

      for (int i=0; i<ncols; i++) {
        QseColumn *col = scan -> column(i);
        QString nam = col->name();

        QseColumn *res = m_Result -> appendColumn(nam, col);
        QseColumn *sum = m_Totals -> appendColumn(nam, col);

        sum -> assignment(1.0);
      }
    } else {
      m_Result -> appendHeader("#C Averaged : "+scan->scanCommand());
      m_Result -> mergeHeaders(scan->header());

      int ncols = scan -> columnCount();

      for (int i=0; i<ncols; i++) {
        QseColumn *col = scan -> column(i);
        QString nam = col->name();

        QseColumn *res = NULL;
        QseColumn *sum = NULL;

        if (m_LookupIndex) {
          res = m_Result -> column(i);
          sum = m_Totals -> column(i);
        } else {
          res = m_Result -> column(nam);
          sum = m_Totals -> column(nam);
        }

        if (res == NULL) {
          res = m_Result -> appendColumn(nam, col);
          if (sum != NULL) {
            printf("Problem: sum != NULL when res == NULL\n");
          }
          sum = m_Totals -> appendColumn(nam, col);
          sum -> assignment(1.0);
        } else {
          if (sum == NULL) {
            printf("Problem: sum == NULL when res != NULL\n");
            sum = m_Totals -> appendColumn(nam, col);
            sum -> assignment(1.0);
          }

          int nRowsIn = col -> rowCount();
          int nRowsOut = res -> rowCount();

          if (nRowsOut < nRowsIn) {
            res -> setRowCount(nRowsIn);
            sum -> setRowCount(nRowsIn);
          }

          for (int r=0; r<nRowsIn; r++) {
            double vres = res->data(r);
            double vsum = sum->data(r);

            res->setData(r, vres+col->data(r));
            sum->setData(r, vsum+1.0);
          }
        }
      }
    }

    m_NScans += 1;
  }
}

void QseScanAverager::finishAveraging()
{
  int ncols = m_Result -> columnCount();

  for (int i = 0; i<ncols; i++) {
    QseColumn *res = m_Result->column(i);
    QseColumn *tot = m_Totals->column(i);

    res->division(tot);
  }

  delete m_Totals;

  m_Result = NULL;
  m_Totals = NULL;

  m_OutputData -> emitOperationCompleted();
}