bonds.h

Go to the documentation of this file.

#ifndef BONDS_H
#define BONDS_H
 
inline int VisualisationState::lua_bond(lua_State * lua)
{
    if (lua_gettop(lua) >= 1 && lua_isstring(lua, 1) && std::string(lua_tostring(lua, 1)) == "help")
    {
        lua_writestring("bond:\n  Arguments:\n    i [number]\n    j [number]\n  Bond i and j.\n");
        return 0;
    };
    int args = lua_gettop(lua);
    if (args != 2)
    {
        const std::string msg = "bond expects two atom indices as argument.\n";
        lua_pushlstring(lua, msg.c_str(), msg.length());
        return lua_error(lua);
    }
 
    LuaNumber li, lj;
    li.read(lua, 1);
    lj.read(lua, 2);
 
    uint64_t i = li.n;
    uint64_t j = lj.n;
 
    if (i >= atomCount || j >= atomCount)
    {
        const std::string msg = "bond atom index larger than atom count.\n";
        lua_pushlstring(lua, msg.c_str(), msg.length());
        return lua_error(lua);
    }
 
    bonds[i].insert(j);
    auto s = bonds[j];
    if (s.find(i) != s.cend())
    {
        bonds[j].erase(i);
    }
    elementsUpdated = true;
 
    return 0;
}
 
inline int VisualisationState::lua_unbond(lua_State * lua)
{
    if (lua_gettop(lua) >= 1 && lua_isstring(lua, 1) && std::string(lua_tostring(lua, 1)) == "help")
    {
        lua_writestring("unbond:\n  Arguments:\n    i [number]\n    j [number]\n  Unbond i and j.\n");
        return 0;
    };
    int args = lua_gettop(lua);
    if (args != 2)
    {
        const std::string msg = "unbond expects two atom indices as argument.\n";
        lua_pushlstring(lua, msg.c_str(), msg.length());
        return lua_error(lua);
    }
 
    LuaNumber li, lj;
    li.read(lua, 1);
    lj.read(lua, 2);
 
    uint64_t i = li.n;
    uint64_t j = lj.n;
 
    if (i >= atomCount || j >= atomCount)
    {
        const std::string msg = "unbond atom index larger than atom count.\n";
        lua_pushlstring(lua, msg.c_str(), msg.length());
        return lua_error(lua);
    }
 
    auto s = bonds[i];
    if (s.find(j) != s.cend())
    {
        bonds[i].erase(j);
    }
 
    s = bonds[j];
    if (s.find(i) != s.cend())
    {
        bonds[j].erase(i);
    }
    elementsUpdated = true;
 
    return 0;
}
 
#endif /* BONDS_H */