Program Listing for File potentialSubstrate.cpp

↰ Return to documentation for file (pymembrane/cppmodule/src/potentials/potentialSubstrate.cpp)

#include "potentialSubstrate.hpp"

void ComputeVertexLimitForce_fn(int Numvertices,
                                HE_VertexProp *vertices,
                                real *_kz1,
                                real *_kz2)
{
    for (int vertex_index = 0; vertex_index < Numvertices; vertex_index++)
    {
        int type = vertices[vertex_index].type;
        auto r = vertices[vertex_index].r;

        auto fval = 0.0;

        if (r.z >= 0.0)
            fval = -_kz1[type] * r.z;
        else
            fval = -_kz2[type] * r.z;

        vertices[vertex_index].forceC.z += fval;
    }
}
void ComputeVertexSubstrateEnergy::compute(void)
{

    ComputeVertexLimitForce_fn(_system.Numvertices,
                               &_system.vertices[0],
                               &m_kz1[0],
                               &m_kz2[0]);
}