llvm/flang/include/flang/Evaluate/logical.h

//===-- include/flang/Evaluate/logical.h ------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef FORTRAN_EVALUATE_LOGICAL_H_
#define FORTRAN_EVALUATE_LOGICAL_H_

#include "integer.h"
#include <cinttypes>

namespace Fortran::evaluate::value {

template <int BITS, bool IS_LIKE_C = true> class Logical {
public:
  static constexpr int bits{BITS};
  using Word = Integer<bits>;

  // Module ISO_C_BINDING kind C_BOOL is LOGICAL(KIND=1) and must have
  // C's bit representation (.TRUE. -> 1, .FALSE. -> 0).
  static constexpr bool IsLikeC{BITS <= 8 || IS_LIKE_C};

  constexpr Logical() {} // .FALSE.
  template <int B, bool C>
  constexpr Logical(Logical<B, C> x) : word_{Represent(x.IsTrue())} {}
  constexpr Logical(bool truth) : word_{Represent(truth)} {}
  // A raw word, for DATA initialization
  constexpr Logical(Word &&w) : word_{std::move(w)} {}

  template <int B, bool C> constexpr Logical &operator=(Logical<B, C> x) {
    word_ = Represent(x.IsTrue());
    return *this;
  }

  Word word() const { return word_; }
  bool IsCanonical() const {
    return word_ == canonicalFalse || word_ == canonicalTrue;
  }

  // Fortran actually has only .EQV. & .NEQV. relational operations
  // for LOGICAL, but this template class supports more so that
  // it can be used with the STL for sorting and as a key type for
  // std::set<> & std::map<>.
  template <int B, bool C>
  constexpr bool operator<(const Logical<B, C> &that) const {
    return !IsTrue() && that.IsTrue();
  }
  template <int B, bool C>
  constexpr bool operator<=(const Logical<B, C> &) const {
    return !IsTrue();
  }
  template <int B, bool C>
  constexpr bool operator==(const Logical<B, C> &that) const {
    return IsTrue() == that.IsTrue();
  }
  template <int B, bool C>
  constexpr bool operator!=(const Logical<B, C> &that) const {
    return IsTrue() != that.IsTrue();
  }
  template <int B, bool C>
  constexpr bool operator>=(const Logical<B, C> &) const {
    return IsTrue();
  }
  template <int B, bool C>
  constexpr bool operator>(const Logical<B, C> &that) const {
    return IsTrue() && !that.IsTrue();
  }

  constexpr bool IsTrue() const {
    if constexpr (IsLikeC) {
      return !word_.IsZero();
    } else {
      return word_.BTEST(0);
    }
  }

  constexpr Logical NOT() const { return {word_.IEOR(canonicalTrue)}; }

  constexpr Logical AND(const Logical &that) const {
    return {word_.IAND(that.word_)};
  }

  constexpr Logical OR(const Logical &that) const {
    return {word_.IOR(that.word_)};
  }

  constexpr Logical EQV(const Logical &that) const { return NEQV(that).NOT(); }

  constexpr Logical NEQV(const Logical &that) const {
    return {word_.IEOR(that.word_)};
  }

private:
  static constexpr Word canonicalTrue{IsLikeC ? 1 : -std::uint64_t{1}};
  static constexpr Word canonicalFalse{0};
  static constexpr Word Represent(bool x) {
    return x ? canonicalTrue : canonicalFalse;
  }
  Word word_;
};

extern template class Logical<8>;
extern template class Logical<16>;
extern template class Logical<32>;
extern template class Logical<64>;
} // namespace Fortran::evaluate::value
#endif // FORTRAN_EVALUATE_LOGICAL_H_