import log_reader
import interactive_host
import sys


def main(args):
  # this advisor just picks the first legal register to evict, which is
  # identifiable by the "mask" feature
  class Advisor:
    to_return = False

    def advice(self, tensor_values: list[log_reader.TensorValue]):
      for tv in tensor_values:
        if tv.spec().name != 'mask':
          continue
        for i, v in enumerate(tv):
          if v == 1:
            return i
      # i.e. invalid:
      return -1


  a = Advisor()
  interactive_host.run_interactive(args[0], a.advice, args[1:])


if __name__ == '__main__':
  main(sys.argv[1:])