AM1RepulsionEnergy.h

Go to the documentation of this file.

#ifndef SPARROW_AM1REPULSIONENERGY_H
#define SPARROW_AM1REPULSIONENERGY_H

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

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

namespace nddo {
class ElementParameters;
class AM1RepulsionEnergy : public Utils::RepulsionCalculator {
 public:
  using PairRepulsionType = std::unique_ptr<AM1PairwiseRepulsion>;
  using Container = std::vector<std::vector<PairRepulsionType>>;

  AM1RepulsionEnergy(const Utils::ElementTypeCollection& elements, const Utils::PositionCollection& positions,
                     const ElementParameters& elementParameters);
  ~AM1RepulsionEnergy() 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 // REPULSIONENERGY_H