Subversion Repositories bluespec-h264

//**********************************************************************

// Deblocking Filter

//----------------------------------------------------------------------

//

//

package mkDeblockFilter;

import H264Types::*;

import IDeblockFilter::*;

import FIFO::*;

import Vector::*;

import Connectable::*;

import GetPut::*;

import ClientServer::*;

//-----------------------------------------------------------

// Local Datatypes

//-----------------------------------------------------------

//-----------------------------------------------------------

// Helper functions

//-----------------------------------------------------------

// Deblocking Filter Module

//-----------------------------------------------------------

(* synthesize *)

module mkDeblockFilter( IDeblockFilter );

   FIFO#(EntropyDecOT) infifo     <- mkFIFO();

   FIFO#(DeblockFilterOT) outfifo <- mkFIFO();

   FIFO#(MemReq#(TAdd#(PicWidthSz,5),32)) dataMemReqQ       <- mkSizedFIFO(1);

   FIFO#(MemReq#(PicWidthSz,13))          parameterMemReqQ  <- mkSizedFIFO(1);

   FIFO#(MemResp#(32))                    dataMemRespQ      <- mkSizedFIFO(1);

   FIFO#(MemResp#(13))                    parameterMemRespQ <- mkSizedFIFO(1);

   Reg#(Bit#(1)) chromaFlag    <- mkReg(0);

   Reg#(Bit#(4)) blockNum      <- mkReg(0);

   Reg#(Bit#(4)) pixelNum      <- mkReg(0);

   Reg#(Bit#(PicWidthSz))  picWidth  <- mkReg(maxPicWidthInMB);

   Reg#(Bit#(PicHeightSz)) picHeight <- mkReg(0);

   Reg#(Bit#(PicAreaSz))   firstMb   <- mkReg(0);

   Reg#(Bit#(PicAreaSz))   currMb    <- mkReg(0);

   Reg#(Bit#(PicAreaSz))   currMbHor <- mkReg(0);//horizontal position of currMb

   Reg#(Bit#(PicHeightSz)) currMbVer <- mkReg(0);//vertical position of currMb

   Vector#(3,Reg#(Bit#(8)))   tempinput  <- replicateM(mkRegU);

   Reg#(Bool) endOfFrame <- mkReg(False);

   //-----------------------------------------------------------

   // Rules

   rule passing (currMbHor

      //$display( "Trace Deblocking Filter: passing infifo packed %h", pack(infifo.first()));

      case (infifo.first()) matches

         tagged NewUnit . xdata :

            begin

               infifo.deq();

               outfifo.enq(EDOT infifo.first());

               $display("ccl5newunit");

               $display("ccl5rbspbyte %h", xdata);

            end

         tagged SPSpic_width_in_mbs .xdata :

            begin

               infifo.deq();

               outfifo.enq(EDOT infifo.first());

               picWidth <= xdata;

            end

         tagged SPSpic_height_in_map_units .xdata :

            begin

               infifo.deq();

               outfifo.enq(EDOT infifo.first());

               picHeight <= xdata;

            end

         tagged SHfirst_mb_in_slice .xdata :

            begin

               infifo.deq();

               outfifo.enq(EDOT infifo.first());

               firstMb   <= xdata;

               currMb    <= xdata;

               currMbHor <= xdata;

               currMbVer <= 0;

            end

         tagged PBoutput .xdata :

            begin

               infifo.deq();

               Bit#(2) blockHor = {blockNum[2],blockNum[0]};

               Bit#(2) blockVer = {blockNum[3],blockNum[1]};

               Bit#(2) pixelHor = {pixelNum[1],pixelNum[0]};

               Bit#(2) pixelVer = {pixelNum[3],pixelNum[2]};

               Bit#(PicWidthSz) currMbHorT = truncate(currMbHor);

               Bit#(32) pixelq = {xdata[3],xdata[2],xdata[1],xdata[0]};

               if(chromaFlag==0)

                  outfifo.enq(DFBLuma {ver:{currMbVer,blockVer,pixelVer},hor:{currMbHorT,blockHor},data:pixelq});

               else

                  outfifo.enq(DFBChroma {uv:blockHor[1],ver:{currMbVer,blockVer[0],pixelVer},hor:{currMbHorT,blockHor[0]},data:pixelq});

               if(pixelNum == 12)

                  begin

                     pixelNum <= 0;

                     if(blockNum == 15)

                        begin

                           blockNum <= 0;

                           chromaFlag <= 1;

                        end

                     else if(blockNum==7 && chromaFlag==1)

                        begin

                           blockNum <= 0;

                           chromaFlag <= 0;

                           currMb <= currMb+1;

                           currMbHor <= currMbHor+1;

                           if(currMbVer==picHeight-1 && currMbHor==zeroExtend(picWidth-1))

                              endOfFrame <= True;

                        end

                     else

                        blockNum <= blockNum+1;

                  end

               else

                  pixelNum <= pixelNum+4;

               //$display( "Trace Deblocking Filter: passing PBoutput %h %h %h %h", blockNum, pixelNum, pixelHor, xdata);

            end

         tagged EndOfFile :

            begin

               infifo.deq();

               outfifo.enq(EDOT infifo.first());

               $display( "ccl5: EndOfFile reached");

               //$finish(0);

            end

         default:

            begin

               infifo.deq();

               outfifo.enq(EDOT infifo.first());

            end

      endcase

   endrule

   rule currMbHorUpdate( !(currMbHor

      Bit#(PicAreaSz) temp = zeroExtend(picWidth);

      if((currMbHor >> 3) >= temp)

         begin

            currMbHor <= currMbHor - (temp << 3);

            currMbVer <= currMbVer + 8;

         end

      else

         begin

            currMbHor <= currMbHor - temp;

            currMbVer <= currMbVer + 1;

         end

      //$display( "Trace Deblocking Filter: currMbHorUpdate %h %h", currMbHor, currMbVer);

   endrule

   rule outputEndOfFrame(endOfFrame);

      outfifo.enq(EndOfFrame);

      endOfFrame <= False;

      //$display( "Trace Deblocking Filter: outputEndOfFrame %h", pack(infifo.first()));

   endrule

   interface Client mem_client_data;

      interface Get request  = fifoToGet(dataMemReqQ);

      interface Put response = fifoToPut(dataMemRespQ);

   endinterface

   interface Client mem_client_parameter;

      interface Get request  = fifoToGet(parameterMemReqQ);

      interface Put response = fifoToPut(parameterMemRespQ);

   endinterface

   interface Put ioin  = fifoToPut(infifo);

   interface Get ioout = fifoToGet(outfifo);

endmodule

endpackage