newmen/versatile-diamond

#include "behavior_tor.h"
#include "smart_atoms_vector3d.h"

namespace vd
{

int BehaviorTor::correctOne(int value, uint max) const
{
    if (value < 0) return (int)max + value;
    else if (value >= (int)max) return (int)max - value;
    return value;
}

Atom *&BehaviorTor::choiseAtom(SmartAtomsVector3d<Atom> *atomsVector, const int3 &coords) const
{
    return atomsVector->BaseVector::operator[](correct(atomsVector, coords));
}

int3 BehaviorTor::correct(const SmartAtomsVector3d<Atom> *atomsVector, const int3 &coords) const
{
    assert(coords.z >= 0);
    assert(coords.z < (int)atomsVector->sizes().z);

    int3 result = coords;
    result.x = correctOne(coords.x, atomsVector->sizes().x);
    result.y = correctOne(coords.y, atomsVector->sizes().y);
    return result;
}

}