finish reconstruction and pass simulation
This commit is contained in:
@@ -2,36 +2,37 @@
|
||||
|
||||
// 三通道图像合成一个RGB图像
|
||||
module ColorBlender #(
|
||||
parameter reg [ 4:0] IN_DEPTH = 12, // 输入图像的色深
|
||||
parameter reg [ 4:0] OUT_DEPTH = 8, // 输出图像的色深
|
||||
parameter reg [16:0] GAIN_RED = 120, // 红色增益系数
|
||||
parameter reg [16:0] GAIN_GREEN = 50, // 绿色增益系数
|
||||
parameter reg [16:0] GAIN_BLUE = 95, // 蓝色增益系数
|
||||
parameter reg [ 4:0] GAIN_OFFSET = 7
|
||||
parameter reg [ 4:0] IN_DEPTH = 12, // 输入图像的色深
|
||||
parameter reg [ 4:0] OUT_DEPTH = 8, // 输出图像的色深
|
||||
parameter reg [ 8:0] BUFF_SIZE = 32
|
||||
) (
|
||||
input clk,
|
||||
input reset,
|
||||
input wire clk,
|
||||
input wire reset,
|
||||
|
||||
input in_en,
|
||||
input [15:0] data_in[2], // 0:R 1:G 2:B
|
||||
output out_ready,
|
||||
output out_receive,
|
||||
input wire in_en,
|
||||
input wire [15:0] in_data[3], // 0:R 1:G 2:B
|
||||
output wire out_ready,
|
||||
output wire out_receive,
|
||||
|
||||
// 输出相关
|
||||
input in_ready,
|
||||
input in_receive,
|
||||
output out_en,
|
||||
output [3 * OUT_DEPTH - 1:0] out_data,
|
||||
input wire in_ready,
|
||||
input wire in_receive,
|
||||
output reg out_en,
|
||||
output reg [3 * OUT_DEPTH - 1:0] out_data,
|
||||
|
||||
// 颜色校正
|
||||
input wire [15:0] gain_red,
|
||||
input wire [15:0] gain_green,
|
||||
input wire [15:0] gain_blue,
|
||||
input wire color_correction
|
||||
);
|
||||
localparam READ_DATA = 0;
|
||||
localparam CALC_DATA = 1;
|
||||
localparam SEND_DATA = 2;
|
||||
localparam reg [2:0] READ_DATA = 0;
|
||||
localparam reg [2:0] CALC_DATA = 1;
|
||||
localparam reg [2:0] SATI_DATA = 2;
|
||||
localparam reg [2:0] SEND_DATA = 3;
|
||||
|
||||
reg [2:0] state, nextState;
|
||||
reg [31:0] data_cal[2]; // 用于保存运算结果,防止溢出
|
||||
reg [BUFF_SIZE - 1:0] data_cal[3]; // 用于保存运算结果,防止溢出
|
||||
|
||||
always @(posedge clk or posedge reset) begin
|
||||
if (reset) begin
|
||||
@@ -44,15 +45,15 @@ module ColorBlender #(
|
||||
always @(*) begin
|
||||
case (state)
|
||||
READ_DATA: nextState = (in_en) ? (color_correction ? CALC_DATA : SEND_DATA) : READ_DATA;
|
||||
CALC_DATA: nextState = SEND_DATA;
|
||||
CALC_DATA: nextState = SATI_DATA;
|
||||
SATI_DATA: nextState = SEND_DATA;
|
||||
SEND_DATA: nextState = (in_receive) ? READ_DATA : SEND_DATA;
|
||||
default: nextState = READ_DATA;
|
||||
default: nextState = READ_DATA;
|
||||
endcase
|
||||
end
|
||||
|
||||
assign out_ready = (state == READ_DATA) ? 1 : 0;
|
||||
assign out_receive = in_en ? 1 : 0;
|
||||
assign out_en = (in_ready && state == SEND_DATA) ? 1 : 0;
|
||||
assign out_ready = (!in_en && state == READ_DATA) ? 1 : 0;
|
||||
assign out_receive = (in_en && state == READ_DATA) ? 1 : 0;
|
||||
|
||||
always @(posedge clk or posedge reset) begin
|
||||
if (reset) begin
|
||||
@@ -60,28 +61,38 @@ module ColorBlender #(
|
||||
data_cal[0] <= 0;
|
||||
data_cal[1] <= 0;
|
||||
data_cal[2] <= 0;
|
||||
|
||||
out_data <= 0;
|
||||
out_en <= 0;
|
||||
end else begin
|
||||
case (state)
|
||||
READ_DATA: begin
|
||||
if (in_en) begin
|
||||
data_cal[0] <= ({16'b0, data_in[0]} << 8;
|
||||
data_cal[1] <= ({16'b0, data_in[1]} << 8;
|
||||
data_cal[2] <= ({16'b0, data_in[2]} << 8;
|
||||
data_cal[0] <= ({{(BUFF_SIZE - 16){1'b0}}, in_data[0]}) << (8 - (IN_DEPTH - OUT_DEPTH));
|
||||
data_cal[1] <= ({{(BUFF_SIZE - 16){1'b0}}, in_data[1]}) << (8 - (IN_DEPTH - OUT_DEPTH));
|
||||
data_cal[2] <= ({{(BUFF_SIZE - 16){1'b0}}, in_data[2]}) << (8 - (IN_DEPTH - OUT_DEPTH));
|
||||
end
|
||||
end
|
||||
|
||||
CALC_DATA: begin
|
||||
data_cal[0] <= data_cal[0] * (GAIN_RED << 8)
|
||||
data_cal[1] <= data_cal[1] * (GAIN_GREEN << 8)
|
||||
data_cal[2] <= data_cal[2] * (GAIN_BLUE << 8)
|
||||
data_cal[0] <= (data_cal[0] * {{(BUFF_SIZE - 16){1'b0}}, gain_red}) >> 16;
|
||||
data_cal[1] <= (data_cal[1] * {{(BUFF_SIZE - 16){1'b0}}, gain_green}) >> 16;
|
||||
data_cal[2] <= (data_cal[2] * {{(BUFF_SIZE - 16){1'b0}}, gain_blue}) >> 16;
|
||||
end
|
||||
|
||||
SATI_DATA: begin
|
||||
data_cal[0] <= |data_cal[0][BUFF_SIZE -1 :OUT_DEPTH] ? {BUFF_SIZE{1'b1}} : data_cal[0];
|
||||
data_cal[1] <= |data_cal[0][BUFF_SIZE -1 :OUT_DEPTH] ? {BUFF_SIZE{1'b1}} : data_cal[1];
|
||||
data_cal[2] <= |data_cal[0][BUFF_SIZE -1 :OUT_DEPTH] ? {BUFF_SIZE{1'b1}} : data_cal[2];
|
||||
end
|
||||
|
||||
SEND_DATA: begin
|
||||
if (in_ready) begin
|
||||
out_data[0] <= data_cal[0] >> (8 + IN_DEPTH - OUT_DEPTH)
|
||||
out_data[1] <= data_cal[1] >> (8 + IN_DEPTH - OUT_DEPTH)
|
||||
out_data[2] <= data_cal[2] >> (8 + IN_DEPTH - OUT_DEPTH)
|
||||
end
|
||||
if (in_ready && !in_receive) begin
|
||||
out_en <= 1;
|
||||
out_data <= {
|
||||
data_cal[0][OUT_DEPTH-1:0], data_cal[1][OUT_DEPTH-1:0], data_cal[2][OUT_DEPTH-1:0]
|
||||
};
|
||||
end else out_en <= 0;
|
||||
end
|
||||
|
||||
default: ;
|
||||
|
||||
Reference in New Issue
Block a user