module heron_fsm(clk, load, reset, flag_z, flag_c, flag_s, state);
  // (C) 1953, Frank Gray
  parameter IDLE           = 4'b0000;
  parameter LD_N_1         = 4'b0001;
  parameter LD_N_2         = 4'b0011;
  parameter I_GT_ZERO      = 4'b0010;
  parameter LD_S_1         = 4'b0110;
  parameter LD_S_2         = 4'b0111;
  parameter S_GT_ONE       = 4'b0101;
  parameter X_1            = 4'b0100;
  parameter DIV_1          = 4'b1100;
  parameter DIV_2          = 4'b1101;
  parameter DIV_3          = 4'b1111;
  parameter DIV_4          = 4'b1110;
  parameter OLD_X_LTE_X_1  = 4'b1010;
  parameter OLD_X_LTE_X_2  = 4'b1011;
  parameter STORE_X        = 4'b1001;
  parameter DEC_I          = 4'b1000;

  input clk;
  input load;
  input reset;

  input flag_z;
  input flag_c;
  input flag_s;

  output [3:0] state;

  reg [3:0] state;
  reg [3:0] next_state;

  always @(posedge clk or posedge reset)
    if(reset) begin
      state <= IDLE;
    end else begin
      state <= next_state;
    end

  always @(state or flag_z or load)
    case(state)
      IDLE:
        if(load) begin
          next_state = LD_N_1;
        end else begin
          next_state = IDLE;
        end
      LD_N_1: next_state = LD_N_2;
      LD_N_2: next_state = I_GT_ZERO;
      I_GT_ZERO:
        if (flag_z) begin
          next_state = LD_S_1;
        end else begin
          next_state = IDLE;
        end
      LD_S_1: next_state = LD_S_2;
      LD_S_2: next_state = S_GT_ONE;
      S_GT_ONE:
        if (flag_s || flag_z) begin
          next_state = STORE_X;
        end else begin
          next_state = X_1;
        end
      X_1: next_state = DIV_2;
      DIV_1: next_state = DIV_2;
      DIV_2: next_state = DIV_3;
      DIV_3: next_state = DIV_4;
      DIV_4: next_state = OLD_X_LTE_X_1;
      OLD_X_LTE_X_1: next_state = OLD_X_LTE_X_2;
      OLD_X_LTE_X_2:
        if (flag_c) begin
          next_state = DIV_1;
        end else begin
          next_state = STORE_X;
        end
      STORE_X: next_state = DEC_I;
      DEC_I: next_state = I_GT_ZERO;
    endcase
    end
endmodule