int CASW_Mission_Chooser_Source_Local::GetNumMissionsInCampaign( int iCampaignIndex )
{
if (!m_bBuiltMapList)
BuildMapList();
ASW_Mission_Chooser_Mission* pCampaign = GetCampaign( iCampaignIndex );
if ( !pCampaign )
return 0;
KeyValues *pCampaignDetails = GetCampaignDetails( pCampaign->m_szMissionName );
if ( !pCampaignDetails )
return 0;
CUtlVector<KeyValues*> aMissionKeys;
bool bSkippedFirst = false;
for ( KeyValues *pMission = pCampaignDetails->GetFirstSubKey(); pMission; pMission = pMission->GetNextKey() )
{
if ( !Q_stricmp( pMission->GetName(), "MISSION" ) )
{
if ( !bSkippedFirst )
{
bSkippedFirst = true;
}
else
{
aMissionKeys.AddToTail( pMission );
}
}
}
return aMissionKeys.Count();
}
static base_t* CreateBase (const char* name, const vec2_t pos, bool fillBase = false)
{
const campaign_t* campaign = GetCampaign();
RS_InitTree(campaign, false);
E_InitialEmployees(campaign);
base_t* base = B_Build(campaign, pos, name, fillBase);
/* First base */
if (ccs.campaignStats.basesBuilt == 1)
B_SetUpFirstBase(campaign, base);
return base;
}
static base_t* CreateBase (const char *name, const vec2_t pos)
{
const campaign_t *campaign = GetCampaign();
base_t *base = B_GetFirstUnfoundedBase();
CU_ASSERT_PTR_NOT_NULL_FATAL(campaign);
CU_ASSERT_PTR_NOT_NULL_FATAL(base);
CU_ASSERT_FALSE(base->founded);
RS_InitTree(campaign, qfalse);
B_SetUpBase(campaign, base, pos, name);
return base;
}
static void ResetCampaignData (void)
{
const campaign_t *campaign;
CL_ResetSinglePlayerData();
CL_ReadSinglePlayerData();
campaign = GetCampaign();
CL_ReadCampaignData(campaign);
ResetInventoryList();
CL_UpdateCredits(MAX_CREDITS);
MAP_Shutdown();
MAP_Init(campaign->map);
}
void CASW_Mission_Chooser_Source_Local::FindMissionsInCampaign( int iCampaignIndex, int nMissionOffset, int iNumSlots )
{
if (!m_bBuiltMapList)
BuildMapList();
ASW_Mission_Chooser_Mission* pCampaign = GetCampaign( iCampaignIndex );
if ( !pCampaign )
return;
KeyValues *pCampaignDetails = GetCampaignDetails( pCampaign->m_szMissionName );
if ( !pCampaignDetails )
return;
CUtlVector<KeyValues*> aMissionKeys;
bool bSkippedFirst = false;
for ( KeyValues *pMission = pCampaignDetails->GetFirstSubKey(); pMission; pMission = pMission->GetNextKey() )
{
if ( !Q_stricmp( pMission->GetName(), "MISSION" ) )
{
if ( !bSkippedFirst )
{
bSkippedFirst = true;
}
else
{
aMissionKeys.AddToTail( pMission );
}
}
}
int max_items = aMissionKeys.Count();
for ( int stored=0;stored<iNumSlots;stored++ )
{
int realoffset = nMissionOffset;
if ( realoffset >= max_items || realoffset < 0 )
{
// no more missions...
Q_snprintf( m_missions[stored].m_szMissionName, sizeof( m_missions[stored].m_szMissionName ), "" );
}
else
{
Q_snprintf( m_missions[stored].m_szMissionName, sizeof( m_missions[stored].m_szMissionName ), "%s", aMissionKeys[realoffset]->GetString( "MapName" ) );
}
nMissionOffset++;
}
}
void SetUp()
{
campaign_t* campaign;
CP_ResetCampaignData();
CP_ParseCampaignData();
campaign = GetCampaign();
CP_ReadCampaignData(campaign);
ResetInventoryList();
CP_UpdateCredits(MAX_CREDITS);
GEO_Shutdown();
GEO_Init(campaign->map);
ccs.curCampaign = campaign;
}
static void testAircraftHandling (void)
{
const vec2_t destination = { 10, 10 };
campaign_t *campaign;
base_t *base;
aircraft_t *aircraft;
aircraft_t *newAircraft;
aircraft_t *aircraftTemplate;
int firstIdx;
int initialCount;
int count;
int newFound;
ResetCampaignData();
campaign = GetCampaign();
base = CreateBase("unittestaircraft", destination);
CU_ASSERT_PTR_NOT_NULL_FATAL(base);
/** @todo we should not assume that initial base has aircraft. It's a campaign parameter */
aircraft = AIR_GetFirstFromBase(base);
CU_ASSERT_PTR_NOT_NULL_FATAL(aircraft);
/* aircraft should have a template */
aircraftTemplate = aircraft->tpl;
CU_ASSERT_PTR_NOT_NULL_FATAL(aircraftTemplate);
firstIdx = aircraft->idx;
initialCount = AIR_BaseCountAircraft(base);
/* test deletion (part 1) */
AIR_DeleteAircraft(aircraft);
count = AIR_BaseCountAircraft(base);
CU_ASSERT_EQUAL(count, initialCount - 1);
/* test addition (part 1) */
newAircraft = AIR_NewAircraft(base, aircraftTemplate);
CU_ASSERT_PTR_NOT_NULL_FATAL(newAircraft);
count = AIR_BaseCountAircraft(base);
CU_ASSERT_EQUAL(count, initialCount);
/* new aircraft assigned to the right base */
CU_ASSERT_EQUAL(newAircraft->homebase, base);
newFound = 0;
AIR_Foreach(aircraft) {
/* test deletion (part 2) */
CU_ASSERT_NOT_EQUAL(firstIdx, aircraft->idx);
/* for test addition (part 2) */
if (aircraft->idx == newAircraft->idx)
newFound++;
}
/* test addition (part 2) */
CU_ASSERT_EQUAL(newFound, 1);
/* check if AIR_Foreach iterates through all aircraft */
AIR_Foreach(aircraft) {
AIR_DeleteAircraft(aircraft);
}
aircraft = AIR_GetFirstFromBase(base);
CU_ASSERT_PTR_NULL_FATAL(aircraft);
count = AIR_BaseCountAircraft(base);
CU_ASSERT_EQUAL(count, 0);
/* cleanup for the following tests */
E_DeleteAllEmployees(NULL);
base->founded = qfalse;
}
