一个C++调用lua文件操作的工具软件

lua代码：

function WriteFengHaoFile(szDate,szSrc,szDst)

   local fileSrc = io.open(szSrc,"r")

   local fileDst = io.open(szDst,"w")

   if fileSrc ~= nil  then

      for line in fileSrc:lines() do

             local start,_ = string.find(line,szDate)

         if start ~= nil then

            fileDst:write(string.sub(line,1,10))

            fileDst:write("\r\n")

          end

       end

       fileSrc:close()

       fileDst:close()

   end

end

mfc代码：

#include "stdafx.h"

#include "ParseAccountFileByTime.h"

#include "ParseAccountFileByTimeDlg.h"

#include "afxdialogex.h"

#ifdef _DEBUG

#define new DEBUG_NEW

#endif

extern "C"

{

#include ".\\lua\\lua.h"

#include ".\\lua\\lauxlib.h"

#include ".\\lua\\lualib.h"

}

// CAboutDlg dialog used for App About

class CAboutDlg : public CDialogEx

{

public:

    CAboutDlg();

// Dialog Data

    enum { IDD = IDD_ABOUTBOX };

    protected:

    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support

// Implementation

protected:

    DECLARE_MESSAGE_MAP()

};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)

{

}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)

{

    CDialogEx::DoDataExchange(pDX);

}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)

END_MESSAGE_MAP()

// CParseAccountFileByTimeDlg dialog

CParseAccountFileByTimeDlg::CParseAccountFileByTimeDlg(CWnd* pParent /*=NULL*/)

    : CDialogEx(CParseAccountFileByTimeDlg::IDD, pParent)

{

    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

    m_pSrc = _T("");

    m_pDst = _T("");

    m_pDate = _T("");

}

void CParseAccountFileByTimeDlg::DoDataExchange(CDataExchange* pDX)

{

    CDialogEx::DoDataExchange(pDX);

    DDX_Text(pDX, IDC_EDIT1, m_pSrc);

    DDV_MaxChars(pDX, m_pSrc, 200);

    DDX_Text(pDX, IDC_EDIT2, m_pDst);

    DDV_MaxChars(pDX, m_pDst, 200);

    DDX_Text(pDX, IDC_EDIT3, m_pDate);

    DDV_MaxChars(pDX, m_pDate, 50);

}

BEGIN_MESSAGE_MAP(CParseAccountFileByTimeDlg, CDialogEx)

    ON_WM_SYSCOMMAND()

    ON_WM_PAINT()

    ON_WM_QUERYDRAGICON()

    ON_BN_CLICKED(IDC_BUTTON1, &CParseAccountFileByTimeDlg::OnBnClickedButton1)

    ON_BN_CLICKED(IDC_BUTTON2, &CParseAccountFileByTimeDlg::OnBnClickedButton2)

    ON_BN_CLICKED(IDOK, &CParseAccountFileByTimeDlg::OnBnClickedOk)

END_MESSAGE_MAP()

// CParseAccountFileByTimeDlg message handlers

BOOL CParseAccountFileByTimeDlg::OnInitDialog()

{

    CDialogEx::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)

    {

        BOOL bNameValid;

        CString strAboutMenu;

        bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);

        ASSERT(bNameValid);

        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 CParseAccountFileByTimeDlg::OnSysCommand(UINT nID, LPARAM lParam)

{

    if ((nID & 0xFFF0) == IDM_ABOUTBOX)

    {

        CAboutDlg dlgAbout;

        dlgAbout.DoModal();

    }

    else

    {

        CDialogEx::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 CParseAccountFileByTimeDlg::OnPaint()

{

    if (IsIconic())

    {

        CPaintDC dc(this); // device context for painting

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast<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

    {

        CDialogEx::OnPaint();

    }

}

// The system calls this function to obtain the cursor to display while the user drags

//  the minimized window.

HCURSOR CParseAccountFileByTimeDlg::OnQueryDragIcon()

{

    return static_cast<HCURSOR>(m_hIcon);

}

//获取源文件路径

void CParseAccountFileByTimeDlg::OnBnClickedButton1()

{

    char szCurrentFile[MAX_PATH];

    GetCurrentDirectory(MAX_PATH,szCurrentFile);

    CFileDialog dlg(TRUE,".txt",szCurrentFile, OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT | OFN_NOCHANGEDIR , "*.txt|*.*||");

    dlg.m_ofn.lpstrInitialDir = szCurrentFile;

    if(dlg.DoModal() == IDOK)

    {

        m_pSrc = dlg.GetPathName();

        UpdateData(FALSE);

    }

}

//设置生成文件路径

void CParseAccountFileByTimeDlg::OnBnClickedButton2()

{

    char szCurrentFile[MAX_PATH];

    GetCurrentDirectory(MAX_PATH,szCurrentFile);

    CFileDialog dlg(TRUE,".txt",szCurrentFile, OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT | OFN_NOCHANGEDIR , "*.txt|*.*||");

    dlg.m_ofn.lpstrInitialDir = szCurrentFile;

    if(dlg.DoModal() == IDOK)

    {

        m_pDst = dlg.GetPathName();

        UpdateData(FALSE);

    }

}

//调用lua函数，解析文件，生成新文件

void CParseAccountFileByTimeDlg::OnBnClickedOk()

{

    UpdateData(TRUE);

    char szErrMsg[MAX_PATH];

    

    lua_State *L =  luaL_newstate();

    luaL_openlibs(L);

    if(0 != luaL_dofile(L,"helloworld.lua"))

    {

        wsprintf(szErrMsg,"err:%s",lua_tostring(L,-1));

        AfxMessageBox(szErrMsg);

    }

 

    lua_getglobal(L,"WriteFengHaoFile");     //lua函数

    lua_pushstring(L,m_pDate.GetBuffer(1));  //参数1

    lua_pushstring(L,m_pSrc.GetBuffer(1));   //参数2

    lua_pushstring(L,m_pDst.GetBuffer(1));   //参数3

    if(lua_pcall(L,3,0,0) != 0)

    {

        wsprintf(szErrMsg,"err:%s",lua_tostring(L,-1));

        AfxMessageBox(szErrMsg);

    }

    AfxMessageBox("按时间抽取结束");

//    CDialogEx::OnOK();

}

4000