selection.cpp

Go to the documentation of this file.

#include <algorithm>
#include <pcl/apps/point_cloud_editor/selection.h>
#include <pcl/apps/point_cloud_editor/cloud.h>
#include <pcl/apps/point_cloud_editor/common.h>

Selection&
Selection::operator= (const Selection& selection)
{
  cloud_ptr_ = selection.cloud_ptr_;
  selected_indices_ = selection.selected_indices_;
  return (*this);
}

void
Selection::addIndex (unsigned int index)
{
  selected_indices_.insert(index);
}

void
Selection::removeIndex (unsigned int index)
{
  selected_indices_.erase(index);
}

void
Selection::addIndex (const IndexVector &indices)
{
  selected_indices_.insert(indices.begin(), indices.end());
}

void
Selection::removeIndex (const IndexVector &indices)
{
  IndexVector::const_iterator it;
  for(it = indices.begin(); it != indices.end(); ++it)
    removeIndex(*it);
}

void
Selection::addIndexRange (unsigned int start, unsigned int num)
{
  for(unsigned int i = start; i < start + num; ++i)
    addIndex(i);
}

void
Selection::removeIndexRange (unsigned int start, unsigned int num)
{
  for(unsigned int i = start; i < start + num; ++i)
    removeIndex(i);
}

bool
Selection::isSelected (unsigned int index) const
{
  if (index >= cloud_ptr_->size())
    return (false);
  iterator it = selected_indices_.find(index);
  if (it != selected_indices_.end())
      return (true);
  return (false);
}

void
Selection::invertSelect ()
{
  std::set<unsigned int> s, complement;
  IncIndex inc;
  for(unsigned int i = 0; i < cloud_ptr_->size(); ++i)
    s.insert(i);
  std::set_difference(s.begin(), s.end(),
                      selected_indices_.begin(), selected_indices_.end(),
                      std::inserter(complement, complement.end()));
  selected_indices_ = complement;
}

std::string
Selection::getStat () const
{
  if (selected_indices_.size() == 0)
    return ("");
  std::string title = "Total number of selected points: ";
  std::string num_str;
  ::toString(selected_indices_.size(), num_str);
  return (title + num_str);
}