Regroup.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
//----------------------------------------------------------------------
// Includes
//----------------------------------------------------------------------
#include "MantidAlgorithms/Regroup.h"
#include "MantidAPI/Axis.h"
#include "MantidAPI/CommonBinsValidator.h"
#include "MantidAPI/HistogramValidator.h"
#include "MantidAPI/WorkspaceFactory.h"
#include "MantidKernel/ArrayProperty.h"
#include "MantidKernel/CompositeValidator.h"
#include "MantidKernel/RebinParamsValidator.h"
 
#include <algorithm>
#include <cmath>
#include <functional>
#include <numeric>
 
namespace Mantid {
using HistogramData::HistogramE;
using HistogramData::HistogramX;
using HistogramData::HistogramY;
namespace Algorithms {
 
// Register the class into the algorithm factory
DECLARE_ALGORITHM(Regroup)
 
using namespace Kernel;
using API::MatrixWorkspace;
using API::MatrixWorkspace_const_sptr;
using API::MatrixWorkspace_sptr;
using API::WorkspaceProperty;
 
void Regroup::init() {
  auto wsVal = std::make_shared<CompositeValidator>();
  wsVal->add<API::HistogramValidator>();
  wsVal->add<API::CommonBinsValidator>();
  declareProperty(std::make_unique<WorkspaceProperty<MatrixWorkspace>>("InputWorkspace", "", Direction::Input, wsVal),
                  "The input workspace.");
  declareProperty(std::make_unique<WorkspaceProperty<MatrixWorkspace>>("OutputWorkspace", "", Direction::Output),
                  "The result of regrouping.");
 
  declareProperty(std::make_unique<ArrayProperty<double>>("Params", std::make_shared<RebinParamsValidator>()),
                  "The new approximate bin boundaries in the form: x1,dx1,x2,dx2,...,xn");
}
 
void Regroup::exec() {
  // retrieve the properties
  std::vector<double> rb_params = getProperty("Params");
 
  // Get the input workspace
  MatrixWorkspace_const_sptr inputW = getProperty("InputWorkspace");
 
  const bool dist = inputW->isDistribution();
 
  const auto histnumber = static_cast<int>(inputW->getNumberHistograms());
  auto &XValues_old = inputW->x(0);
  std::vector<int> xoldIndex; // indeces of new x in XValues_old
  // create new output X axis
  std::vector<double> xAxisTmp;
  int ntcnew = newAxis(rb_params, XValues_old.rawData(), xAxisTmp, xoldIndex);
  HistogramData::BinEdges XValues_new(std::move(xAxisTmp));
 
  // make output Workspace the same type is the input, but with new length of
  // signal array
  API::MatrixWorkspace_sptr outputW = API::WorkspaceFactory::Instance().create(inputW, histnumber, ntcnew, ntcnew - 1);
 
  int progress_step = histnumber / 100;
  if (progress_step == 0)
    progress_step = 1;
  for (int hist = 0; hist < histnumber; hist++) {
    // get const references to input Workspace arrays (no copying)
    auto &XValues = inputW->x(hist);
    auto &YValues = inputW->y(hist);
    auto &YErrors = inputW->e(hist);
 
    // get references to output workspace data (no copying)
    auto &YValues_new = outputW->mutableY(hist);
    auto &YErrors_new = outputW->mutableE(hist);
 
    // output data arrays are implicitly filled by function
    rebin(XValues, YValues, YErrors, xoldIndex, YValues_new, YErrors_new, dist);
 
    outputW->setBinEdges(hist, XValues_new);
 
    if (hist % progress_step == 0) {
      progress(double(hist) / histnumber);
      interruption_point();
    }
  }
 
  outputW->setDistribution(dist);
 
  // Copy units
  if (outputW->getAxis(0)->unit().get())
    outputW->getAxis(0)->unit() = inputW->getAxis(0)->unit();
  try {
    if (inputW->getAxis(1)->unit().get())
      outputW->getAxis(1)->unit() = inputW->getAxis(1)->unit();
  } catch (Exception::IndexError &) {
    // OK, so this isn't a Workspace2D
  }
 
  // Assign it to the output workspace property
  setProperty("OutputWorkspace", outputW);
}
 
void Regroup::rebin(const HistogramX &xold, const HistogramY &yold, const HistogramE &eold,
                    const std::vector<int> &xoldIndex, HistogramY &ynew, HistogramE &enew, bool distribution) {
 
  for (int i = 0; i < int(xoldIndex.size() - 1); i++) {
 
    int n = xoldIndex[i];             // start the group
    int m = xoldIndex[i + 1];         // end the group
    double width = xold[m] - xold[n]; // width of the group
 
    if (width == 0.) {
      g_log.error("Zero bin width");
      throw std::runtime_error("Zero bin width");
    }
    /*
     *        yold contains counts/unit time, ynew contains counts
     *         enew contains counts**2
     */
    if (distribution) {
      ynew[i] = 0.;
      enew[i] = 0.;
      for (int j = n; j < m; j++) {
        double wdt = xold[j + 1] - xold[j]; // old bin width
        ynew[i] += yold[j] * wdt;
        enew[i] += eold[j] * eold[j] * wdt * wdt;
      }
      ynew[i] /= width;
      enew[i] = sqrt(enew[i]) / width;
    } else // yold,eold data is not distribution but counts
    {
      ynew[i] = 0.;
      enew[i] = 0.;
      for (int j = n; j < m; j++) {
        ynew[i] += yold[j];
        enew[i] += eold[j] * eold[j];
      }
      enew[i] = sqrt(enew[i]);
    }
  }
}
 
int Regroup::newAxis(const std::vector<double> &params, const std::vector<double> &xold, std::vector<double> &xnew,
                     std::vector<int> &xoldIndex) {
  double xcurr, xs;
  int ibound(2), istep(1), inew(0);
  auto ibounds = static_cast<int>(params.size()); // highest index in params array containing a bin boundary
  int isteps = ibounds - 1;                       // highest index in params array containing a step
 
  xcurr = params[0];
  using std::placeholders::_1;
  auto iup = std::find_if(xold.cbegin(), xold.cend(), std::bind(std::greater_equal<double>(), _1, xcurr));
  if (iup != xold.end()) {
    xcurr = *iup;
    xnew.emplace_back(xcurr);
    xoldIndex.emplace_back(inew);
    inew++;
  } else
    return 0;
 
  while ((ibound <= ibounds) && (istep <= isteps)) {
    // if step is negative then it is logarithmic step
    if (params[istep] >= 0.0)
      xs = params[istep];
    else
      xs = xcurr * fabs(params[istep]);
 
    // xcurr += xs;
 
    // find nearest x_i that is >= xcurr
    iup = std::find_if(xold.begin(), xold.end(), std::bind(std::greater_equal<double>(), _1, xcurr + xs));
    if (iup != xold.end()) {
      if (*iup <= params[ibound]) {
        xcurr = *iup;
        xnew.emplace_back(xcurr);
        xoldIndex.emplace_back(inew);
        inew++;
      } else {
        ibound += 2;
        istep += 2;
      }
    } else
      return inew;
  }
  // returns length of new x array or -1 if failure
  return inew;
  // return( (ibound == ibounds) && (istep == isteps) ? inew : -1 );
}
 
} // namespace Algorithms
} // namespace Mantid