source: src/Descriptors/AtomOrderDescriptor.cpp@ eb0d77

/*
 * Project: MoleCuilder
 * Description: creates and alters _atomecular systems
 * Copyright (C)  2010-2012 University of Bonn. All rights reserved.
 * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
 */

/*
 * AtomOrderDescriptor.cpp
 *
 *  Created on: Mar 22, 2012
 *      Author: heber
 */

// include config.h
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include "CodePatterns/MemDebug.hpp"

#include "AtomOrderDescriptor.hpp"
#include "AtomOrderDescriptor_impl.hpp"

#include "Atom/atom.hpp"
#include "CodePatterns/Observer/ObservedContainer_impl.hpp"

using namespace std;


AtomOrderDescriptor_impl::AtomOrderDescriptor_impl(int _id) :
  id(_id)
{}

AtomOrderDescriptor_impl::~AtomOrderDescriptor_impl()
{}

bool AtomOrderDescriptor_impl::predicate(std::pair<atomId_t,atom*> _atom)
{
  atom *Walker = find();
  return (Walker == _atom.second);
}

AtomDescriptor AtomByOrder(int id){
  return AtomDescriptor(AtomDescriptor::impl_ptr(new AtomOrderDescriptor_impl(id)));
}

World::AtomSet& AtomOrderDescriptor_impl::getAtoms()
{
  return World::getInstance().atoms;
}


atom *AtomOrderDescriptor_impl::find(){
  World::AtomSet &atoms = getAtoms();

  int i=0;
  atom *_atom = NULL;
  if (id == 0) {
    return NULL;
  } else if (id > 0) {
    World::AtomSet::internal_iterator res = atoms.begin_internal();
    for (; res != atoms.end_internal(); ++res) { // when iterator is normal, ++ goes forward!
      ++i;
      if (id == i) {
        _atom = res->second;
        break;
      }
    }
  } else {
    World::AtomSet::reverse_internal_iterator res = atoms.rbegin_internal();
    for (; res != atoms.rend_internal(); ++res) {  // when iterator is reverse, ++ goes backward!
      --i;
      if (id == i) {
        _atom = res->second;
        break;
      }
    }
  }

  return _atom;
}

vector<atom*> AtomOrderDescriptor_impl::findAll(){
  atom *res = find();
  return (res)?(vector<atom*>(1,res)):(vector<atom*>());
}