Skip to content

Commit

Permalink
improve doc
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed Jan 13, 2024
1 parent d911a22 commit 0a52f7f
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions source/api_cc/include/DeepPot.h
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,7 @@ class DeepPotBase {
* nframes x natoms x dim_aparam.
* natoms x dim_aparam. Then all frames are assumed to be provided with the
*same aparam.
* @{
**/
virtual void computew(
double& ener,
Expand Down Expand Up @@ -94,6 +95,7 @@ class DeepPotBase {
const std::vector<float>& box,
const std::vector<float>& fparam = std::vector<float>(),
const std::vector<float>& aparam = std::vector<float>()) = 0;
/** @} */
/**
* @brief Evaluate the energy, force and virial by using this DP.
* @param[out] ener The system energy.
Expand All @@ -115,6 +117,7 @@ class DeepPotBase {
* nframes x natoms x dim_aparam.
* natoms x dim_aparam. Then all frames are assumed to be provided with the
*same aparam.
* @{
**/
virtual void computew(
double& ener,
Expand Down Expand Up @@ -165,6 +168,8 @@ class DeepPotBase {
const int& ago,
const std::vector<float>& fparam = std::vector<float>(),
const std::vector<float>& aparam = std::vector<float>()) = 0;
/** @} */

/**
* @brief Evaluate the energy, force, virial, atomic energy, and atomic virial
*by using this DP.
Expand All @@ -186,6 +191,7 @@ class DeepPotBase {
* nframes x natoms x dim_aparam.
* natoms x dim_aparam. Then all frames are assumed to be provided with the
*same aparam.
* @{
**/
virtual void computew(
double& ener,
Expand Down Expand Up @@ -232,6 +238,7 @@ class DeepPotBase {
const std::vector<float>& box,
const std::vector<float>& fparam = std::vector<float>(),
const std::vector<float>& aparam = std::vector<float>()) = 0;
/** @} */
/**
* @brief Evaluate the energy, force, virial, atomic energy, and atomic virial
*by using this DP.
Expand All @@ -256,6 +263,7 @@ class DeepPotBase {
* nframes x natoms x dim_aparam.
* natoms x dim_aparam. Then all frames are assumed to be provided with the
*same aparam.
* @{
**/
virtual void computew(
double& ener,
Expand Down Expand Up @@ -314,6 +322,7 @@ class DeepPotBase {
const int& ago,
const std::vector<float>& fparam = std::vector<float>(),
const std::vector<float>& aparam = std::vector<float>()) = 0;
/** @} */
/**
* @brief Evaluate the energy, force, and virial with the mixed type
*by using this DP.
Expand All @@ -335,6 +344,7 @@ class DeepPotBase {
* nframes x natoms x dim_aparam.
* natoms x dim_aparam. Then all frames are assumed to be provided with the
*same aparam.
* @{
**/
virtual void computew_mixed_type(
double& ener,
Expand Down Expand Up @@ -376,6 +386,7 @@ class DeepPotBase {
const std::vector<float>& box,
const std::vector<float>& fparam = std::vector<float>(),
const std::vector<float>& aparam = std::vector<float>()) = 0;
/** @} */
/**
* @brief Evaluate the energy, force, and virial with the mixed type
*by using this DP.
Expand All @@ -399,6 +410,7 @@ class DeepPotBase {
* nframes x natoms x dim_aparam.
* natoms x dim_aparam. Then all frames are assumed to be provided with the
*same aparam.
* @{
**/
virtual void computew_mixed_type(
double& ener,
Expand Down Expand Up @@ -448,6 +460,7 @@ class DeepPotBase {
const std::vector<float>& box,
const std::vector<float>& fparam = std::vector<float>(),
const std::vector<float>& aparam = std::vector<float>()) = 0;
/** @} */
/**
* @brief Get the cutoff radius.
* @return The cutoff radius.
Expand Down

0 comments on commit 0a52f7f

Please sign in to comment.