source: src/Actions/FillAction/FillRegularGridAction.cpp@ 5a4cbc

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

/*
 * FillRegularGridAction.cpp
 *
 *  Created on: Jan 12, 2012
 *      Author: heber
 */

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

#include "CodePatterns/MemDebug.hpp"

#include "Atom/atom.hpp"
#include "Atom/CopyAtoms/CopyAtoms_withBonds.hpp"
#include "CodePatterns/Log.hpp"
#include "Descriptors/MoleculeOrderDescriptor.hpp"
#include "Filling/Cluster.hpp"
#include "Filling/Filler.hpp"
#include "Filling/Mesh/CubeMesh.hpp"
#include "Filling/Predicates/IsVoidNode_FillPredicate.hpp"
#include "MoleculeListClass.hpp"
#include "Parser/FormatParserInterface.hpp"
#include "Parser/FormatParserStorage.hpp"
#include "Shapes/BaseShapes.hpp"
#include "World.hpp"


#include <algorithm>
#include <iostream>
#include <string>
#include <vector>

#include "Actions/FillAction/FillRegularGridAction.hpp"

using namespace MoleCuilder;

// and construct the stuff
#include "FillRegularGridAction.def"
#include "Action_impl_pre.hpp"
/** =========== define the function ====================== */
Action::state_ptr FillRegularGridAction::performCall() {
  typedef std::vector<atom*> AtomVector;

  // obtain information
  getParametersfromValueStorage();

  // get all present atoms for UndoState
  AtomVector presentatoms = World::getInstance().getAllAtoms();

  // get selected atoms and find containing sphere
  World::AtomComposite atoms = World::getInstance().getSelectedAtoms();
  if (atoms.size() == 0) {
    ELOG(2, "No atoms selected, aborting,");
    return Action::failure;
  }

  Vector center;
  double radius = 0.;
  {
    center.Zero();
    for(World::AtomComposite::const_iterator iter = atoms.begin();
        iter != atoms.end(); ++iter)
      center += (*iter)->getPosition();
    center *= 1./(double)atoms.size();
    for(World::AtomComposite::const_iterator iter = atoms.begin();
        iter != atoms.end(); ++iter) {
      const Vector &position = (*iter)->getPosition();
      for (size_t i=0;i<NDIM;++i) {
        const double temp_distance = fabs(position[i] - center[i]);
        if (temp_distance > radius)
          radius = temp_distance;
      }
    }
  }
  // always move cluster to be centered at origin
  for(World::AtomComposite::iterator iter = atoms.begin();
      iter != atoms.end(); ++iter)
    (*iter)->setPosition((*iter)->getPosition() - center);
  if (radius == 0.) { // never have zero radius: set to one or less
    ASSERT( (params.counts[0] != 0.) && (params.counts[1] != 0.) && (params.counts[2] != 0.),
        "FillRegularGridAction::performCall() - counts must not be zero.");
    const Vector diag(1./(double)(params.counts[0]+1), 1./(double)(params.counts[1]+1), 1./(double)(params.counts[2]+1));
    radius = std::min(1.,World::getInstance().getDomain().translateIn(diag).Norm()/2.);
  }
  // add some small boundary
  radius += 1e+6*std::numeric_limits<double>::epsilon();
  radius = std::max(params.mindistance, radius);
  LOG(2, "INFO: The " << atoms.size() << " are contained in a sphere at "
      << center << " with radius " << radius << ".");
  Shape *s = new Shape(Sphere(zeroVec, radius));
  LOG(2, "INFO: Created sphere at " << s->getCenter() << " and radius " << s->getRadius() << ".");

  // create set of atomic ids
  std::vector<atomId_t> atomIds(atoms.size(), (size_t)-1);
  std::transform(atoms.begin(), atoms.end(), atomIds.begin(),
      boost::bind(&atom::getId, _1) );
  Cluster::atomIdSet atomset(atomIds.begin(), atomIds.end());
  LOG(2, "INFO: Created atomset of size " << atomset.size() << ".");

  // create cluster
  ClusterInterface::Cluster_impl cluster(new Cluster(atomset, *s));
  LOG(2, "INFO: Created cluster of size " << atomset.size() << ".");

  // create predicate, mesh, and filler
  {
    FillPredicate *predicate = new FillPredicate( IsVoidNode_FillPredicate(*s) );
    Mesh *mesh = new CubeMesh(params.counts, params.offset, World::getInstance().getDomain().getM());
    Filler *fillerFunction = new Filler(*mesh, *predicate);

    // fill
    CopyAtoms_withBonds copyMethod;
    (*fillerFunction)(copyMethod, cluster);

    // remove
    delete fillerFunction;
    delete mesh;
    delete predicate;
    delete s;
  }

  // generate list of newly created atoms
  // (we can in general remove more quickly from a list than a vector)
  AtomVector filleratoms = World::getInstance().getAllAtoms();
//  LOG(0, filleratoms.size() << " atoms are present.");
  std::list<atom *> filleratoms_list;
  std::copy( filleratoms.begin(),  filleratoms.end(), std::back_inserter( filleratoms_list ));
//  LOG(0, filleratoms_list.size() << " atoms have been copied.");
  for (AtomVector::const_iterator iter = presentatoms.begin();
      iter != presentatoms.end();
      ++iter) {
    filleratoms_list.remove(*iter);
  }
//  LOG(0, filleratoms_list.size() << " atoms left after removal.");
  filleratoms.clear();
  std::copy(filleratoms_list.begin(), filleratoms_list.end(), std::back_inserter( filleratoms ));

//  LOG(0, filleratoms.size() << " atoms have been inserted.");

  return Action::state_ptr(new FillRegularGridState(filleratoms,params));
}

Action::state_ptr FillRegularGridAction::performUndo(Action::state_ptr _state) {
//  FillRegularGridState *state = assert_cast<FillRegularGridState*>(_state.get());
//
//  BOOST_FOREACH(atom *_atom, state->filleratoms) {
//    World::getInstance().destroyAtom(Walker);
//  }
//
//  // as atoms and atoms from state are removed, we have to create a new one
//  std::vector<atom*> filleratoms;
//  return Action::state_ptr(new FillRegularGridState(filleratoms,state->params));
  return Action::failure;
}

Action::state_ptr FillRegularGridAction::performRedo(Action::state_ptr _state){
  //FillRegularGridState *state = assert_cast<FillRegularGridState*>(_state.get());

  return Action::failure;
  //return Action::state_ptr(_state);
}

bool FillRegularGridAction::canUndo() {
  return false;
}

bool FillRegularGridAction::shouldUndo() {
  return false;
}
/** =========== end of function ====================== */