DeleteTableRows.cpp

//----------------------------------------------------------------------
// Includes
//----------------------------------------------------------------------
#include "MantidDataHandling/DeleteTableRows.h"
#include "MantidAPI/IWorkspaceProperty.h"
#include "MantidAPI/ITableWorkspace.h"
#include "MantidAPI/IPeaksWorkspace.h"
#include "MantidKernel/ArrayProperty.h"

#include <set>
#include <functional>

namespace Mantid
{
  namespace DataHandling
  {
    // Register the algorithm into the algorithm factory
    DECLARE_ALGORITHM(DeleteTableRows);
    
    /// Sets documentation strings for this algorithm
    void DeleteTableRows::initDocs()
    {
      this->setWikiSummary("Deletes a row from a TableWorkspace.");
      this->setOptionalMessage("Deletes a row from a TableWorkspace.");
    }
    

    using namespace Kernel;
    using namespace API;

    /// Initialisation method.
    void DeleteTableRows::init()
    {
      declareProperty(new WorkspaceProperty<API::ITableWorkspace>("TableWorkspace", "",Direction::InOut), 
        "The name of the workspace that will be modified.");
      declareProperty(new ArrayProperty<size_t> ("Rows"),"The rows to delete. Numbering starts with 0.");
    }

    /** 
    *   Executes the algorithm.
    */
    void DeleteTableRows::exec()
    {
      API::ITableWorkspace_sptr tw = getProperty("TableWorkspace");
      API::IPeaksWorkspace_sptr pw = boost::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());
      std::set<size_t,std::greater<size_t> >::iterator it = sortedRows.begin();
      for(; it != sortedRows.end(); ++it)
      {
        int i = static_cast<int>(*it);
        if (i < 0 || i >= tw->rowCount()) continue;
        if (pw)
        {
          pw->removePeak(i);
        }
        else
        {
          tw->removeRow(i);
        }
      }
      setProperty("TableWorkspace",tw);
    }

  } // namespace DataHandling
} // namespace Mantid