【C++】插值函数代码分析学习

插值函数代码分析学习，用的CImage类load图像。

// Imagejoint.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include "Imagejoint.h"
#include "math.h"
#include <afxwin.h>
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
#include <atlimage.h>//CImage类
#include <locale.h>

// The one and only application object

CWinApp theApp;

using namespace std;
//双三次插值系数
double fs(double w)
{
        double a=-0.5;
        double fs;
        if (abs(w)<=)
            fs=(a+)*pow(abs(w),)-(a+)*pow(abs(w),)+;
        else if (abs(w)>&&abs(w)<=)
            fs=a*pow(abs(w),)-*a*pow(abs(w),)+*a*abs(w)-*a;
        else
            fs=;
        return fs;
}

HRESULT Imagejoint(PBYTE pbSrc,int iWidth,int iHeight,double dbRotate,double dbZoom,PBYTE pbTag);

int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{

    int nRetCode = ;//表示整数类型的函数返回码。n表示整数类型,Ret是Return的缩写,表示返回值,Code表示代码。
    setlocale(LC_ALL,"chs");
    HMODULE hModule = ::GetModuleHandle(NULL);

    if (hModule != NULL)
    {
        // initialize MFC and print and error on failure
        CImage cImage;
        HRESULT hResult;

        //初始化一些变量
        int        iWidth,iHeight,iBytePerPixel,iPitch;
        int        x,y;
        PBYTE    pbSrc=NULL,pbTag=NULL;//源图、目标图
        PBYTE    pbImage=NULL;//load图像后存在这
        PDWORD    pdwImage=NULL;//用于保存图像
        double    dbRotate=,dbZoom=;

        do{
            if (!AfxWinInit(hModule, NULL, ::GetCommandLine(), ))
            {
                // TODO: change error code to suit your needs
                _tprintf(_T("Fatal Error: MFC initialization failed\n"));
                nRetCode = ;
                break;
            }
            // TODO: code your application's behavior here.
            if(argc<)
            {
                _tprintf(_T("使用方法：Imagejoint 源图像文件 旋转角度 缩放倍数 输出文件\n"));
                nRetCode= -;
                break;
            }

            _stscanf(argv[],_T("%lf"),&dbRotate);
            _stscanf(argv[],_T("%lf"),&dbZoom);
            if(!((dbRotate>= && dbRotate<) && (dbZoom>=0.25 && dbZoom<=)))
            {
                _tprintf(_T("“旋转角度” 或 “缩放倍数”参数错误！\n"));
                nRetCode= -;
                break;
            }
            //Load 图像到cImage对象中
            hResult=cImage.Load(argv[]);
            if(hResult!=ERROR_SUCCESS)
            {
                _tprintf(_T("源图像文件名错误！\n"));
                nRetCode= -;
                break;
            }

            iWidth=cImage.GetWidth();
            iHeight=cImage.GetHeight();
            //分配源图内存
            pbSrc = (PBYTE)malloc(iWidth*iHeight);
            //分配目标图内存
            if(dbZoom>) pbTag = (PBYTE)malloc(ceil(iWidth*dbZoom)*ceil(iHeight*dbZoom));
            else         pbTag = (PBYTE)malloc(iWidth*iHeight);
            if(pbSrc==NULL || pbTag==NULL )
            {
                _tprintf(_T("内存申请错误！\n"));
                nRetCode= -;
                break;
            }
            //cImage数据存到pbImage，后再转换为源灰度图pbSrc
            iPitch=cImage.GetPitch();
            iBytePerPixel=(cImage.GetBPP()+)/;
            if(iBytePerPixel==)
            {
                for(y=;y<iHeight;y++)
                {   //load的图像数据放到pbImage
                    pbImage=(PBYTE)(PBYTE(cImage.GetBits())+iPitch*y);//得到的是图像初始像素地址
                    for(x=;x<iWidth;x++)
                    {
                        //pbImage转换为灰度图pbSrc
                        pbSrc[y*iWidth+x]=(pbImage[*x]*0.15+pbImage[*x+]*0.55+pbImage[*x+]*0.3);
                    }
                }
            }
            cImage.Destroy();
            //TO DO:执行操作
            hResult=Imagejoint(pbSrc,iWidth,iHeight,dbRotate,dbZoom,pbTag);
            if(hResult!=ERROR_SUCCESS)
            {
                _tprintf(_T("图像处理错误！\n"));
                nRetCode= -;
                break;
            }

            //处理后保存图像
            iWidth=ceil(iWidth*dbZoom);
            iHeight=ceil(iHeight*dbZoom);
            cImage.Create(iWidth,-iHeight,);
            iPitch=cImage.GetPitch();
            for(y=;y<iHeight;y++)
            {
                pdwImage=(PDWORD)(PBYTE(cImage.GetBits())+iPitch*y);
                for(x=;x<iWidth;x++)
                {
                    pdwImage[x]=pbTag[y*iWidth+x]*0x10101;
                }
            }
            //待保存图像文件名
            CString csTagName=argv[];
            csTagName.Trim();
            csTagName.MakeUpper();
            if(csTagName.Right()!=_T(".BMP") ) csTagName.Append(_T(".BMP"));
            hResult=cImage.Save(csTagName);
            if(hResult!=ERROR_SUCCESS)
            {
                _tprintf(_T("图像结果保存错误！\n"));
                nRetCode= -;
                break;
            }
            _tprintf(_T("图像处理成功！\n"));
            nRetCode= ERROR_SUCCESS;
            break;
        }while();
        if(pbSrc) free(pbSrc);
        if(pbTag) free(pbTag);
    }
    else
    {
        //  可以根据需求更改错误提示
        _tprintf(_T("Fatal Error: GetModuleHandle failed\n"));
        nRetCode = ;
    }
    getchar();
    return nRetCode;
}

HRESULT Imagejoint(PBYTE pbSrc,int iWidth,int iHeight,double dbRotate,double dbZoom,PBYTE pbTag)
{
    int size;
    if(dbZoom>)
    {
        size=iWidth*iHeight;
    }
    else
    {
        size=ceil(iWidth*dbZoom)*ceil(iHeight*dbZoom);
    }
    //旋转中心为图像中心
    double rx0=iWidth;
    double ry0=iHeight;
    double srcx,srcy,u,v;
    int xOr,yOr;
    //int zoom=ceil(dbZoom);
    dbRotate=dbRotate*3.1415926/180.0;
    for (int y=;y<dbZoom*iHeight;y++)
    {
        for (int x=;x<dbZoom*iWidth;x++)
        {
            srcx=(double)((x-rx0)*cos(dbRotate) - (y-ry0)*sin(dbRotate) + rx0) ;
            srcy=(double)((x-rx0)*sin(dbRotate) + (y-ry0)*cos(dbRotate) + ry0) ;
            srcx=srcx*/dbZoom;
            srcy=srcy*/dbZoom;
            xOr = floor(srcx);
            yOr = floor(srcy);
            u=srcx-xOr;
            v=srcy-yOr;

            if( !(srcx>= && srcx<=iWidth && srcy>= && srcy<=iHeight))
                {
                    pbTag[y**iWidth+x]=;//
                }
            else
            {
                double middle=
                    pbSrc[(yOr-)*iWidth+(xOr-)]*fs(+u)*fs(+v)+
                    pbSrc[(yOr)*iWidth+(xOr-)]*fs(+u)*fs(v)+
                    pbSrc[(yOr+)*iWidth+(xOr-)]*fs(+u)*fs(-v)+
                    pbSrc[(yOr+)*iWidth+(xOr-)]*fs(+u)*fs(-v)+

                    pbSrc[(yOr-)*iWidth+(xOr)]*fs(u)*fs(+v)+
                    pbSrc[(yOr)*iWidth+(xOr)]*fs(u)*fs(v)+
                    pbSrc[(yOr+)*iWidth+(xOr)]*fs(u)*fs(-v)+
                    pbSrc[(yOr+)*iWidth+(xOr)]*fs(u)*fs(-v)+

                    pbSrc[(yOr-)*iWidth+(xOr+)]*fs(-u)*fs(+v)+
                    pbSrc[(yOr)*iWidth+(xOr+)]*fs(-u)*fs(v)+
                    pbSrc[(yOr+)*iWidth+(xOr+)]*fs(-u)*fs(-v)+
                    pbSrc[(yOr+)*iWidth+(xOr+)]*fs(-u)*fs(-v)+

                    pbSrc[(yOr-)*iWidth+(xOr+)]*fs(-u)*fs(+v)+
                    pbSrc[(yOr)*iWidth+(xOr+)]*fs(-u)*fs(v)+
                    pbSrc[(yOr+)*iWidth+(xOr+)]*fs(-u)*fs(-v)+
                    pbSrc[(yOr+)*iWidth+(xOr+)]*fs(-u)*fs(-v);

                if(middle<=&&middle>=)
                    pbTag[y**iWidth+x]=middle;
                else if(middle>)
                    pbTag[y**iWidth+x]=;
                else
                    pbTag[y**iWidth+x]=;
            }
        }
    }
    //memcpy(pbTag,pbSrc,size);
    return ERROR_SUCCESS;
}