// Copyright 2017 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef V8_IC_BINARY_OP_ASSEMBLER_H_
#define V8_IC_BINARY_OP_ASSEMBLER_H_

#include <functional>

#include "src/codegen/code-stub-assembler.h"

namespace v8 {
namespace internal {

namespace compiler {
class CodeAssemblerState;
}  // namespace compiler

class BinaryOpAssembler : public CodeStubAssembler {
 public:
  explicit BinaryOpAssembler(compiler::CodeAssemblerState* state)
      : CodeStubAssembler(state) {}

  FeedbackUpdater MakeVectorSlotFeedbackUpdater(
      TNode<UintPtrT> slot, const LazyNode<HeapObject>& maybe_feedback_vector,
      UpdateFeedbackMode update_feedback_mode) {
    return [=, this](TNode<Smi> feedback) {
      UpdateFeedback(feedback, maybe_feedback_vector(), slot,
                     update_feedback_mode);
    };
  }

  FeedbackUpdater MakeEmbeddedFeedbackUpdater(
      TNode<BytecodeArray> bytecode_array, TNode<IntPtrT> feedback_offset) {
    return [=, this](TNode<Smi> feedback) {
      UpdateEmbeddedFeedback<BinaryOperationFeedback>(feedback, bytecode_array,
                                                      feedback_offset);
    };
  }

  TNode<Object> Generate_AddWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      TNode<UintPtrT> slot, const LazyNode<HeapObject>& maybe_feedback_vector,
      UpdateFeedbackMode update_feedback_mode, bool rhs_known_smi) {
    return Generate_AddWithFeedback(
        context, left, right,
        MakeVectorSlotFeedbackUpdater(slot, maybe_feedback_vector,
                                      update_feedback_mode),
        rhs_known_smi);
  }

  TNode<Object> Generate_AddWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi);

  TNode<Object> Generate_AddLhsIsStringConstantInternalizeWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      TNode<UintPtrT> slot, const LazyNode<HeapObject>& maybe_feedback_vector,
      UpdateFeedbackMode update_feedback_mode, bool rhs_known_smi);
  TNode<Object> Generate_AddRhsIsStringConstantInternalizeWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      TNode<UintPtrT> slot, const LazyNode<HeapObject>& maybe_feedback_vector,
      UpdateFeedbackMode update_feedback_mode, bool rhs_known_smi);
  TNode<Object> Generate_AddStringConstantInternalizeWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      TNode<UintPtrT> slot, const LazyNode<HeapObject>& maybe_feedback_vector,
      UpdateFeedbackMode update_feedback_mode, bool rhs_known_smi,
      AddStringConstantAndInternalizeVariant as_variant);

  TNode<Object> Generate_SubtractWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi);

  TNode<Object> Generate_MultiplyWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi);

  TNode<Object> Generate_DivideWithFeedback(
      const LazyNode<Context>& context, TNode<Object> dividend,
      TNode<Object> divisor, const FeedbackUpdater& feedback_updater,
      bool rhs_known_smi);

  TNode<Object> Generate_ModulusWithFeedback(
      const LazyNode<Context>& context, TNode<Object> dividend,
      TNode<Object> divisor, const FeedbackUpdater& feedback_updater,
      bool rhs_known_smi);

  TNode<Object> Generate_ExponentiateWithFeedback(
      const LazyNode<Context>& context, TNode<Object> base,
      TNode<Object> exponent, const FeedbackUpdater& feedback_updater,
      bool rhs_known_smi);

  TNode<Object> Generate_BitwiseOrWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi) {
    return Generate_BitwiseBinaryOpWithFeedback(
        Operation::kBitwiseOr, left, right, context, feedback_updater,
        rhs_known_smi);
  }

  TNode<Object> Generate_BitwiseXorWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi) {
    return Generate_BitwiseBinaryOpWithFeedback(
        Operation::kBitwiseXor, left, right, context, feedback_updater,
        rhs_known_smi);
  }

  TNode<Object> Generate_BitwiseAndWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi) {
    return Generate_BitwiseBinaryOpWithFeedback(
        Operation::kBitwiseAnd, left, right, context, feedback_updater,
        rhs_known_smi);
  }

  TNode<Object> Generate_ShiftLeftWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi) {
    return Generate_BitwiseBinaryOpWithFeedback(
        Operation::kShiftLeft, left, right, context, feedback_updater,
        rhs_known_smi);
  }

  TNode<Object> Generate_ShiftRightWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi) {
    return Generate_BitwiseBinaryOpWithFeedback(
        Operation::kShiftRight, left, right, context, feedback_updater,
        rhs_known_smi);
  }

  TNode<Object> Generate_ShiftRightLogicalWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, bool rhs_known_smi) {
    return Generate_BitwiseBinaryOpWithFeedback(
        Operation::kShiftRightLogical, left, right, context, feedback_updater,
        rhs_known_smi);
  }

  TNode<Object> Generate_BitwiseBinaryOpWithFeedback(
      Operation bitwise_op, TNode<Object> left, TNode<Object> right,
      const LazyNode<Context>& context, const FeedbackUpdater& feedback_updater,
      bool rhs_known_smi) {
    return rhs_known_smi
               ? Generate_BitwiseBinaryOpWithSmiOperandAndOptionalFeedback(
                     bitwise_op, left, right, context, &feedback_updater)
               : Generate_BitwiseBinaryOpWithOptionalFeedback(
                     bitwise_op, left, right, context, &feedback_updater);
  }

  TNode<Object> Generate_BitwiseBinaryOp(Operation bitwise_op,
                                         TNode<Object> left,
                                         TNode<Object> right,
                                         TNode<Context> context) {
    return Generate_BitwiseBinaryOpWithOptionalFeedback(
        bitwise_op, left, right, [&] { return context; }, nullptr);
  }

 private:
  using SmiOperation =
      std::function<TNode<Object>(TNode<Smi>, TNode<Smi>, TVariable<Smi>*)>;
  using FloatOperation =
      std::function<TNode<Float64T>(TNode<Float64T>, TNode<Float64T>)>;

  TNode<Object> Generate_BinaryOperationWithFeedback(
      const LazyNode<Context>& context, TNode<Object> left, TNode<Object> right,
      const FeedbackUpdater& feedback_updater, const SmiOperation& smiOperation,
      const FloatOperation& floatOperation, Operation op, bool rhs_known_smi);

  TNode<Object> Generate_BitwiseBinaryOpWithOptionalFeedback(
      Operation bitwise_op, TNode<Object> left, TNode<Object> right,
      const LazyNode<Context>& context,
      const FeedbackUpdater* feedback_updater);

  TNode<Object> Generate_BitwiseBinaryOpWithSmiOperandAndOptionalFeedback(
      Operation bitwise_op, TNode<Object> left, TNode<Object> right,
      const LazyNode<Context>& context,
      const FeedbackUpdater* feedback_updater);

  // Check if output is known to be Smi when both operands of bitwise operation
  // are Smi.
  bool IsBitwiseOutputKnownSmi(Operation bitwise_op) {
    switch (bitwise_op) {
      case Operation::kBitwiseAnd:
      case Operation::kBitwiseOr:
      case Operation::kBitwiseXor:
      case Operation::kShiftRight:
        return true;
      default:
        return false;
    }
  }
};

}  // namespace internal
}  // namespace v8

#endif  // V8_IC_BINARY_OP_ASSEMBLER_H_