宁波猫的窝

int IM_AddBlindWaterMark(unsigned char *Src, unsigned char *WaterMark, unsigned char *Dest, int Width, int Height, int Stride, int WidthW, int HeightW, int StrideW)
{    int Channel = Stride / Width, ChannelW = StrideW / WidthW;    if ((Src == NULL) || (Dest == NULL))                        return IM_STATUS_NULLREFRENCE;    if ((Width <= 0) || (Height <= 0))                            return IM_STATUS_INVALIDPARAMETER;    if ((Channel != 1) && (Channel != 3) && (Channel != 4))        return IM_STATUS_INVALIDPARAMETER;    if ((ChannelW != 1) && (ChannelW != 3) && (ChannelW != 4))    return IM_STATUS_INVALIDPARAMETER;    if ((WidthW >= Width / 4) || (HeightW >= Height / 4))        return IM_STATUS_INVALIDPARAMETER;        //    水印图不能大于原图尺寸的一半
            
    int Status = IM_STATUS_OK;    
    int OptimalW = IM_GetOptimalDftSize(Width),    OptimalH = IM_GetOptimalDftSize(Height);    int OffsetX = (OptimalW - Width) / 2,        OffsetY = (OptimalH - Height) / 2;    int HalfW = OptimalW / 2,                    HalfH = OptimalH / 2;    
    if (Channel == 1)
    {
        Complex    *Data = (Complex *)malloc(OptimalW * OptimalH * sizeof(Complex));        if ((Data == NULL))    return IM_STATUS_OUTOFMEMORY;        
        for (int Y = 0; Y < Height; Y++)                                //    我们把数据居中布置，边缘用重复像素的方式        {
            unsigned char *LinePS = Src + Y * Stride;
            Complex *LinePD = Data + (Y + OffsetY) * OptimalW;            for (int X = 0; X < OffsetX; X++)
            {
                LinePD[X].Real = LinePS[0];
                LinePD[X].Imag = 0;
            }            for (int X = OffsetX; X < OffsetX + Width; X++)
            {
                LinePD[X].Real = LinePS[X - OffsetX];            
                LinePD[X].Imag = 0;
            }            for (int X = OffsetX + Width; X < OptimalW; X++)
            {
                LinePD[X].Real = LinePS[Width - 1];
                LinePD[X].Imag = 0;
            }
        }        for (int Y = 0; Y < OffsetY; Y++)
        {
            memcpy(Data + Y * OptimalW, Data + OffsetY * OptimalW, OptimalW * sizeof(Complex));
        }        for (int Y = OffsetY + Height; Y < OptimalH; Y++)
        {
            memcpy(Data + Y * OptimalW, Data + (OffsetY + Height - 1) * OptimalW, OptimalW * sizeof(Complex));
        }

        IM_FFT2D(Data, Data, OptimalW, OptimalH, false, 0, 0);
        IM_FFTShift(Data, Data, OptimalW, OptimalH);                //    数据偏移到中心

        for (int Y = 0; Y < HeightW; Y++)
        {
            Complex *LineLT = Data + (Y + OffsetY + Height / 16) * OptimalW + OffsetX + Width / 16;            //    确保在可见的范围内添加，左上角和右下角都镜像添加
            Complex *LineRB = Data + (OffsetY + Height - 1 - Height / 16 - Y) * OptimalW + OffsetX + Width - 1 - Width / 16;    //    再稍微往内部移动一点，可以适当增强抵抗变形的能力，但是越往中心其对最终结果的影响越大。
            unsigned char *LinePS = WaterMark + Y * StrideW;            if (ChannelW == 1)
            {                for (int X = 0; X < WidthW; X++)
                {                    float Cof = ((LinePS[X] * 4) >> 8) + 1;
                    LineLT[X].Real *= Cof;    LineLT[X].Imag *= Cof;
                    LineRB[-X].Real *= Cof;    LineRB[-X].Imag *= Cof;
                }
            }            else if (ChannelW == 3)
            {                for (int X = 0; X < WidthW; X++)
                {                    float Cof = ((((LinePS[0] + LinePS[1] + LinePS[1] + LinePS[2]) >> 2) * 4) >> 8) + 1;
                    LineLT[X].Real *= Cof;    LineLT[X].Imag *= Cof;
                    LineRB[-X].Real *= Cof;    LineRB[-X].Imag *= Cof;
                    LinePS += 3;
                }
            }            else if (ChannelW == 4)
            {                for (int X = 0; X < WidthW; X++)
                {                    float Cof = ((IM_Div255(((LinePS[0] + LinePS[1] + LinePS[1] + LinePS[2]) >> 2) * LinePS[3]) * 4) >> 8) + 1;
                    LineLT[X].Real *= Cof;    LineLT[X].Imag *= Cof;
                    LineRB[-X].Real *= Cof;    LineRB[-X].Imag *= Cof;
                    LinePS += 4;
                }
            }
        }
        
        IM_IFFTShift(Data, Data, OptimalW, OptimalH);
        IM_FFT2D(Data, Data, OptimalW, OptimalH, true, 0, 0);        for (int Y = 0; Y < Height; Y++)
        {
            Complex *LinePS = Data + (Y + OffsetY) * OptimalW + OffsetX;
            unsigned char *LinePD = Dest + Y * Stride;            for (int X = 0; X < Width; X++)
            {
                LinePD[X] = IM_ClampToByte(LinePS[X].Real);
            }
        }        if (Data != NULL)        free(Data);        return IM_STATUS_OK;
    }    else
    {
        
    
    }
}