torch/csrc/jit/mobile/frame.h

#pragma once

#include <cstddef>

#include <c10/util/Optional.h>
#include <torch/csrc/jit/mobile/code.h>

namespace torch {
namespace jit {
namespace mobile {

class Frame {
 public:
  explicit Frame(const Code& code) : code_(code) {}
  const Code& getCode() const {
    return code_;
  }

  void step() {
    pc_++;
  }

  void jump(size_t n) {
    pc_ += n;
  }

  size_t getPC() const {
    return pc_;
  }

  const Instruction& getInstruction() const {
    return code_.instructions_.at(pc_);
  }

  c10::optional<int64_t> getDebugHandle() const {
    return getDebugHandle(pc_);
  }

  c10::optional<int64_t> getDebugHandle(size_t pc) const {
    if (pc >= code_.debug_handles_.size()) {
      return {};
    }
    return code_.debug_handles_[pc];
  }

 private:
  const Code& code_;
  size_t pc_{0};
};

} // namespace mobile
} // namespace jit
} // namespace torch