C语言实现数字雨效果

#include <windows.h>
#include <windowsx.h>

#define ID_TIMER 1
#define STRMAXLEN 25 //一个显示列的最大长度
#define STRMINLEN 8 //一个显示列的最小长度

LRESULT CALLBACK WndProc (HWND, UINT, WPARAM, LPARAM) ;
//////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////
typedef struct tagCharChain //整个当作屏幕的一个显示列,这是个双向列表
{
  struct tagCharChain *prev; //链表的前个元素
  TCHAR ch;     //一个显示列中的一个字符
  struct tagCharChain *next; //链表的后个元素
}CharChain, *pCharChain;

typedef struct tagCharColumn
{
  CharChain *head, *current, *point;
  int x, y, iStrLen; //显示列的开始显示的x,y坐标,iStrLen是这个列的长度
  int iStopTimes, iMustStopTimes; //已经停滞的次数和必须停滞的次数,必须停滞的次数是随机的
}CharColumn, *pCharColumn;

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
     PSTR szCmdLine, int iCmdShow)
{
  static TCHAR szAppName[] = TEXT ("matrix") ;
  HWND   hwnd ;
  MSG   msg ;
  WNDCLASS wndclass ;

  wndclass.style    = CS_HREDRAW | CS_VREDRAW ; // class style
  wndclass.lpfnWndProc  = WndProc ;
  wndclass.cbClsExtra  = 0 ;
  wndclass.cbWndExtra  = 0 ;
  wndclass.hInstance  = hInstance ;
  wndclass.hIcon    = LoadIcon (NULL, IDI_APPLICATION) ;
  wndclass.hCursor  = LoadCursor (NULL, IDC_ARROW) ;
  wndclass.hbrBackground  = (HBRUSH) GetStockObject (BLACK_BRUSH) ;
  wndclass.lpszMenuName  = NULL ;
  wndclass.lpszClassName  = szAppName ;

  if(!RegisterClass (&wndclass))
  {
   MessageBox (NULL, TEXT ("此程序必须运行在NT下!"), szAppName, MB_ICONERROR) ;
   return 0;
  }

  hwnd = CreateWindow (szAppName, NULL,
       WS_DLGFRAME | WS_THICKFRAME | WS_POPUP, // windows style
       0, 0,
       GetSystemMetrics(SM_CXSCREEN), GetSystemMetrics(SM_CYSCREEN),
       NULL, NULL, hInstance,
       NULL) ;

  ShowWindow (hwnd, SW_SHOWMAXIMIZED) ; //最大化显示 show windows
  UpdateWindow (hwnd) ;
  ShowCursor(FALSE); //隐藏鼠标光标
  
  srand ((int) GetCurrentTime ()) ; //初始化随机数发生器
  while (GetMessage (&msg, NULL, 0, 0))
  {
   TranslateMessage (&msg) ;
   DispatchMessage (&msg) ;
  }
  ShowCursor(TRUE); //显示鼠标光标
  return msg.wParam ;
}

TCHAR randomChar() //随机字符产生函数
{
  return (TCHAR)(rand()%(126-33)+33); //33到126之间
}

int init(CharColumn *cc, int cyScreen, int x) //初始化
{
  int j;
  cc->iStrLen = rand()%(STRMAXLEN-STRMINLEN) + STRMINLEN; //显示列的长度
  cc->x = x+3 ;  //显示列的开始显示的x坐标
  cc->y =rand()%3?rand()%cyScreen:0; //显示列的开始显示的y坐标
  cc->iMustStopTimes = rand()%6 ;
  cc->iStopTimes = 0 ;
  cc->head = cc->current =
      (pCharChain)calloc(cc->iStrLen, sizeof(CharChain)); //生成显示列
  for(j=0; j<cc->iStrLen-1; j++)
  {
    cc->current->prev = cc->point; //cc->point一个显示列的前个元素
    cc->current->ch = '\0';
    cc->current->next = cc->current+1; //cc->current+1一个显示列的后个元素
    cc->point   = cc->current++; //cc->point = cc->current; cc->current++;
  }
  cc->current->prev = cc->point; //最后一个节点
  cc->current->ch = '\0';
  cc->current->next = cc->head;
  cc->head->prev = cc->current; //头节点的前一个为此链的最后一个元素

  cc->current = cc->point = cc->head; //free掉申请的内存要用current当参数
  cc->head->ch = randomChar(); // 对链表头的 元素填充
  return 0;
}

LRESULT CALLBACK WndProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
  HDC  hdc ;
  //ctn 用来确定一个显示链是否 向下前进,如果等待次数超过必须等待的次数,ctn就代表要向下前进
  int i, j, temp, ctn; //j为一个显示链中除链表头外的在屏幕上显示的y坐标,temp绿色过度到黑色之用
  static HDC hdcMem;
  HFONT hFont;
  static HBITMAP hBitmap;
  static int cxScreen, cyScreen; //屏幕的宽度 高度.
  static int iFontWidth=10, iFontHeight=15, iColumnCount; //字体的宽度 高度, 列数
  static CharColumn *ccChain;

  switch (message)
  {
  case WM_CREATE:
   cxScreen = GetSystemMetrics(SM_CXSCREEN) ; //屏幕宽度
   cyScreen = GetSystemMetrics(SM_CYSCREEN) ;
   SetTimer (hwnd, ID_TIMER, 10, NULL) ;

   hdc = GetDC(hwnd);
   hdcMem = CreateCompatibleDC(hdc);
   hBitmap = CreateCompatibleBitmap(hdc, cxScreen, cyScreen);
   SelectObject(hdcMem, hBitmap);
   ReleaseDC(hwnd, hdc);
   //创建字体
   hFont = CreateFont(iFontHeight, iFontWidth-5, 0, 0, FW_BOLD, 0, 0, 0,
        DEFAULT_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS,
        DRAFT_QUALITY, FIXED_PITCH | FF_SWISS, TEXT("Fixedsys"));
   SelectObject(hdcMem, hFont);
   DeleteObject (hFont) ;
   SetBkMode(hdcMem, TRANSPARENT); //设置背景模式为 透明
   iColumnCount = cxScreen/(iFontWidth*3/2); //屏幕所显示字母雨的列数

   ccChain = (pCharColumn)calloc(iColumnCount, sizeof(CharColumn));
   for(i=0; i<iColumnCount; i++)
   {
    init(ccChain+i, cyScreen, (iFontWidth*3/2)*i);
   }
   return 0 ;

  case WM_TIMER:
   hdc = GetDC(hwnd);
   PatBlt (hdcMem, 0, 0, cxScreen, cyScreen, BLACKNESS) ; //将内存设备映像刷成黑色
   for(i=0; i<iColumnCount; i++)
   {
    ctn = (ccChain+i)->iStopTimes++ > (ccChain+i)->iMustStopTimes;
    //
    (ccChain+i)->point = (ccChain+i)->head; //point用于遍历整个显示列
    
    //第一个字符显示为 白色
    SetTextColor(hdcMem, RGB(255, 255, 255));
    TextOut(hdcMem, (ccChain+i)->x, (ccChain+i)->y, &((ccChain+i)->point->ch), 1);
    j = (ccChain+i)->y; 
    (ccChain+i)->point = (ccChain+i)->point->next;
    //遍历整个显示列,将这个显示列里的字符从下往上显示
    temp = 0 ; //temp绿色过度到黑色之用
    while((ccChain+i)->point != (ccChain+i)->head && (ccChain+i)->point->ch)
    {
      SetTextColor(hdcMem, RGB(0, 255-(255*(temp++)/(ccChain+i)->iStrLen), 0));
      TextOut(hdcMem, (ccChain+i)->x, j-=iFontHeight, &((ccChain+i)->point->ch), 1);
      (ccChain+i)->point = (ccChain+i)->point->next;
    }
    if(ctn)
     (ccChain+i)->iStopTimes = 0 ;
    else continue;
    (ccChain+i)->y += iFontHeight; //下次开始显示的y坐标 为当前的y坐标加上 一个字符的高度
    //如果开始显示的y坐标减去 整个显示列的长度超过了屏幕的高度
    if( (ccChain+i)->y-(ccChain+i)->iStrLen*iFontHeight > cyScreen)
    {
      free( (ccChain+i)->current );
      init(ccChain+i, cyScreen, (iFontWidth*3/2)*i);
    }
    //链表的头 为此链表的前个元素,因为下次开始显示的时候 就相当与在整个显示列的开头添加个元素,然后在开始往上显示
    (ccChain+i)->head = (ccChain+i)->head->prev;
    (ccChain+i)->head->ch = randomChar();
   }
   
   BitBlt(hdc, 0, 0, cxScreen, cyScreen, hdcMem, 0, 0, SRCCOPY);
   ReleaseDC(hwnd, hdc);
   return 0;

  case WM_RBUTTONDOWN:
   KillTimer (hwnd, ID_TIMER) ;
   return 0;

  case WM_RBUTTONUP:
   SetTimer (hwnd, ID_TIMER, 10, NULL) ;
   return 0;

  //处理善后工作
  case WM_KEYDOWN:
  case WM_LBUTTONDOWN:
  case WM_DESTROY:
   KillTimer (hwnd, ID_TIMER) ;
   DeleteObject(hBitmap);
   DeleteDC(hdcMem);
   for(i=0; i<iColumnCount; i++)
   {
    free( (ccChain+i)->current );
   }
   free(ccChain);
   PostQuitMessage (0) ;
   return 0 ;
  }
  return DefWindowProc (hwnd, message, wParam, lParam) ;
}