Cluster.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 +
#include "MantidCrystal/Cluster.h"
#include "MantidAPI/IMDHistoWorkspace.h"
 
namespace {
using VecElements = std::vector<Mantid::Crystal::DisjointElement>;
}
 
namespace Mantid::Crystal {
 
Cluster::Cluster(const size_t &label) : m_originalLabel(label), m_rootCluster(this) { m_indexes.reserve(1000); }
 
size_t Cluster::getLabel() const {
  if (m_rootCluster != this) {
    return m_rootCluster->getLabel();
  }
 
  return m_originalLabel;
}
 
size_t Cluster::getOriginalLabel() const { return m_originalLabel; }
 
size_t Cluster::size() const { return m_indexes.size(); }
 
void Cluster::addIndex(const size_t &index) { m_indexes.emplace_back(index); }
 
void Cluster::writeTo(Mantid::API::IMDHistoWorkspace_sptr ws) const {
  const size_t label = this->getLabel();
  for (auto index : m_indexes) {
    ws->setSignalAt(index, static_cast<Mantid::signal_t>(label));
    ws->setErrorSquaredAt(index, 0);
  }
}
 
ICluster::ClusterIntegratedValues Cluster::integrate(Mantid::API::IMDHistoWorkspace_const_sptr ws) const {
  double errorIntSQ = 0;
  double sigInt = 0;
  // Integrate accross indexes owned by this workspace.
  for (auto index : m_indexes) {
    sigInt += ws->getSignalAt(index);
    double errorSQ = ws->getErrorAt(index);
    errorSQ *= errorSQ;
    errorIntSQ += errorSQ;
  }
  return ClusterIntegratedValues(sigInt, errorIntSQ);
}
 
void Cluster::toUniformMinimum(VecElements &disjointSet) {
  if (!m_indexes.empty()) {
    size_t parentIndex = m_rootCluster->getRepresentitiveIndex();
 
    for (size_t i = 1; i < m_indexes.size(); ++i) {
      disjointSet[m_indexes[i]].unionWith(disjointSet[parentIndex].getParent());
    }
  }
}
 
size_t Cluster::getRepresentitiveIndex() const { return m_indexes.front(); }
 
void Cluster::setRootCluster(ICluster const *root) { m_rootCluster = root; }
 
bool Cluster::operator==(const Cluster &other) const { return getLabel() == other.getLabel(); }
 
bool Cluster::containsLabel(const size_t &label) const { return (label == this->getLabel()); }
 
} // namespace Mantid::Crystal