Program Listing for File result_expr.cpp

↰ Return to documentation for file (SeQuant/core/result_expr.cpp)

#include <SeQuant/core/result_expr.hpp>
#include <SeQuant/core/tensor.hpp>
#include <SeQuant/core/utility/indices.hpp>

namespace sequant {

bool ResultExpr::ResultCmp::operator()(const ResultExpr &lhs,
                                       const ResultExpr &rhs) const {
  if (lhs.has_label() != rhs.has_label()) {
    return false;
  }

  if (lhs.has_label() && lhs.label() != rhs.label()) {
    return false;
  }

  return lhs.symmetry() == rhs.symmetry() &&
         lhs.braket_symmetry() == rhs.braket_symmetry() &&
         lhs.particle_symmetry() == rhs.particle_symmetry() &&
         lhs.bra() == rhs.bra() && lhs.ket() == rhs.ket() &&
         lhs.aux() == rhs.aux();
}

ResultExpr::ResultExpr(const Tensor &tensor, ExprPtr expression)
    : m_expr(std::move(expression)),
      m_symm(tensor.symmetry()),
      m_bksymm(tensor.braket_symmetry()),
      m_psymm(tensor.particle_symmetry()),
      m_braIndices(tensor.bra().begin(), tensor.bra().end()),
      m_ketIndices(tensor.ket().begin(), tensor.ket().end()),
      m_label(tensor.label()) {}

ResultExpr::ResultExpr(const Variable &variable, ExprPtr expression)
    : m_expr(std::move(expression)), m_label(variable.label()) {}

ResultExpr::ResultExpr(IndexContainer bra, IndexContainer ket,
                       IndexContainer aux, Symmetry symm,
                       BraKetSymmetry braket_symm,
                       ParticleSymmetry particle_symm,
                       std::optional<std::wstring> label, ExprPtr expression)
    : m_expr(std::move(expression)),
      m_braIndices(std::move(bra)),
      m_ketIndices(std::move(ket)),
      m_auxIndices(std::move(aux)),
      m_symm(symm),
      m_bksymm(braket_symm),
      m_psymm(particle_symm),
      m_label(std::move(label)) {}

ResultExpr &ResultExpr::operator=(ExprPtr expression) {
  m_expr = std::move(expression);

  return *this;
}

bool ResultExpr::has_label() const { return m_label.has_value(); }

const std::wstring &ResultExpr::label() const { return m_label.value(); }

void ResultExpr::set_label(std::wstring label) { m_label = std::move(label); }

Symmetry ResultExpr::symmetry() const { return m_symm; }

void ResultExpr::set_symmetry(Symmetry symm) { m_symm = symm; }

BraKetSymmetry ResultExpr::braket_symmetry() const { return m_bksymm; }

void ResultExpr::set_braket_symmetry(BraKetSymmetry symm) { m_bksymm = symm; }

ParticleSymmetry ResultExpr::particle_symmetry() const { return m_psymm; }

void ResultExpr::set_particle_symmetry(ParticleSymmetry symm) {
  m_psymm = symm;
}

const ResultExpr::IndexContainer &ResultExpr::bra() const {
  return m_braIndices;
}

const ResultExpr::IndexContainer &ResultExpr::ket() const {
  return m_ketIndices;
}

const ResultExpr::IndexContainer &ResultExpr::aux() const {
  return m_auxIndices;
}

const ExprPtr &ResultExpr::expression() const { return m_expr; }

ExprPtr &ResultExpr::expression() { return m_expr; }

ResultExpr ResultExpr::clone() const {
  return ResultExpr(m_braIndices, m_ketIndices, m_auxIndices, m_symm, m_bksymm,
                    m_psymm, m_label, m_expr->clone());
}

}  // namespace sequant