117 lines
3.4 KiB
C++
117 lines
3.4 KiB
C++
/*
|
|
* This file is part of the Colobot: Gold Edition source code
|
|
* Copyright (C) 2001-2015, Daniel Roux, EPSITEC SA & TerranovaTeam
|
|
* http://epsitec.ch; http://colobot.info; http://github.com/colobot
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
* See the GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see http://gnu.org/licenses
|
|
*/
|
|
|
|
// Modules inlcude
|
|
#include "CBot/CBotInstr/CBotReturn.h"
|
|
|
|
#include "CBot/CBotInstr/CBotInstrUtils.h"
|
|
|
|
#include "CBot/CBotInstr/CBotExpression.h"
|
|
|
|
#include "CBot/CBotStack.h"
|
|
#include "CBot/CBotCStack.h"
|
|
|
|
// Local include
|
|
|
|
// Global include
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
CBotReturn::CBotReturn()
|
|
{
|
|
m_Instr = nullptr;
|
|
name = "CBotReturn"; // debug
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
CBotReturn::~CBotReturn()
|
|
{
|
|
delete m_Instr;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
CBotInstr* CBotReturn::Compile(CBotToken* &p, CBotCStack* pStack)
|
|
{
|
|
CBotToken* pp = p;
|
|
|
|
if (!IsOfType(p, ID_RETURN)) return nullptr; // should never happen
|
|
|
|
CBotReturn* inst = new CBotReturn(); // creates the object
|
|
inst->SetToken( pp );
|
|
|
|
CBotTypResult type = pStack->GetRetType();
|
|
|
|
if ( type.GetType() == 0 ) // returned void ?
|
|
{
|
|
if ( IsOfType( p, ID_SEP ) ) return inst;
|
|
pStack->SetError( TX_BADTYPE, pp );
|
|
return nullptr;
|
|
}
|
|
|
|
inst->m_Instr = CBotExpression::Compile(p, pStack);
|
|
if ( pStack->IsOk() )
|
|
{
|
|
CBotTypResult retType = pStack->GetTypResult(2);
|
|
if (TypeCompatible(retType, type, ID_ASS))
|
|
{
|
|
if ( IsOfType( p, ID_SEP ) )
|
|
return inst;
|
|
|
|
pStack->SetError(TX_ENDOF, p->GetStart());
|
|
}
|
|
pStack->SetError(TX_BADTYPE, p->GetStart());
|
|
}
|
|
|
|
delete inst;
|
|
return nullptr; // no object, the error is on the stack
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
bool CBotReturn::Execute(CBotStack* &pj)
|
|
{
|
|
CBotStack* pile = pj->AddStack(this);
|
|
// if ( pile == EOX ) return true;
|
|
|
|
if ( pile->GetState() == 0 )
|
|
{
|
|
if ( m_Instr != nullptr && !m_Instr->Execute(pile) ) return false; // evaluate the result
|
|
// the result is on the stack
|
|
pile->IncState();
|
|
}
|
|
|
|
if ( pile->IfStep() ) return false;
|
|
|
|
pile->SetBreak(3, CBotString());
|
|
return pj->Return(pile);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
void CBotReturn::RestoreState(CBotStack* &pj, bool bMain)
|
|
{
|
|
if ( !bMain ) return;
|
|
CBotStack* pile = pj->RestoreStack(this);
|
|
if ( pile == nullptr ) return;
|
|
|
|
if ( pile->GetState() == 0 )
|
|
{
|
|
if ( m_Instr != nullptr ) m_Instr->RestoreState(pile, bMain); // evaluate the result
|
|
return;
|
|
}
|
|
}
|