qseaverager.cpp

Go to the documentation of this file.

#include "qseaverager.h"
#include "qseaveragerrole.h"
#include "qsedataset.h"
// #include "qsefilereader.h"
#include "qsescan.h"
#include "qsecolumn.h"

QseAverager::QseAverager(QObject *parent)
        : QObject(parent)
{}

QseAverager::~QseAverager()
{}

void QseAverager::exec(QseDataSet      *inFile,
           QseAveragerRole *role,
           QseDataSet      *outFile)
{
  outFile -> clear();

  int ns = inFile -> scanCount();

  for (int s = 0; s < ns; s++) {
    QseScan *sc = inFile -> scan(s);

    if (role -> matchesScan(sc)) {
      QseScan *osc = outFile -> appendScan();

      normalizeScan(inFile, role, osc, s);
    }
  }

  outFile -> emitLoadCompleted();
}


void QseAverager::normalizeScan(QseDataSet      *inFile,
              QseAveragerRole *role,
              QseScan         *osc,
              int n)
{
  QseScan *sc = inFile->scan(n);

  if (sc) {
    int nc = sc -> columnCount();
    int energycol = -1;

    for (int c = 0; c < nc; c++) {
      QseColumn *col = sc -> column(c);

      if (role -> matchesEnergyColumn(col)) {
  energycol = c;
      }
    }

    if (energycol < 0) {
      emit message(QString("No energy column found for scan %1").arg(n));
    } else {
      mergeHeaders(osc, sc);

      osc -> setScanNumber(sc->scanNumber());

      QseColumn *energy = osc -> appendColumn("Energy", sc -> column(energycol));
      QseColumn *normal = osc -> appendColumn("Norm");
      QseColumn *detect = osc -> appendColumn("Detector");
      QseColumn *seconds = osc -> appendColumn("Seconds");

      energy -> multiplication(role->energyScalingRatio());

      int nnorm = 0, ndet = 0, nsecs = 0;
      
      for (int c = 0; c < nc; c++) {
  QseColumn *col = sc -> column(c);

  if (role -> matchesNormalizationColumn(col)) {
    nnorm += 1;
    normal -> addition(col);
  } else if (role -> matchesDetectorColumn(col)) {
    ndet += 1;
    detect -> addition(col);
  } else if (role -> matchesSecondsColumn(col)) {
    nsecs += 1;
    seconds -> addition(col);
  }
      }
      
      if (nnorm > 0) {
  QseColumn *ratio = osc -> appendColumn("Ratio", detect);
  
  ratio -> division(normal);
  
  emit message(QString("Scan %1 had %2 detector columns, was normalized by %3 ion chambers").arg(n).arg(ndet).arg(nnorm));
      } else if (nsecs > 0) {
  QseColumn *ratio = osc -> appendColumn("Ratio", detect);
  
  ratio -> division(seconds);
  emit message(QString("Scan %1 had %2 detector columns, was normalized by counting time").arg(n).arg(ndet));
      } else {
  QseColumn *ratio = osc -> appendColumn("Ratio", detect);
  
  emit message(QString("Scan %1 had %2 detector columns, no normalization performed").arg(n).arg(ndet));
      }
    }
  }
}

bool QseAverager::addScans(QseScan* s1, const QseScan* s2)
{
  if (s1 && s2) {
    int nc1 = s1->columnCount();
    int nc2 = s2->columnCount();

    if (nc1 == 0) {
      for (int c = 0; c < nc2; c++) {
  QseColumn *c2 = s2->column(c);
  QseColumn *c1 = s1->appendColumn(c2->name(), c2);
      }

      mergeHeaders(s1, s2);

      emit message(QString("Started new average with scan %1").arg(s1->scanCommand()));

      return true;
    } else {
      if (nc1 != nc2) {
  emit message(QString("Scans have different numbers of columns (%1,%2)").arg(nc1).arg(nc2));
  return false;
      }
      
      for (int c = 0; c < nc1; c++) {
  QseColumn *c1 = s1->column(c);
  QseColumn *c2 = s2->column(c);

  if (c1 && c2) {
    int nr1 = c1->rowCount();
    int nr2 = c2->rowCount();
    
    if (nr1 != nr2) {
      emit message(QString("Scan columns have differing numbers of points"));
      return false;
    }
  } else {
    emit message(QString("Scan column missing in one or other scan"));
    return false;
  }
      }

      for (int c = 0; c < nc1; c++) {
  QseColumn *c1 = s1->column(c);
  QseColumn *c2 = s2->column(c);
  
  if (c1 && c2) {
    c1 -> addition(c2);
  }
      }

      mergeHeaders(s1, s2);

      emit message(QString("Scan %1 added to scan %2").arg(s2->scanCommand()).arg(s1->scanCommand()));

      return true;
    }
  } else {
    emit message(QString("Scan was missing"));
    return false;
  }
}

bool QseAverager::divideScan(QseScan* s, double v)
{
  if (s) {
    int nc = s->columnCount();

    for (int c = 0; c < nc; c++) {
      QseColumn *col = s->column(c);

      if (col) {
  col->division(v);
      }
    }

    return true;
  } else {
    return false;
  }
}

void QseAverager::mergeHeaders(QseScan *s1, const QseScan *s2)
{
  if (s1 && s2) {
//     printf("set name %s\n", qPrintable(s2->name()));
    s1->setName(s2->name());
    s1->mergeHeaders(s2->header());
    s1->setScanDateTime(QDateTime::currentDateTime());
  }
}

---