zc
This commit is contained in:
@@ -16,6 +16,35 @@ namespace WulaFallenEmpire.EventSystem.AI.Utils
|
||||
public float Score;
|
||||
}
|
||||
|
||||
public static List<SearchResult> Search(string query, int maxResults = 20, bool itemsOnly = false, float minScore = 0.15f)
|
||||
{
|
||||
var results = new List<SearchResult>();
|
||||
if (string.IsNullOrWhiteSpace(query)) return results;
|
||||
|
||||
query = query.Trim();
|
||||
string lowerQuery = query.ToLowerInvariant();
|
||||
|
||||
string normalizedQuery = NormalizeKey(lowerQuery);
|
||||
var tokens = TokenizeQuery(lowerQuery);
|
||||
|
||||
foreach (var def in DefDatabase<ThingDef>.AllDefs)
|
||||
{
|
||||
if (def == null || def.label == null) continue;
|
||||
if (def.category != ThingCategory.Item && def.category != ThingCategory.Building) continue;
|
||||
if (itemsOnly && def.category != ThingCategory.Item) continue;
|
||||
|
||||
float score = ScoreThingDef(def, lowerQuery, normalizedQuery, tokens);
|
||||
if (score >= minScore)
|
||||
{
|
||||
results.Add(new SearchResult { Def = def, Count = 1, Score = score });
|
||||
}
|
||||
}
|
||||
|
||||
results.Sort((a, b) => b.Score.CompareTo(a.Score));
|
||||
if (results.Count > maxResults) results.RemoveRange(maxResults, results.Count - maxResults);
|
||||
return results;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Parses a natural language request string into a list of spawnable items.
|
||||
/// Example: "5 beef, 1 persona core, 30 wood"
|
||||
@@ -25,9 +54,7 @@ namespace WulaFallenEmpire.EventSystem.AI.Utils
|
||||
var results = new List<SearchResult>();
|
||||
if (string.IsNullOrEmpty(request)) return results;
|
||||
|
||||
// Split by common separators
|
||||
var parts = request.Split(new[] { ',', '，', ';', '、', '\n' }, StringSplitOptions.RemoveEmptyEntries);
|
||||
|
||||
foreach (var part in parts)
|
||||
{
|
||||
var result = ParseSingleItem(part.Trim());
|
||||
@@ -40,27 +67,28 @@ namespace WulaFallenEmpire.EventSystem.AI.Utils
|
||||
return results;
|
||||
}
|
||||
|
||||
public static bool TryFindBestThingDef(string query, out ThingDef bestDef, out float bestScore, bool itemsOnly = false, float minScore = 0.4f)
|
||||
{
|
||||
bestDef = null;
|
||||
bestScore = 0f;
|
||||
var results = Search(query, maxResults: 1, itemsOnly: itemsOnly, minScore: minScore);
|
||||
if (results.Count == 0) return false;
|
||||
bestDef = results[0].Def;
|
||||
bestScore = results[0].Score;
|
||||
return true;
|
||||
}
|
||||
|
||||
private static SearchResult ParseSingleItem(string itemRequest)
|
||||
{
|
||||
// Extract count and name
|
||||
// Regex to match "number name" or "name number" or "number个name"
|
||||
// Supports Chinese and English numbers
|
||||
|
||||
int count = 1;
|
||||
string nameQuery = itemRequest;
|
||||
|
||||
// Try to find digits
|
||||
var match = Regex.Match(itemRequest, @"(\d+)");
|
||||
if (match.Success)
|
||||
if (match.Success && int.TryParse(match.Value, out int parsedCount))
|
||||
{
|
||||
if (int.TryParse(match.Value, out int parsedCount))
|
||||
{
|
||||
count = parsedCount;
|
||||
// Remove the number from the query string to get the name
|
||||
nameQuery = itemRequest.Replace(match.Value, "").Trim();
|
||||
// Remove common quantifiers
|
||||
nameQuery = Regex.Replace(nameQuery, @"[个只把张条xX×]", "").Trim();
|
||||
}
|
||||
count = parsedCount;
|
||||
nameQuery = itemRequest.Replace(match.Value, "").Trim();
|
||||
nameQuery = Regex.Replace(nameQuery, @"[个只把张条xX×]", "").Trim();
|
||||
}
|
||||
|
||||
if (string.IsNullOrWhiteSpace(nameQuery))
|
||||
@@ -68,66 +96,106 @@ namespace WulaFallenEmpire.EventSystem.AI.Utils
|
||||
return new SearchResult { Def = null, Count = 0, Score = 0 };
|
||||
}
|
||||
|
||||
// Search for the Def
|
||||
var bestMatch = FindBestThingDef(nameQuery);
|
||||
|
||||
TryFindBestThingDef(nameQuery, out ThingDef bestDef, out float bestScore, itemsOnly: false, minScore: 0.15f);
|
||||
return new SearchResult
|
||||
{
|
||||
Def = bestMatch.Def,
|
||||
Def = bestDef,
|
||||
Count = count,
|
||||
Score = bestMatch.Score
|
||||
Score = bestScore
|
||||
};
|
||||
}
|
||||
|
||||
private static (ThingDef Def, float Score) FindBestThingDef(string query)
|
||||
private static float ScoreThingDef(ThingDef def, string lowerQuery, string normalizedQuery, List<string> tokens)
|
||||
{
|
||||
ThingDef bestDef = null;
|
||||
float bestScore = 0f;
|
||||
string lowerQuery = query.ToLower();
|
||||
string label = def.label?.ToLowerInvariant() ?? "";
|
||||
string defName = def.defName?.ToLowerInvariant() ?? "";
|
||||
string normalizedLabel = NormalizeKey(label);
|
||||
string normalizedDefName = NormalizeKey(defName);
|
||||
|
||||
foreach (var def in DefDatabase<ThingDef>.AllDefs)
|
||||
float score = 0f;
|
||||
|
||||
if (!string.IsNullOrEmpty(normalizedQuery))
|
||||
{
|
||||
// Filter out non-items or abstract defs
|
||||
if (def.category != ThingCategory.Item && def.category != ThingCategory.Building) continue;
|
||||
if (def.label == null) continue;
|
||||
if (normalizedLabel == normalizedQuery) score = Math.Max(score, 1.00f);
|
||||
if (normalizedDefName == normalizedQuery) score = Math.Max(score, 0.98f);
|
||||
}
|
||||
|
||||
float score = 0f;
|
||||
string label = def.label.ToLower();
|
||||
string defName = def.defName.ToLower();
|
||||
if (!string.IsNullOrEmpty(lowerQuery))
|
||||
{
|
||||
if (label == lowerQuery) score = Math.Max(score, 0.95f);
|
||||
if (defName == lowerQuery) score = Math.Max(score, 0.93f);
|
||||
|
||||
// Exact match
|
||||
if (label == lowerQuery) score = 1.0f;
|
||||
else if (defName == lowerQuery) score = 0.9f;
|
||||
// Starts with match (High priority for defNames like "WoodLog" when searching "Wood")
|
||||
else if (defName.StartsWith(lowerQuery))
|
||||
{
|
||||
score = 0.8f + (0.1f * ((float)lowerQuery.Length / defName.Length));
|
||||
}
|
||||
// Contains match
|
||||
else if (label.Contains(lowerQuery))
|
||||
{
|
||||
// Shorter labels that contain the query are better matches
|
||||
score = 0.6f + (0.2f * ((float)lowerQuery.Length / label.Length));
|
||||
}
|
||||
else if (defName.Contains(lowerQuery))
|
||||
{
|
||||
score = 0.5f + (0.2f * ((float)lowerQuery.Length / defName.Length));
|
||||
}
|
||||
|
||||
// Bonus for tradeability (more likely to be what player wants)
|
||||
if (def.tradeability != Tradeability.None) score += 0.05f;
|
||||
if (label.StartsWith(lowerQuery)) score = Math.Max(score, 0.80f);
|
||||
if (defName.StartsWith(lowerQuery)) score = Math.Max(score, 0.85f);
|
||||
|
||||
if (score > bestScore)
|
||||
if (label.Contains(lowerQuery)) score = Math.Max(score, 0.65f + 0.15f * ((float)lowerQuery.Length / Math.Max(1, label.Length)));
|
||||
if (defName.Contains(lowerQuery)) score = Math.Max(score, 0.60f + 0.15f * ((float)lowerQuery.Length / Math.Max(1, defName.Length)));
|
||||
}
|
||||
|
||||
if (tokens.Count > 0)
|
||||
{
|
||||
int matchedInLabel = tokens.Count(t => normalizedLabel.Contains(NormalizeKey(t)));
|
||||
int matchedInDefName = tokens.Count(t => normalizedDefName.Contains(NormalizeKey(t)));
|
||||
int matched = Math.Max(matchedInLabel, matchedInDefName);
|
||||
float coverage = (float)matched / tokens.Count;
|
||||
|
||||
if (matched > 0)
|
||||
{
|
||||
bestScore = score;
|
||||
bestDef = def;
|
||||
score = Math.Max(score, 0.45f + 0.35f * coverage);
|
||||
}
|
||||
|
||||
if (matched == tokens.Count && tokens.Count >= 2)
|
||||
{
|
||||
score = Math.Max(score, 0.80f);
|
||||
}
|
||||
}
|
||||
|
||||
// Threshold
|
||||
if (bestScore < 0.4f) return (null, 0f);
|
||||
bool queryLooksLikeFood = tokens.Any(t => t == "meal" || t == "food" || t.Contains("meal") || t.Contains("food")) ||
|
||||
lowerQuery.Contains("食") || lowerQuery.Contains("饭") || lowerQuery.Contains("餐");
|
||||
if (queryLooksLikeFood && def.ingestible != null)
|
||||
{
|
||||
score += 0.05f;
|
||||
}
|
||||
|
||||
return (bestDef, bestScore);
|
||||
if (def.tradeability != Tradeability.None) score += 0.03f;
|
||||
if (!def.IsStuff) score += 0.01f;
|
||||
|
||||
if (score > 1.0f) score = 1.0f;
|
||||
return score;
|
||||
}
|
||||
|
||||
private static List<string> TokenizeQuery(string lowerQuery)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(lowerQuery)) return new List<string>();
|
||||
|
||||
string q = lowerQuery.Trim();
|
||||
q = q.Replace('_', ' ').Replace('-', ' ');
|
||||
var rawTokens = Regex.Split(q, @"\s+").Where(t => !string.IsNullOrWhiteSpace(t)).ToList();
|
||||
|
||||
var tokens = new List<string>();
|
||||
foreach (var token in rawTokens)
|
||||
{
|
||||
string cleaned = Regex.Replace(token, @"[^\p{L}\p{N}]+", "");
|
||||
if (string.IsNullOrWhiteSpace(cleaned)) continue;
|
||||
tokens.Add(cleaned);
|
||||
}
|
||||
|
||||
// For queries like "fine meal" also consider the normalized concatenation for matching "MealFine".
|
||||
if (tokens.Count >= 2)
|
||||
{
|
||||
tokens.Add(string.Concat(tokens));
|
||||
}
|
||||
|
||||
return tokens.Distinct().ToList();
|
||||
}
|
||||
|
||||
private static string NormalizeKey(string s)
|
||||
{
|
||||
if (string.IsNullOrEmpty(s)) return "";
|
||||
string t = s.ToLowerInvariant();
|
||||
t = Regex.Replace(t, @"[\s_\-]+", "");
|
||||
t = Regex.Replace(t, @"[^\p{L}\p{N}]+", "");
|
||||
return t;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user