// PartitionType.cpp : implementation file
//

#include "stdafx.h"
#include "ext2mgr.h"
#include "PartitionType.h"

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

/////////////////////////////////////////////////////////////////////////////
// CPartitionType dialog


CPartitionType::CPartitionType(CWnd* pParent /*=NULL*/)
        : CDialog(CPartitionType::IDD, pParent)
{
    //{{AFX_DATA_INIT(CPartitionType)
    // NOTE: the ClassWizard will add member initialization here
    m_Part = NULL;
    m_cPartType = 0;
    m_sDevice = _T("");
    m_sPartType = _T("");
    //}}AFX_DATA_INIT
}


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


BEGIN_MESSAGE_MAP(CPartitionType, CDialog)
    //{{AFX_MSG_MAP(CPartitionType)
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CPartitionType message handlers

void CPartitionType::OnOK()
{
    CString str;
    CComboBox *cbList = (CComboBox *)GetDlgItem(IDC_PARTTION_TYPE_LIST);
    m_cPartType = (UCHAR) cbList->GetCurSel();
    if (m_cPartType != m_Part->Entry->Mbr.PartitionType) {
        if (Ext2SetPartitionType(m_Part, m_cPartType)) {
            str.Format("Succeed to set partition type to %2.2X: %s",
                       m_cPartType, PartitionString(m_cPartType));
            AfxMessageBox(str, MB_OK | MB_ICONINFORMATION);
        } else {
            AfxMessageBox("Failed to set the partition type!",
                          MB_OK | MB_ICONWARNING);
            m_cPartType = 0;
            return;
        }
    } else {
        AfxMessageBox("Same partition type to the previous. Nothing is changed !",
                      MB_OK | MB_ICONWARNING);
    }

    CDialog::OnOK();
}

BOOL CPartitionType::OnInitDialog()
{
    CString str, type;
    CDialog::OnInitDialog();

    SET_TEXT(IDC_PARTITION_NAME, m_sDevice);

    CComboBox   *cbList = (CComboBox *)GetDlgItem(IDC_PARTTION_TYPE_LIST);
    for (unsigned int i=0;  i < 0x100; i++) {
        type = PartitionString(i);
        str.Format("%2.2X ", i);
        if (type.CompareNoCase("UNKNOWN")) {
            str += type;
        }
        cbList->AddString(str);
    }

    m_cPartType = m_Part->Entry->Mbr.PartitionType;
    cbList->SetCurSel((int) m_cPartType);

    return TRUE;  // return TRUE unless you set the focus to a control
    // EXCEPTION: OCX Property Pages should return FALSE
}