shieldsysdlg.cpp

Go to the documentation of this file.

/*
 * Copyright (C) Volition, Inc. 1999.  All rights reserved.
 *
 * All source code herein is the property of Volition, Inc. You may not sell 
 * or otherwise commercially exploit the source or things you created based on the 
 * source.
 *
*/



#include "stdafx.h"
#include "FRED.h"
#include "ShieldSysDlg.h"
#include "ship/ship.h"
#include "mission/missionparse.h"
#include "iff_defs/iff_defs.h"

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

int Shield_sys_teams[MAX_IFFS] = {0};
int Shield_sys_types[MAX_SHIP_CLASSES] = {0};

// shield_sys_dlg dialog

shield_sys_dlg::shield_sys_dlg(CWnd* pParent /*=NULL*/)
        : CDialog(shield_sys_dlg::IDD, pParent)
{
        //{{AFX_DATA_INIT(shield_sys_dlg)
        m_team = 0;
        m_type = 0;
        //}}AFX_DATA_INIT
}

void shield_sys_dlg::DoDataExchange(CDataExchange* pDX)
{
        CDialog::DoDataExchange(pDX);
        //{{AFX_DATA_MAP(shield_sys_dlg)
        DDX_CBIndex(pDX, IDC_TEAM, m_team);
        DDX_CBIndex(pDX, IDC_TYPE, m_type);
        //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(shield_sys_dlg, CDialog)
        //{{AFX_MSG_MAP(shield_sys_dlg)
        ON_CBN_SELCHANGE(IDC_TEAM, OnSelchangeTeam)
        ON_CBN_SELCHANGE(IDC_TYPE, OnSelchangeType)
        //}}AFX_MSG_MAP
END_MESSAGE_MAP()

// shield_sys_dlg message handlers

BOOL shield_sys_dlg::OnInitDialog() 
{
        int i, z;
        int teams[MAX_IFFS];
        int types[MAX_SHIP_CLASSES];
        CComboBox *box;

        for (i=0; i<MAX_IFFS; i++)
                teams[i] = 0;

        for (i=0; i<MAX_SHIP_CLASSES; i++)
                types[i] = 0;

        for (i=0; i<MAX_SHIPS; i++)
                if (Ships[i].objnum >= 0) {
                        z = (Objects[Ships[i].objnum].flags & OF_NO_SHIELDS) ? 1 : 0;
                        if (!teams[Ships[i].team])
                                Shield_sys_teams[Ships[i].team] = z;
                        else if (Shield_sys_teams[Ships[i].team] != z)
                                Shield_sys_teams[Ships[i].team] = 2;

                        if (!types[Ships[i].ship_info_index])
                                Shield_sys_types[Ships[i].ship_info_index] = z;
                        else if (Shield_sys_types[Ships[i].ship_info_index] != z)
                                Shield_sys_types[Ships[i].ship_info_index] = 2;

                        teams[Ships[i].team]++;
                        types[Ships[i].ship_info_index]++;
                }

        box = (CComboBox *) GetDlgItem(IDC_TYPE);
        box->ResetContent();
        for (auto it = Ship_info.cbegin(); it != Ship_info.cend(); ++it)
                box->AddString(it->name);

        box = (CComboBox *) GetDlgItem(IDC_TEAM);
        box->ResetContent();
        for (i=0; i<Num_iffs; i++)
                box->AddString(Iff_info[i].iff_name);

        CDialog::OnInitDialog();
        set_team();
        set_type();
        return TRUE;
}

void shield_sys_dlg::OnOK() 
{
        int i, z;

        OnSelchangeTeam();
        OnSelchangeType();
        for (i=0; i<MAX_SHIPS; i++)
                if (Ships[i].objnum >= 0) {
                        z = Shield_sys_teams[Ships[i].team];
                        if (!Shield_sys_types[Ships[i].ship_info_index])
                                z = 0;
                        else if (Shield_sys_types[Ships[i].ship_info_index] == 1)
                                z = 1;

                        if (!z)
                                Objects[Ships[i].objnum].flags &= ~OF_NO_SHIELDS;
                        else if (z == 1)
                                Objects[Ships[i].objnum].flags |= OF_NO_SHIELDS;
                }

        CDialog::OnOK();
}

void shield_sys_dlg::OnSelchangeTeam()
{
        Assert(m_team >= 0);
        if (((CButton *) GetDlgItem(IDC_TEAM_YES))->GetCheck())
                Shield_sys_teams[m_team] = 0;
        else if (((CButton *) GetDlgItem(IDC_TEAM_NO))->GetCheck())
                Shield_sys_teams[m_team] = 1;

        UpdateData(TRUE);
        set_team();
}

void shield_sys_dlg::set_team()
{
        if (!Shield_sys_teams[m_team])
                ((CButton *) GetDlgItem(IDC_TEAM_YES))->SetCheck(TRUE);
        else
                ((CButton *) GetDlgItem(IDC_TEAM_YES))->SetCheck(FALSE);

        if (Shield_sys_teams[m_team] == 1)
                ((CButton *) GetDlgItem(IDC_TEAM_NO))->SetCheck(TRUE);
        else
                ((CButton *) GetDlgItem(IDC_TEAM_NO))->SetCheck(FALSE);
}

void shield_sys_dlg::OnSelchangeType()
{
        Assert(m_type >= 0);
        if (((CButton *) GetDlgItem(IDC_TYPE_YES))->GetCheck())
                Shield_sys_types[m_type] = 0;
        else if (((CButton *) GetDlgItem(IDC_TYPE_NO))->GetCheck())
                Shield_sys_types[m_type] = 1;

        UpdateData(TRUE);
        set_type();
}

void shield_sys_dlg::set_type()
{
        if (!Shield_sys_types[m_type])
                ((CButton *) GetDlgItem(IDC_TYPE_YES))->SetCheck(TRUE);
        else
                ((CButton *) GetDlgItem(IDC_TYPE_YES))->SetCheck(FALSE);

        if (Shield_sys_types[m_type] == 1)
                ((CButton *) GetDlgItem(IDC_TYPE_NO))->SetCheck(TRUE);
        else
                ((CButton *) GetDlgItem(IDC_TYPE_NO))->SetCheck(FALSE);
}