verilog 写一个数码管闪烁，一旦按键摁到第三个摁键时，数码管开始从9到0倒计时

以下是一个用Verilog编写的数码管闪烁代码。在接收到第三个按键按下的信号时，数码管将反转并开始倒计时。倒计时完成后，数字9将重新出现并重新开始闪烁。

module digit_display(
    input clk, // 时钟信号
    input [2:0] btn, // 按键信号
    output reg [6:0] seg // 数码管信号
);

reg [3:0] count; // 倒计时计数器
reg [3:0] max_count; // 倒计时上限

always @(posedge clk) begin
    if(btn[2] == 1) begin
        // 第三个按键按下，开始倒计时
        max_count <= 10;
        count <= count + 1;
        if(count == max_count) begin
            // 倒计时结束
            count <= 0;
        end
    end else begin
        // 闪烁模式
        max_count <= 2;
        count <= count + 1;
        if(count == max_count) begin
            count <= 0;
            if (seg == 7'b111_1110) begin
                seg <= 7'b111_1111; // 切换到下一个数字
            end else begin
                seg <= 7'b111_1110; // 切换到上一个数字
            end
        end
    end
end

assign seg = (btn[2] && count != max_count) ? 7'b111_1001 : seg;

endmodule