MNDORepulsionEnergy.h

Go to the documentation of this file.

#ifndef SPARROW_MNDOREPULSIONENERGY_H
#define SPARROW_MNDOREPULSIONENERGY_H

#include "MNDOPairwiseRepulsion.h"
#include <Utils/Scf/MethodInterfaces/RepulsionCalculator.h>
#include <memory>
#include <vector>

namespace Scine {
namespace Utils {
enum class DerivativeOrder;
}
namespace Sparrow {

namespace nddo {
class ElementParameters;

class MNDORepulsionEnergy : public Utils::RepulsionCalculator {
 public:
  using pairRepulsion_t = std::unique_ptr<MNDOPairwiseRepulsion>;
  using Container = std::vector<std::vector<pairRepulsion_t>>;

  MNDORepulsionEnergy(const Utils::ElementTypeCollection& elements, const Utils::PositionCollection& positions,
                      const ElementParameters& elementParameters);
  ~MNDORepulsionEnergy() override;

  void initialize() override;
  void calculateRepulsion(Utils::DerivativeOrder order) override;
  double getRepulsionEnergy() const override;
  void addRepulsionDerivatives(Utils::AutomaticDifferentiation::DerivativeContainerType<Utils::Derivative::First>& derivatives) const override;
  void addRepulsionDerivatives(
      Utils::AutomaticDifferentiation::DerivativeContainerType<Utils::Derivative::SecondAtomic>& derivatives) const override;
  void addRepulsionDerivatives(
      Utils::AutomaticDifferentiation::DerivativeContainerType<Utils::Derivative::SecondFull>& derivatives) const override;

 private:
  template<Utils::Derivative O>
  void addRepulsionDerivativesImpl(Utils::AutomaticDifferentiation::DerivativeContainerType<O>& derivatives) const;

  void calculatePairRepulsion(int i, int j, Utils::DerivativeOrder order);
  void initializePair(int i, int j);

  const ElementParameters& elementParameters_;
  Container rep_;
  int nAtoms_;
};

} // namespace nddo

} // namespace Sparrow
} // namespace Scine

#endif // SPARROW_REPULSIONENERGY_H