2020-04-07 22:19:49 +00:00
/*
* * zcc_compile . cpp
* *
* * - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
* * Copyright - 2016 Randy Heit
* * Copyright 2016 Christoph Oelckers
* * All rights reserved .
* *
* * Redistribution and use in source and binary forms , with or without
* * modification , are permitted provided that the following conditions
* * are met :
* *
* * 1. Redistributions of source code must retain the above copyright
* * notice , this list of conditions and the following disclaimer .
* * 2. Redistributions in binary form must reproduce the above copyright
* * notice , this list of conditions and the following disclaimer in the
* * documentation and / or other materials provided with the distribution .
* * 3. The name of the author may not be used to endorse or promote products
* * derived from this software without specific prior written permission .
* *
* * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ` ` AS IS ' ' AND ANY EXPRESS OR
* * IMPLIED WARRANTIES , INCLUDING , BUT NOT LIMITED TO , THE IMPLIED WARRANTIES
* * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED .
* * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT , INDIRECT ,
* * INCIDENTAL , SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES ( INCLUDING , BUT
* * NOT LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ; LOSS OF USE ,
* * DATA , OR PROFITS ; OR BUSINESS INTERRUPTION ) HOWEVER CAUSED AND ON ANY
* * THEORY OF LIABILITY , WHETHER IN CONTRACT , STRICT LIABILITY , OR TORT
* * ( INCLUDING NEGLIGENCE OR OTHERWISE ) ARISING IN ANY WAY OUT OF THE USE OF
* * THIS SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
* * - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
* *
*/
# include "c_console.h"
# include "filesystem.h"
# include "zcc_parser.h"
# include "zcc-parse.h"
# include "zcc_compile.h"
# include "printf.h"
# include "symbols.h"
FSharedStringArena VMStringConstants ;
2020-04-08 16:02:53 +00:00
int GetIntConst ( FxExpression * ex , FCompileContext & ctx )
2020-04-07 22:19:49 +00:00
{
ex = new FxIntCast ( ex , false ) ;
ex = ex - > Resolve ( ctx ) ;
return ex ? static_cast < FxConstant * > ( ex ) - > GetValue ( ) . GetInt ( ) : 0 ;
}
2020-04-08 16:02:53 +00:00
double GetFloatConst ( FxExpression * ex , FCompileContext & ctx )
2020-04-07 22:19:49 +00:00
{
ex = new FxFloatCast ( ex ) ;
ex = ex - > Resolve ( ctx ) ;
return ex ? static_cast < FxConstant * > ( ex ) - > GetValue ( ) . GetFloat ( ) : 0 ;
}
const char * ZCCCompiler : : GetStringConst ( FxExpression * ex , FCompileContext & ctx )
{
ex = new FxStringCast ( ex ) ;
ex = ex - > Resolve ( ctx ) ;
if ( ! ex ) return " " ;
// The string here must be stored in a persistent place that lasts long enough to have it processed.
return AST . Strings . Alloc ( static_cast < FxConstant * > ( ex ) - > GetValue ( ) . GetString ( ) ) - > GetChars ( ) ;
}
int ZCCCompiler : : IntConstFromNode ( ZCC_TreeNode * node , PContainerType * cls )
{
FCompileContext ctx ( OutNamespace , cls , false ) ;
FxExpression * ex = new FxIntCast ( ConvertNode ( node ) , false ) ;
ex = ex - > Resolve ( ctx ) ;
if ( ex = = nullptr ) return 0 ;
if ( ! ex - > isConstant ( ) )
{
ex - > ScriptPosition . Message ( MSG_ERROR , " Expression is not constant " ) ;
return 0 ;
}
return static_cast < FxConstant * > ( ex ) - > GetValue ( ) . GetInt ( ) ;
}
FString ZCCCompiler : : StringConstFromNode ( ZCC_TreeNode * node , PContainerType * cls )
{
FCompileContext ctx ( OutNamespace , cls , false ) ;
FxExpression * ex = new FxStringCast ( ConvertNode ( node ) ) ;
ex = ex - > Resolve ( ctx ) ;
if ( ex = = nullptr ) return " " ;
if ( ! ex - > isConstant ( ) )
{
ex - > ScriptPosition . Message ( MSG_ERROR , " Expression is not constant " ) ;
return " " ;
}
return static_cast < FxConstant * > ( ex ) - > GetValue ( ) . GetString ( ) ;
}
ZCC_MixinDef * ZCCCompiler : : ResolveMixinStmt ( ZCC_MixinStmt * mixinStmt , EZCCMixinType type )
{
for ( auto mx : Mixins )
{
if ( mx - > mixin - > NodeName = = mixinStmt - > MixinName )
{
if ( mx - > mixin - > MixinType ! = type )
{
Error ( mixinStmt , " Mixin %s is a %s mixin cannot be used here. " , FName ( mixinStmt - > MixinName ) . GetChars ( ) , GetMixinTypeString ( type ) ) ;
return nullptr ;
}
return mx - > mixin ;
}
}
Error ( mixinStmt , " Mixin %s does not exist. " , FName ( mixinStmt - > MixinName ) . GetChars ( ) ) ;
return nullptr ;
}
//==========================================================================
//
// ZCCCompiler :: ProcessClass
//
//==========================================================================
void ZCCCompiler : : ProcessClass ( ZCC_Class * cnode , PSymbolTreeNode * treenode )
{
ZCC_ClassWork * cls = nullptr ;
// If this is a class extension, put the new node directly into the existing class.
if ( cnode - > Flags = = ZCC_Extension )
{
for ( auto clss : Classes )
{
if ( clss - > NodeName ( ) = = cnode - > NodeName )
{
cls = clss ;
break ;
}
}
if ( cls = = nullptr )
{
Error ( cnode , " Class %s cannot be found in the current translation unit. " , FName ( cnode - > NodeName ) . GetChars ( ) ) ;
return ;
}
}
else
{
Classes . Push ( new ZCC_ClassWork ( static_cast < ZCC_Class * > ( cnode ) , treenode ) ) ;
cls = Classes . Last ( ) ;
}
auto node = cnode - > Body ;
PSymbolTreeNode * childnode ;
ZCC_Enum * enumType = nullptr ;
// [pbeta] Handle mixins here for the sake of simplifying things.
if ( node ! = nullptr )
{
bool mixinError = false ;
TArray < ZCC_MixinStmt * > mixinStmts ;
mixinStmts . Clear ( ) ;
// Gather all mixin statement nodes.
do
{
if ( node - > NodeType = = AST_MixinStmt )
{
mixinStmts . Push ( static_cast < ZCC_MixinStmt * > ( node ) ) ;
}
node = node - > SiblingNext ;
}
while ( node ! = cnode - > Body ) ;
for ( auto mixinStmt : mixinStmts )
{
ZCC_MixinDef * mixinDef = ResolveMixinStmt ( mixinStmt , ZCC_Mixin_Class ) ;
if ( mixinDef = = nullptr )
{
mixinError = true ;
continue ;
}
// Insert the mixin if there's a body. If not, just remove this node.
if ( mixinDef - > Body ! = nullptr )
{
auto newNode = TreeNodeDeepCopy ( & AST , mixinDef - > Body , true ) ;
if ( mixinStmt - > SiblingNext ! = mixinStmt & & mixinStmt - > SiblingPrev ! = mixinStmt )
{
auto prevSibling = mixinStmt - > SiblingPrev ;
auto nextSibling = mixinStmt - > SiblingNext ;
auto newFirst = newNode ;
auto newLast = newNode - > SiblingPrev ;
newFirst - > SiblingPrev = prevSibling ;
newLast - > SiblingNext = nextSibling ;
prevSibling - > SiblingNext = newFirst ;
nextSibling - > SiblingPrev = newLast ;
}
if ( cnode - > Body = = mixinStmt )
{
cnode - > Body = newNode ;
}
}
else
{
if ( mixinStmt - > SiblingNext ! = mixinStmt & & mixinStmt - > SiblingPrev ! = mixinStmt )
{
auto prevSibling = mixinStmt - > SiblingPrev ;
auto nextSibling = mixinStmt - > SiblingNext ;
prevSibling - > SiblingNext = nextSibling ;
nextSibling - > SiblingPrev = prevSibling ;
if ( cnode - > Body = = mixinStmt )
{
cnode - > Body = nextSibling ;
}
}
else if ( cnode - > Body = = mixinStmt )
{
cnode - > Body = nullptr ;
}
}
}
mixinStmts . Clear ( ) ;
if ( mixinError )
{
return ;
}
}
node = cnode - > Body ;
// Need to check if the class actually has a body.
if ( node ! = nullptr ) do
{
switch ( node - > NodeType )
{
case AST_MixinStmt :
assert ( 0 & & " Unhandled mixin statement in class parsing loop. If this has been reached, something is seriously wrong " ) ;
Error ( node , " Internal mixin error. " ) ;
break ;
case AST_Struct :
case AST_ConstantDef :
case AST_Enum :
if ( ( childnode = AddTreeNode ( static_cast < ZCC_NamedNode * > ( node ) - > NodeName , node , & cls - > TreeNodes ) ) )
{
switch ( node - > NodeType )
{
case AST_Enum :
enumType = static_cast < ZCC_Enum * > ( node ) ;
cls - > Enums . Push ( enumType ) ;
break ;
case AST_Struct :
if ( static_cast < ZCC_Struct * > ( node ) - > Flags & VARF_Native )
{
Error ( node , " Cannot define native structs inside classes " ) ;
static_cast < ZCC_Struct * > ( node ) - > Flags & = ~ VARF_Native ;
}
ProcessStruct ( static_cast < ZCC_Struct * > ( node ) , childnode , cls - > cls ) ;
break ;
case AST_ConstantDef :
cls - > Constants . Push ( static_cast < ZCC_ConstantDef * > ( node ) ) ;
cls - > Constants . Last ( ) - > Type = enumType ;
break ;
default :
assert ( 0 & & " Default case is just here to make GCC happy. It should never be reached " ) ;
}
}
break ;
case AST_Property :
cls - > Properties . Push ( static_cast < ZCC_Property * > ( node ) ) ;
break ;
case AST_FlagDef :
cls - > FlagDefs . Push ( static_cast < ZCC_FlagDef * > ( node ) ) ;
break ;
case AST_VarDeclarator :
cls - > Fields . Push ( static_cast < ZCC_VarDeclarator * > ( node ) ) ;
break ;
case AST_EnumTerminator :
enumType = nullptr ;
break ;
case AST_States :
cls - > States . Push ( static_cast < ZCC_States * > ( node ) ) ;
break ;
case AST_FuncDeclarator :
cls - > Functions . Push ( static_cast < ZCC_FuncDeclarator * > ( node ) ) ;
break ;
case AST_Default :
cls - > Defaults . Push ( static_cast < ZCC_Default * > ( node ) ) ;
break ;
case AST_StaticArrayStatement :
if ( AddTreeNode ( static_cast < ZCC_StaticArrayStatement * > ( node ) - > Id , node , & cls - > TreeNodes ) )
{
cls - > Arrays . Push ( static_cast < ZCC_StaticArrayStatement * > ( node ) ) ;
}
break ;
default :
assert ( 0 & & " Unhandled AST node type " ) ;
break ;
}
node = node - > SiblingNext ;
}
while ( node ! = cnode - > Body ) ;
}
//==========================================================================
//
// ZCCCompiler :: ProcessMixin
//
//==========================================================================
void ZCCCompiler : : ProcessMixin ( ZCC_MixinDef * cnode , PSymbolTreeNode * treenode )
{
ZCC_MixinWork * cls = new ZCC_MixinWork ( cnode , treenode ) ;
auto node = cnode - > Body ;
// Need to check if the mixin actually has a body.
if ( node ! = nullptr ) do
{
if ( cnode - > MixinType = = ZCC_Mixin_Class )
{
switch ( node - > NodeType )
{
case AST_Struct :
case AST_ConstantDef :
case AST_Enum :
case AST_Property :
case AST_FlagDef :
case AST_VarDeclarator :
case AST_EnumTerminator :
case AST_States :
case AST_FuncDeclarator :
case AST_Default :
case AST_StaticArrayStatement :
break ;
default :
assert ( 0 & & " Unhandled AST node type " ) ;
break ;
}
}
node = node - > SiblingNext ;
} while ( node ! = cnode - > Body ) ;
Mixins . Push ( cls ) ;
}
//==========================================================================
//
// ZCCCompiler :: ProcessStruct
//
//==========================================================================
void ZCCCompiler : : ProcessStruct ( ZCC_Struct * cnode , PSymbolTreeNode * treenode , ZCC_Class * outer )
{
Structs . Push ( new ZCC_StructWork ( static_cast < ZCC_Struct * > ( cnode ) , treenode , outer ) ) ;
ZCC_StructWork * cls = Structs . Last ( ) ;
auto node = cnode - > Body ;
PSymbolTreeNode * childnode ;
ZCC_Enum * enumType = nullptr ;
// Need to check if the struct actually has a body.
if ( node ! = nullptr ) do
{
switch ( node - > NodeType )
{
case AST_ConstantDef :
case AST_Enum :
if ( ( childnode = AddTreeNode ( static_cast < ZCC_NamedNode * > ( node ) - > NodeName , node , & cls - > TreeNodes ) ) )
{
switch ( node - > NodeType )
{
case AST_Enum :
enumType = static_cast < ZCC_Enum * > ( node ) ;
cls - > Enums . Push ( enumType ) ;
break ;
case AST_ConstantDef :
cls - > Constants . Push ( static_cast < ZCC_ConstantDef * > ( node ) ) ;
cls - > Constants . Last ( ) - > Type = enumType ;
break ;
default :
assert ( 0 & & " Default case is just here to make GCC happy. It should never be reached " ) ;
}
}
break ;
case AST_VarDeclarator :
cls - > Fields . Push ( static_cast < ZCC_VarDeclarator * > ( node ) ) ;
break ;
case AST_FuncDeclarator :
cls - > Functions . Push ( static_cast < ZCC_FuncDeclarator * > ( node ) ) ;
break ;
case AST_EnumTerminator :
enumType = nullptr ;
break ;
case AST_StaticArrayStatement :
if ( AddTreeNode ( static_cast < ZCC_StaticArrayStatement * > ( node ) - > Id , node , & cls - > TreeNodes ) )
{
cls - > Arrays . Push ( static_cast < ZCC_StaticArrayStatement * > ( node ) ) ;
}
break ;
default :
assert ( 0 & & " Unhandled AST node type " ) ;
break ;
}
node = node - > SiblingNext ;
}
while ( node ! = cnode - > Body ) ;
}
//==========================================================================
//
// ZCCCompiler Constructor
//
//==========================================================================
ZCCCompiler : : ZCCCompiler ( ZCC_AST & ast , DObject * _outer , PSymbolTable & _symbols , PNamespace * _outnamespc , int lumpnum , const VersionInfo & ver )
: mVersion ( ver ) , Outer ( _outer ) , ConvertClass ( nullptr ) , GlobalTreeNodes ( & _symbols ) , OutNamespace ( _outnamespc ) , AST ( ast ) , Lump ( lumpnum )
{
FScriptPosition : : ResetErrorCounter ( ) ;
// Group top-level nodes by type
if ( ast . TopNode ! = NULL )
{
ZCC_TreeNode * node = ast . TopNode ;
PSymbolTreeNode * tnode ;
// [pbeta] Anything that must be processed before classes, structs, etc. should go here.
do
{
switch ( node - > NodeType )
{
// [pbeta] Mixins must be processed before everything else.
case AST_MixinDef :
if ( ( tnode = AddTreeNode ( static_cast < ZCC_NamedNode * > ( node ) - > NodeName , node , GlobalTreeNodes ) ) )
{
ProcessMixin ( static_cast < ZCC_MixinDef * > ( node ) , tnode ) ;
break ;
}
break ;
default :
break ; // Shut GCC up.
}
node = node - > SiblingNext ;
} while ( node ! = ast . TopNode ) ;
node = ast . TopNode ;
PType * enumType = nullptr ;
ZCC_Enum * zenumType = nullptr ;
do
{
switch ( node - > NodeType )
{
case AST_MixinDef :
// [pbeta] We already processed mixins, ignore them here.
break ;
case AST_Class :
// a class extension should not check the tree node symbols.
if ( static_cast < ZCC_Class * > ( node ) - > Flags = = ZCC_Extension )
{
ProcessClass ( static_cast < ZCC_Class * > ( node ) , tnode ) ;
break ;
}
case AST_Struct :
case AST_ConstantDef :
case AST_Enum :
if ( ( tnode = AddTreeNode ( static_cast < ZCC_NamedNode * > ( node ) - > NodeName , node , GlobalTreeNodes ) ) )
{
switch ( node - > NodeType )
{
case AST_Enum :
zenumType = static_cast < ZCC_Enum * > ( node ) ;
enumType = NewEnum ( zenumType - > NodeName , OutNamespace ) ;
OutNamespace - > Symbols . AddSymbol ( Create < PSymbolType > ( zenumType - > NodeName , enumType ) ) ;
break ;
case AST_Class :
ProcessClass ( static_cast < ZCC_Class * > ( node ) , tnode ) ;
break ;
case AST_Struct :
ProcessStruct ( static_cast < ZCC_Struct * > ( node ) , tnode , nullptr ) ;
break ;
case AST_ConstantDef :
Constants . Push ( static_cast < ZCC_ConstantDef * > ( node ) ) ;
Constants . Last ( ) - > Type = zenumType ;
break ;
default :
assert ( 0 & & " Default case is just here to make GCC happy. It should never be reached " ) ;
}
}
break ;
case AST_EnumTerminator :
zenumType = nullptr ;
break ;
default :
assert ( 0 & & " Unhandled AST node type " ) ;
break ;
}
node = node - > SiblingNext ;
} while ( node ! = ast . TopNode ) ;
}
}
ZCCCompiler : : ~ ZCCCompiler ( )
{
for ( auto s : Structs )
{
delete s ;
}
for ( auto c : Classes )
{
delete c ;
}
Structs . Clear ( ) ;
Classes . Clear ( ) ;
}
//==========================================================================
//
// ZCCCompiler :: AddTreeNode
//
// Keeps track of definition nodes by their names. Ensures that all names
// in this scope are unique.
//
//==========================================================================
PSymbolTreeNode * ZCCCompiler : : AddTreeNode ( FName name , ZCC_TreeNode * node , PSymbolTable * treenodes , bool searchparents )
{
PSymbol * check = treenodes - > FindSymbol ( name , searchparents ) ;
if ( check ! = NULL )
{
assert ( check - > IsA ( RUNTIME_CLASS ( PSymbolTreeNode ) ) ) ;
Error ( node , " Attempt to redefine '%s' " , name . GetChars ( ) ) ;
Error ( static_cast < PSymbolTreeNode * > ( check ) - > Node , " Original definition is here " ) ;
return nullptr ;
}
else
{
auto sy = Create < PSymbolTreeNode > ( name , node ) ;
FString name ;
treenodes - > AddSymbol ( sy ) ;
return sy ;
}
}
//==========================================================================
//
// ZCCCompiler :: Warn
//
// Prints a warning message, and increments WarnCount.
//
//==========================================================================
void ZCCCompiler : : Warn ( ZCC_TreeNode * node , const char * msg , . . . )
{
va_list argptr ;
va_start ( argptr , msg ) ;
MessageV ( node , TEXTCOLOR_ORANGE , msg , argptr ) ;
va_end ( argptr ) ;
FScriptPosition : : WarnCounter + + ;
}
//==========================================================================
//
// ZCCCompiler :: Error
//
// Prints an error message, and increments ErrorCount.
//
//==========================================================================
void ZCCCompiler : : Error ( ZCC_TreeNode * node , const char * msg , . . . )
{
va_list argptr ;
va_start ( argptr , msg ) ;
MessageV ( node , TEXTCOLOR_RED , msg , argptr ) ;
va_end ( argptr ) ;
FScriptPosition : : ErrorCounter + + ;
}
//==========================================================================
//
// ZCCCompiler :: MessageV
//
// Prints a message, annotated with the source location for the tree node.
//
//==========================================================================
void ZCCCompiler : : MessageV ( ZCC_TreeNode * node , const char * txtcolor , const char * msg , va_list argptr )
{
FString composed ;
composed . Format ( " %s%s, line %d: " , txtcolor , node - > SourceName - > GetChars ( ) , node - > SourceLoc ) ;
composed . VAppendFormat ( msg , argptr ) ;
composed + = ' \n ' ;
PrintString ( PRINT_HIGH , composed ) ;
}
//==========================================================================
//
// ZCCCompiler :: Compile
//
// Compile everything defined at this level.
2020-04-08 16:02:53 +00:00
// This can be overridden to add custom content.
2020-04-07 22:19:49 +00:00
//
//==========================================================================
int ZCCCompiler : : Compile ( )
{
CreateClassTypes ( ) ;
CreateStructTypes ( ) ;
CompileAllConstants ( ) ;
CompileAllFields ( ) ;
2020-06-14 16:57:55 +00:00
InitDefaults ( ) ;
2020-04-07 22:19:49 +00:00
InitFunctions ( ) ;
return FScriptPosition : : ErrorCounter ;
}
//==========================================================================
//
// ZCCCompiler :: CreateStructTypes
//
// Creates a PStruct for every struct.
//
//==========================================================================
void ZCCCompiler : : CreateStructTypes ( )
{
for ( auto s : Structs )
{
PTypeBase * outer ;
PSymbolTable * syms ;
s - > Outer = s - > OuterDef = = nullptr ? nullptr : s - > OuterDef - > CType ( ) ;
if ( s - > Outer )
{
outer = s - > Outer - > VMType ;
syms = & s - > Outer - > VMType - > Symbols ;
}
else
{
outer = OutNamespace ;
syms = & OutNamespace - > Symbols ;
}
if ( s - > NodeName ( ) = = NAME__ & & fileSystem . GetFileContainer ( Lump ) = = 0 )
{
// This is just a container for syntactic purposes.
s - > strct - > Type = nullptr ;
continue ;
}
else if ( s - > strct - > Flags & ZCC_Native )
{
s - > strct - > Type = NewStruct ( s - > NodeName ( ) , outer , true ) ;
}
else
{
s - > strct - > Type = NewStruct ( s - > NodeName ( ) , outer ) ;
}
if ( s - > strct - > Flags & ZCC_Version )
{
s - > strct - > Type - > mVersion = s - > strct - > Version ;
}
auto & sf = s - > Type ( ) - > ScopeFlags ;
if ( mVersion > = MakeVersion ( 2 , 4 , 0 ) )
{
if ( ( s - > strct - > Flags & ( ZCC_UIFlag | ZCC_Play ) ) = = ( ZCC_UIFlag | ZCC_Play ) )
{
Error ( s - > strct , " Struct %s has incompatible flags " , s - > NodeName ( ) . GetChars ( ) ) ;
}
if ( outer ! = OutNamespace ) sf = FScopeBarrier : : ChangeSideInObjectFlags ( sf , FScopeBarrier : : SideFromObjectFlags ( static_cast < PType * > ( outer ) - > ScopeFlags ) ) ;
else if ( s - > strct - > Flags & ZCC_ClearScope ) Warn ( s - > strct , " Useless 'ClearScope' on struct %s not inside a class " , s - > NodeName ( ) . GetChars ( ) ) ;
if ( s - > strct - > Flags & ZCC_UIFlag )
sf = FScopeBarrier : : ChangeSideInObjectFlags ( sf , FScopeBarrier : : Side_UI ) ;
if ( s - > strct - > Flags & ZCC_Play )
sf = FScopeBarrier : : ChangeSideInObjectFlags ( sf , FScopeBarrier : : Side_Play ) ;
if ( s - > strct - > Flags & ZCC_ClearScope )
sf = FScopeBarrier : : ChangeSideInObjectFlags ( sf , FScopeBarrier : : Side_PlainData ) ; // don't inherit the scope from the outer class
}
else
{
// old versions force 'play'.
sf = FScopeBarrier : : ChangeSideInObjectFlags ( sf , FScopeBarrier : : Side_Play ) ;
}
s - > strct - > Symbol = Create < PSymbolType > ( s - > NodeName ( ) , s - > Type ( ) ) ;
syms - > AddSymbol ( s - > strct - > Symbol ) ;
for ( auto e : s - > Enums )
{
auto etype = NewEnum ( e - > NodeName , s - > Type ( ) ) ;
s - > Type ( ) - > Symbols . AddSymbol ( Create < PSymbolType > ( e - > NodeName , etype ) ) ;
}
}
}
//==========================================================================
//
// ZCCCompiler :: CreateClassTypes
//
// Creates a PClass for every class so that we get access to the symbol table
// These will be created with unknown size because for that we need to
// process all fields first, but to do that we need the PClass and some
// other info depending on the PClass.
//
//==========================================================================
void ZCCCompiler : : CreateClassTypes ( )
{
// we are going to sort the classes array so that entries are sorted in order of inheritance.
auto OrigClasses = std : : move ( Classes ) ;
Classes . Clear ( ) ;
bool donesomething = true ;
while ( donesomething )
{
donesomething = false ;
for ( unsigned i = 0 ; i < OrigClasses . Size ( ) ; i + + )
{
auto c = OrigClasses [ i ] ;
// Check if we got the parent already defined.
PClass * parent ;
auto ParentName = c - > cls - > ParentName ;
if ( ParentName ! = nullptr & & ParentName - > SiblingNext = = ParentName ) parent = PClass : : FindClass ( ParentName - > Id ) ;
else if ( ParentName = = nullptr ) parent = RUNTIME_CLASS ( DObject ) ;
else
{
// The parent is a dotted name which the type system currently does not handle.
// Once it does this needs to be implemented here.
auto p = ParentName ;
FString build ;
do
{
if ( build . IsNotEmpty ( ) ) build + = ' . ' ;
build + = FName ( p - > Id ) . GetChars ( ) ;
p = static_cast < decltype ( p ) > ( p - > SiblingNext ) ;
} while ( p ! = ParentName ) ;
Error ( c - > cls , " Qualified name '%s' for base class not supported in '%s' " , build . GetChars ( ) , FName ( c - > NodeName ( ) ) . GetChars ( ) ) ;
parent = RUNTIME_CLASS ( DObject ) ;
}
if ( parent ! = nullptr & & ( parent - > VMType ! = nullptr | | c - > NodeName ( ) = = NAME_Object ) )
{
// The parent exists, we may create a type for this class
if ( c - > cls - > Flags & ZCC_Native )
{
// If this is a native class, its own type must also already exist and not be a runtime class.
auto me = PClass : : FindClass ( c - > NodeName ( ) ) ;
if ( me = = nullptr )
{
Error ( c - > cls , " Unknown native class %s " , c - > NodeName ( ) . GetChars ( ) ) ;
// Create a placeholder so that the compiler can continue looking for errors.
me = parent - > FindClassTentative ( c - > NodeName ( ) ) ;
}
else if ( me - > bRuntimeClass )
{
Error ( c - > cls , " %s is not a native class " , c - > NodeName ( ) . GetChars ( ) ) ;
}
else
{
DPrintf ( DMSG_SPAMMY , " Registered %s as native with parent %s \n " , me - > TypeName . GetChars ( ) , parent - > TypeName . GetChars ( ) ) ;
}
c - > cls - > Type = NewClassType ( me ) ;
me - > SourceLumpName = * c - > cls - > SourceName ;
}
else
{
// We will never get here if the name is a duplicate, so we can just do the assignment.
try
{
if ( parent - > VMType - > mVersion > mVersion )
{
Error ( c - > cls , " Parent class %s of %s not accessible to ZScript version %d.%d.%d " , parent - > TypeName . GetChars ( ) , c - > NodeName ( ) . GetChars ( ) , mVersion . major , mVersion . minor , mVersion . revision ) ;
}
auto newclass = parent - > CreateDerivedClass ( c - > NodeName ( ) , TentativeClass ) ;
if ( newclass = = nullptr )
{
Error ( c - > cls , " Class name %s already exists " , c - > NodeName ( ) . GetChars ( ) ) ;
}
else
{
c - > cls - > Type = NewClassType ( newclass ) ;
DPrintf ( DMSG_SPAMMY , " Created class %s with parent %s \n " , c - > Type ( ) - > TypeName . GetChars ( ) , c - > ClassType ( ) - > ParentClass - > TypeName . GetChars ( ) ) ;
}
}
catch ( CRecoverableError & err )
{
Error ( c - > cls , " %s " , err . GetMessage ( ) ) ;
c - > cls - > Type = nullptr ;
}
}
if ( c - > Type ( ) = = nullptr )
{
// create a placeholder so that the compiler can continue looking for errors.
c - > cls - > Type = NewClassType ( parent - > FindClassTentative ( c - > NodeName ( ) ) ) ;
}
if ( c - > cls - > Flags & ZCC_Abstract )
c - > ClassType ( ) - > bAbstract = true ;
if ( c - > cls - > Flags & ZCC_Version )
{
c - > Type ( ) - > mVersion = c - > cls - > Version ;
}
//
if ( mVersion > = MakeVersion ( 2 , 4 , 0 ) )
{
static int incompatible [ ] = { ZCC_UIFlag , ZCC_Play , ZCC_ClearScope } ;
int incompatiblecnt = 0 ;
for ( size_t k = 0 ; k < countof ( incompatible ) ; k + + )
if ( incompatible [ k ] & c - > cls - > Flags ) incompatiblecnt + + ;
if ( incompatiblecnt > 1 )
{
Error ( c - > cls , " Class %s has incompatible flags " , c - > NodeName ( ) . GetChars ( ) ) ;
}
if ( c - > cls - > Flags & ZCC_UIFlag )
c - > Type ( ) - > ScopeFlags = EScopeFlags ( ( c - > Type ( ) - > ScopeFlags & ~ Scope_Play ) | Scope_UI ) ;
if ( c - > cls - > Flags & ZCC_Play )
c - > Type ( ) - > ScopeFlags = EScopeFlags ( ( c - > Type ( ) - > ScopeFlags & ~ Scope_UI ) | Scope_Play ) ;
if ( parent - > VMType - > ScopeFlags & ( Scope_UI | Scope_Play ) ) // parent is either ui or play
{
if ( c - > cls - > Flags & ( ZCC_UIFlag | ZCC_Play ) )
{
Error ( c - > cls , " Can't change class scope in class %s " , c - > NodeName ( ) . GetChars ( ) ) ;
}
c - > Type ( ) - > ScopeFlags = FScopeBarrier : : ChangeSideInObjectFlags ( c - > Type ( ) - > ScopeFlags , FScopeBarrier : : SideFromObjectFlags ( parent - > VMType - > ScopeFlags ) ) ;
}
}
else
{
c - > Type ( ) - > ScopeFlags = FScopeBarrier : : ChangeSideInObjectFlags ( c - > Type ( ) - > ScopeFlags , FScopeBarrier : : Side_Play ) ;
}
c - > cls - > Symbol = Create < PSymbolType > ( c - > NodeName ( ) , c - > Type ( ) ) ;
OutNamespace - > Symbols . AddSymbol ( c - > cls - > Symbol ) ;
Classes . Push ( c ) ;
OrigClasses . Delete ( i - - ) ;
donesomething = true ;
}
else if ( c - > cls - > ParentName ! = nullptr )
{
// No base class found. Now check if something in the unprocessed classes matches.
// If not, print an error. If something is found let's retry again in the next iteration.
bool found = false ;
for ( auto d : OrigClasses )
{
if ( d - > NodeName ( ) = = c - > cls - > ParentName - > Id )
{
found = true ;
break ;
}
}
if ( ! found )
{
Error ( c - > cls , " Class %s has unknown base class %s " , c - > NodeName ( ) . GetChars ( ) , FName ( c - > cls - > ParentName - > Id ) . GetChars ( ) ) ;
// create a placeholder so that the compiler can continue looking for errors.
c - > cls - > Type = NewClassType ( RUNTIME_CLASS ( DObject ) - > FindClassTentative ( c - > NodeName ( ) ) ) ;
c - > cls - > Symbol = Create < PSymbolType > ( c - > NodeName ( ) , c - > Type ( ) ) ;
OutNamespace - > Symbols . AddSymbol ( c - > cls - > Symbol ) ;
Classes . Push ( c ) ;
OrigClasses . Delete ( i - - ) ;
donesomething = true ;
}
}
}
}
// What's left refers to some other class in the list but could not be resolved.
// This normally means a circular reference.
for ( auto c : OrigClasses )
{
Error ( c - > cls , " Class %s has circular inheritance " , FName ( c - > NodeName ( ) ) . GetChars ( ) ) ;
c - > cls - > Type = NewClassType ( RUNTIME_CLASS ( DObject ) - > FindClassTentative ( c - > NodeName ( ) ) ) ;
c - > cls - > Symbol = Create < PSymbolType > ( c - > NodeName ( ) , c - > Type ( ) ) ;
OutNamespace - > Symbols . AddSymbol ( c - > cls - > Symbol ) ;
Classes . Push ( c ) ;
}
// Last but not least: Now that all classes have been created, we can create the symbols for the internal enums and link the treenode symbol tables.
for ( auto cd : Classes )
{
for ( auto e : cd - > Enums )
{
auto etype = NewEnum ( e - > NodeName , cd - > Type ( ) ) ;
cd - > Type ( ) - > Symbols . AddSymbol ( Create < PSymbolType > ( e - > NodeName , etype ) ) ;
}
// Link the tree node tables. We only can do this after we know the class relations.
for ( auto cc : Classes )
{
if ( cc - > ClassType ( ) = = cd - > ClassType ( ) - > ParentClass )
{
cd - > TreeNodes . SetParentTable ( & cc - > TreeNodes ) ;
break ;
}
}
}
}
//==========================================================================
//
// ZCCCompiler :: AddConstants
//
// Helper for CompileAllConstants
//
//==========================================================================
void ZCCCompiler : : CopyConstants ( TArray < ZCC_ConstantWork > & dest , TArray < ZCC_ConstantDef * > & Constants , PContainerType * cls , PSymbolTable * ot )
{
for ( auto c : Constants )
{
dest . Push ( { c , cls , ot } ) ;
}
}
//==========================================================================
//
// ZCCCompiler :: CompileAllConstants
//
// Make symbols from every constant defined at all levels.
// Since constants may only depend on other constants this can be done
// without any more involved processing of the AST as a first step.
//
//==========================================================================
void ZCCCompiler : : CompileAllConstants ( )
{
// put all constants in one list to make resolving this easier.
TArray < ZCC_ConstantWork > constantwork ;
CopyConstants ( constantwork , Constants , nullptr , & OutNamespace - > Symbols ) ;
for ( auto c : Classes )
{
CopyConstants ( constantwork , c - > Constants , c - > Type ( ) , & c - > Type ( ) - > Symbols ) ;
}
for ( auto s : Structs )
{
if ( s - > Type ( ) ! = nullptr )
CopyConstants ( constantwork , s - > Constants , s - > Type ( ) , & s - > Type ( ) - > Symbols ) ;
}
// Before starting to resolve the list, let's create symbols for all already resolved ones first (i.e. all literal constants), to reduce work.
for ( unsigned i = 0 ; i < constantwork . Size ( ) ; i + + )
{
if ( constantwork [ i ] . node - > Value - > NodeType = = AST_ExprConstant )
{
AddConstant ( constantwork [ i ] ) ;
// Remove the constant from the list
constantwork . Delete ( i ) ;
i - - ;
}
}
bool donesomething = true ;
// Now go through this list until no more constants can be resolved. The remaining ones will be non-constant values.
while ( donesomething & & constantwork . Size ( ) > 0 )
{
donesomething = false ;
for ( unsigned i = 0 ; i < constantwork . Size ( ) ; i + + )
{
if ( CompileConstant ( & constantwork [ i ] ) )
{
AddConstant ( constantwork [ i ] ) ;
// Remove the constant from the list
constantwork . Delete ( i ) ;
i - - ;
donesomething = true ;
}
}
}
for ( unsigned i = 0 ; i < constantwork . Size ( ) ; i + + )
{
Error ( constantwork [ i ] . node , " %s is not a constant " , FName ( constantwork [ i ] . node - > NodeName ) . GetChars ( ) ) ;
}
for ( auto s : Structs )
{
CompileArrays ( s ) ;
}
for ( auto c : Classes )
{
CompileArrays ( c ) ;
}
}
//==========================================================================
//
// ZCCCompiler :: AddConstant
//
// Adds a constant to its assigned symbol table
//
//==========================================================================
void ZCCCompiler : : AddConstant ( ZCC_ConstantWork & constant )
{
auto def = constant . node ;
auto val = def - > Value ;
ExpVal & c = constant . constval ;
// This is for literal constants.
if ( val - > NodeType = = AST_ExprConstant )
{
ZCC_ExprConstant * cval = static_cast < ZCC_ExprConstant * > ( val ) ;
if ( cval - > Type = = TypeString )
{
def - > Symbol = Create < PSymbolConstString > ( def - > NodeName , * ( cval - > StringVal ) ) ;
}
else if ( cval - > Type - > isInt ( ) )
{
// How do we get an Enum type in here without screwing everything up???
//auto type = def->Type != nullptr ? def->Type : cval->Type;
def - > Symbol = Create < PSymbolConstNumeric > ( def - > NodeName , cval - > Type , cval - > IntVal ) ;
}
else if ( cval - > Type - > isFloat ( ) )
{
if ( def - > Type ! = nullptr )
{
Error ( def , " Enum members must be integer values " ) ;
}
def - > Symbol = Create < PSymbolConstNumeric > ( def - > NodeName , cval - > Type , cval - > DoubleVal ) ;
}
else
{
Error ( def - > Value , " Bad type for constant definiton " ) ;
def - > Symbol = nullptr ;
}
}
else
{
if ( c . Type = = TypeString )
{
def - > Symbol = Create < PSymbolConstString > ( def - > NodeName , c . GetString ( ) ) ;
}
else if ( c . Type - > isInt ( ) )
{
// How do we get an Enum type in here without screwing everything up???
//auto type = def->Type != nullptr ? def->Type : cval->Type;
def - > Symbol = Create < PSymbolConstNumeric > ( def - > NodeName , c . Type , c . GetInt ( ) ) ;
}
else if ( c . Type - > isFloat ( ) )
{
if ( def - > Type ! = nullptr )
{
Error ( def , " Enum members must be integer values " ) ;
}
def - > Symbol = Create < PSymbolConstNumeric > ( def - > NodeName , c . Type , c . GetFloat ( ) ) ;
}
else
{
Error ( def - > Value , " Bad type for constant definiton " ) ;
def - > Symbol = nullptr ;
}
}
if ( def - > Symbol = = nullptr )
{
// Create a dummy constant so we don't make any undefined value warnings.
def - > Symbol = Create < PSymbolConstNumeric > ( def - > NodeName , TypeError , 0 ) ;
}
constant . Outputtable - > ReplaceSymbol ( def - > Symbol ) ;
}
//==========================================================================
//
// ZCCCompiler :: CompileConstant
//
// For every constant definition, evaluate its value (which should result
// in a constant), and create a symbol for it.
//
//==========================================================================
bool ZCCCompiler : : CompileConstant ( ZCC_ConstantWork * work )
{
FCompileContext ctx ( OutNamespace , work - > cls , false ) ;
FxExpression * exp = ConvertNode ( work - > node - > Value ) ;
try
{
FScriptPosition : : errorout = true ;
exp = exp - > Resolve ( ctx ) ;
if ( exp = = nullptr ) return false ;
FScriptPosition : : errorout = false ;
if ( ! exp - > isConstant ( ) )
{
delete exp ;
return false ;
}
work - > constval = static_cast < FxConstant * > ( exp ) - > GetValue ( ) ;
delete exp ;
return true ;
}
catch ( . . . )
{
// eat the reported error and treat this as a temorary failure. All unresolved contants will be reported at the end.
FScriptPosition : : errorout = false ;
return false ;
}
}
void ZCCCompiler : : CompileArrays ( ZCC_StructWork * work )
{
for ( auto sas : work - > Arrays )
{
PType * ztype = DetermineType ( work - > Type ( ) , sas , sas - > Id , sas - > Type , false , true ) ;
PType * ctype = ztype ;
FArgumentList values ;
// Don't use narrow typea for casting.
if ( ctype - > isInt ( ) ) ctype = static_cast < PInt * > ( ztype ) - > Unsigned ? TypeUInt32 : TypeSInt32 ;
else if ( ctype = = TypeFloat32 ) ctype = TypeFloat64 ;
ConvertNodeList ( values , sas - > Values ) ;
bool fail = false ;
FCompileContext ctx ( OutNamespace , work - > Type ( ) , false ) ;
char * destmem = ( char * ) ClassDataAllocator . Alloc ( values . Size ( ) * ztype - > Align ) ;
memset ( destmem , 0 , values . Size ( ) * ztype - > Align ) ;
char * copyp = destmem ;
for ( unsigned i = 0 ; i < values . Size ( ) ; i + + )
{
values [ i ] = new FxTypeCast ( values [ i ] , ctype , false ) ;
values [ i ] = values [ i ] - > Resolve ( ctx ) ;
if ( values [ i ] = = nullptr ) fail = true ;
else if ( ! values [ i ] - > isConstant ( ) )
{
Error ( sas , " Initializer must be constant " ) ;
fail = true ;
}
else
{
ExpVal val = static_cast < FxConstant * > ( values [ i ] ) - > GetValue ( ) ;
switch ( ztype - > GetRegType ( ) )
{
default :
// should never happen
Error ( sas , " Non-integral type in constant array " ) ;
return ;
case REGT_INT :
ztype - > SetValue ( copyp , val . GetInt ( ) ) ;
break ;
case REGT_FLOAT :
ztype - > SetValue ( copyp , val . GetFloat ( ) ) ;
break ;
case REGT_POINTER :
* ( void * * ) copyp = val . GetPointer ( ) ;
break ;
case REGT_STRING :
: : new ( copyp ) FString ( val . GetString ( ) ) ;
break ;
}
copyp + = ztype - > Align ;
}
}
work - > Type ( ) - > Symbols . AddSymbol ( Create < PField > ( sas - > Id , NewArray ( ztype , values . Size ( ) ) , VARF_Static | VARF_ReadOnly , ( size_t ) destmem ) ) ;
}
}
//==========================================================================
//
// ZCCCompiler :: NodeFromSymbol
//
//==========================================================================
ZCC_Expression * ZCCCompiler : : NodeFromSymbol ( PSymbol * sym , ZCC_Expression * source , PSymbolTable * table )
{
assert ( sym ! = nullptr ) ;
if ( sym - > IsKindOf ( RUNTIME_CLASS ( PSymbolConst ) ) )
{
return NodeFromSymbolConst ( static_cast < PSymbolConst * > ( sym ) , source ) ;
}
else if ( sym - > IsKindOf ( RUNTIME_CLASS ( PSymbolType ) ) )
{
return NodeFromSymbolType ( static_cast < PSymbolType * > ( sym ) , source ) ;
}
return NULL ;
}
//==========================================================================
//
// ZCCCompiler :: NodeFromSymbolConst
//
// Returns a new AST constant node with the symbol's content.
//
//==========================================================================
ZCC_ExprConstant * ZCCCompiler : : NodeFromSymbolConst ( PSymbolConst * sym , ZCC_Expression * idnode )
{
ZCC_ExprConstant * val = static_cast < ZCC_ExprConstant * > ( AST . InitNode ( sizeof ( * val ) , AST_ExprConstant , idnode ) ) ;
val - > Operation = PEX_ConstValue ;
if ( sym = = NULL )
{
val - > Type = TypeError ;
val - > IntVal = 0 ;
}
else if ( sym - > IsKindOf ( RUNTIME_CLASS ( PSymbolConstString ) ) )
{
val - > StringVal = AST . Strings . Alloc ( static_cast < PSymbolConstString * > ( sym ) - > Str ) ;
val - > Type = TypeString ;
}
else
{
val - > Type = sym - > ValueType ;
if ( val - > Type ! = TypeError )
{
assert ( sym - > IsKindOf ( RUNTIME_CLASS ( PSymbolConstNumeric ) ) ) ;
if ( sym - > ValueType - > isIntCompatible ( ) )
{
val - > IntVal = static_cast < PSymbolConstNumeric * > ( sym ) - > Value ;
}
else
{
assert ( sym - > ValueType - > isFloat ( ) ) ;
val - > DoubleVal = static_cast < PSymbolConstNumeric * > ( sym ) - > Float ;
}
}
}
return val ;
}
//==========================================================================
//
// ZCCCompiler :: NodeFromSymbolType
//
// Returns a new AST type ref node with the symbol's content.
//
//==========================================================================
ZCC_ExprTypeRef * ZCCCompiler : : NodeFromSymbolType ( PSymbolType * sym , ZCC_Expression * idnode )
{
ZCC_ExprTypeRef * ref = static_cast < ZCC_ExprTypeRef * > ( AST . InitNode ( sizeof ( * ref ) , AST_ExprTypeRef , idnode ) ) ;
ref - > Operation = PEX_TypeRef ;
ref - > RefType = sym - > Type ;
ref - > Type = NewClassPointer ( RUNTIME_CLASS ( DObject ) ) ;
return ref ;
}
//==========================================================================
//
// ZCCCompiler :: CompileAllFields
//
// builds the internal structure of all classes and structs
//
//==========================================================================
void ZCCCompiler : : CompileAllFields ( )
{
// Create copies of the arrays which can be altered
auto Classes = this - > Classes ;
auto Structs = this - > Structs ;
TMap < FName , bool > HasNativeChildren ;
// first step: Look for native classes with native children.
// These may not have any variables added to them because it'd clash with the native definitions.
for ( unsigned i = 0 ; i < Classes . Size ( ) ; i + + )
{
auto c = Classes [ i ] ;
if ( c - > Type ( ) - > Size ! = TentativeClass & & c - > Fields . Size ( ) > 0 )
{
// We need to search the global class table here because not all children may have a scripted definition attached.
for ( auto ac : PClass : : AllClasses )
{
if ( ac - > ParentClass ! = nullptr & & ac - > ParentClass - > VMType = = c - > Type ( ) & & ac - > Size ! = TentativeClass )
{
// Only set a marker here, so that we can print a better message when the actual fields get added.
HasNativeChildren . Insert ( c - > Type ( ) - > TypeName , true ) ;
break ;
}
}
}
}
bool donesomething = true ;
while ( donesomething & & ( Structs . Size ( ) > 0 | | Classes . Size ( ) > 0 ) )
{
donesomething = false ;
for ( unsigned i = 0 ; i < Structs . Size ( ) ; i + + )
{
if ( CompileFields ( Structs [ i ] - > Type ( ) , Structs [ i ] - > Fields , Structs [ i ] - > Outer , Structs [ i ] - > Type ( ) = = 0 ? GlobalTreeNodes : & Structs [ i ] - > TreeNodes , true ) )
{
// Remove from the list if all fields got compiled.
Structs . Delete ( i - - ) ;
donesomething = true ;
}
}
for ( unsigned i = 0 ; i < Classes . Size ( ) ; i + + )
{
auto type = Classes [ i ] - > ClassType ( ) ;
if ( type - > Size = = TentativeClass )
{
if ( type - > ParentClass - > Size = = TentativeClass )
{
// we do not know the parent class's size yet, so skip this class for now.
continue ;
}
else
{
// Inherit the size of the parent class
type - > Size = Classes [ i ] - > ClassType ( ) - > ParentClass - > Size ;
}
}
2020-04-08 16:02:53 +00:00
if ( ! PrepareMetaData ( type ) )
2020-04-07 22:19:49 +00:00
{
2020-04-08 16:02:53 +00:00
if ( Classes [ i ] - > ClassType ( ) - > ParentClass )
type - > MetaSize = Classes [ i ] - > ClassType ( ) - > ParentClass - > MetaSize ;
else
type - > MetaSize = 0 ;
2020-04-07 22:19:49 +00:00
}
if ( CompileFields ( type - > VMType , Classes [ i ] - > Fields , nullptr , & Classes [ i ] - > TreeNodes , false , ! ! HasNativeChildren . CheckKey ( type - > TypeName ) ) )
{
// Remove from the list if all fields got compiled.
Classes . Delete ( i - - ) ;
donesomething = true ;
}
}
}
// This really should never happen, but if it does, let's better print an error.
for ( auto s : Structs )
{
Error ( s - > strct , " Unable to resolve all fields for struct %s " , FName ( s - > NodeName ( ) ) . GetChars ( ) ) ;
}
for ( auto s : Classes )
{
Error ( s - > cls , " Unable to resolve all fields for class %s " , FName ( s - > NodeName ( ) ) . GetChars ( ) ) ;
}
}
//==========================================================================
//
// ZCCCompiler :: CompileFields
//
// builds the internal structure of a single class or struct
//
//==========================================================================
bool ZCCCompiler : : CompileFields ( PContainerType * type , TArray < ZCC_VarDeclarator * > & Fields , PClass * Outer , PSymbolTable * TreeNodes , bool forstruct , bool hasnativechildren )
{
while ( Fields . Size ( ) > 0 )
{
auto field = Fields [ 0 ] ;
FieldDesc * fd = nullptr ;
PType * fieldtype = DetermineType ( type , field , field - > Names - > Name , field - > Type , true , true ) ;
// For structs only allow 'deprecated', for classes exclude function qualifiers.
int notallowed = forstruct ?
ZCC_Latent | ZCC_Final | ZCC_Action | ZCC_Static | ZCC_FuncConst | ZCC_Abstract | ZCC_Virtual | ZCC_Override | ZCC_Meta | ZCC_Extension | ZCC_VirtualScope | ZCC_ClearScope :
ZCC_Latent | ZCC_Final | ZCC_Action | ZCC_Static | ZCC_FuncConst | ZCC_Abstract | ZCC_Virtual | ZCC_Override | ZCC_Extension | ZCC_VirtualScope | ZCC_ClearScope ;
// Some internal fields need to be set to clearscope.
if ( fileSystem . GetFileContainer ( Lump ) = = 0 ) notallowed & = ~ ZCC_ClearScope ;
if ( field - > Flags & notallowed )
{
Error ( field , " Invalid qualifiers for %s (%s not allowed) " , FName ( field - > Names - > Name ) . GetChars ( ) , FlagsToString ( field - > Flags & notallowed ) . GetChars ( ) ) ;
field - > Flags & = notallowed ;
}
uint32_t varflags = 0 ;
// These map directly to implementation flags.
if ( field - > Flags & ZCC_Private ) varflags | = VARF_Private ;
if ( field - > Flags & ZCC_Protected ) varflags | = VARF_Protected ;
if ( field - > Flags & ZCC_Deprecated ) varflags | = VARF_Deprecated ;
if ( field - > Flags & ZCC_ReadOnly ) varflags | = VARF_ReadOnly ;
if ( field - > Flags & ZCC_Internal ) varflags | = VARF_InternalAccess ;
if ( field - > Flags & ZCC_Transient ) varflags | = VARF_Transient ;
if ( mVersion > = MakeVersion ( 2 , 4 , 0 ) )
{
if ( type ! = nullptr )
{
if ( type - > ScopeFlags & Scope_UI )
varflags | = VARF_UI ;
if ( type - > ScopeFlags & Scope_Play )
varflags | = VARF_Play ;
}
if ( field - > Flags & ZCC_UIFlag )
varflags = FScopeBarrier : : ChangeSideInFlags ( varflags , FScopeBarrier : : Side_UI ) ;
if ( field - > Flags & ZCC_Play )
varflags = FScopeBarrier : : ChangeSideInFlags ( varflags , FScopeBarrier : : Side_Play ) ;
if ( field - > Flags & ZCC_ClearScope )
varflags = FScopeBarrier : : ChangeSideInFlags ( varflags , FScopeBarrier : : Side_PlainData ) ;
}
else
{
varflags | = VARF_Play ;
}
if ( field - > Flags & ZCC_Native )
{
varflags | = VARF_Native | VARF_Transient ;
}
static int excludescope [ ] = { ZCC_UIFlag , ZCC_Play , ZCC_ClearScope } ;
int excludeflags = 0 ;
int fc = 0 ;
for ( size_t i = 0 ; i < countof ( excludescope ) ; i + + )
{
if ( field - > Flags & excludescope [ i ] )
{
fc + + ;
excludeflags | = excludescope [ i ] ;
}
}
if ( fc > 1 )
{
Error ( field , " Invalid combination of scope qualifiers %s on field %s " , FlagsToString ( excludeflags ) . GetChars ( ) , FName ( field - > Names - > Name ) . GetChars ( ) ) ;
varflags & = ~ ( VARF_UI | VARF_Play ) ; // make plain data
}
if ( field - > Flags & ZCC_Meta )
{
varflags | = VARF_Meta | VARF_Static | VARF_ReadOnly ; // metadata implies readonly
}
if ( field - > Type - > ArraySize ! = nullptr )
{
bool nosize ;
fieldtype = ResolveArraySize ( fieldtype , field - > Type - > ArraySize , type , & nosize ) ;
if ( nosize )
{
Error ( field , " Must specify array size " ) ;
}
}
auto name = field - > Names ;
do
{
2020-09-27 08:47:11 +00:00
if ( ( fieldtype - > Size = = 0 | | ! fieldtype - > SizeKnown ) & & ! ( varflags & VARF_Native ) ) // Size not known yet.
2020-04-07 22:19:49 +00:00
{
2020-09-27 08:47:11 +00:00
if ( type ! = nullptr )
{
type - > SizeKnown = false ;
}
2020-04-07 22:19:49 +00:00
return false ;
}
if ( AddTreeNode ( name - > Name , name , TreeNodes , ! forstruct ) )
{
auto thisfieldtype = fieldtype ;
if ( name - > ArraySize ! = nullptr )
{
bool nosize ;
thisfieldtype = ResolveArraySize ( thisfieldtype , name - > ArraySize , type , & nosize ) ;
if ( nosize )
{
Error ( field , " Must specify array size " ) ;
}
}
PField * f = nullptr ;
if ( varflags & VARF_Native )
{
if ( varflags & VARF_Meta )
{
Error ( field , " Native meta variable %s not allowed " , FName ( name - > Name ) . GetChars ( ) ) ;
}
else
{
fd = FindField ( type , FName ( name - > Name ) . GetChars ( ) ) ;
if ( fd = = nullptr )
{
Error ( field , " The member variable '%s.%s' has not been exported from the executable. " , type = = nullptr ? " " : type - > TypeName . GetChars ( ) , FName ( name - > Name ) . GetChars ( ) ) ;
}
// For native structs a size check cannot be done because they normally have no size. But for a native reference they are still fine.
else if ( thisfieldtype - > Size ! = ~ 0u & & fd - > FieldSize ! = ~ 0u & & thisfieldtype - > Size ! = fd - > FieldSize & & fd - > BitValue = = 0 & &
( ! thisfieldtype - > isStruct ( ) | | ! static_cast < PStruct * > ( thisfieldtype ) - > isNative ) )
{
Error ( field , " The member variable '%s.%s' has mismatching sizes in internal and external declaration. (Internal = %d, External = %d) " , type = = nullptr ? " " : type - > TypeName . GetChars ( ) , FName ( name - > Name ) . GetChars ( ) , fd - > FieldSize , thisfieldtype - > Size ) ;
}
// Q: Should we check alignment, too? A mismatch may be an indicator for bad assumptions.
else if ( type ! = nullptr )
{
// for bit fields the type must point to the source variable.
if ( fd - > BitValue ! = 0 ) thisfieldtype = fd - > FieldSize = = 1 ? TypeUInt8 : fd - > FieldSize = = 2 ? TypeUInt16 : TypeUInt32 ;
f = type - > AddNativeField ( name - > Name , thisfieldtype , fd - > FieldOffset , varflags , fd - > BitValue ) ;
}
else
{
// This is a global variable.
if ( fd - > BitValue ! = 0 ) thisfieldtype = fd - > FieldSize = = 1 ? TypeUInt8 : fd - > FieldSize = = 2 ? TypeUInt16 : TypeUInt32 ;
f = Create < PField > ( name - > Name , thisfieldtype , varflags | VARF_Native | VARF_Static , fd - > FieldOffset , fd - > BitValue ) ;
if ( OutNamespace - > Symbols . AddSymbol ( f ) = = nullptr )
{ // name is already in use
2020-09-27 08:47:11 +00:00
if ( type ! = nullptr )
{
type - > SizeKnown = false ;
}
2020-04-07 22:19:49 +00:00
f - > Destroy ( ) ;
return false ;
}
}
}
}
else if ( hasnativechildren & & ! ( varflags & VARF_Meta ) )
{
Error ( field , " Cannot add field %s to %s. %s has native children which means it size may not change " , FName ( name - > Name ) . GetChars ( ) , type - > TypeName . GetChars ( ) , type - > TypeName . GetChars ( ) ) ;
}
else if ( type ! = nullptr )
{
f = type - > AddField ( name - > Name , thisfieldtype , varflags ) ;
}
else
{
Error ( field , " Cannot declare non-native global variables. Tried to declare %s " , FName ( name - > Name ) . GetChars ( ) ) ;
}
if ( ( field - > Flags & ( ZCC_Version | ZCC_Deprecated ) ) & & f ! = nullptr )
{
f - > mVersion = field - > Version ;
if ( field - > DeprecationMessage ! = nullptr )
{
f - > DeprecationMessage = * field - > DeprecationMessage ;
}
}
}
name = static_cast < ZCC_VarName * > ( name - > SiblingNext ) ;
} while ( name ! = field - > Names ) ;
Fields . Delete ( 0 ) ;
}
2020-09-27 08:47:11 +00:00
if ( type ! = nullptr )
{
type - > SizeKnown = Fields . Size ( ) = = 0 ;
}
2020-04-07 22:19:49 +00:00
return Fields . Size ( ) = = 0 ;
}
//==========================================================================
//
// ZCCCompiler :: FieldFlagsToString
//
// creates a string for a field's flags
//
//==========================================================================
FString ZCCCompiler : : FlagsToString ( uint32_t flags )
{
const char * flagnames [ ] = { " native " , " static " , " private " , " protected " , " latent " , " final " , " meta " , " action " , " deprecated " , " readonly " , " const " , " abstract " , " extend " , " virtual " , " override " , " transient " , " vararg " , " ui " , " play " , " clearscope " , " virtualscope " } ;
FString build ;
for ( size_t i = 0 ; i < countof ( flagnames ) ; i + + )
{
if ( flags & ( 1 < < i ) )
{
if ( build . IsNotEmpty ( ) ) build + = " , " ;
build + = flagnames [ i ] ;
}
}
return build ;
}
//==========================================================================
//
// ZCCCompiler :: DetermineType
//
// retrieves the type for this field, for arrays the type of a single entry.
//
//==========================================================================
PType * ZCCCompiler : : DetermineType ( PType * outertype , ZCC_TreeNode * field , FName name , ZCC_Type * ztype , bool allowarraytypes , bool formember )
{
PType * retval = TypeError ;
if ( ! allowarraytypes & & ztype - > ArraySize ! = nullptr )
{
Error ( field , " %s: Array type not allowed " , name . GetChars ( ) ) ;
return TypeError ;
}
switch ( ztype - > NodeType )
{
case AST_BasicType :
{
auto btype = static_cast < ZCC_BasicType * > ( ztype ) ;
switch ( btype - > Type )
{
case ZCC_SInt8 :
retval = formember ? TypeSInt8 : ( PType * ) TypeError ;
break ;
case ZCC_UInt8 :
retval = formember ? TypeUInt8 : ( PType * ) TypeError ;
break ;
case ZCC_SInt16 :
retval = formember ? TypeSInt16 : ( PType * ) TypeError ;
break ;
case ZCC_UInt16 :
retval = formember ? TypeUInt16 : ( PType * ) TypeError ;
break ;
case ZCC_SInt32 :
case ZCC_IntAuto : // todo: for enums, autoselect appropriately sized int
retval = TypeSInt32 ;
break ;
case ZCC_UInt32 :
retval = TypeUInt32 ;
break ;
case ZCC_Bool :
retval = TypeBool ;
break ;
case ZCC_FloatAuto :
retval = formember ? TypeFloat32 : TypeFloat64 ;
break ;
case ZCC_Float64 :
retval = TypeFloat64 ;
break ;
case ZCC_String :
retval = TypeString ;
break ;
case ZCC_Name :
retval = TypeName ;
break ;
case ZCC_Vector2 :
retval = TypeVector2 ;
break ;
case ZCC_Vector3 :
retval = TypeVector3 ;
break ;
case ZCC_State :
retval = TypeState ;
break ;
case ZCC_Color :
retval = TypeColor ;
break ;
case ZCC_Sound :
retval = TypeSound ;
break ;
case ZCC_Let :
retval = TypeAuto ;
break ;
case ZCC_NativeType :
// Creating an instance of a native struct is only allowed for internal definitions of native variables.
if ( fileSystem . GetFileContainer ( Lump ) ! = 0 | | ! formember )
{
Error ( field , " %s: @ not allowed for user scripts " , name . GetChars ( ) ) ;
}
retval = ResolveUserType ( btype , outertype ? & outertype - > Symbols : nullptr , true ) ;
break ;
case ZCC_UserType :
// statelabel et.al. are not tokens - there really is no need to, it works just as well as an identifier. Maybe the same should be done for some other types, too?
switch ( btype - > UserType - > Id )
{
case NAME_Voidptr :
retval = TypeVoidPtr ;
break ;
case NAME_StateLabel :
retval = TypeStateLabel ;
break ;
case NAME_SpriteID :
retval = TypeSpriteID ;
break ;
case NAME_TextureID :
retval = TypeTextureID ;
break ;
default :
retval = ResolveUserType ( btype , outertype ? & outertype - > Symbols : nullptr , false ) ;
break ;
}
break ;
default :
break ;
}
break ;
}
case AST_MapType :
if ( allowarraytypes )
{
Error ( field , " %s: Map types not implemented yet " , name . GetChars ( ) ) ;
// Todo: Decide what we allow here and if it makes sense to allow more complex constructs.
auto mtype = static_cast < ZCC_MapType * > ( ztype ) ;
retval = NewMap ( DetermineType ( outertype , field , name , mtype - > KeyType , false , false ) , DetermineType ( outertype , field , name , mtype - > ValueType , false , false ) ) ;
break ;
}
break ;
case AST_DynArrayType :
{
auto atype = static_cast < ZCC_DynArrayType * > ( ztype ) ;
auto ftype = DetermineType ( outertype , field , name , atype - > ElementType , false , true ) ;
if ( ftype - > GetRegType ( ) = = REGT_NIL | | ftype - > GetRegCount ( ) > 1 )
{
if ( field - > NodeType = = AST_VarDeclarator & & ( static_cast < ZCC_VarDeclarator * > ( field ) - > Flags & ZCC_Native ) & & fileSystem . GetFileContainer ( Lump ) = = 0 )
{
// the internal definitions may declare native arrays to complex types.
// As long as they can be mapped to a static array type the VM can handle them, in a limited but sufficient fashion.
retval = NewPointer ( NewStaticArray ( ftype ) , false ) ;
retval - > Size = ~ 0u ; // don't check for a size match, it's likely to fail anyway.
retval - > Align = ~ 0u ;
}
else
{
Error ( field , " %s: Base type for dynamic array types must be integral, but got %s " , name . GetChars ( ) , ftype - > DescriptiveName ( ) ) ;
}
}
else
{
retval = NewDynArray ( ftype ) ;
}
break ;
}
case AST_ClassType :
{
auto ctype = static_cast < ZCC_ClassType * > ( ztype ) ;
if ( ctype - > Restriction = = nullptr )
{
retval = NewClassPointer ( RUNTIME_CLASS ( DObject ) ) ;
}
else
{
// This doesn't check the class list directly but the current symbol table to ensure that
// this does not reference a type that got shadowed by a more local definition.
// We first look in the current class and its parents, and then in the current namespace and its parents.
auto sym = outertype ? outertype - > Symbols . FindSymbol ( ctype - > Restriction - > Id , true ) : nullptr ;
if ( sym = = nullptr ) sym = OutNamespace - > Symbols . FindSymbol ( ctype - > Restriction - > Id , true ) ;
if ( sym = = nullptr )
{
// A symbol with a given name cannot be reached from this definition point, so
// even if a class with the given name exists, it is not accessible.
Error ( field , " %s: Unknown identifier " , FName ( ctype - > Restriction - > Id ) . GetChars ( ) ) ;
return TypeError ;
}
auto typesym = dyn_cast < PSymbolType > ( sym ) ;
if ( typesym = = nullptr | | ! typesym - > Type - > isClass ( ) )
{
Error ( field , " %s does not represent a class type " , FName ( ctype - > Restriction - > Id ) . GetChars ( ) ) ;
return TypeError ;
}
if ( typesym - > Type - > mVersion > mVersion )
{
Error ( field , " Class %s not accessible to ZScript version %d.%d.%d " , FName ( ctype - > Restriction - > Id ) . GetChars ( ) , mVersion . major , mVersion . minor , mVersion . revision ) ;
return TypeError ;
}
retval = NewClassPointer ( static_cast < PClassType * > ( typesym - > Type ) - > Descriptor ) ;
}
break ;
}
default :
break ;
}
if ( retval ! = TypeError & & retval - > MemberOnly & & ! formember )
{
Error ( field , " Invalid type %s " , retval - > DescriptiveName ( ) ) ;
return TypeError ;
}
return retval ;
}
//==========================================================================
//
// ZCCCompiler :: ResolveUserType
//
// resolves a user type and returns a matching PType
//
//==========================================================================
PType * ZCCCompiler : : ResolveUserType ( ZCC_BasicType * type , PSymbolTable * symt , bool nativetype )
{
// Check the symbol table for the identifier.
PSymbol * sym = nullptr ;
// We first look in the current class and its parents, and then in the current namespace and its parents.
if ( symt ! = nullptr ) sym = symt - > FindSymbol ( type - > UserType - > Id , true ) ;
if ( sym = = nullptr ) sym = OutNamespace - > Symbols . FindSymbol ( type - > UserType - > Id , true ) ;
if ( sym ! = nullptr & & sym - > IsKindOf ( RUNTIME_CLASS ( PSymbolType ) ) )
{
auto ptype = static_cast < PSymbolType * > ( sym ) - > Type ;
if ( ptype - > mVersion > mVersion )
{
Error ( type , " Type %s not accessible to ZScript version %d.%d.%d " , FName ( type - > UserType - > Id ) . GetChars ( ) , mVersion . major , mVersion . minor , mVersion . revision ) ;
return TypeError ;
}
if ( ptype - > isEnum ( ) )
{
if ( ! nativetype ) return TypeSInt32 ; // hack this to an integer until we can resolve the enum mess.
}
else if ( ptype - > isClass ( ) ) // classes cannot be instantiated at all, they always get used as references.
{
return NewPointer ( ptype , type - > isconst ) ;
}
else if ( ptype - > isStruct ( ) & & static_cast < PStruct * > ( ptype ) - > isNative ) // native structs and classes cannot be instantiated, they always get used as reference.
{
if ( ! nativetype ) return NewPointer ( ptype , type - > isconst ) ;
return ptype ; // instantiation of native structs. Only for internal use.
}
if ( ! nativetype ) return ptype ;
}
Error ( type , " Unable to resolve %s%s as type. " , nativetype ? " @ " : " " , FName ( type - > UserType - > Id ) . GetChars ( ) ) ;
return TypeError ;
}
//==========================================================================
//
// ZCCCompiler :: ResolveArraySize
//
// resolves the array size and returns a matching type.
//
//==========================================================================
PType * ZCCCompiler : : ResolveArraySize ( PType * baseType , ZCC_Expression * arraysize , PContainerType * cls , bool * nosize )
{
TArray < ZCC_Expression * > indices ;
// Simplify is too broken to resolve this inside the ring list so unravel the list into an array before starting to simplify its components.
auto node = arraysize ;
do
{
indices . Push ( node ) ;
node = static_cast < ZCC_Expression * > ( node - > SiblingNext ) ;
} while ( node ! = arraysize ) ;
if ( indices . Size ( ) = = 1 & & indices [ 0 ] - > Operation = = PEX_Nil )
{
* nosize = true ;
return baseType ;
}
if ( mVersion > = MakeVersion ( 3 , 7 , 2 ) )
{
TArray < ZCC_Expression * > fixedIndices ;
for ( auto node : indices )
{
fixedIndices . Insert ( 0 , node ) ;
}
indices = std : : move ( fixedIndices ) ;
}
FCompileContext ctx ( OutNamespace , cls , false ) ;
for ( auto node : indices )
{
// There is no float->int casting here.
FxExpression * ex = ConvertNode ( node ) ;
ex = ex - > Resolve ( ctx ) ;
if ( ex = = nullptr ) return TypeError ;
if ( ! ex - > isConstant ( ) | | ! ex - > ValueType - > isInt ( ) )
{
Error ( arraysize , " Array index must be an integer constant " ) ;
return TypeError ;
}
int size = static_cast < FxConstant * > ( ex ) - > GetValue ( ) . GetInt ( ) ;
if ( size < 1 )
{
Error ( arraysize , " Array size must be positive " ) ;
return TypeError ;
}
baseType = NewArray ( baseType , size ) ;
}
* nosize = false ;
return baseType ;
}
//==========================================================================
//
2020-04-08 16:02:53 +00:00
// SetImplicitArgs
2020-04-07 22:19:49 +00:00
//
2020-04-08 16:02:53 +00:00
// Adds the parameters implied by the function flags.
2020-04-07 22:19:49 +00:00
//
//==========================================================================
2020-04-08 16:02:53 +00:00
void ZCCCompiler : : SetImplicitArgs ( TArray < PType * > * args , TArray < uint32_t > * argflags , TArray < FName > * argnames , PContainerType * cls , uint32_t funcflags , int useflags )
{
// Must be called before adding any other arguments.
assert ( args = = nullptr | | args - > Size ( ) = = 0 ) ;
assert ( argflags = = nullptr | | argflags - > Size ( ) = = 0 ) ;
if ( funcflags & VARF_Method )
{
// implied self pointer
if ( args ! = nullptr ) args - > Push ( NewPointer ( cls , ! ! ( funcflags & VARF_ReadOnly ) ) ) ;
if ( argflags ! = nullptr ) argflags - > Push ( VARF_Implicit | VARF_ReadOnly ) ;
if ( argnames ! = nullptr ) argnames - > Push ( NAME_self ) ;
}
}
2020-04-07 22:19:49 +00:00
2020-06-14 16:57:55 +00:00
void ZCCCompiler : : InitDefaults ( )
{
for ( auto c : Classes )
{
if ( c - > ClassType ( ) - > ParentClass )
{
auto ti = c - > ClassType ( ) ;
ti - > InitializeDefaults ( ) ;
}
}
}
2020-04-08 16:02:53 +00:00
//==========================================================================
//
//
//
//==========================================================================
2020-04-07 22:19:49 +00:00
void ZCCCompiler : : CompileFunction ( ZCC_StructWork * c , ZCC_FuncDeclarator * f , bool forclass )
{
TArray < PType * > rets ( 1 ) ;
TArray < PType * > args ;
TArray < uint32_t > argflags ;
TArray < TypedVMValue > argdefaults ;
TArray < FName > argnames ;
rets . Clear ( ) ;
args . Clear ( ) ;
argflags . Clear ( ) ;
bool hasdefault = false ;
// For the time being, let's not allow overloading. This may be reconsidered later but really just adds an unnecessary amount of complexity here.
if ( AddTreeNode ( f - > Name , f , & c - > TreeNodes , false ) )
{
auto t = f - > Type ;
if ( t ! = nullptr )
{
do
{
auto type = DetermineType ( c - > Type ( ) , f , f - > Name , t , false , false ) ;
if ( type - > isContainer ( ) & & type ! = TypeVector2 & & type ! = TypeVector3 )
{
// structs and classes only get passed by pointer.
type = NewPointer ( type ) ;
}
else if ( type - > isDynArray ( ) )
{
Error ( f , " The return type of a function cannot be a dynamic array " ) ;
break ;
}
// TBD: disallow certain types? For now, let everything pass that isn't an array.
rets . Push ( type ) ;
t = static_cast < decltype ( t ) > ( t - > SiblingNext ) ;
} while ( t ! = f - > Type ) ;
}
int notallowed = ZCC_Latent | ZCC_Meta | ZCC_ReadOnly | ZCC_Abstract | ZCC_Internal ;
if ( f - > Flags & notallowed )
{
Error ( f , " Invalid qualifiers for %s (%s not allowed) " , FName ( f - > Name ) . GetChars ( ) , FlagsToString ( f - > Flags & notallowed ) . GetChars ( ) ) ;
f - > Flags & = notallowed ;
}
uint32_t varflags = VARF_Method ;
int implicitargs = 1 ;
AFuncDesc * afd = nullptr ;
int useflags = SUF_ACTOR | SUF_OVERLAY | SUF_WEAPON | SUF_ITEM ;
if ( f - > UseFlags ! = nullptr )
{
useflags = 0 ;
auto p = f - > UseFlags ;
do
{
switch ( p - > Id )
{
case NAME_Actor :
useflags | = SUF_ACTOR ;
break ;
case NAME_Overlay :
useflags | = SUF_OVERLAY ;
break ;
case NAME_Weapon :
useflags | = SUF_WEAPON ;
break ;
case NAME_Item :
useflags | = SUF_ITEM ;
break ;
default :
Error ( p , " Unknown Action qualifier %s " , FName ( p - > Id ) . GetChars ( ) ) ;
break ;
}
p = static_cast < decltype ( p ) > ( p - > SiblingNext ) ;
} while ( p ! = f - > UseFlags ) ;
}
// map to implementation flags.
if ( f - > Flags & ZCC_Private ) varflags | = VARF_Private ;
if ( f - > Flags & ZCC_Protected ) varflags | = VARF_Protected ;
if ( f - > Flags & ZCC_Deprecated ) varflags | = VARF_Deprecated ;
if ( f - > Flags & ZCC_Virtual ) varflags | = VARF_Virtual ;
if ( f - > Flags & ZCC_Override ) varflags | = VARF_Override ;
if ( f - > Flags & ZCC_VarArg ) varflags | = VARF_VarArg ;
if ( f - > Flags & ZCC_FuncConst ) varflags | = VARF_ReadOnly ; // FuncConst method is internally marked as VARF_ReadOnly
if ( mVersion > = MakeVersion ( 2 , 4 , 0 ) )
{
if ( c - > Type ( ) - > ScopeFlags & Scope_UI )
varflags | = VARF_UI ;
if ( c - > Type ( ) - > ScopeFlags & Scope_Play )
varflags | = VARF_Play ;
//if (f->Flags & ZCC_FuncConst)
// varflags = FScopeBarrier::ChangeSideInFlags(varflags, FScopeBarrier::Side_PlainData); // const implies clearscope. this is checked a bit later to also not have ZCC_Play/ZCC_UIFlag.
if ( f - > Flags & ZCC_UIFlag )
varflags = FScopeBarrier : : ChangeSideInFlags ( varflags , FScopeBarrier : : Side_UI ) ;
if ( f - > Flags & ZCC_Play )
varflags = FScopeBarrier : : ChangeSideInFlags ( varflags , FScopeBarrier : : Side_Play ) ;
if ( f - > Flags & ZCC_ClearScope )
varflags = FScopeBarrier : : ChangeSideInFlags ( varflags , FScopeBarrier : : Side_Clear ) ;
if ( f - > Flags & ZCC_VirtualScope )
varflags = FScopeBarrier : : ChangeSideInFlags ( varflags , FScopeBarrier : : Side_Virtual ) ;
}
else
{
varflags | = VARF_Play ;
}
if ( ( f - > Flags & ZCC_VarArg ) & & ! ( f - > Flags & ZCC_Native ) )
{
Error ( f , " 'VarArg' can only be used with native methods " ) ;
}
if ( f - > Flags & ZCC_Action )
{
2020-04-08 16:02:53 +00:00
implicitargs = CheckActionKeyword ( f , varflags , useflags , c ) ;
if ( implicitargs < 0 )
2020-04-07 22:19:49 +00:00
{
2020-04-08 16:02:53 +00:00
Error ( f , " 'Action' not allowed as a function qualifier " ) ;
// Set state to allow continued compilation to find more errors.
varflags & = ~ VARF_ReadOnly ;
implicitargs = 1 ;
2020-04-07 22:19:49 +00:00
}
}
if ( f - > Flags & ZCC_Static ) varflags = ( varflags & ~ VARF_Method ) | VARF_Final , implicitargs = 0 ; // Static implies Final.
if ( varflags & VARF_Override ) varflags & = ~ VARF_Virtual ; // allow 'virtual override'.
// Only one of these flags may be used.
static int exclude [ ] = { ZCC_Virtual , ZCC_Override , ZCC_Action , ZCC_Static } ;
int excludeflags = 0 ;
int fc = 0 ;
for ( size_t i = 0 ; i < countof ( exclude ) ; i + + )
{
if ( f - > Flags & exclude [ i ] )
{
fc + + ;
excludeflags | = exclude [ i ] ;
}
}
if ( fc > 1 )
{
Error ( f , " Invalid combination of qualifiers %s on function %s " , FlagsToString ( excludeflags ) . GetChars ( ) , FName ( f - > Name ) . GetChars ( ) ) ;
varflags | = VARF_Method ;
}
if ( varflags & VARF_Override ) varflags | = VARF_Virtual ; // Now that the flags are checked, make all override functions virtual as well.
// [ZZ] this doesn't make sense either.
if ( ( varflags & ( VARF_ReadOnly | VARF_Method ) ) = = VARF_ReadOnly ) // non-method const function
{
Error ( f , " 'Const' on a static method is not supported " ) ;
}
// [ZZ] neither this
if ( ( varflags & ( VARF_VirtualScope | VARF_Method ) ) = = VARF_VirtualScope ) // non-method virtualscope function
{
Error ( f , " 'VirtualScope' on a static method is not supported " ) ;
}
static int excludescope [ ] = { ZCC_UIFlag , ZCC_Play , ZCC_ClearScope , ZCC_VirtualScope } ;
excludeflags = 0 ;
fc = 0 ;
for ( size_t i = 0 ; i < countof ( excludescope ) ; i + + )
{
if ( f - > Flags & excludescope [ i ] )
{
fc + + ;
excludeflags | = excludescope [ i ] ;
}
}
if ( fc > 1 )
{
Error ( f , " Invalid combination of scope qualifiers %s on function %s " , FlagsToString ( excludeflags ) . GetChars ( ) , FName ( f - > Name ) . GetChars ( ) ) ;
varflags & = ~ ( VARF_UI | VARF_Play ) ; // make plain data
}
if ( f - > Flags & ZCC_Native )
{
varflags | = VARF_Native ;
afd = FindFunction ( c - > Type ( ) , FName ( f - > Name ) . GetChars ( ) ) ;
if ( afd = = nullptr )
{
Error ( f , " The function '%s.%s' has not been exported from the executable " , c - > Type ( ) - > TypeName . GetChars ( ) , FName ( f - > Name ) . GetChars ( ) ) ;
}
else
{
( * afd - > VMPointer ) - > ImplicitArgs = uint8_t ( implicitargs ) ;
}
}
SetImplicitArgs ( & args , & argflags , & argnames , c - > Type ( ) , varflags , useflags ) ;
argdefaults . Resize ( argnames . Size ( ) ) ;
auto p = f - > Params ;
bool hasoptionals = false ;
if ( p ! = nullptr )
{
bool overridemsg = false ;
do
{
int elementcount = 1 ;
TypedVMValue vmval [ 3 ] ; // default is REGT_NIL which means 'no default value' here.
if ( p - > Type ! = nullptr )
{
auto type = DetermineType ( c - > Type ( ) , p , f - > Name , p - > Type , false , false ) ;
int flags = 0 ;
if ( ( type - > isStruct ( ) & & type ! = TypeVector2 & & type ! = TypeVector3 ) | | type - > isDynArray ( ) )
{
// Structs are being passed by pointer, but unless marked 'out' that pointer must be readonly.
type = NewPointer ( type /*, !(p->Flags & ZCC_Out)*/ ) ;
flags | = VARF_Ref ;
}
else if ( type - > GetRegType ( ) ! = REGT_NIL )
{
if ( p - > Flags & ZCC_Out ) flags | = VARF_Out ;
if ( type = = TypeVector2 )
{
elementcount = 2 ;
}
else if ( type = = TypeVector3 )
{
elementcount = 3 ;
}
}
if ( type - > GetRegType ( ) = = REGT_NIL & & type ! = TypeVector2 & & type ! = TypeVector3 )
{
Error ( p , " Invalid type %s for function parameter " , type - > DescriptiveName ( ) ) ;
}
else if ( p - > Default ! = nullptr )
{
if ( flags & VARF_Out )
{
Error ( p , " Out parameters cannot have default values " ) ;
}
flags | = VARF_Optional ;
hasoptionals = true ;
if ( ( varflags & VARF_Override ) & & ! overridemsg )
{
// This is illegal, but in older compilers wasn't checked, so there it has to be demoted to a warning.
// Virtual calls always need to get their defaults from the base virtual method.
if ( mVersion > = MakeVersion ( 3 , 3 ) )
{
Error ( p , " Default values for parameter of virtual override not allowed " ) ;
}
else
{
Warn ( p , " Default values for parameter of virtual override will be ignored! " ) ;
}
overridemsg = true ;
}
FxExpression * x = new FxTypeCast ( ConvertNode ( p - > Default ) , type , false ) ;
FCompileContext ctx ( OutNamespace , c - > Type ( ) , false ) ;
x = x - > Resolve ( ctx ) ;
if ( x ! = nullptr )
{
// Vectors need special treatment because they use more than one entry in the Defaults and do not report as actual constants
if ( type = = TypeVector2 & & x - > ExprType = = EFX_VectorValue & & static_cast < FxVectorValue * > ( x ) - > isConstVector ( 2 ) )
{
auto vx = static_cast < FxVectorValue * > ( x ) ;
vmval [ 0 ] = static_cast < FxConstant * > ( vx - > xyz [ 0 ] ) - > GetValue ( ) . GetFloat ( ) ;
vmval [ 1 ] = static_cast < FxConstant * > ( vx - > xyz [ 1 ] ) - > GetValue ( ) . GetFloat ( ) ;
}
else if ( type = = TypeVector3 & & x - > ExprType = = EFX_VectorValue & & static_cast < FxVectorValue * > ( x ) - > isConstVector ( 3 ) )
{
auto vx = static_cast < FxVectorValue * > ( x ) ;
vmval [ 0 ] = static_cast < FxConstant * > ( vx - > xyz [ 0 ] ) - > GetValue ( ) . GetFloat ( ) ;
vmval [ 1 ] = static_cast < FxConstant * > ( vx - > xyz [ 1 ] ) - > GetValue ( ) . GetFloat ( ) ;
vmval [ 2 ] = static_cast < FxConstant * > ( vx - > xyz [ 2 ] ) - > GetValue ( ) . GetFloat ( ) ;
}
else if ( ! x - > isConstant ( ) )
{
Error ( p , " Default parameter %s is not constant in %s " , FName ( p - > Name ) . GetChars ( ) , FName ( f - > Name ) . GetChars ( ) ) ;
}
else if ( x - > ValueType ! = type )
{
Error ( p , " Default parameter %s could not be converted to target type %s " , FName ( p - > Name ) . GetChars ( ) , c - > Type ( ) - > TypeName . GetChars ( ) ) ;
}
else
{
auto cnst = static_cast < FxConstant * > ( x ) ;
switch ( type - > GetRegType ( ) )
{
case REGT_INT :
vmval [ 0 ] = cnst - > GetValue ( ) . GetInt ( ) ;
break ;
case REGT_FLOAT :
vmval [ 0 ] = cnst - > GetValue ( ) . GetFloat ( ) ;
break ;
case REGT_POINTER :
if ( type - > isClassPointer ( ) )
vmval [ 0 ] = ( DObject * ) cnst - > GetValue ( ) . GetPointer ( ) ;
else
vmval [ 0 ] = cnst - > GetValue ( ) . GetPointer ( ) ;
break ;
case REGT_STRING :
// We need a reference to something permanently stored here.
vmval [ 0 ] = VMStringConstants . Alloc ( cnst - > GetValue ( ) . GetString ( ) ) ;
break ;
default :
assert ( 0 & & " no valid type for constant " ) ;
break ;
}
}
hasdefault = true ;
}
if ( x ! = nullptr ) delete x ;
}
else if ( hasoptionals )
{
Error ( p , " All arguments after the first optional one need also be optional " ) ;
}
// TBD: disallow certain types? For now, let everything pass that isn't an array.
args . Push ( type ) ;
argflags . Push ( flags ) ;
argnames . Push ( p - > Name ) ;
}
else
{
args . Push ( nullptr ) ;
argflags . Push ( 0 ) ;
argnames . Push ( NAME_None ) ;
}
for ( int i = 0 ; i < elementcount ; i + + ) argdefaults . Push ( vmval [ i ] ) ;
p = static_cast < decltype ( p ) > ( p - > SiblingNext ) ;
} while ( p ! = f - > Params ) ;
}
PFunction * sym = Create < PFunction > ( c - > Type ( ) , f - > Name ) ;
sym - > AddVariant ( NewPrototype ( rets , args ) , argflags , argnames , afd = = nullptr ? nullptr : * ( afd - > VMPointer ) , varflags , useflags ) ;
c - > Type ( ) - > Symbols . ReplaceSymbol ( sym ) ;
if ( f - > DeprecationMessage ! = nullptr )
{
sym - > Variants [ 0 ] . DeprecationMessage = * f - > DeprecationMessage ;
}
auto vcls = PType : : toClass ( c - > Type ( ) ) ;
auto cls = vcls ? vcls - > Descriptor : nullptr ;
PFunction * virtsym = nullptr ;
if ( cls ! = nullptr & & cls - > ParentClass ! = nullptr ) virtsym = dyn_cast < PFunction > ( cls - > ParentClass - > VMType - > Symbols . FindSymbol ( FName ( f - > Name ) , true ) ) ;
unsigned vindex = ~ 0u ;
if ( virtsym ! = nullptr )
{
auto imp = virtsym - > Variants [ 0 ] . Implementation ;
if ( imp ! = nullptr ) vindex = imp - > VirtualIndex ;
else Error ( f , " Virtual base function %s not found in %s " , FName ( f - > Name ) . GetChars ( ) , cls - > ParentClass - > TypeName . GetChars ( ) ) ;
}
if ( f - > Flags & ( ZCC_Version | ZCC_Deprecated ) )
{
sym - > mVersion = f - > Version ;
if ( varflags & VARF_Override )
{
Error ( f , " Overridden function %s may not alter version restriction in %s " , FName ( f - > Name ) . GetChars ( ) , cls - > ParentClass - > TypeName . GetChars ( ) ) ;
}
}
VMFunction * newfunc = nullptr ;
if ( ! ( f - > Flags & ZCC_Native ) )
{
if ( f - > Body = = nullptr )
{
Error ( f , " Empty function %s " , FName ( f - > Name ) . GetChars ( ) ) ;
return ;
}
else
{
auto code = ConvertAST ( c - > Type ( ) , f - > Body ) ;
if ( code ! = nullptr )
{
newfunc = FunctionBuildList . AddFunction ( OutNamespace , mVersion , sym , code , FStringf ( " %s.%s " , c - > Type ( ) - > TypeName . GetChars ( ) , FName ( f - > Name ) . GetChars ( ) ) , false , - 1 , 0 , Lump ) ;
}
}
}
if ( sym - > Variants [ 0 ] . Implementation ! = nullptr & & hasdefault ) // do not copy empty default lists, they only waste space and processing time.
{
sym - > Variants [ 0 ] . Implementation - > DefaultArgs = std : : move ( argdefaults ) ;
}
if ( sym - > Variants [ 0 ] . Implementation ! = nullptr )
{
// [ZZ] unspecified virtual function inherits old scope. virtual function scope can't be changed.
sym - > Variants [ 0 ] . Implementation - > VarFlags = sym - > Variants [ 0 ] . Flags ;
}
2020-09-27 14:16:53 +00:00
bool exactReturnType = mVersion < MakeVersion ( 4 , 4 ) ;
2020-04-07 22:19:49 +00:00
PClass * clstype = forclass ? static_cast < PClassType * > ( c - > Type ( ) ) - > Descriptor : nullptr ;
if ( varflags & VARF_Virtual )
{
if ( sym - > Variants [ 0 ] . Implementation = = nullptr )
{
Error ( f , " Virtual function %s.%s not present " , c - > Type ( ) - > TypeName . GetChars ( ) , FName ( f - > Name ) . GetChars ( ) ) ;
return ;
}
if ( forclass )
{
auto parentfunc = clstype - > ParentClass ? dyn_cast < PFunction > ( clstype - > ParentClass - > VMType - > Symbols . FindSymbol ( sym - > SymbolName , true ) ) : nullptr ;
2020-09-27 14:16:53 +00:00
int vindex = clstype - > FindVirtualIndex ( sym - > SymbolName , & sym - > Variants [ 0 ] , parentfunc , exactReturnType ) ;
2020-04-07 22:19:49 +00:00
// specifying 'override' is necessary to prevent one of the biggest problem spots with virtual inheritance: Mismatching argument types.
if ( varflags & VARF_Override )
{
if ( vindex = = - 1 )
{
Error ( f , " Attempt to override non-existent virtual function %s " , FName ( f - > Name ) . GetChars ( ) ) ;
}
else
{
auto oldfunc = clstype - > Virtuals [ vindex ] ;
if ( parentfunc & & parentfunc - > mVersion > mVersion )
{
Error ( f , " Attempt to override function %s which is incompatible with version %d.%d.%d " , FName ( f - > Name ) . GetChars ( ) , mVersion . major , mVersion . minor , mVersion . revision ) ;
}
if ( oldfunc - > VarFlags & VARF_Final )
{
Error ( f , " Attempt to override final function %s " , FName ( f - > Name ) . GetChars ( ) ) ;
}
// you can't change ui/play/clearscope for a virtual method.
if ( f - > Flags & ( ZCC_UIFlag | ZCC_Play | ZCC_ClearScope | ZCC_VirtualScope ) )
{
Error ( f , " Attempt to change scope for virtual function %s " , FName ( f - > Name ) . GetChars ( ) ) ;
}
// you can't change const qualifier for a virtual method
if ( ( sym - > Variants [ 0 ] . Implementation - > VarFlags & VARF_ReadOnly ) & & ! ( oldfunc - > VarFlags & VARF_ReadOnly ) )
{
Error ( f , " Attempt to add const qualifier to virtual function %s " , FName ( f - > Name ) . GetChars ( ) ) ;
}
// you can't change protected qualifier for a virtual method (i.e. putting private), because this cannot be reliably checked without runtime stuff
if ( f - > Flags & ( ZCC_Private | ZCC_Protected ) )
{
Error ( f , " Attempt to change private/protected qualifiers for virtual function %s " , FName ( f - > Name ) . GetChars ( ) ) ;
}
// inherit scope of original function if override not specified
sym - > Variants [ 0 ] . Flags = FScopeBarrier : : ChangeSideInFlags ( sym - > Variants [ 0 ] . Flags , FScopeBarrier : : SideFromFlags ( oldfunc - > VarFlags ) ) ;
// inherit const from original function
if ( oldfunc - > VarFlags & VARF_ReadOnly )
sym - > Variants [ 0 ] . Flags | = VARF_ReadOnly ;
if ( oldfunc - > VarFlags & VARF_Protected )
sym - > Variants [ 0 ] . Flags | = VARF_Protected ;
clstype - > Virtuals [ vindex ] = sym - > Variants [ 0 ] . Implementation ;
sym - > Variants [ 0 ] . Implementation - > VirtualIndex = vindex ;
sym - > Variants [ 0 ] . Implementation - > VarFlags = sym - > Variants [ 0 ] . Flags ;
// Defaults must be identical to parent class
if ( parentfunc - > Variants [ 0 ] . Implementation - > DefaultArgs . Size ( ) > 0 )
{
sym - > Variants [ 0 ] . Implementation - > DefaultArgs = parentfunc - > Variants [ 0 ] . Implementation - > DefaultArgs ;
sym - > Variants [ 0 ] . ArgFlags = parentfunc - > Variants [ 0 ] . ArgFlags ;
}
// Update argument flags for VM function if needed as their list could be incomplete
// At the moment of function creation, arguments with default values were not copied yet from the parent function
if ( newfunc ! = nullptr & & sym - > Variants [ 0 ] . ArgFlags . Size ( ) ! = newfunc - > ArgFlags . Size ( ) )
{
for ( unsigned i = newfunc - > ArgFlags . Size ( ) , count = sym - > Variants [ 0 ] . ArgFlags . Size ( ) ; i < count ; + + i )
{
newfunc - > ArgFlags . Push ( sym - > Variants [ 0 ] . ArgFlags [ i ] ) ;
}
}
}
}
else
{
if ( vindex ! = - 1 )
{
Error ( f , " Function %s attempts to override parent function without 'override' qualifier " , FName ( f - > Name ) . GetChars ( ) ) ;
}
sym - > Variants [ 0 ] . Implementation - > VirtualIndex = clstype - > Virtuals . Push ( sym - > Variants [ 0 ] . Implementation ) ;
}
}
else
{
Error ( p , " Virtual functions can only be defined for classes " ) ;
}
}
else if ( forclass )
{
2020-09-27 14:16:53 +00:00
int vindex = clstype - > FindVirtualIndex ( sym - > SymbolName , & sym - > Variants [ 0 ] , nullptr , exactReturnType ) ;
2020-04-07 22:19:49 +00:00
if ( vindex ! = - 1 )
{
Error ( f , " Function %s attempts to override parent function without 'override' qualifier " , FName ( f - > Name ) . GetChars ( ) ) ;
}
}
}
}
//==========================================================================
//
// Parses the functions list
//
//==========================================================================
void ZCCCompiler : : InitFunctions ( )
{
for ( auto s : Structs )
{
for ( auto f : s - > Functions )
{
CompileFunction ( s , f , false ) ;
}
}
for ( auto c : Classes )
{
// cannot be done earlier because it requires the parent class to be processed by this code, too.
if ( c - > ClassType ( ) - > ParentClass ! = nullptr )
{
c - > ClassType ( ) - > Virtuals = c - > ClassType ( ) - > ParentClass - > Virtuals ;
}
for ( auto f : c - > Functions )
{
CompileFunction ( c , f , true ) ;
}
}
}
//==========================================================================
//
// Convert the AST data for the code generator.
//
//==========================================================================
FxExpression * ZCCCompiler : : ConvertAST ( PContainerType * cls , ZCC_TreeNode * ast )
{
ConvertClass = cls ;
// there are two possibilities here: either a single function call or a compound statement. For a compound statement we also need to check if the last thing added was a return.
if ( ast - > NodeType = = AST_ExprFuncCall )
{
auto cp = new FxCompoundStatement ( * ast ) ;
cp - > Add ( new FxReturnStatement ( ConvertNode ( ast ) , * ast ) ) ;
return cp ;
}
else
{
// This must be done here so that we can check for a trailing return statement.
auto x = new FxCompoundStatement ( * ast ) ;
auto compound = static_cast < ZCC_CompoundStmt * > ( ast ) ;
//bool isreturn = false;
auto node = compound - > Content ;
if ( node ! = nullptr ) do
{
x - > Add ( ConvertNode ( node ) ) ;
//isreturn = node->NodeType == AST_ReturnStmt;
node = static_cast < decltype ( node ) > ( node - > SiblingNext ) ;
} while ( node ! = compound - > Content ) ;
//if (!isreturn) x->Add(new FxReturnStatement(nullptr, *ast));
return x ;
}
}
# define xx(a,z) z,
static int Pex2Tok [ ] = {
# include "zcc_exprlist.h"
} ;
//==========================================================================
//
// Helper for modify/assign operators
//
//==========================================================================
static FxExpression * ModifyAssign ( FxBinary * operation , FxExpression * left )
{
auto assignself = static_cast < FxAssignSelf * > ( operation - > left ) ;
auto assignment = new FxAssign ( left , operation , true ) ;
assignself - > Assignment = assignment ;
return assignment ;
}
//==========================================================================
//
// Convert an AST node and its children
//
//==========================================================================
FxExpression * ZCCCompiler : : ConvertNode ( ZCC_TreeNode * ast , bool substitute )
{
if ( ast = = nullptr ) return nullptr ;
switch ( ast - > NodeType )
{
case AST_ExprFuncCall :
{
auto fcall = static_cast < ZCC_ExprFuncCall * > ( ast ) ;
// function names can either be
// - plain identifiers
// - class members
// - array syntax for random() calls.
// Everything else coming here is a syntax error.
FArgumentList args ;
switch ( fcall - > Function - > NodeType )
{
case AST_ExprID :
// The function name is a simple identifier.
return new FxFunctionCall ( static_cast < ZCC_ExprID * > ( fcall - > Function ) - > Identifier , NAME_None , ConvertNodeList ( args , fcall - > Parameters ) , * ast ) ;
case AST_ExprMemberAccess :
{
auto ema = static_cast < ZCC_ExprMemberAccess * > ( fcall - > Function ) ;
return new FxMemberFunctionCall ( ConvertNode ( ema - > Left , true ) , ema - > Right , ConvertNodeList ( args , fcall - > Parameters ) , * ast ) ;
}
case AST_ExprBinary :
// Array syntax for randoms. They are internally stored as ExprBinary with both an identifier on the left and right side.
if ( fcall - > Function - > Operation = = PEX_ArrayAccess )
{
auto binary = static_cast < ZCC_ExprBinary * > ( fcall - > Function ) ;
if ( binary - > Left - > NodeType = = AST_ExprID & & binary - > Right - > NodeType = = AST_ExprID )
{
return new FxFunctionCall ( static_cast < ZCC_ExprID * > ( binary - > Left ) - > Identifier , static_cast < ZCC_ExprID * > ( binary - > Right ) - > Identifier , ConvertNodeList ( args , fcall - > Parameters ) , * ast ) ;
}
}
// fall through if this isn't an array access node.
default :
Error ( fcall , " Invalid function identifier " ) ;
return new FxNop ( * ast ) ; // return something so that the compiler can continue.
}
break ;
}
case AST_ClassCast :
{
auto cc = static_cast < ZCC_ClassCast * > ( ast ) ;
if ( cc - > Parameters = = nullptr | | cc - > Parameters - > SiblingNext ! = cc - > Parameters )
{
Error ( cc , " Class type cast requires exactly one parameter " ) ;
return new FxNop ( * ast ) ; // return something so that the compiler can continue.
}
auto cls = PClass : : FindClass ( cc - > ClassName ) ;
if ( cls = = nullptr | | cls - > VMType = = nullptr )
{
Error ( cc , " Unknown class %s " , FName ( cc - > ClassName ) . GetChars ( ) ) ;
return new FxNop ( * ast ) ; // return something so that the compiler can continue.
}
return new FxClassPtrCast ( cls , ConvertNode ( cc - > Parameters ) ) ;
}
case AST_StaticArrayStatement :
{
auto sas = static_cast < ZCC_StaticArrayStatement * > ( ast ) ;
PType * ztype = DetermineType ( ConvertClass , sas , sas - > Id , sas - > Type , false , false ) ;
FArgumentList args ;
ConvertNodeList ( args , sas - > Values ) ;
// This has to let the code generator resolve the constants, not the Simplifier, which lacks all the necessary type info.
return new FxStaticArray ( ztype , sas - > Id , args , * ast ) ;
}
case AST_ExprMemberAccess :
{
auto memaccess = static_cast < ZCC_ExprMemberAccess * > ( ast ) ;
return new FxMemberIdentifier ( ConvertNode ( memaccess - > Left ) , memaccess - > Right , * ast ) ;
}
case AST_FuncParm :
{
auto fparm = static_cast < ZCC_FuncParm * > ( ast ) ;
auto node = ConvertNode ( fparm - > Value ) ;
if ( fparm - > Label ! = NAME_None ) node = new FxNamedNode ( fparm - > Label , node , * ast ) ;
return node ;
}
case AST_ExprID :
{
auto id = static_cast < ZCC_ExprID * > ( ast ) - > Identifier ;
if ( id = = NAME_LevelLocals & & substitute ) id = NAME_Level ; // All static methods of FLevelLocals are now non-static so remap the name right here before passing it to the backend.
return new FxIdentifier ( id , * ast ) ;
}
case AST_ExprConstant :
{
auto cnst = static_cast < ZCC_ExprConstant * > ( ast ) ;
if ( cnst - > Type = = TypeName )
{
return new FxConstant ( FName ( ENamedName ( cnst - > IntVal ) ) , * ast ) ;
}
else if ( cnst - > Type - > isInt ( ) )
{
return new FxConstant ( cnst - > IntVal , * ast ) ;
}
else if ( cnst - > Type = = TypeBool )
{
return new FxConstant ( ! ! cnst - > IntVal , * ast ) ;
}
else if ( cnst - > Type - > isFloat ( ) )
{
return new FxConstant ( cnst - > DoubleVal , * ast ) ;
}
else if ( cnst - > Type = = TypeString )
{
return new FxConstant ( * cnst - > StringVal , * ast ) ;
}
else if ( cnst - > Type = = TypeNullPtr )
{
return new FxConstant ( * ast ) ;
}
else
{
// can there be other types?
Error ( cnst , " Unknown constant type %s " , cnst - > Type - > DescriptiveName ( ) ) ;
return new FxConstant ( 0 , * ast ) ;
}
}
case AST_ExprUnary :
{
auto unary = static_cast < ZCC_ExprUnary * > ( ast ) ;
auto operand = ConvertNode ( unary - > Operand ) ;
auto op = unary - > Operation ;
switch ( op )
{
case PEX_PostDec :
case PEX_PostInc :
return new FxPostIncrDecr ( operand , Pex2Tok [ op ] ) ;
case PEX_PreDec :
case PEX_PreInc :
return new FxPreIncrDecr ( operand , Pex2Tok [ op ] ) ;
case PEX_Negate :
return new FxMinusSign ( operand ) ;
case PEX_AntiNegate :
return new FxPlusSign ( operand ) ;
case PEX_BitNot :
return new FxUnaryNotBitwise ( operand ) ;
case PEX_BoolNot :
return new FxUnaryNotBoolean ( operand ) ;
case PEX_SizeOf :
case PEX_AlignOf :
return new FxSizeAlign ( operand , Pex2Tok [ op ] ) ;
default :
assert ( 0 & & " Unknown unary operator. " ) ; // should never happen
Error ( unary , " Unknown unary operator ID #%d " , op ) ;
return new FxNop ( * ast ) ;
}
break ;
}
case AST_ExprBinary :
{
auto binary = static_cast < ZCC_ExprBinary * > ( ast ) ;
auto left = ConvertNode ( binary - > Left ) ;
auto right = ConvertNode ( binary - > Right ) ;
auto op = binary - > Operation ;
auto tok = Pex2Tok [ op ] ;
switch ( op )
{
case PEX_Add :
case PEX_Sub :
return new FxAddSub ( tok , left , right ) ;
case PEX_Mul :
case PEX_Div :
case PEX_Mod :
return new FxMulDiv ( tok , left , right ) ;
case PEX_Pow :
return new FxPow ( left , right ) ;
case PEX_LeftShift :
case PEX_RightShift :
case PEX_URightShift :
return new FxShift ( tok , left , right ) ;
case PEX_BitAnd :
case PEX_BitOr :
case PEX_BitXor :
return new FxBitOp ( tok , left , right ) ;
case PEX_BoolOr :
case PEX_BoolAnd :
return new FxBinaryLogical ( tok , left , right ) ;
case PEX_LT :
case PEX_LTEQ :
case PEX_GT :
case PEX_GTEQ :
return new FxCompareRel ( tok , left , right ) ;
case PEX_EQEQ :
case PEX_NEQ :
case PEX_APREQ :
return new FxCompareEq ( tok , left , right ) ;
case PEX_Assign :
return new FxAssign ( left , right ) ;
case PEX_AddAssign :
case PEX_SubAssign :
return ModifyAssign ( new FxAddSub ( tok , new FxAssignSelf ( * ast ) , right ) , left ) ;
case PEX_MulAssign :
case PEX_DivAssign :
case PEX_ModAssign :
return ModifyAssign ( new FxMulDiv ( tok , new FxAssignSelf ( * ast ) , right ) , left ) ;
case PEX_LshAssign :
case PEX_RshAssign :
case PEX_URshAssign :
return ModifyAssign ( new FxShift ( tok , new FxAssignSelf ( * ast ) , right ) , left ) ;
case PEX_AndAssign :
case PEX_OrAssign :
case PEX_XorAssign :
return ModifyAssign ( new FxBitOp ( tok , new FxAssignSelf ( * ast ) , right ) , left ) ;
case PEX_LTGTEQ :
return new FxLtGtEq ( left , right ) ;
case PEX_ArrayAccess :
return new FxArrayElement ( left , right ) ;
case PEX_CrossProduct :
case PEX_DotProduct :
return new FxDotCross ( tok , left , right ) ;
case PEX_Is :
return new FxTypeCheck ( left , right ) ;
case PEX_Concat :
return new FxConcat ( left , right ) ;
default :
I_Error ( " Binary operator %d not implemented yet " , op ) ;
}
break ;
}
case AST_ExprTrinary :
{
auto trinary = static_cast < ZCC_ExprTrinary * > ( ast ) ;
auto condition = ConvertNode ( trinary - > Test ) ;
auto left = ConvertNode ( trinary - > Left ) ;
auto right = ConvertNode ( trinary - > Right ) ;
return new FxConditional ( condition , left , right ) ;
}
case AST_VectorValue :
{
auto vecini = static_cast < ZCC_VectorValue * > ( ast ) ;
auto xx = ConvertNode ( vecini - > X ) ;
auto yy = ConvertNode ( vecini - > Y ) ;
auto zz = ConvertNode ( vecini - > Z ) ;
return new FxVectorValue ( xx , yy , zz , * ast ) ;
}
case AST_LocalVarStmt :
{
auto loc = static_cast < ZCC_LocalVarStmt * > ( ast ) ;
auto node = loc - > Vars ;
FxSequence * list = new FxSequence ( * ast ) ;
PType * ztype = DetermineType ( ConvertClass , node , node - > Name , loc - > Type , true , false ) ;
if ( loc - > Type - > ArraySize ! = nullptr )
{
bool nosize ;
ztype = ResolveArraySize ( ztype , loc - > Type - > ArraySize , ConvertClass , & nosize ) ;
if ( nosize )
{
Error ( node , " Must specify array size " ) ;
}
}
do
{
PType * type ;
bool nosize = false ;
if ( node - > ArraySize ! = nullptr )
{
type = ResolveArraySize ( ztype , node - > ArraySize , ConvertClass , & nosize ) ;
if ( nosize & & ! node - > InitIsArray )
{
Error ( node , " Must specify array size for non-initialized arrays " ) ;
}
}
else
{
type = ztype ;
}
if ( node - > InitIsArray & & ( type - > isArray ( ) | | type - > isDynArray ( ) | | nosize ) )
{
auto arrtype = static_cast < PArray * > ( type ) ;
if ( ! nosize & & ( arrtype - > ElementType - > isArray ( ) | | arrtype - > ElementType - > isDynArray ( ) ) )
{
Error ( node , " Compound initializer not implemented yet for multi-dimensional arrays " ) ;
}
FArgumentList args ;
ConvertNodeList ( args , node - > Init ) ;
if ( nosize )
{
type = NewArray ( type , args . Size ( ) ) ;
}
list - > Add ( new FxLocalArrayDeclaration ( type , node - > Name , args , 0 , * node ) ) ;
}
else
{
FxExpression * val = node - > Init ? ConvertNode ( node - > Init ) : nullptr ;
list - > Add ( new FxLocalVariableDeclaration ( type , node - > Name , val , 0 , * node ) ) ; // todo: Handle flags in the grammar.
}
node = static_cast < decltype ( node ) > ( node - > SiblingNext ) ;
} while ( node ! = loc - > Vars ) ;
return list ;
}
case AST_Expression :
{
auto ret = static_cast < ZCC_Expression * > ( ast ) ;
if ( ret - > Operation = = PEX_Super )
{
return new FxSuper ( * ast ) ;
}
break ;
}
case AST_ExpressionStmt :
return ConvertNode ( static_cast < ZCC_ExpressionStmt * > ( ast ) - > Expression ) ;
case AST_ReturnStmt :
{
auto ret = static_cast < ZCC_ReturnStmt * > ( ast ) ;
FArgumentList args ;
ConvertNodeList ( args , ret - > Values ) ;
if ( args . Size ( ) = = 0 )
{
return new FxReturnStatement ( nullptr , * ast ) ;
}
else
{
return new FxReturnStatement ( args , * ast ) ;
}
}
case AST_BreakStmt :
case AST_ContinueStmt :
return new FxJumpStatement ( ast - > NodeType = = AST_BreakStmt ? TK_Break : TK_Continue , * ast ) ;
case AST_IfStmt :
{
auto iff = static_cast < ZCC_IfStmt * > ( ast ) ;
FxExpression * const truePath = ConvertImplicitScopeNode ( ast , iff - > TruePath ) ;
FxExpression * const falsePath = ConvertImplicitScopeNode ( ast , iff - > FalsePath ) ;
return new FxIfStatement ( ConvertNode ( iff - > Condition ) , truePath , falsePath , * ast ) ;
}
case AST_IterationStmt :
{
auto iter = static_cast < ZCC_IterationStmt * > ( ast ) ;
if ( iter - > CheckAt = = ZCC_IterationStmt : : End )
{
assert ( iter - > LoopBumper = = nullptr ) ;
FxExpression * const loop = ConvertImplicitScopeNode ( ast , iter - > LoopStatement ) ;
return new FxDoWhileLoop ( ConvertNode ( iter - > LoopCondition ) , loop , * ast ) ;
}
else if ( iter - > LoopBumper ! = nullptr )
{
FArgumentList bumper ;
ConvertNodeList ( bumper , iter - > LoopBumper ) ;
FxCompoundStatement * bumps = new FxCompoundStatement ( * ast ) ;
for ( auto & ex : bumper )
{
bumps - > Add ( ex ) ;
ex = nullptr ;
}
return new FxForLoop ( nullptr , ConvertNode ( iter - > LoopCondition ) , bumps , ConvertNode ( iter - > LoopStatement ) , * ast ) ;
}
else
{
FxExpression * const loop = ConvertImplicitScopeNode ( ast , iter - > LoopStatement ) ;
return new FxWhileLoop ( ConvertNode ( iter - > LoopCondition ) , loop , * ast ) ;
}
}
// not yet done
case AST_SwitchStmt :
{
auto swtch = static_cast < ZCC_SwitchStmt * > ( ast ) ;
if ( swtch - > Content - > NodeType ! = AST_CompoundStmt )
{
Error ( ast , " Expecting { after 'switch' " ) ;
return new FxNop ( * ast ) ; // allow compiler to continue looking for errors.
}
else
{
// The switch content is wrapped into a compound statement which needs to be unraveled here.
auto cmpnd = static_cast < ZCC_CompoundStmt * > ( swtch - > Content ) ;
FArgumentList args ;
return new FxSwitchStatement ( ConvertNode ( swtch - > Condition ) , ConvertNodeList ( args , cmpnd - > Content ) , * ast ) ;
}
}
case AST_CaseStmt :
{
auto cases = static_cast < ZCC_CaseStmt * > ( ast ) ;
return new FxCaseStatement ( ConvertNode ( cases - > Condition ) , * ast ) ;
}
case AST_CompoundStmt :
{
auto x = new FxCompoundStatement ( * ast ) ;
auto compound = static_cast < ZCC_CompoundStmt * > ( ast ) ;
auto node = compound - > Content ;
if ( node ! = nullptr ) do
{
x - > Add ( ConvertNode ( node ) ) ;
node = static_cast < decltype ( node ) > ( node - > SiblingNext ) ;
} while ( node ! = compound - > Content ) ;
return x ;
}
case AST_AssignStmt :
{
auto ass = static_cast < ZCC_AssignStmt * > ( ast ) ;
FArgumentList args ;
ConvertNodeList ( args , ass - > Dests ) ;
assert ( ass - > Sources - > SiblingNext = = ass - > Sources ) ; // right side should be a single function call - nothing else
if ( ass - > Sources - > NodeType ! = AST_ExprFuncCall )
{
// don't let this through to the code generator. This node is only used to assign multiple returns of a function to more than one variable.
Error ( ass , " Right side of multi-assignment must be a function call " ) ;
return new FxNop ( * ast ) ; // allow compiler to continue looking for errors.
}
return new FxMultiAssign ( args , ConvertNode ( ass - > Sources ) , * ast ) ;
}
default :
break ;
}
// only for development. I_Error is more convenient here than a normal error.
I_Error ( " ConvertNode encountered unsupported node of type %d " , ast - > NodeType ) ;
return nullptr ;
}
//==========================================================================
//
// Wrapper around ConvertNode() that adds a scope (a compound statement)
// when needed to avoid leaking of variable or contant to an outer scope:
//
// if (true) int i; else bool b[1];
// while (false) readonly<Actor> a;
// do static const float f[] = {0}; while (false);
//
// Accessing such variables outside of their statements is now an error
//
//==========================================================================
FxExpression * ZCCCompiler : : ConvertImplicitScopeNode ( ZCC_TreeNode * node , ZCC_Statement * nested )
{
assert ( nullptr ! = node ) ;
if ( nullptr = = nested )
{
return nullptr ;
}
FxExpression * nestedExpr = ConvertNode ( nested ) ;
assert ( nullptr ! = nestedExpr ) ;
const EZCCTreeNodeType nestedType = nested - > NodeType ;
const bool needScope = AST_LocalVarStmt = = nestedType | | AST_StaticArrayStatement = = nestedType ;
if ( needScope )
{
FxCompoundStatement * implicitCompound = new FxCompoundStatement ( * node ) ;
implicitCompound - > Add ( nestedExpr ) ;
nestedExpr = implicitCompound ;
}
return nestedExpr ;
}
FArgumentList & ZCCCompiler : : ConvertNodeList ( FArgumentList & args , ZCC_TreeNode * head )
{
if ( head ! = nullptr )
{
auto node = head ;
do
{
args . Push ( ConvertNode ( node ) ) ;
node = node - > SiblingNext ;
} while ( node ! = head ) ;
}
return args ;
}