Mandelbrot

// MandelbrotDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Mandelbrot.h"
#include "MandelbrotDlg.h"
#include <complex>

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
    CAboutDlg();

// Dialog Data
    //{{AFX_DATA(CAboutDlg)
    enum { IDD = IDD_ABOUTBOX };
    //}}AFX_DATA

    // ClassWizard generated virtual function overrides
    //{{AFX_VIRTUAL(CAboutDlg)
    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
    //}}AFX_VIRTUAL

// Implementation
protected:
    //{{AFX_MSG(CAboutDlg)
    //}}AFX_MSG
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
    //{{AFX_DATA_INIT(CAboutDlg)
    //}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CAboutDlg)
    //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
    //{{AFX_MSG_MAP(CAboutDlg)
        // No message handlers
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMandelbrotDlg dialog

CMandelbrotDlg::CMandelbrotDlg(CWnd* pParent /*=NULL*/)
    : CDialog(CMandelbrotDlg::IDD, pParent)
{
    //{{AFX_DATA_INIT(CMandelbrotDlg)
        // NOTE: the ClassWizard will add member initialization here
    //}}AFX_DATA_INIT
    // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CMandelbrotDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CMandelbrotDlg)
        // NOTE: the ClassWizard will add DDX and DDV calls here
    //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CMandelbrotDlg, CDialog)
    //{{AFX_MSG_MAP(CMandelbrotDlg)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_WM_LBUTTONDOWN()
    ON_WM_LBUTTONUP()
    ON_WM_MOUSEWHEEL()
    ON_WM_MOUSEMOVE()
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMandelbrotDlg message handlers

BOOL CMandelbrotDlg::OnInitDialog()
{
    CDialog::OnInitDialog();

    // Add "About..." menu item to system menu.

    // IDM_ABOUTBOX must be in the system command range.
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

    CMenu* pSysMenu = GetSystemMenu(FALSE);
    if (pSysMenu != NULL)
    {
        CString strAboutMenu;
        strAboutMenu.LoadString(IDS_ABOUTBOX);
        if (!strAboutMenu.IsEmpty())
        {
            pSysMenu->AppendMenu(MF_SEPARATOR);
            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
        }
    }

    // Set the icon for this dialog.  The framework does this automatically
    //  when the application's main window is not a dialog
    SetIcon(m_hIcon, TRUE);            // Set big icon
    SetIcon(m_hIcon, FALSE);        // Set small icon
    
    // TODO: Add extra initialization here
    maxIterations = 4096;
    maxMod = 10000.0;

    CPaintDC dc(this);
    bitmap.CreateCompatibleBitmap(&dc,WIDTH,HEIGHT);

    xMin = -2.2;
    yMax = 2;
    h = 0.1;

/*    for (int i=0; i<HEIGHT; i++)
        for (int j=0; j<WIDTH; j++)
            iterColor(xMin+i*h, yMax-j*h, bits[i][j*3], bits[i][j*3+1], bits[i][j*3+2]);*/
    for (int i=0; i<300; i++)
        for (int j=0; j<500; j++)
        {
            bits[i][j*3]=255;
            bits[i][j*3+1]=0;
            bits[i][j*3+2]=0;
        }
    bitmap.SetBitmapBits(HEIGHT*WIDTH*3,bits);
    
    return TRUE;  // return TRUE  unless you set the focus to a control
}

void CMandelbrotDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialog::OnSysCommand(nID, lParam);
    }
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CMandelbrotDlg::OnPaint() 
{
    if (IsIconic())
    {
        CPaintDC dc(this); // device context for painting

        SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

        // Center icon in client rectangle
        int cxIcon = GetSystemMetrics(SM_CXICON);
        int cyIcon = GetSystemMetrics(SM_CYICON);
        CRect rect;
        GetClientRect(&rect);
        int x = (rect.Width() - cxIcon + 1) / 2;
        int y = (rect.Height() - cyIcon + 1) / 2;

        // Draw the icon
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CPaintDC dc(this);
        CDC memDC;
        memDC.CreateCompatibleDC(&dc);

//        bitmap.SetBitmapBits(HEIGHT*WIDTH*3,bits);
        memDC.SelectObject(&bitmap);
        dc.BitBlt(0,0,WIDTH,HEIGHT,&memDC,0,0,SRCCOPY);

        CDialog::OnPaint();
    }
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CMandelbrotDlg::OnQueryDragIcon()
{
    return (HCURSOR) m_hIcon;
}

void CMandelbrotDlg::OnLButtonDown(UINT nFlags, CPoint point) 
{
    // TODO: Add your message handler code here and/or call default
    
    CDialog::OnLButtonDown(nFlags, point);
}

void CMandelbrotDlg::OnLButtonUp(UINT nFlags, CPoint point) 
{
    // TODO: Add your message handler code here and/or call default
    
    CDialog::OnLButtonUp(nFlags, point);
}

BOOL CMandelbrotDlg::OnMouseWheel(UINT nFlags, short zDelta, CPoint pt) 
{
    // TODO: Add your message handler code here and/or call default
    
    return CDialog::OnMouseWheel(nFlags, zDelta, pt);
}

void CMandelbrotDlg::OnMouseMove(UINT nFlags, CPoint point) 
{
    // TODO: Add your message handler code here and/or call default
    
    CDialog::OnMouseMove(nFlags, point);
}

void CMandelbrotDlg::iterColor(double re0, double im0, BYTE& R, BYTE& G, BYTE& B)
{
/*    int color = 0;
    double real = 0;
    double image = 0;
    while (color < maxIterations && real*real+image*image < maxMod)
    {
        double re = real*real - image*image + re0;
        double im = 2*real*image + im0;
        real = re;
        image = im;
        color++;
    }

    R = (color & 0xf) * 16;
    G = (color>>4 &0xf) * 16;
    B = (color>>8) * 16;*/
}
原文地址:https://www.cnblogs.com/saieuler/p/2730336.html