#pragma once

#include <ostream>
#include <string_view>

#include "visitor.hpp"
#include "expr.hpp"
#include "num.hpp"
#include "stmt.hpp"

namespace dsp {

struct ast_printer_t : visitor_t<void>
{
  ast_printer_t(std::ostream& os)
    : os(os) {}

  std::ostream& os;

  void visit(const unary_t * unary) const;
  void visit(const binary_t * binary) const;
  void visit(const grouping_t * grouping) const;
  void visit(const literal_t * literal) const;

  void visit(const op::alu_t * alu) const;
  void visit(const op::mov_ram_x_t * mov_ram_x) const;
  void visit(const op::mov_mul_p_t * mov_mul_p) const;
  void visit(const op::mov_ram_p_t * mov_ram_p) const;
  void visit(const op::mov_ram_y_t * mov_ram_y) const;
  void visit(const op::clr_a_t * clr_a) const;
  void visit(const op::mov_alu_a_t * mov_alu_a) const;
  void visit(const op::mov_ram_a_t * mov_ram_a) const;
  void visit(const op::mov_imm_d1_t * mov_imm_) const;
  void visit(const op::mov_ram_d1_t * mov_ram_) const;
  void visit(const op::instruction_t * instruction) const;

  void visit(const load::mvi_t * mvi) const;
  void visit(const load::mvi_cond_t * mvi_cond) const;

  void visit(const dma::ingress_imm_t * ingress_imm) const;
  void visit(const dma::egress_imm_t * egress_imm) const;
  void visit(const dma::ingress_ram_t * ingress_ram) const;
  void visit(const dma::egress_ram_t * egress_ram) const;

  void visit(const jump::jmp_t * jmp) const;
  void visit(const jump::jmp_cond_t * jmp_cond) const;

  void visit(const loop::btm_t * btm) const;
  void visit(const loop::lps_t * lps) const;

  void visit(const end::end_t * end) const;
  void visit(const end::endi_t  * endi) const;

  void parenthesize(const std::string_view s, const expr_t * a) const;
  void parenthesize(const std::string_view s, const expr_t * a, const expr_t * b) const;
};

}