videomux_env.cc

Go to the documentation of this file.

#include <stdio.h>
#include <assert.h>
#include "videomux_env.h"

VideoMuxEnv::VideoMuxEnv
(
        const Id&                                     n,
        In<videomux_args_t>&    muxArguments,   
        In<pT_header>&                    seqInfoF,
        Out<imagePosition>&     YimgPosF0,
        Out<imagePosition>&     UVimgPosF0,
        Out<imagePosition>&     YimgPosF1,
        Out<imagePosition>&     UVimgPosF1      
)
:       // member initialisation
        // process network
        Process(n),
        // ports
  muxArgumentsP(id("muxArgumentsP"),muxArguments),
  seqInfoP(id("seqInfoP"),seqInfoF),
        YimgPosP0(id("YimgPosP0"),YimgPosF0),   
        UVimgPosP0(id("UVimgPosP0"),UVimgPosF0),        
        YimgPosP1(id("YimgPosP1"),YimgPosF1),   
        UVimgPosP1(id("UVimgPosP1"),UVimgPosF1)
{
}

void VideoMuxEnv::main()
{
        while(true)
        {
                const int N = 2;

    pT_header               seqInfo;
    pT_color                colorInfo;
                int                           nrImages;
                int                           interlaceFactor;
                videomux_args_t arguments;

                imagePosition     YimgPos[N];
                imagePosition     UVimgPos[N];
                
                read(muxArgumentsP, arguments);
    assert(arguments.NSTRM == N);

                #ifdef VERBOSE
                        printf("%s:\n",fullName());
                #endif
        
                for(int i=0; i<N; i++)
                {
                        YimgPos[i].x = arguments.hpos[i];
                        YimgPos[i].y = arguments.vpos[i];
                        YimgPos[i].z = arguments.depth[i];
                        YimgPos[i].enabled = true; /*arguments.depth[i];*/
                
                        #ifdef VERBOSE
                                printf("    Stream %d position=(%d,%d), depth=%d\n",
                                                i, YimgPos[i].x, YimgPos[i].y, YimgPos[i].z);
                        #endif
                }
                #ifdef VERBOSE
                        printf("\n");
                #endif
        
                read(seqInfoP, seqInfo);
                if (p_is_interlaced(&seqInfo))
                        interlaceFactor = 2;
                else
                        interlaceFactor = 1;
                colorInfo = p_get_color_format(&seqInfo);
                
                nrImages = interlaceFactor * p_get_num_frames(&seqInfo);
                for (int i=0; i<N; i++)
                {
                        YimgPos[i].y = YimgPos[i].y / interlaceFactor;
        
                        if ((colorInfo == P_COLOR_420) || (colorInfo == P_COLOR_420_PL))
                        {
                                UVimgPos[i].x = YimgPos[i].x / 2;
                                UVimgPos[i].y = YimgPos[i].y / 2;
                                UVimgPos[i].z = YimgPos[i].z;
                                UVimgPos[i].enabled = YimgPos[i].enabled;
                        }
                        else if ((colorInfo == P_COLOR_422) || (colorInfo == P_COLOR_422_PL))
                        {
                                UVimgPos[i].x = YimgPos[i].x / 2;
                                UVimgPos[i].y = YimgPos[i].y;
                                UVimgPos[i].z = YimgPos[i].z;
                                UVimgPos[i].enabled = YimgPos[i].enabled;
                        } 
                        else
                        {
                                UVimgPos[i].x = YimgPos[i].x;
                                UVimgPos[i].y = YimgPos[i].y;
                                UVimgPos[i].z = YimgPos[i].z;
                                UVimgPos[i].enabled = YimgPos[i].enabled;
                        }
                }
        
    for (int i=0; i<nrImages; i++)
                {
      write(YimgPosP0, YimgPos[0]);
      write(UVimgPosP0, UVimgPos[0]);
      write(YimgPosP1, YimgPos[1]);
      write(UVimgPosP1, UVimgPos[1]);
                }
        }
}

---