QueryEngine<T0>.SetDefaultParamFilter

参数

描述

为未注册的函数筛选器设置默认筛选器处理程序。

using System.Collections.Generic;
using System.Linq;
using UnityEditor;
using UnityEditor.Search;
using UnityEngine;

static class Example_QueryEngine_SetDefaultParamFilter
{
    static List<MyObjectType> s_Data;

    [MenuItem("Examples/QueryEngine/SetDefaultParamFilter")]
    public static void RunExample()
    {
        // Set up the query engine
        var queryEngine = new QueryEngine<MyObjectType>();
        queryEngine.SetSearchDataCallback(myObj => new[] { myObj.id.ToString(), myObj.name });

        // Add a filter for MyObjectType.id that supports all operators
        queryEngine.AddFilter("id", myObj => myObj.id);

        // Register a default filter handler for any filter with parameters.
        queryEngine.SetDefaultParamFilter((myObj, filterId, paramValue, operatorToken, filterValue) =>
        {
            if (myObj.property.name != filterId)
                return false;

            if (myObj.property.type == PropertyType.Integer_Array)
            {
                if (!(myObj.property.value is int[] integerArray))
                    return false;
                if (!int.TryParse(paramValue, out var indexValue))
                    return false;
                if (!int.TryParse(filterValue, out var filterValueInt))
                    return false;
                if (indexValue < 0 || indexValue >= integerArray.Length)
                    return false;
                switch (operatorToken)
                {
                    case "=": return integerArray[indexValue] == filterValueInt;
                    case "!=": return integerArray[indexValue] != filterValueInt;
                    case "<": return integerArray[indexValue] < filterValueInt;
                    case ">": return integerArray[indexValue] > filterValueInt;
                    case "<=": return integerArray[indexValue] <= filterValueInt;
                    case ">=": return integerArray[indexValue] >= filterValueInt;
                    default: return false;
                }
            }

            return false;
        });

        s_Data = new List<MyObjectType>()
        {
            new MyObjectType { id = 0, property = new Property("values", PropertyType.Integer_Array, new int[] {0, 2, 42}) },
            new MyObjectType { id = 1, property = new Property("weights", PropertyType.Integer_Array, new int[] {1000, 250, 400}) }
        };

        // Find all items that have the property "values" higher than 10 at index 2
        var query = queryEngine.ParseQuery("values(2)>10");
        var filteredData = query.Apply(s_Data).ToList();
        Debug.Assert(filteredData.Count == 1, $"There should be 1 item in the filtered list but found {filteredData.Count} items.");
        Debug.Assert(filteredData.Contains(s_Data[0]), "The first item should be in the filtered list since its property \"values\" has a value higher than 10 at index 2.");
    }

    enum PropertyType
    {
        Integer,
        String,
        Integer_Array
    }
    struct Property
    {
        public string name { get; }
        public PropertyType type { get; }
        public object value { get; set; }

        public Property(string name, PropertyType type, object value)
        {
            this.name = name;
            this.type = type;
            this.value = value;
        }
    }

    class MyObjectType
    {
        public int id { get; set; }
        public string name { get; set; } = string.Empty;
        public Vector2 position { get; set; } = Vector2.zero;
        public bool active { get; set; }
        public Property property { get; set; }

        public override string ToString()
        {
            return $"({id}, {name}, ({position.x}, {position.y}), {active})";
        }
    }
}