DeleteTableRows.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 "MantidDataHandling/DeleteTableRows.h"
#include "MantidAPI/IPeaksWorkspace.h"
#include "MantidAPI/ITableWorkspace.h"
#include "MantidAPI/IWorkspaceProperty.h"
#include "MantidKernel/ArrayProperty.h"
 
#include <functional>
#include <set>
 
namespace Mantid::DataHandling {
// Register the algorithm into the algorithm factory
DECLARE_ALGORITHM(DeleteTableRows)
 
using namespace Kernel;
using namespace API;
 
void DeleteTableRows::init() {
  declareProperty(std::make_unique<WorkspaceProperty<API::ITableWorkspace>>("TableWorkspace", "", Direction::InOut),
                  "The name of the workspace that will be modified.");
  declareProperty(std::make_unique<ArrayProperty<size_t>>("Rows"),
                  "A comma-separated list of row numbers. Row numbering starts with 0.");
}
 
void DeleteTableRows::exec() {
  API::ITableWorkspace_sptr tw = getProperty("TableWorkspace");
  API::IPeaksWorkspace_sptr pw = std::dynamic_pointer_cast<API::IPeaksWorkspace>(tw);
  std::vector<size_t> rows = getProperty("Rows");
  // sort the row indices in reverse order
  std::set<size_t, std::greater<size_t>> sortedRows(rows.begin(), rows.end());
  auto it = sortedRows.begin();
  for (; it != sortedRows.end(); ++it) {
    if (*it >= tw->rowCount())
      continue;
    if (pw) {
      pw->removePeak(static_cast<int>(*it));
    } else {
      tw->removeRow(*it);
    }
  }
  setProperty("TableWorkspace", tw);
}
 
} // namespace Mantid::DataHandling