枚举的定义

Author: Alumopper

TODO_CN.md

@@ -19,6 +19,7 @@
 *  基本库
 *  枚举
 *  异常非中断编译
+*  函数返回值重载
 
 ## 已知问题
 * [ ] 递归调用的时候，错误地修改了本层函数的值

src/main/antlr/mcfppLexer.g4

@@ -103,6 +103,7 @@ OBJECT:'object';
 INTERFACE:'interface';
 DATA:'data';
 FUNCTION:'func';
+ENUM:'enum';
 
 GLOBAL:'global';
 VAR:'var';

src/main/antlr/mcfppParser.g4

@@ -73,6 +73,7 @@ declarations
     |   extensionFunctionDeclaration
     |   interfaceDeclaration
     |   globalDeclaration
+    |   enumDeclaration
     ;
 
 //全局声明
@@ -190,6 +191,20 @@ extensionFunctionDeclaration
     :   funcAnnoation? STATIC? FUNCTION (type '.')? Identifier functionParams (ARROW functionReturnType)? '{' functionBody '}'
     ;
 
+//枚举
+enumDeclaration
+    :   ENUM Identifier '{' enumBody '}'
+    ;
+
+enumBody
+    :   enumMember (',' enumMember)*
+    ;
+
+enumMember
+    :   Identifier ('=' intValue)?
+    ;
+
+
 namespaceID
     : (Identifier ( '.' Identifier)* ':')? Identifier
     ;

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

@@ -3,6 +3,7 @@ package top.mcfpp.antlr
 import top.mcfpp.Project
 import top.mcfpp.lang.MCAny
 import top.mcfpp.model.*
+import top.mcfpp.model.Enum
 import top.mcfpp.model.field.GlobalField
 import top.mcfpp.model.generic.ClassParam
 import top.mcfpp.model.generic.GenericClass
@@ -131,9 +132,9 @@ class McfppTypeVisitor: mcfppParserBaseVisitor<Unit>() {
         //注册类
         val id = ctx.classWithoutNamespace().text
         val nsp = GlobalField.localNamespaces[Project.currNamespace]!!
-        if (nsp.field.hasInterface(id)) {
+        if (nsp.field.hasDeclaredType(id)) {
             //重复声明
-            LogProcessor.error("Interface has been defined: $id in namespace ${Project.currNamespace}")
+            LogProcessor.error("Type has been defined: $id in namespace ${Project.currNamespace}")
             Interface.currInterface = nsp.field.getInterface(id)
         } else {
             //如果没有声明过这个类
@@ -180,8 +181,8 @@ class McfppTypeVisitor: mcfppParserBaseVisitor<Unit>() {
             //如果没有声明过这个类
             Class(id, Project.currNamespace)
         }
-        if(!nsp.field.addClass(id, cls)){
-            LogProcessor.error("Class has been defined: $id in namespace ${Project.currNamespace}")
+        if(!nsp.field.hasDeclaredType(id)){
+            LogProcessor.error("Type has been defined: $id in namespace ${Project.currNamespace}")
             return
         }
         cls.initialize()
@@ -211,14 +212,14 @@ class McfppTypeVisitor: mcfppParserBaseVisitor<Unit>() {
     /**
      *
      */
-    override fun visitTemplateDeclaration(ctx: mcfppParser.TemplateDeclarationContext?){
-        Project.ctx = ctx!!
+    override fun visitTemplateDeclaration(ctx: mcfppParser.TemplateDeclarationContext){
+        Project.ctx = ctx
         //注册模板
         val id = ctx.classWithoutNamespace().text
         val nsp = GlobalField.localNamespaces[Project.currNamespace]!!
-        if (nsp.field.hasTemplate(id)) {
+        if (nsp.field.hasDeclaredType(id)) {
             //重复声明
-            LogProcessor.error("Template has been defined: $id in namespace ${Project.currNamespace}")
+            LogProcessor.error("Type has been defined: $id in namespace ${Project.currNamespace}")
             DataTemplate.currTemplate = nsp.field.getTemplate(id)
         }
         val template = DataTemplate(id,Project.currNamespace)
@@ -246,4 +247,28 @@ class McfppTypeVisitor: mcfppParserBaseVisitor<Unit>() {
         }
         nsp.field.addTemplate(id, template)
     }
+
+    override fun visitEnumDeclaration(ctx: mcfppParser.EnumDeclarationContext) {
+        Project.ctx = ctx
+        //注册枚举
+        val id = ctx.Identifier().text
+        val nsp = GlobalField.localNamespaces[Project.currNamespace]!!
+        if (nsp.field.hasDeclaredType(id)) {
+            //重复声明
+            LogProcessor.error("Type has been defined: $id in namespace ${Project.currNamespace}")
+        }
+        val enum = Enum(id, Project.currNamespace)
+        nsp.field.addEnum(id, enum)
+        //添加成员
+        for (m in ctx.enumBody().enumMember()) {
+            val value = if(m.intValue() != null){
+                //获取enum的int值
+                m.intValue().text.toInt()
+            }else{
+                enum.getNextMemberValue()
+            }
+            val member = EnumMember(m.Identifier().text, value)
+            enum.addMember(member)
+        }
+    }
 }

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

@@ -0,0 +1,33 @@
+package top.mcfpp.model
+
+class Enum(var identifier: String, var namespace: String){
+
+    var members: HashMap<String, EnumMember> = HashMap()
+
+    fun addMember(member: EnumMember){
+        members[member.identifier] = member
+    }
+
+    fun addMember(identifier: String){
+        members[identifier] = EnumMember(identifier, getNextMemberValue())
+    }
+
+    fun getNextMemberValue(): Int{
+        val values = members.map { it.value.value }
+        var i = values.last() + 1
+        while (values.contains(i)){
+            i++
+        }
+        return i
+    }
+}
+
+class EnumMember{
+    var identifier: String
+    var value: Int
+
+    constructor(identifier: String, value: Int) {
+        this.identifier = identifier
+        this.value = value
+    }
+}

src/main/kotlin/top/mcfpp/model/field/IFieldWithEnum.kt

@@ -0,0 +1,49 @@
+package top.mcfpp.model.field
+
+import top.mcfpp.model.Enum
+
+interface IFieldWithEnum {
+    /**
+     * 向域中添加一个枚举
+     *
+     * @param identifier 枚举的标识符
+     * @param enum 枚举
+     * @param force 是否强制添加。如果为true，则即使已经添加过相同标识符的枚举，也会覆盖原来的枚举进行添加。
+     * @return 是否添加成功。如果已经存在相同标识符的枚举，且不是强制添加则为false
+     */
+    fun addEnum(identifier: String, enum: Enum, force: Boolean = false): Boolean
+
+    /**
+     * 移除一个枚举
+     *
+     * @param identifier 这个枚举的标识符
+     * @return 是否移除成功。如果不存在此枚举，则返回false
+     */
+    fun removeEnum(identifier: String):Boolean
+
+    /**
+     * 获取一个非泛型枚举。可能不存在
+     *
+     * @param identifier 枚举的标识符
+     * @return 获取到的枚举。如果不存在，则返回null
+     */
+    fun getEnum(identifier: String): Enum?
+
+    /**
+     * 是否存在此枚举
+     *
+     * @param identifier 枚举的标识符
+     * @return
+     */
+    fun hasEnum(identifier: String):Boolean
+
+    /**
+     * 是否存在此枚举
+     *
+     * @param enum 枚举
+     * @return
+     */
+    fun hasEnum(enum: Enum): Boolean
+
+    fun forEachEnum(operation: (Enum) -> Any?)
+}

src/main/kotlin/top/mcfpp/model/field/NamespaceField.kt

@@ -30,7 +30,9 @@ import kotlin.collections.HashMap
  *
  * 函数储存在一个列表中
  */
-open class NamespaceField: IFieldWithClass, IFieldWithFunction, IFieldWithTemplate, IFieldWithInterface, SimpleFieldWithType {
+open class NamespaceField(private val simpleFieldWithType: SimpleFieldWithType = SimpleFieldWithType(), private val simpleFieldWithEnum: SimpleFieldWithEnum = SimpleFieldWithEnum())
+    : IFieldWithClass, IFieldWithFunction, IFieldWithTemplate, IFieldWithInterface,
+    IFieldWithType by simpleFieldWithType, IFieldWithEnum by simpleFieldWithEnum {
     /**
      * 变量
      */
@@ -93,13 +95,11 @@ open class NamespaceField: IFieldWithClass, IFieldWithFunction, IFieldWithTempla
     @Nullable
     var container: FieldContainer? = GlobalField
 
-    constructor()
-
     /**
      * 复制一个缓存。
      * @param cache 原来的缓存
      */
-    constructor(cache: NamespaceField) {
+    constructor(cache: NamespaceField) : this(cache.simpleFieldWithType, cache.simpleFieldWithEnum) {
         parent = cache.parent
         //变量复制
         for (key in cache.vars.keys) {
@@ -420,4 +420,8 @@ open class NamespaceField: IFieldWithClass, IFieldWithFunction, IFieldWithTempla
     }
 
 
+    fun hasDeclaredType(identifier: String): Boolean{
+        return hasEnum(identifier) || hasTemplate(identifier) || hasInterface(identifier) || hasClass(identifier)
+    }
+
 }

src/main/kotlin/top/mcfpp/model/field/SimpleFieldWithEnum.kt

@@ -0,0 +1,70 @@
+package top.mcfpp.model.field
+
+import org.jetbrains.annotations.Nullable
+import top.mcfpp.model.Enum
+
+class SimpleFieldWithEnum : IFieldWithEnum {
+
+    /**
+     * 方法
+     */
+    private var enums: ArrayList<Enum> = ArrayList()
+
+    /**
+     * 遍历每一个方法
+     *
+     * @param operation 要对方法进行的操作
+     * @receiver
+     */
+    override fun forEachEnum(operation: (Enum) -> Any?){
+        for (enum in enums){
+            operation(enum)
+        }
+    }
+
+
+    @Nullable
+    override fun getEnum(identifier: String): Enum? {
+        for (e in enums) {
+            if(e.identifier == identifier){
+                return e
+            }
+        }
+        return null
+    }
+
+    override fun addEnum(identifier: String, enum: Enum, force: Boolean): Boolean {
+        if(hasEnum(enum)){
+            if(force){
+                enums[enums.indexOf(enum)] = enum
+                return true
+            }
+            return false
+        }
+        enums.add(enum)
+        return true
+    }
+
+    override fun removeEnum(identifier: String): Boolean {
+        for (e in enums) {
+            if(e.identifier == identifier){
+                enums.remove(e)
+                return true
+            }
+        }
+        return false
+    }
+
+    override fun hasEnum(enum: Enum): Boolean{
+        return enums.contains(enum)
+    }
+
+    override fun hasEnum(identifier: String): Boolean {
+        for (e in enums) {
+            if(e.identifier == identifier){
+                return true
+            }
+        }
+        return false
+    }
+}

src/test/java/top/mcfpp/test/QAQ.java

@@ -1,16 +1,35 @@
 package top.mcfpp.test;
 
+import java.util.Arrays;
+
 public class QAQ {
     public static void main(String[] args) {
-        System.out.println(removeTrailingZeros("51230100"));
+        System.out.println(findTargetSumWays(new int[]{1,0},1));
+    }
+
+    public static int findTargetSumWays(int[] nums, int target) {
+        //求和
+        int sum = 0;
+        for (int num : nums) {
+            sum += num;
+        }
+        //如果sum小于target或者sum+target为奇数，直接返回0
+        if (sum < target || (sum - target) % 2 == 1) {
+            return 0;
+        }
+        //求差
+        target = (sum - target) / 2;
+        //从小往大找
+        return solve(nums, target);
     }
 
-    public static String removeTrailingZeros(String num) {
-        for (int i = num.length()-1; i > 0; i --){
-            if(num.charAt(i) != '0'){
-                return num.substring(0, i + 1);
-            }
+    public static int solve(int[] nums, int target){
+        //从小往大找
+        int count = 0;
+        if(target == 0) count++;
+        for (int i = 0; i < nums.length; i++) {
+            count += solve(Arrays.copyOfRange(nums, i + 1, nums.length), target - nums[i]);
         }
-        return num;
+        return count;
     }
 }