/*
 * Copyright (C) 2024 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef ART_COMPILER_OPTIMIZING_NODES_RISCV64_H_
#define ART_COMPILER_OPTIMIZING_NODES_RISCV64_H_

namespace art HIDDEN {

class HRiscv64ShiftAdd final : public HBinaryOperation {
 public:
  HRiscv64ShiftAdd(HInstruction* left,
                   HInstruction* right,
                   uint32_t distance,
                   uint32_t dex_pc = kNoDexPc)
      : HBinaryOperation(
            kRiscv64ShiftAdd, DataType::Type::kInt64, left, right, SideEffects::None(), dex_pc) {
    DCHECK_GE(distance, 1u);
    DCHECK_LE(distance, 3u);

    SetPackedField<DistanceField>(distance);
  }

  uint32_t GetDistance() const { return GetPackedField<DistanceField>(); }

  bool InstructionDataEquals(const HInstruction* other) const override {
    return GetPackedFields() == other->AsRiscv64ShiftAdd()->GetPackedFields();
  }

  HConstant* Evaluate(HLongConstant* x, HLongConstant* y) const override {
    const int64_t value = y->GetValue() + (x->GetValue() << GetDistance());
    return GetBlock()->GetGraph()->GetLongConstant(value);
  }

  DECLARE_INSTRUCTION(Riscv64ShiftAdd);

 protected:
  DEFAULT_COPY_CONSTRUCTOR(Riscv64ShiftAdd);

 private:
  static constexpr size_t kFieldDistance = kNumberOfGenericPackedBits;
  static constexpr size_t kFieldDistanceSize = MinimumBitsToStore(3u);
  static constexpr size_t kNumberOfRiscv64ShiftAddPackedBits =
      kFieldDistance + kFieldDistanceSize;
  static_assert(kNumberOfRiscv64ShiftAddPackedBits <= kMaxNumberOfPackedBits,
                "Too many packed fields.");
  using DistanceField = BitField<uint32_t, kFieldDistance, kFieldDistanceSize>;
};

class HRiscv64BitSet final : public HBinaryOperation {
 public:
  HRiscv64BitSet(HInstruction* left, HInstruction* right, uint32_t dex_pc = kNoDexPc)
      : HBinaryOperation(
            kRiscv64BitSet, DataType::Type::kInt64, left, right, SideEffects::None(), dex_pc) {}

  HConstant* Evaluate(HLongConstant* x, HIntConstant* y) const override {
    int64_t value = x->GetValue() | (UINT64_C(1) << (y->GetValue() & kMaxLongShiftDistance));
    return GetBlock()->GetGraph()->GetLongConstant(value);
  }

  DECLARE_INSTRUCTION(Riscv64BitSet);

 protected:
  DEFAULT_COPY_CONSTRUCTOR(Riscv64BitSet);
};

class HRiscv64BitClear final : public HBinaryOperation {
 public:
  HRiscv64BitClear(HInstruction* left, HInstruction* right, uint32_t dex_pc = kNoDexPc)
      : HBinaryOperation(
            kRiscv64BitClear, DataType::Type::kInt64, left, right, SideEffects::None(), dex_pc) {}

  HConstant* Evaluate(HLongConstant* x, HIntConstant* y) const override {
    int64_t value = x->GetValue() & (~(UINT64_C(1) << (y->GetValue() & kMaxLongShiftDistance)));
    return GetBlock()->GetGraph()->GetLongConstant(value);
  }

  DECLARE_INSTRUCTION(Riscv64BitClear);

 protected:
  DEFAULT_COPY_CONSTRUCTOR(Riscv64BitClear);
};

class HRiscv64BitExtract final : public HBinaryOperation {
 public:
  HRiscv64BitExtract(HInstruction* left, HInstruction* right, uint32_t dex_pc = kNoDexPc)
      : HBinaryOperation(
            kRiscv64BitExtract, DataType::Type::kInt64, left, right, SideEffects::None(), dex_pc) {}

  HConstant* Evaluate(HLongConstant* x, HIntConstant* y) const override {
    int64_t value = (x->GetValue() >> (y->GetValue() & kMaxLongShiftDistance)) & UINT64_C(1);
    return GetBlock()->GetGraph()->GetLongConstant(value);
  }

  DECLARE_INSTRUCTION(Riscv64BitExtract);

 protected:
  DEFAULT_COPY_CONSTRUCTOR(Riscv64BitExtract);
};

class HRiscv64BitInvert final : public HBinaryOperation {
 public:
  HRiscv64BitInvert(HInstruction* left, HInstruction* right, uint32_t dex_pc = kNoDexPc)
      : HBinaryOperation(
            kRiscv64BitInvert, DataType::Type::kInt64, left, right, SideEffects::None(), dex_pc) {}

  HConstant* Evaluate(HLongConstant* x, HIntConstant* y) const override {
    int64_t value = x->GetValue() ^ (UINT64_C(1) << (y->GetValue() & kMaxLongShiftDistance));
    return GetBlock()->GetGraph()->GetLongConstant(value);
  }

  DECLARE_INSTRUCTION(Riscv64BitInvert);

 protected:
  DEFAULT_COPY_CONSTRUCTOR(Riscv64BitInvert);
};

}  // namespace art

#endif  // ART_COMPILER_OPTIMIZING_NODES_RISCV64_H_