/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef jit_ABIArgGenerator_h
#define jit_ABIArgGenerator_h

#include "mozilla/Assertions.h"

#include <stdint.h>

#include "jit/ABIFunctionType.h"
#include "jit/Assembler.h"
#include "jit/IonTypes.h"
#include "jit/RegisterSets.h"
#include "wasm/WasmFrame.h"

namespace js::jit {

static inline MIRType ToMIRType(MIRType t) { return t; }

static inline MIRType ToMIRType(ABIType argType) {
  switch (argType) {
    case ABIType::General:
      return MIRType::Pointer;
    case ABIType::Float64:
      return MIRType::Double;
    case ABIType::Float32:
      return MIRType::Float32;
    case ABIType::Int32:
      return MIRType::Int32;
    case ABIType::Int64:
      return MIRType::Int64;
    case ABIType::Void:
      return MIRType::None;
    default:
      break;
  }
  MOZ_CRASH("unexpected argType");
}

template <class VecT>
class ABIArgIter {
  ABIArgGenerator gen_;
  const VecT& types_;
  unsigned i_;

  void settle() {
    if (!done()) gen_.next(ToMIRType(types_[i_]));
  }

 public:
  ABIArgIter(const VecT& types, ABIKind kind)
      : gen_(kind), types_(types), i_(0) {
    settle();
  }
  void operator++(int) {
    MOZ_ASSERT(!done());
    i_++;
    settle();
  }
  bool done() const { return i_ == types_.length(); }

  ABIArg* operator->() {
    MOZ_ASSERT(!done());
    return &gen_.current();
  }
  ABIArg& operator*() {
    MOZ_ASSERT(!done());
    return gen_.current();
  }

  unsigned index() const {
    MOZ_ASSERT(!done());
    return i_;
  }
  MIRType mirType() const {
    MOZ_ASSERT(!done());
    return ToMIRType(types_[i_]);
  }
  uint32_t stackBytesConsumedSoFar() const {
    return gen_.stackBytesConsumedSoFar();
  }
};

}  // namespace js::jit

#endif /* jit_ABIArgGenerator_h */