类初始化器

Author: Alumopper

src/main/antlr/mcfppParser.g4

@@ -367,7 +367,11 @@ var
 
 identifierSuffix
     :   '[' conditionalExpression ']'
-    |   '[' Identifier '=' expression  (',' basicExpression '=' expression)* ']'
+    |   '[' objectInitializer  (',' objectInitializer)* ']'
+    ;
+
+objectInitializer
+    :   Identifier '=' expression
     ;
 
 selector

src/main/kotlin/top/mcfpp/antlr/McfppExprVisitor.kt

@@ -9,9 +9,12 @@ import top.mcfpp.lang.type.MCFPPEnumType
 import top.mcfpp.lang.type.MCFPPGenericClassType
 import top.mcfpp.lang.type.MCFPPType
 import top.mcfpp.lang.value.MCFPPValue
-import top.mcfpp.model.*
-import top.mcfpp.model.function.Function
+import top.mcfpp.model.CanSelectMember
+import top.mcfpp.model.Class
+import top.mcfpp.model.CompoundDataCompanion
+import top.mcfpp.model.Namespace
 import top.mcfpp.model.field.GlobalField
+import top.mcfpp.model.function.Function
 import top.mcfpp.model.function.FunctionParam
 import top.mcfpp.model.function.NoStackFunction
 import top.mcfpp.model.function.UnknownFunction
@@ -20,11 +23,7 @@ import top.mcfpp.model.generic.GenericClass
 import top.mcfpp.util.LogProcessor
 import top.mcfpp.util.StringHelper
 import top.mcfpp.util.Utils
-import java.lang.StringBuilder
 import java.util.*
-import kotlin.IllegalArgumentException
-import kotlin.collections.ArrayList
-import kotlin.collections.HashMap
 import kotlin.system.exitProcess
 
 /**
@@ -535,19 +534,32 @@ class McfppExprVisitor(private var defaultGenericClassType : MCFPPGenericClassTy
             if (ctx.identifierSuffix() == null || ctx.identifierSuffix().size == 0) {
                 return re
             } else {
-                if(re is Indexable<*>){
-                    for (value in ctx.identifierSuffix()) {
-                        if(value.conditionalExpression() != null){
-                            //索引
-                            val index = visit(value.conditionalExpression())!!
-                            re = (re as Indexable<*>).getByIndex(index)
-                        }else{
-                            //初始化
-                            TODO()
+                for (value in ctx.identifierSuffix()) {
+                    if(value.conditionalExpression() != null){
+                        if(re !is Indexable<*>){
+                            LogProcessor.error("Cannot index ${re.type}")
+                            return UnknownVar("${re.identifier}_index_" + UUID.randomUUID())
+                        }
+                        //索引
+                        val index = visit(value.conditionalExpression())!!
+                        re = (re as Indexable<*>).getByIndex(index)
+                    }else{
+                        if(!re.isTemp) re = re.getTempVar()
+                        //初始化
+                        for (initializer in value.objectInitializer()){
+                            val id = initializer.Identifier().text
+                            val v = visit(initializer.expression())
+                            val (m, b) = re.getMemberVar(id, re.getAccess(Function.currFunction))
+                            if(!b){
+                                LogProcessor.error("Cannot access member $id")
+                            }
+                            if(m == null) {
+                                LogProcessor.error("Member $id not found")
+                                continue
+                            }
+                            m.replacedBy(m.assign(v))
                         }
                     }
-                }else{
-                    throw IllegalArgumentException("Cannot index ${re.type}")
                 }
                 return re
             }

src/main/kotlin/top/mcfpp/lang/DataTemplateObject.kt

@@ -4,8 +4,8 @@ import net.querz.nbt.io.SNBTUtil
 import net.querz.nbt.tag.CompoundTag
 import top.mcfpp.command.Command
 import top.mcfpp.command.Commands
-import top.mcfpp.lang.type.MCFPPDataTemplateType
 import top.mcfpp.lang.type.MCFPPClassType
+import top.mcfpp.lang.type.MCFPPDataTemplateType
 import top.mcfpp.lang.type.MCFPPNBTType
 import top.mcfpp.lang.type.MCFPPType
 import top.mcfpp.lang.value.MCFPPValue
@@ -195,7 +195,7 @@ open class DataTemplateObject : Var<DataTemplateObject> {
         }
     }
 
-    override fun onMemberChanged(member: Member) {
+    override fun onMemberVarChanged(member: Var<*>) {
         if(member is MCFPPValue<*> && isConcrete()){
             this.replacedBy(this.toConcrete())
         }
@@ -264,6 +264,10 @@ class DataTemplateObjectConcrete: DataTemplateObject, MCFPPValue<CompoundTag>{
         return DataTemplateObject(this)
     }
 
+    override fun getTempVar(): DataTemplateObjectConcrete {
+        return DataTemplateObjectConcrete(super.getTempVar(), this.value.clone())
+    }
+
     override fun toDynamic(replace: Boolean): Var<*> {
         val parent = this.parent
 
@@ -292,15 +296,18 @@ class DataTemplateObjectConcrete: DataTemplateObject, MCFPPValue<CompoundTag>{
         return re
     }
 
-    override fun getMemberVar(key: String, accessModifier: Member.AccessModifier): Pair<Var<*>?, Boolean> {
-        val re = super.getMemberVar(key, accessModifier)
-        var qwq = re.first
-        if(re.first != null) {
-            //re是已知的
-            val data = value.get(key)
-             qwq = (re.first as Var<*>).assign(NBTBasedDataConcrete(data))
+    override fun toString(): String {
+        return "[$type,value=${SNBTUtil.toSNBT(value)}]"
+    }
+
+    override fun onMemberVarChanged(member: Var<*>) {
+        if(member !is MCFPPValue<*>){
+            toDynamic(true)
+        }else{
+            val key = member.identifier
+            val data = NBTUtil.toNBT(member)
+            value.put(key, data)
         }
-        return qwq to re.second
     }
 
 }

src/main/kotlin/top/mcfpp/lang/Var.kt

@@ -438,6 +438,7 @@ abstract class Var<Self: Var<Self>> : Member, Cloneable, CanSelectMember, Serial
         if(this.parent != other.parent) return false
         if(this.name != other.name) return false
         if(this is MCFPPValue<*> != other is MCFPPValue<*>) return false
+        if(this is MCFPPValue<*> && other is MCFPPValue<*> && this.value != other.value) return false
         return true
     }
 
@@ -455,7 +456,9 @@ abstract class Var<Self: Var<Self>> : Member, Cloneable, CanSelectMember, Serial
     fun replacedBy(v : Var<*>){
         if(v == this) return
         if(parent == null){
-            Function.currFunction.field.putVar(identifier, v, true)
+            if(Function.currFunction.field.containVar(identifier)){
+                Function.currFunction.field.putVar(identifier, v, true)
+            }
         }else{
             v.parent = this.parent
             when (val parent = parent){
@@ -478,7 +481,7 @@ abstract class Var<Self: Var<Self>> : Member, Cloneable, CanSelectMember, Serial
                 }
                 else -> {}
             }
-            parent?.onMemberChanged(v)
+            parent?.onMemberVarChanged(v)
         }
     }
 

src/main/kotlin/top/mcfpp/model/CanSelectMember.kt

@@ -33,5 +33,10 @@ interface CanSelectMember{
 
     fun getAccess(function: Function): Member.AccessModifier
 
-    fun onMemberChanged(member: Member) {}
+    /**
+     * @param member 更改之后的成员变量
+     *
+     * 当成员变量发生变化时，会调用此方法
+     */
+    fun onMemberVarChanged(member: Var<*>) {}
 }

test/test.mcfpp

@@ -9,5 +9,7 @@ data Test{
 
 func main(){
     Test t = {a:1, b:2, c:3, d:4};
+    t = t[a = 100];
+    var a = @a[tag="qwq", distance=1..2]
     print(t.toText());
 }