ca_testdlg.cpp

用遗传算法实现函数优化

// CA_TestDlg.cpp : implementation file
//

#include "stdafx.h"
#include "CA_Test.h"
#include "CA_TestDlg.h"

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

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <math.h>
#define POPSIZE 500                       
#define maximization 1                     
#define minimization 2                    
#define cmax 100                         
#define cmin 0                          
//#define length1 10                        
//#define length2 10                        
//#define chromlength length1+length2    
int functionmode=maximization;
int popsize;         
int maxgeneration; 
double pc; //交叉概论         
double pm; //变异概论
double min,max;
double aver1[10000];
double curbest[10000];
double xx1;
double xx2;
double x1,x2;
double  p;//自适应小常数
struct individual              
{
double x[2];    
double value;                   
double fitness;                
};
int generation;                      
int best_index;                      
//int worst_index;                      
struct individual bestindividual;     
struct individual worstindividual; 
struct individual currentbest;        
struct individual population[10][10];
struct individual agent[10][10];
struct individual bestneighborindividual[10][10];
int best_indexi;                     
int best_indexj; 
int worst_indexi;                     
int worst_indexj;
#define xmin -2.048
#define xmax 2.048
#define L 4




/////////////////////////////////////////////////////////////////////////////
// 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()

/////////////////////////////////////////////////////////////////////////////
// CCA_TestDlg dialog

CCA_TestDlg::CCA_TestDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CCA_TestDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CCA_TestDlg)
	m_edit1 = 80;
	m_edit2 = 40;
	m_edit3 = 0.6;
	m_edit4 = 0.001;
	m_edit5 = 0.0;
	m_edit6 = 0;
	m_edit7 = 0.0;
	m_edit8 = 0.0;
	m_edit9 = _T("");
	m_edit10 = 0.0;
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CCA_TestDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CCA_TestDlg)
	DDX_Text(pDX, IDC_EDIT1, m_edit1);
	DDX_Text(pDX, IDC_EDIT2, m_edit2);
	DDX_Text(pDX, IDC_EDIT3, m_edit3);
	DDX_Text(pDX, IDC_EDIT4, m_edit4);
	DDX_Text(pDX, IDC_EDIT5, m_edit5);
	DDX_Text(pDX, IDC_EDIT6, m_edit6);
	DDX_Text(pDX, IDC_EDIT7, m_edit7);
	DDX_Text(pDX, IDC_EDIT8, m_edit8);
	DDX_Text(pDX, IDC_EDIT9, m_edit9);
	DDX_Text(pDX, IDC_EDIT10, m_edit10);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CCA_TestDlg, CDialog)
	//{{AFX_MSG_MAP(CCA_TestDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_EXIT, OnExit)
	ON_BN_CLICKED(IDC_RUN, OnRun)
	ON_WM_TIMER()
	ON_BN_CLICKED(IDC_DISPLAY, OnDisplay)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CCA_TestDlg message handlers

BOOL CCA_TestDlg::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
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CCA_TestDlg::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 CCA_TestDlg::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
	{
		CDialog::OnPaint();
	}
}

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

/*void CCA_TestDlg::OnRun() 
{
	// TODO: Add your control notification handler code here
// TODO: Add your control notification handler code here
//	int i;
	
	UpdateData(true);
	if(m_edit1<=0 || m_edit2<=0 || m_edit3<=0 || m_edit3>=1 || m_edit4<=0 || m_edit4>1)
	{
		MessageBox("请正确输入参数！",NULL,MB_OK);
	}
	else
	{
		generation=0;
		input();
		generateinitialpopulation();
		//evaluatepopulation();
		SetTimer(1,10, NULL);
	}
	
}
	
}*/

void CCA_TestDlg::OnExit() 
{
	// TODO: Add your control notification handler code here
	PostQuitMessage(0);

}

void CCA_TestDlg::OnRun() 
{
	// TODO: Add your control notification handler code here
		UpdateData(true);
	if(m_edit1<=0 || m_edit2<=0 || m_edit3<=0 || m_edit3>=1 || m_edit4<=0 || m_edit4>1)
	{
		MessageBox("请正确输入参数！",NULL,MB_OK);
	}
	else
	{
		generation=0;
		p=0.3;
		input();
		generateinitialpopulation();
		evaluatepopulation();
		performevolution();
		SetTimer(1,10, NULL);
	}
	
}


void CCA_TestDlg::OnTimer(UINT nIDEvent) 
{
	// TODO: Add your message handler code here and/or call default
	
	if(nIDEvent==1)
	{
		generation++;
		p=p*0.95;
		if (p<0.08)
		{
			p=0.2;
		}
        generatenextpopulation();
        evaluatepopulation();
        performevolution();
        outputtextreport();
		
        while(generation>=maxgeneration)
        {
			generation=0;
			KillTimer(1);
        }
	}
	
	CDialog::OnTimer(nIDEvent);
}

void CCA_TestDlg::outputtextreport()
{
   //输出要显示的内容如：最优个体，世代数
	int i,j;
	double sum;        
	double average;       
	sum=0.0;

	for(i=0;i<L;i++)
	{
       for(j=0;j<L;j++)
	   {
		   sum+=population[i][j].fitness;
	   }
	}
	average=sum/16;
    
      //generation;
	m_edit6=generation;                
	m_edit7=average;      //currentbest.value;
	m_edit8=currentbest.fitness;
	UpdateData(false);
	aver1[generation]=average;
    curbest[generation]=currentbest.value;
}

void CCA_TestDlg::input()
{
   UpdateData(true);
	popsize=m_edit1;
	maxgeneration=m_edit2;
	pc=m_edit3;
	pm=m_edit4;
    p=m_edit5;
}

void CCA_TestDlg::generateinitialpopulation()
{
	int i,j,k;
	for(i=0; i<L; i++)
	{
       for(j=0; j<L; j++)
	   {
		   for(k=0; k<2; k++)
		   {
				population[i][j].x[k]=double(rand())/RAND_MAX*4.096-2.048;
		   }
	   }
	}
	FILE *fp1;
	fp1=fopen("initialpopulation.txt","w+");
	for(i=0;i<L;i++)
	{
		for(j=0;j<L;j++)
		{
			for(k=0;k<2;k++)
			{
				fprintf(fp1,"%lf  ",population[i][j].x[k]);
			}
			fprintf(fp1,"\n");
		}
	}
	fclose(fp1);
}

void CCA_TestDlg::generatenextpopulation()
{
    compete();
	auto_mutation();
	crossover();
}

void CCA_TestDlg::evaluatepopulation()
{
	calculateobjectvalue();       
	calculatefitnessvalue();      
	findneighborbestindividual();
	findbestandworstindividual();        
}

void CCA_TestDlg::calculatefunctionvalue()
{
	int i,j;
    for (i=0; i<L; i++)
	{
       for(j=0;j<L;j++)

	   {	x1=population[i][j].x[0];
		    x2=population[i][j].x[1];
	        population[i][j].value=100*(x1*x1-x2)* (x1*x1-x2)+(1-x1)*(1-x1);
	   }	
	}
	
	
}

void CCA_TestDlg::calculatefitnessvalue()
{
/*	int i,j;
	double temp;
	for(i=0;i<L;i++)
	{
		for(j=0;j<L;j++)
		{
			if(functionmode==maximization)
			{ 
			    if((population[i][j].value+cmin)>0.0)
			   {
				temp=cmin+population[i][j].value;
			   }
		    	else
				{
				temp=0.0;
				}
			}
	     	else if (functionmode==minimization)
			{
			 if(population[i][j].value<cmax)
			 {
				temp=cmax-population[i][j].value;
			 }
			 else
			 { 
			 	temp=0.0;
			 }
			}
		    population[i][j].fitness=temp;
		}
	}
*/
	int i,j;	
	for(i=0;i<L;i++)
	{
		for(j=0;j<L;j++)
		{
//			if(functionmode==1)
			{				
			   population[i][j].fitness=population[i][j].value;			
			}
//			else if (functionmode==2)
//			{			  
//			  agent[i][j].fitness=-agent[i][j].value;
//			}			  
		}
	}

	FILE *fp1;
	fp1=fopen("fitness.txt","w+");
	for(i=0;i<L;i++)
	{
		for(int j=0;j<L;j++)
		{
			fprintf(fp1,"%lf\n",population[i][j].fitness);
		}
	}
	fclose(fp1);
}



void CCA_TestDlg::findbestandworstindividual()
{
	int i,j;
	bestindividual=population[0][0];
	worstindividual=population[0][0];
	for(i=0;i<L;i++)
	{
		for(j=0;j<L;j++)
		{
			if(population[i][j].fitness>bestindividual.fitness)
			{
				bestindividual=population[i][j];
				best_indexi=i;                     
                best_indexj=j; 
			}
			else if(agent[i][j].fitness<worstindividual.fitness)
			{
				worstindividual=population[i][j];
				worst_indexi=i;                     
                worst_indexj=j;
			}
		}
	}
	if(generation==0)
	{
		currentbest=bestindividual;
	}
	else 
	{
		if(currentbest.fitness<bestindividual.fitness)
		{
		  currentbest=bestindividual;
		}
	}
}

void CCA_TestDlg::OnDisplay() 
{
	CString str;
	str.Format("x1=%lf;x2=%lf;",population[best_indexi][best_indexj].x[0],population[best_indexi][best_indexj].x[1]);
	m_edit9=str;
	m_edit10=currentbest.value;  //应该是.value，而不是currentbest.fitness
	UpdateData(false);
	
	// TODO: Add your control notification handler code here
	FILE *fp;
	fp=fopen("result_aver_best.txt","w+");
	for(int ii=1;ii<=maxgeneration;ii++)
	{
		fprintf(fp,"%d  %lf  %lf\n",ii,aver1[ii],curbest[ii]);
	}
	fclose(fp);

	FILE *fp1;
	fp1=fopen("result_fitness.txt","w+");
	for(ii=0;ii<L;ii++)
	{
		for(int j=0;j<L;j++)
		{
			fprintf(fp1,"%lf\n",population[ii][j].fitness);
		}
	}
	fclose(fp1);

	int lMaxCount = 0;
	CPaintDC dc(this);
	CWnd* pWnd = GetDlgItem(IDC_EDIT11);
	CDC* pDC = pWnd->GetDC();
	pWnd->Invalidate();
	pWnd->UpdateWindow();
	int LeftTop_x=40;
	int LeftTop_y=20;
	int RightButtom_x=380;
	int RightButtom_y=390;
	int SegCount=10;
	int DataLength=m_edit2;
	double lMaxCount1=aver1[1];
	double lMaxCount2=curbest[1];
	double lMinCount1=aver1[1];
	double lMinCount2=curbest[1];
	for (int i=1; i<=DataLength; i++)
	{
		if (aver1[i] > lMaxCount1)
		{
			lMaxCount1 = aver1[i];
		}
	}
	for (i=1; i<=DataLength; i++)
	{
		if (aver1[i] < lMinCount1)
		{
			lMinCount1 = aver1[i];
		}
	}
	for (i=0; i<DataLength; i++)
	{
		if (curbest[i] > lMaxCount2)
		{
			lMaxCount2 = curbest[i];
		}
	}
	for (i=0; i<DataLength; i++)
	{
		if (curbest[i] < lMinCount2)
		{
			lMinCount2 = curbest[i];
		}
	}