Subversion Repositories bilinear_demosaic

/*-----------------------------------------------------------------------------

                                                                Video Stream Scaler testbench

                                                        Author: David Kronstein

Copyright 2011, David Kronstein, and individual contributors as indicated

by the @authors tag.

This is free software; you can redistribute it and/or modify it

under the terms of the GNU Lesser General Public License as

published by the Free Software Foundation; either version 2.1 of

the License, or (at your option) any later version.

This software is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public

License along with this software; if not, write to the Free

Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA

02110-1301 USA, or see the FSF site: http://www.fsf.org.

-------------------------------------------------------------------------------

Testbench for streamScaler V1.0.0

*/

`default_nettype none

//Input files. Raw data format, no header. 8 bits per pixel, 3 color channels.

`define INPUT640x512                    "src/input640x512RGB.raw"

`define INPUT1280x1024                  "src/input1280x1024RGB.raw"

`define INPUT1280x1024_21EXTRA  "src/input640x512_21extraRGB.raw"       //21 extra pixels at the start to be discarded

module scalerTestbench;

parameter BUFFER_SIZE = 4;

wire done;

        initial

        begin

          #10

          while(done != 1'b1)

           #10

           ;

                $stop;

        end

//640x512 to 1280x1024

        demosaicTest #(

        .X_RES ( 640-1 ),

        .Y_RES ( 512-1 ),

        .DATA_WIDTH ( 8 ),

        .X_RES_WIDTH ( 11 ),

        .Y_RES_WIDTH ( 11 ),

        .BUFFER_SIZE ( BUFFER_SIZE )                            //Number of RAMs in RAM ring buffer

        ) dt_640x512 (

        .inputFilename( "src/input640x512RGBBlur.raw" ),

        .outputFilename( "out/output640x512.raw" ),

        //Control

        .done ( done )

        );

endmodule

module demosaicTest #(

parameter X_RES = 640-1,

parameter Y_RES = 512-1,

parameter DATA_WIDTH = 8,

parameter X_RES_WIDTH = 11,

parameter Y_RES_WIDTH = 11,

parameter BUFFER_SIZE = 5                               //Number of RAMs in RAM ring buffer

)(

input wire [50*8:0] inputFilename, outputFilename,

//Control

output reg done

);

reg clk;

reg rst;

reg [DATA_WIDTH-1:0] dIn;

reg             dInValid;

wire    nextDin;

reg             start;

wire [DATA_WIDTH-1:0] rOut;

wire [DATA_WIDTH-1:0] gOut;

wire [DATA_WIDTH-1:0] bOut;

wire    dOutValid;

reg             nextDout;

reg [X_RES_WIDTH-1:0] xCount;

reg [Y_RES_WIDTH-1:0] yCount;

integer r, rfile, wfile;

initial // Clock generator

  begin

    #10 //Delay to allow filename to get here

    clk = 0;

    #5 forever #5 clk = !clk;

  end

initial // Reset

  begin

        done = 0;

    #10 //Delay to allow filename to get here

    rst = 0;

    #5 rst = 1;

    #4 rst = 0;

   // #50000 $stop;

  end

reg eof;

reg [24-1:0] readMem [0:0];

wire [1:0] quadSelect = {yCount[0], xCount[0]};

initial // Input file read, generates dIn data

begin

  #10 //Delay to allow filename to get here

        rfile = $fopen("src/input640x512RGBBlur.raw"/*inputFilename*/, "rb");

        dIn = 0;

        dInValid = 0;

        start = 0;

        xCount = 0;

        yCount = 0;

        #41

        start = 1;

        #10

        start = 0;

        #20

        r = $fread(readMem, rfile);

        dIn =   quadSelect == 2'b00 ? readMem[0][23:16] :               //RG    [23:16] is first data (R)

                        quadSelect == 2'b01 ? readMem[0][15:8] :        //GB

                        quadSelect == 2'b10 ? readMem[0][15:8] :

                                                                  readMem[0][7:0];

        xCount = xCount + 1;

        while(! $feof(rfile))

        begin

                dInValid = 1;

                #10

                if(nextDin)

                begin

                        r = $fread(readMem, rfile);

                        dIn =   quadSelect == 2'b00 ? readMem[0][23:16] :               //RG    [23:16] is first data (R)

                                        quadSelect == 2'b01 ? readMem[0][15:8] :        //GB

                                        quadSelect == 2'b10 ? readMem[0][15:8] :

                                                                                  readMem[0][7:0];

                        if(xCount == X_RES)

                        begin

                                xCount = 0;

                                yCount = yCount + 1;

                        end

                        else

                                xCount = xCount + 1;

                end

        end

  $fclose(rfile);

end

//Generate nextDout request signal

initial

begin

  #10 //Delay to match filename arrival delay

        nextDout = 0;

        #140001

        forever

        begin

                //This can be used to slow down the read to simulate live read-out. This basically inserts H blank periods.

                #(10*(X_RES+1)*4)

                nextDout = 1;

                #(10*(X_RES+1))

                nextDout = 1;

        end

end

//Read dOut and write to file

integer dOutCount;

initial

begin

  #10 //Delay to allow filename to get here

        wfile = $fopen("dst/output.raw", "wb");

        nextDout = 0;

        dOutCount = 0;

        #1

        while(dOutCount < ((X_RES+1) * (Y_RES+1)))

        begin

                #10

                if(dOutValid == 1)

                begin

                        $fwrite(wfile, "%c", rOut[7:0]);

                        $fwrite(wfile, "%c", gOut[7:0]);

                        $fwrite(wfile, "%c", bOut[7:0]);

                        dOutCount = dOutCount + 1;

                end

        end

        $fclose(wfile);

        done = 1;

end

bilinearDemosaic #(

.DATA_WIDTH( DATA_WIDTH ),

.X_RES_WIDTH( X_RES_WIDTH ),

.Y_RES_WIDTH( Y_RES_WIDTH ),

.BUFFER_SIZE( BUFFER_SIZE )                             //Number of RAMs in RAM ring buffer

) demosaic_inst (

.clk( clk ),

.rst( rst ),

.dIn( dIn ),

.dInValid( dInValid ),

.nextDin( nextDin ),

.start( start ),

.rOut( rOut ),

.gOut( gOut ),

.bOut( bOut ),

.dOutValid( dOutValid ),

.nextDout( nextDout ),

.xRes( X_RES ),                         //Input data number of pixels per line

.yRes( Y_RES )

);

endmodule