找回密码
 立即注册→加入我们

QQ登录

只需一步,快速开始

搜索
热搜: 下载 VB C 实现 编写
查看: 8908|回复: 0

【图形】八叉树算法取得图像最适应调色板

[复制链接]

1109

主题

1649

回帖

7万

积分

用户组: 管理员

一只技术宅

UID
1
精华
244
威望
743 点
宅币
24180 个
贡献
46222 次
宅之契约
0 份
在线时间
2294 小时
注册时间
2014-1-26
发表于 2015-1-29 06:00:45 | 显示全部楼层 |阅读模式

欢迎访问技术宅的结界,请注册或者登录吧。

您需要 登录 才可以下载或查看,没有账号?立即注册→加入我们

×
这次的算法是我的上一个帖子中的程序的完全重写版。之前的程序参考了微软的相关代码,那代码可读性不高,而且用起来麻烦。因此我重新写了这个代码。
八叉树算法用于产生最适合一副图像的256色调色板。这个算法使用一种八叉树的结构,通过建立八叉树,将颜色筛选出来,接近的颜色会被合并,而在图像中出现次数极少的颜色和出现次数很多的颜色都会被照顾到。
这次我写了个窗口,我用它来显示整个八叉树的产生过程。大家可以自己下载了然后修改使其能发展出一个完整的八叉树。因为我的代码是为了达到取得最适应调色板的目的(而不单纯只是为了建立八叉树),所以在树的产生过程中就有“砍树”的行为,一个是减少内存占用,一个是加速程序运行。
Bin在运行的时候,大家可以直接拖拽一个24位BMP文件到窗口内,然后它就会分析这个文件,取得它的像素,再产生八叉树、调色板。
技术结界.png
20150129052319.png
201307020106_002.png
20150129052744.png
颜色越多,最终产生的八叉树的深度就越小。
同时八叉树的展开程度决定了图像颜色的明暗度分布情况。
现在我写的这个库是这么用的:
1、包含我的.h
2、编译我的.c
#include"OctPal.h"
定义一个PalGenP类型的变量:PalGenP pGen;
初始化它。pGen=CreatePalGen(NULL);//其中的NULL可以换成你自己的错误处理程序。
然后不断调用AddColor(pGen,红色值,绿色值,蓝色值);将颜色添加到八叉树中。返回值是当前的八叉树所含颜色数。
可以通过检测pGen->NbLeaf判断颜色数量,然后调用ReduceColor(pGen);减少颜色的数量。ReduceColor的行为就是合并一些八叉树的树叶,将八叉树的级数一点点缩小,以筛选出应该被放入调色板的颜色。
最后调用GenPalette(pGen,接收颜色的回调函数,回调函数的参数);生成调色板。
详细的请看源码的内容。
OPConfig.h:
  1. //=============================================================================
  2. //作者:0xAA55
  3. //网站:http://www.0xaa55.com
  4. //请保留原作者信息,否则视为侵权。
  5. //-----------------------------------------------------------------------------
  6. #ifndef        _OCTREE_PALETTE_GENERATOR_CONFIG_
  7. #define        _OCTREE_PALETTE_GENERATOR_CONFIG_

  8. #ifndef        OPCall//调用约定
  9. #define        OPCall                _cdecl
  10. #endif

  11. #ifndef        OPCallBack//回调函数调用约定
  12. #define        OPCallBack        _cdecl
  13. #endif

  14. #ifndef        OPExpImp//导出导入的前缀
  15. #define        OPExpImp
  16. #endif

  17. #ifndef        OPFunc//函数
  18. #define        OPFunc(r,f)        OPExpImp r OPCall f
  19. #endif

  20. #endif
复制代码
OctPal.h:
  1. //=============================================================================
  2. //作者:0xAA55
  3. //网站:http://www.0xaa55.com
  4. //请保留原作者信息,否则视为侵权。
  5. //-----------------------------------------------------------------------------
  6. #ifndef        _OCTREE_PALETTE_GENERATOR_
  7. #define        _OCTREE_PALETTE_GENERATOR_

  8. #include"OPConfig.h"
  9. #include<stddef.h>

  10. #define        ColorBits 8

  11. typedef enum
  12. {
  13.         OPErr_NoError,                        //没有出错
  14.         OPErr_NoMemory,                        //内存不足
  15.         OPErr_NoReducibleNodes        //没有可删减节点
  16. }OPError,*OPErrorP;

  17. typedef void(OPCallBack*FnOnError)(OPError);

  18. typedef        unsigned char        ColorVal;//颜色值
  19. typedef        unsigned int        ColorSum;//颜色统计值
  20. typedef        struct _OctNode
  21. {
  22.         ColorSum        NbColors;
  23.         ColorSum        RSum;
  24.         ColorSum        GSum;
  25.         ColorSum        BSum;
  26.         int                        IsLeaf;
  27.         struct _OctNode*pChild[8];//八叉树子节点(Index=红绿蓝三个颜色值各贡献一位)
  28.         struct _OctNode*pNext;//下一个可删减节点
  29. }OctreeNode,*OctreeNodeP;//八叉树节点
  30. typedef        struct
  31. {
  32.         OctreeNodeP        pRoot;
  33.         OctreeNodeP        pReducibleNodes[ColorBits];//可删减节点
  34.         size_t                NbLeaf;
  35.         ColorSum        NbColors;
  36.         FnOnError        pfnOnError;
  37. }PalGen,*PalGenP;

  38. //=============================================================================
  39. //函数:FnOnGetColor
  40. //描述:取得调色板颜色的时候调用的用户回调函数
  41. //-----------------------------------------------------------------------------
  42. typedef void(OPCallBack*FnOnGetColor)
  43. (
  44.         ColorVal R,
  45.         ColorVal G,
  46.         ColorVal B,
  47.         void*pUserData
  48. );

  49. //=============================================================================
  50. //函数:CreatePalGen
  51. //描述:分配内存,创建一个PalGen
  52. //-----------------------------------------------------------------------------
  53. OPFunc(PalGenP,CreatePalGen)(FnOnError);

  54. //=============================================================================
  55. //函数:DestroyPalGen
  56. //描述:释放内存,彻底删除PalGen
  57. //-----------------------------------------------------------------------------
  58. OPFunc(void,DestroyPalGen)(PalGenP);

  59. //=============================================================================
  60. //函数:AddColor
  61. //描述:将一个颜色添加到PalGen的八叉树里。返回当前颜色数
  62. //-----------------------------------------------------------------------------
  63. OPFunc(size_t,AddColor)
  64. (
  65.         PalGenP                pGen,
  66.         ColorVal        R,
  67.         ColorVal        G,
  68.         ColorVal        B
  69. );

  70. //=============================================================================
  71. //函数:ReduceColor
  72. //描述:删减PalGen的八叉树颜色,返回剩余颜色数。
  73. //-----------------------------------------------------------------------------
  74. OPFunc(size_t,ReduceColor)(PalGenP pGen);

  75. //=============================================================================
  76. //函数:GenPalette
  77. //描述:从PalGen中取得调色板
  78. //-----------------------------------------------------------------------------
  79. OPFunc(void,GenPalette)
  80. (
  81.         PalGenP                        pGen,
  82.         FnOnGetColor        pfnOnGetColor,
  83.         void                        *pUserData
  84. );

  85. #endif
复制代码
OctPal.c:
  1. //=============================================================================
  2. //作者:0xAA55
  3. //网站:http://www.0xaa55.com
  4. //请保留原作者信息,否则视为侵权。
  5. //-----------------------------------------------------------------------------
  6. #include"OctPal.h"
  7. #include<stdio.h>
  8. #include<malloc.h>
  9. #include<memory.h>

  10. //=============================================================================
  11. //函数:DefOnError
  12. //描述:默认的错误处理函数
  13. //-----------------------------------------------------------------------------
  14. static void OPCallBack DefOnError(OPError ErrNumber)
  15. {
  16.         //什么也不做
  17. }

  18. //=============================================================================
  19. //函数:CreateNode
  20. //描述:建立节点,函数会根据级数来判断它是否为叶子节点,如果不是,则“可删减”
  21. //-----------------------------------------------------------------------------
  22. static OctreeNodeP CreateNode
  23. (
  24.         size_t                Level,
  25.         PalGenP                pGen
  26. )
  27. {
  28.         OctreeNodeP pNode;
  29.         pNode=(OctreeNodeP)malloc(sizeof(OctreeNode));
  30.         if(!pNode)//错误汇报机制由调用者完成。
  31.                 return NULL;

  32.         memset(pNode,0,sizeof(OctreeNode));
  33.         if(Level==ColorBits)
  34.         {
  35.                 //判断是否最深一级
  36.                 pNode->IsLeaf=1;
  37.                 pGen->NbLeaf++;
  38.         }
  39.         else
  40.         {
  41.                 //横向连结所有的节点
  42.                 pNode->pNext=pGen->pReducibleNodes[Level];
  43.                 pGen->pReducibleNodes[Level]=pNode;
  44.         }
  45.         return pNode;
  46. }

  47. //=============================================================================
  48. //函数:GetTreePal
  49. //描述:遍历八叉树,从中取得颜色,建立调色板。
  50. //-----------------------------------------------------------------------------
  51. static void GetTreePal
  52. (
  53.         OctreeNodeP                pNode,
  54.         FnOnGetColor        pfnOnGetColor,
  55.         void*                        pUserData
  56. )
  57. {
  58.         if(pNode->IsLeaf)
  59.                 pfnOnGetColor(
  60.                         (ColorVal)(pNode->RSum/pNode->NbColors),
  61.                         (ColorVal)(pNode->GSum/pNode->NbColors),
  62.                         (ColorVal)(pNode->BSum/pNode->NbColors),
  63.                         pUserData);
  64.         else
  65.         {
  66.                 unsigned i;
  67.                 for(i=0;i<8;i++)
  68.                 {
  69.                         if(pNode->pChild[i])
  70.                                 GetTreePal(pNode->pChild[i],pfnOnGetColor,pUserData);
  71.                 }
  72.         }
  73. }

  74. //=============================================================================
  75. //函数:DestroyTree
  76. //描述:遍历八叉树,从最深级开始删除整个八叉树。
  77. //-----------------------------------------------------------------------------
  78. static void DestroyTree(OctreeNodeP pTree)
  79. {
  80.         unsigned i;
  81.         for(i=0;i<8;i++)
  82.         {
  83.                 if(pTree->pChild[i])
  84.                         DestroyTree(pTree->pChild[i]);
  85.         }
  86.         free(pTree);
  87. }

  88. //=============================================================================
  89. //函数:CreatePalGen
  90. //描述:分配内存,创建一个PalGen
  91. //-----------------------------------------------------------------------------
  92. OPFunc(PalGenP,CreatePalGen)(FnOnError pfnOnError)
  93. {
  94.         PalGenP pGen;
  95.        
  96.         pGen=(PalGenP)malloc(sizeof(PalGen));
  97.         if(!pGen)
  98.                 return NULL;
  99.         memset(pGen,0,sizeof(PalGen));
  100.        
  101.         if(pfnOnError)
  102.                 pGen->pfnOnError=pfnOnError;
  103.         else
  104.                 pGen->pfnOnError=DefOnError;

  105.         //建立树根节点
  106.         pGen->pRoot=CreateNode(0,pGen);
  107.         if(!pGen->pRoot)
  108.         {
  109.                 free(pGen);
  110.                 if(pfnOnError)
  111.                         pfnOnError(OPErr_NoMemory);//汇报错误
  112.                 return NULL;
  113.         }

  114.         return pGen;
  115. }

  116. //=============================================================================
  117. //函数:AddColor
  118. //描述:将一个颜色添加到PalGen的八叉树里。返回当前颜色数。
  119. //-----------------------------------------------------------------------------
  120. OPFunc(size_t,AddColor)
  121. (
  122.         PalGenP                pGen,
  123.         ColorVal        R,
  124.         ColorVal        G,
  125.         ColorVal        B
  126. )
  127. {
  128.         unsigned i,Index,Shift;
  129.         OctreeNodeP pNode;
  130.         OPError ErrNumber=OPErr_NoError;

  131.         //树根节点必须存在,否则pGen无法初始化成功。见CreatePalGen函数实现。
  132.         pNode=pGen->pRoot;

  133.         //往下扩展树枝
  134.         for(i=1;i<=ColorBits;i++)
  135.         {
  136.                 if(pNode->IsLeaf)//如果已经是树叶则不再扩展树枝。
  137.                         break;
  138.                 Shift=ColorBits-i;
  139.                 Index=        ((R>>Shift)&1)<<0|
  140.                                 ((G>>Shift)&1)<<1|
  141.                                 ((B>>Shift)&1)<<2;
  142.                 if(!(pNode->pChild[Index]))
  143.                 {
  144.                         pNode->pChild[Index]=CreateNode(i,pGen);
  145.                         if(!(pNode->pChild[Index]))
  146.                         {
  147.                                 ErrNumber=OPErr_NoMemory;
  148.                                 goto ErrorReturn;
  149.                         }
  150.                 }
  151.                 pNode=pNode->pChild[Index];
  152.         }

  153.         //树叶统计颜色
  154.         pNode->RSum+=(ColorSum)R;
  155.         pNode->GSum+=(ColorSum)G;
  156.         pNode->BSum+=(ColorSum)B;
  157.         pNode->NbColors++;
  158.         pGen->NbColors++;

  159.         return 1;
  160. ErrorReturn:
  161.         pGen->pfnOnError(ErrNumber);
  162.         return 0;
  163. }

  164. //=============================================================================
  165. //函数:ReduceColor
  166. //描述:删减PalGen的八叉树颜色,返回剩余颜色数。
  167. //-----------------------------------------------------------------------------
  168. OPFunc(size_t,ReduceColor)(PalGenP pGen)
  169. {
  170.         unsigned i;
  171.         OctreeNodeP pNode;
  172.         OPError ErrNumber=OPErr_NoError;

  173.         //找到最深一级的可删减节点。
  174.         for(i=ColorBits-1;i;i--)
  175.         {
  176.                 if(pGen->pReducibleNodes[i])
  177.                         break;
  178.         }
  179.         if(!pGen->pReducibleNodes[i])
  180.         {
  181.                 ErrNumber=OPErr_NoReducibleNodes;
  182.                 goto ErrorReturn;
  183.         }

  184.         //横向指定同级下一个可删减节点
  185.         pNode=pGen->pReducibleNodes[i];
  186.         pGen->pReducibleNodes[i]=pNode->pNext;

  187.         //删减节点
  188.         for(i=0;i<8;i++)
  189.         {
  190.                 if(pNode->pChild[i])
  191.                 {
  192.                         pNode->RSum+=pNode->pChild[i]->RSum;
  193.                         pNode->GSum+=pNode->pChild[i]->GSum;
  194.                         pNode->BSum+=pNode->pChild[i]->BSum;
  195.                         pNode->NbColors+=pNode->pChild[i]->NbColors;
  196.                         free(pNode->pChild[i]);
  197.                         pNode->pChild[i]=NULL;
  198.                         pGen->NbLeaf--;
  199.                 }
  200.         }
  201.         pNode->IsLeaf=1;
  202.         pGen->NbLeaf++;

  203.         return pGen->NbLeaf;
  204. ErrorReturn:
  205.         pGen->pfnOnError(ErrNumber);
  206.         return pGen->NbLeaf;
  207. }

  208. //=============================================================================
  209. //函数:GenPalette
  210. //描述:从PalGen中取得调色板
  211. //-----------------------------------------------------------------------------
  212. OPFunc(void,GenPalette)
  213. (
  214.         PalGenP                        pGen,
  215.         FnOnGetColor        pfnOnGetColor,
  216.         void                        *pUserData
  217. )
  218. {
  219.         if(pGen->pRoot)
  220.                 GetTreePal(pGen->pRoot,pfnOnGetColor,pUserData);
  221. }


  222. //=============================================================================
  223. //函数:DestroyPalGen
  224. //描述:释放内存,彻底删除PalGen
  225. //-----------------------------------------------------------------------------
  226. OPFunc(void,DestroyPalGen)(PalGenP pGen)
  227. {
  228.         if(pGen->pRoot)
  229.                 DestroyTree(pGen->pRoot);
  230.         free(pGen);
  231. }
复制代码
Entry.c:
  1. //=============================================================================
  2. //作者:0xAA55
  3. //网站:http://www.0xaa55.com
  4. //请保留原作者信息,否则视为侵权。
  5. //-----------------------------------------------------------------------------
  6. #define        _USE_MATH_DEFINES
  7. #include<math.h>

  8. #include<tchar.h>
  9. #include<stdio.h>
  10. #include<stdarg.h>
  11. #include<Windows.h>

  12. #include"OctPal.h"
  13. HWND        g_hWnd=NULL;                        //主窗口
  14. HDC                g_hBackgroundDC=NULL;        //后台绘图句柄
  15. HBITMAP        g_hBackgroundBMP=NULL;        //后台绘图的位图
  16. PalGenP        g_pGen=NULL;                        //调色板生成器的存储数据

  17. //=============================================================================
  18. //函数:DbgPrint
  19. //描述:打印调试信息。
  20. //-----------------------------------------------------------------------------
  21. void DbgPrint(char*szFmt,...)
  22. {
  23.         va_list ap;
  24.         char szBuf[0x1000];

  25.         va_start(ap,szFmt);

  26.         vsprintf(szBuf,szFmt,ap);
  27.         MessageBoxA(g_hWnd,szBuf,"DbgPrint",MB_ICONINFORMATION);

  28.         va_end(ap);
  29. }

  30. //=============================================================================
  31. //函数:DoEvents
  32. //描述:处理窗口消息
  33. //-----------------------------------------------------------------------------
  34. int DoEvents()
  35. {
  36.         MSG msg;
  37.         if(PeekMessage(&msg,g_hWnd,0,0,PM_REMOVE))
  38.         {
  39.                 if(msg.message==WM_QUIT)//遇到WM_QUIT消息返回0
  40.                         return 0;
  41.                 TranslateMessage(&msg);
  42.                 DispatchMessage(&msg);
  43.         }
  44.         return 1;
  45. }

  46. //=============================================================================
  47. //函数:OnGetColor
  48. //描述:取得颜色的回调函数,用于生成调色板。
  49. //-----------------------------------------------------------------------------
  50. RGBQUAD g_Palette[256];
  51. void OPCallBack OnGetColor
  52. (
  53.         ColorVal R,
  54.         ColorVal G,
  55.         ColorVal B,
  56.         void*pUserData
  57. )
  58. {
  59.         size_t Index=*(size_t*)pUserData;
  60.         if(Index>=256)
  61.                 return;
  62.         g_Palette[Index].rgbRed=R;
  63.         g_Palette[Index].rgbGreen=G;
  64.         g_Palette[Index].rgbBlue=B;
  65.         (*(size_t*)pUserData)++;
  66. }

  67. //=============================================================================
  68. //函数:DrawTree
  69. //描述:绘制产生的八叉树。八叉树足够大的时候绘图会变得很慢。
  70. //-----------------------------------------------------------------------------
  71. void DrawTree
  72. (
  73.         HDC hDC,
  74.         OctreeNodeP pNode,
  75.         float fAngle,
  76.         float fX,
  77.         float fY,
  78.         float fLength,
  79.         float fRadius,
  80.         float fMaxSubAngle,
  81.         int LeafMatch
  82. )
  83. {
  84.         unsigned i;
  85.        
  86.         if(pNode->IsLeaf && LeafMatch)//如果是树叶而且只画树叶
  87.         {
  88.                 BYTE R,G,B;
  89.                 HBRUSH hClrFill;
  90.                 LOGBRUSH ClrFill;
  91.                 R=(BYTE)(pNode->RSum/pNode->NbColors);
  92.                 G=(BYTE)(pNode->GSum/pNode->NbColors);
  93.                 B=(BYTE)(pNode->BSum/pNode->NbColors);
  94.                 ClrFill.lbStyle=BS_SOLID;
  95.                 ClrFill.lbColor=RGB(R,G,B);
  96.                 hClrFill=CreateBrushIndirect(&ClrFill);
  97.                 hClrFill=(HBRUSH)SelectObject(hDC,hClrFill);
  98.                 Ellipse(hDC,
  99.                         (int)(fX-fRadius),(int)(fY-fRadius),
  100.                         (int)(fX+fRadius),(int)(fY+fRadius));
  101.                 hClrFill=(HBRUSH)SelectObject(hDC,hClrFill);
  102.                 DeleteObject(hClrFill);
  103.         }
  104.         else if(!pNode->IsLeaf)//如果不是树叶则处理下一级
  105.         {
  106.                 for(i=0;i<8;i++)
  107.                 {
  108.                         if(pNode->pChild[i])
  109.                         {
  110.                                 //计算下一个节点的位置
  111.                                 float fSubAng=fAngle+((float)i-3.5f)*fMaxSubAngle/7.0f;
  112.                                 float fSubX=fX+cosf(fAngle+fSubAng)*fLength;
  113.                                 float fSubY=fY+sinf(fAngle+fSubAng)*fLength;
  114.                                 if(!LeafMatch)//如果画树枝,则进行绘图操作。
  115.                                 {
  116.                                         MoveToEx(hDC,(int)fX,(int)fY,NULL);
  117.                                         LineTo(hDC,
  118.                                                 (int)fSubX,
  119.                                                 (int)fSubY);
  120.                                 }
  121.                                 DrawTree(hDC,pNode->pChild[i],fSubAng,//递归下一个节点
  122.                                         fSubX,fSubY,fLength,fRadius,fMaxSubAngle,LeafMatch);
  123.                         }
  124.                 }
  125.         }
  126. }

  127. //=============================================================================
  128. //函数:ParseBMPFile
  129. //描述:分析BMP文件,产生八叉树,然后产生调色板
  130. //-----------------------------------------------------------------------------
  131. void ParseBMPFile(LPCTSTR szBMPFile)
  132. {
  133.         BITMAPFILEHEADER BMFH;//BMP文件头
  134.         BITMAPINFOHEADER BMIF;//BMP信息头

  135.         size_t Pitch;//每行字节数
  136.         RGBTRIPLE*LineData=NULL;//行像素数据
  137.         size_t x,y,Index;//用于遍历的变量

  138.         RECT rc;//窗口客户区域的左、上、右、下
  139.         DWORD dwLastTick=0;//用于粗略计时

  140.         FILE*fp=NULL;
  141.         //打开文件进行读取
  142.         fp=_tfopen(szBMPFile,TEXT("rb"));
  143.         if(!fp)
  144.                 goto ReadFail;
  145.        
  146.         //读取文件头
  147.         if(fread(&BMFH,1,sizeof(BMFH),fp)!=sizeof(BMFH))goto ReadFail;
  148.         if(fread(&BMIF,1,sizeof(BMIF),fp)!=sizeof(BMIF))goto ReadFail;

  149.         //判断文件格式
  150.         if(BMFH.bfType!='MB')
  151.                 goto BadFormat;
  152.         if(BMIF.biBitCount!=24)
  153.                 goto Not24Bit;

  154.         fseek(fp,BMFH.bfOffBits,SEEK_SET);

  155.         //清空调色板
  156.         memset(g_Palette,0,sizeof(g_Palette));

  157.         //重置调色板生成器数据
  158.         if(g_pGen)
  159.                 DestroyPalGen(g_pGen);
  160.         g_pGen=CreatePalGen(NULL);
  161.         if(!g_pGen)
  162.                 goto NoMem;
  163.        
  164.         //分配一行像素的内存用于读取
  165.         Pitch=((BMIF.biWidth*BMIF.biBitCount-1)/32+1)*4;//4字节对齐
  166.         LineData=(RGBTRIPLE*)malloc(Pitch);
  167.         if(!LineData)
  168.                 goto NoMem;

  169.         //重绘窗口
  170.         GetClientRect(g_hWnd,&rc);
  171.         InvalidateRect(g_hWnd,&rc,TRUE);

  172.         //遍历每一行
  173.         y=BMIF.biHeight;
  174.         do
  175.         {
  176.                 RGBTRIPLE*pPixel;//像素指针
  177.                 fread(LineData,1,Pitch,fp);//读取一行像素
  178.                 pPixel=LineData;//准备遍历这行的每一个像素

  179.                 x=BMIF.biWidth;//遍历每一个像素
  180.                 do
  181.                 {
  182.                         //添加像素颜色到八叉树
  183.                         AddColor(g_pGen,pPixel->rgbtRed,
  184.                                                         pPixel->rgbtGreen,
  185.                                                         pPixel->rgbtBlue);
  186.                         while(g_pGen->NbLeaf>256)//使眼色数量不超过256色
  187.                                 ReduceColor(g_pGen);
  188.                         pPixel++;
  189.                 }while(--x);

  190.                 //每处理一行像素,判断是否应该重绘屏幕
  191.                 //最大重绘频率每秒钟100下。因为绘图需要消耗时间。
  192.                 if(GetTickCount()-dwLastTick>=10)
  193.                 {
  194.                         dwLastTick=GetTickCount();
  195.                         InvalidateRect(g_hWnd,&rc,FALSE);//重绘屏幕
  196.                         if(!DoEvents())
  197.                                 break;
  198.                 }
  199.         }while(--y);

  200.         Index=0;
  201.         GenPalette(g_pGen,OnGetColor,&Index);//产生调色板
  202.         InvalidateRect(g_hWnd,&rc,FALSE);//最后重绘屏幕

  203. CleanReturn:
  204.         if(fp)//返回前清理内存
  205.                 fclose(fp);
  206.         free(LineData);
  207.         return;
  208. NoMem:
  209.         MessageBox(g_hWnd,TEXT("内存不足。"),NULL,MB_ICONEXCLAMATION);
  210.         goto CleanReturn;
  211. ReadFail:
  212.         MessageBox(g_hWnd,TEXT("读取文件失败。"),NULL,MB_ICONEXCLAMATION);
  213.         goto CleanReturn;
  214. BadFormat:
  215.         MessageBox(g_hWnd,TEXT("不是BMP文件或格式错误。"),NULL,MB_ICONEXCLAMATION);
  216.         goto CleanReturn;
  217. Not24Bit:
  218.         MessageBox(g_hWnd,TEXT("不是24位BMP文件。"),NULL,MB_ICONEXCLAMATION);
  219.         goto CleanReturn;
  220. }

  221. LRESULT CALLBACK WndProc(HWND hWnd,UINT uMsg,WPARAM wp,LPARAM lp);//消息处理器
  222. //=============================================================================
  223. //函数:_tWinMain
  224. //描述:整个程序的入口点
  225. //-----------------------------------------------------------------------------
  226. int APIENTRY _tWinMain
  227. (
  228.         HINSTANCE hInst,
  229.         HINSTANCE hPrevInst,
  230.         LPTSTR szCmdLine,
  231.         int nShowCmd
  232. )
  233. {
  234.         WNDCLASSEX WndClass;
  235.         ATOM ClassAtom=0;
  236.         MSG msg;

  237.         //注册窗口类
  238.         memset(&WndClass,0,sizeof(WndClass));
  239.         WndClass.cbSize=sizeof(WndClass);
  240.     WndClass.lpfnWndProc=WndProc;
  241.     WndClass.hInstance=hInst;
  242.         WndClass.hIcon=LoadIcon(NULL,MAKEINTRESOURCE(IDI_APPLICATION));
  243.     WndClass.hCursor=LoadCursor(NULL,MAKEINTRESOURCE(IDC_ARROW));
  244.     WndClass.hbrBackground=(HBRUSH)(COLOR_BTNFACE+1);
  245.     WndClass.lpszClassName=TEXT("NormalWindow");
  246.     WndClass.hIconSm=LoadIcon(NULL,MAKEINTRESOURCE(IDI_APPLICATION));
  247.         ClassAtom=RegisterClassEx(&WndClass);
  248.         if(!ClassAtom)
  249.                 goto Cleanup;

  250.         //创建窗口
  251.         g_hWnd=CreateWindowEx(0,(LPCTSTR)ClassAtom,TEXT("主窗口"),
  252.                 WS_OVERLAPPEDWINDOW,
  253.                 CW_USEDEFAULT,CW_USEDEFAULT,CW_USEDEFAULT,CW_USEDEFAULT,
  254.                 NULL,NULL,hInst,NULL);
  255.         if(!g_hWnd)
  256.                 goto Cleanup;

  257.         //显示窗口
  258.         ShowWindow(g_hWnd,nShowCmd);
  259.         UpdateWindow(g_hWnd);

  260.         //设置窗口允许接收拖拽文件。
  261.         DragAcceptFiles(g_hWnd,TRUE);

  262.         //消息循环
  263.         while(GetMessage(&msg,NULL,0,0))
  264.         {
  265.                 TranslateMessage(&msg);
  266.                 DispatchMessage(&msg);
  267.         }

  268.         if(g_hBackgroundBMP)
  269.                 DeleteObject(g_hBackgroundBMP);
  270.         if(g_hBackgroundDC)
  271.                 ReleaseDC(g_hWnd,g_hBackgroundDC);

  272.         if(g_pGen)
  273.                 DestroyPalGen(g_pGen);
  274.         UnregisterClass((LPCTSTR)ClassAtom,hInst);
  275.         return 0;
  276. Cleanup:
  277.         if(ClassAtom)
  278.                 UnregisterClass((LPCTSTR)ClassAtom,hInst);
  279.         MessageBox(NULL,TEXT("启动程序失败。"),
  280.                 TEXT("无法启动程序"),MB_ICONEXCLAMATION);
  281.         return 1;
  282. }

  283. //=============================================================================
  284. //函数:WndProc
  285. //描述:窗口消息处理程序
  286. //-----------------------------------------------------------------------------
  287. LRESULT CALLBACK WndProc(HWND hWnd,UINT uMsg,WPARAM wp,LPARAM lp)
  288. {
  289.         switch(uMsg)
  290.         {
  291.         case WM_CREATE:
  292.         case WM_SIZING:
  293.         case WM_SIZE:
  294.                 {
  295.                         RECT rc;
  296.                         HDC hDC;

  297.                         hDC=GetDC(hWnd);
  298.                         GetClientRect(hWnd,&rc);

  299.                         //重新生成后台画板
  300.                         if(g_hBackgroundBMP)
  301.                                 DeleteObject(g_hBackgroundBMP);
  302.                         if(g_hBackgroundDC)
  303.                                 ReleaseDC(hWnd,g_hBackgroundDC);
  304.                         g_hBackgroundDC=CreateCompatibleDC(hDC);
  305.                         g_hBackgroundBMP=CreateCompatibleBitmap
  306.                                 (hDC,rc.right-rc.left,rc.bottom-rc.top);
  307.                         SelectObject(g_hBackgroundDC,g_hBackgroundBMP);
  308.                         if(hDC)
  309.                                 ReleaseDC(hWnd,hDC);
  310.                        
  311.                         //刷白画板
  312.                         FillRect(g_hBackgroundDC,&rc,(HBRUSH)(COLOR_BTNFACE+1));
  313.                         InvalidateRect(hWnd,&rc,FALSE);
  314.                         break;
  315.                 }
  316.         case WM_PAINT:
  317.                 {
  318.                         PAINTSTRUCT ps;
  319.                         HDC hDC;
  320.                         HBRUSH hBrush;
  321.                         RECT rc;
  322.                         TCHAR szTitle[256];

  323.                         size_t x,y,Index;
  324.                        
  325.                         //绘图时,先绘制到画板上,再把画板覆盖到窗口上。

  326.                         GetClientRect(hWnd,&rc);
  327.                         hDC=BeginPaint(hWnd,&ps);

  328.                         FillRect(g_hBackgroundDC,&rc,(HBRUSH)(COLOR_BTNFACE+1));

  329.                         Index=0;
  330.                         //先画调色板
  331.                         for(y=0;y<16;y++)
  332.                         {
  333.                                 for(x=0;x<16;x++)
  334.                                 {
  335.                                         hBrush=CreateSolidBrush(RGB(
  336.                                                 g_Palette[Index].rgbRed,
  337.                                                 g_Palette[Index].rgbGreen,
  338.                                                 g_Palette[Index].rgbBlue));
  339.                                         hBrush=(HBRUSH)SelectObject(g_hBackgroundDC,hBrush);

  340.                                         //调色板,每个颜色8x8
  341.                                         Rectangle(g_hBackgroundDC,x*8,y*8,x*8+8,y*8+8);

  342.                                         hBrush=(HBRUSH)SelectObject(g_hBackgroundDC,hBrush);
  343.                                         DeleteObject(hBrush);
  344.                                         Index++;
  345.                                 }
  346.                         }
  347.                         //然后画八叉树
  348.                         if(g_pGen && g_pGen->pRoot)
  349.                         {
  350.                                 float        fRootX=(float)(rc.right)*0.5f,
  351.                                                 fRootY=(float)(rc.bottom)*0.75f;
  352.                                 //先画树枝
  353.                                 DrawTree(g_hBackgroundDC,g_pGen->pRoot,
  354.                                         -(float)M_PI_4,fRootX,fRootY,40,4,(float)(M_PI*0.125),0);
  355.                                 //后画树叶
  356.                                 DrawTree(g_hBackgroundDC,g_pGen->pRoot,
  357.                                         -(float)M_PI_4,fRootX,fRootY,40,4,(float)(M_PI*0.125),1);
  358.                         }
  359.                        
  360.                         BitBlt(hDC,0,0,rc.right-rc.left,rc.bottom-rc.top,
  361.                                 g_hBackgroundDC,0,0,SRCCOPY);
  362.                         EndPaint(hWnd,&ps);

  363.                         _stprintf(szTitle,TEXT("%u"),g_pGen->NbLeaf);
  364.                         SetWindowText(hWnd,szTitle);
  365.                 }
  366.                 break;
  367.         case WM_DESTROY:
  368.                 PostQuitMessage(0);
  369.                 return 0;
  370.         case WM_DROPFILES:
  371.                 {
  372.                         TCHAR*pszDropFileName;
  373.                         size_t cbDropFileName;
  374.                         //取得第0个拖入文件的文件名长度
  375.                         cbDropFileName=DragQueryFile((HDROP)wp,0,NULL,0)+1;//添加'\0'结尾
  376.                         //分配内存并取得第0个拖入文件名
  377.                         pszDropFileName=(TCHAR*)malloc(cbDropFileName*sizeof(TCHAR));
  378.                         if(pszDropFileName)
  379.                         {
  380.                                 DragQueryFile((HDROP)wp,0,pszDropFileName,cbDropFileName);
  381.                                 DragFinish((HDROP)wp);
  382.                                 ParseBMPFile(pszDropFileName);
  383.                                 free(pszDropFileName);
  384.                         }
  385.                 }
  386.                 return 0;
  387.         }
  388.         return DefWindowProc(hWnd,uMsg,wp,lp);
  389. }
复制代码
BIN: OctreePalette_exe.7z (44.1 KB, 下载次数: 58)
SRC: OctreePalette.7z (90.99 KB, 下载次数: 85)

本帖被以下淘专辑推荐:

回复

使用道具 举报

QQ|Archiver|小黑屋|技术宅的结界 ( 滇ICP备16008837号 )|网站地图

GMT+8, 2024-3-29 17:51 , Processed in 0.042326 second(s), 37 queries , Gzip On.

Powered by Discuz! X3.5

© 2001-2024 Discuz! Team.

快速回复 返回顶部 返回列表