WeSign/unity-application
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 6 Wiki Activity

Wes xx mediapipe integration

Browse Source
This commit is contained in:
Jelle De Geest
2023-03-12 20:34:16 +00:00
parent 8349b5f149
commit b11eeb465c
975 changed files with 192230 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

149
Packages/com.github.homuler.mediapipe/Runtime/Scripts/Framework/Tool/NameUtil.cs Normal file
View File

@@ -0,0 +1,149 @@
#pragma warning disable IDE0073
// Copyright 2019 The MediaPipe Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
using System.Collections.Generic;
using System.Linq;
namespace Mediapipe
{
/// <summary>
/// translated version of mediapipe/framework/tool/name_util.cc
/// <summary/>
public static partial class Tool
{
public static string GetUnusedNodeName(CalculatorGraphConfig config, string nodeNameBase)
{
var nodeNames = new HashSet<string>(config.Node.Select(node => node.Name).Where(name => name.Length > 0));
var candidate = nodeNameBase;
var iter = 1;
while (nodeNames.Contains(candidate))
{
candidate = $"{nodeNameBase}_{++iter:D2}";
}
return candidate;
}
public static string GetUnusedSidePacketName(CalculatorGraphConfig config, string inputSidePacketNameBase)
{
var inputSidePackets = new HashSet<string>(
config.Node.SelectMany(node => node.InputSidePacket)
.Select(sidePacket =>
{
ParseTagIndexName(sidePacket, out var tag, out var index, out var name);
return name;
}));
var candidate = inputSidePacketNameBase;
var iter = 1;
while (inputSidePackets.Contains(candidate))
{
candidate = $"{inputSidePacketNameBase}_{++iter:D2}";
}
return candidate;
}
public static string GetUnusedStreamName(CalculatorGraphConfig config, string streamNameBase)
{
var outputStreamNames = config.Node.SelectMany(node => node.OutputStream)
.Select(outputStream =>
{
ParseTagIndexName(outputStream, out var tag, out var index, out var name);
return name;
});
var candidate = streamNameBase;
var iter = 1;
while (config.InputStream.Contains(candidate))
{
candidate = $"{streamNameBase}_{++iter:D2}";
}
while (outputStreamNames.Contains(candidate))
{
candidate = $"{streamNameBase}_{++iter:D2}";
}
return candidate;
}
/// <exception cref="ArgumentOutOfRangeException">
/// Thrown when <paramref name="nodeId" /> is invalid
/// </exception>
public static string CanonicalNodeName(CalculatorGraphConfig graphConfig, int nodeId)
{
var nodeConfig = graphConfig.Node[nodeId];
var nodeName = nodeConfig.Name.Length == 0 ? nodeConfig.Calculator : nodeConfig.Name;
var nodesWithSameName = graphConfig.Node
.Select((node, i) => (node.Name.Length == 0 ? node.Calculator : node.Name, i))
.Where(pair => pair.Item1 == nodeName);
if (nodesWithSameName.Count() <= 1)
{
return nodeName;
}
var seq = nodesWithSameName.Count(pair => pair.i <= nodeId);
return $"{nodeName}_{seq}";
}
/// <exception cref="ArgumentException">
/// Thrown when the format of <paramref cref="stream" /> is invalid
/// </exception>
public static string ParseNameFromStream(string stream)
{
ParseTagIndexName(stream, out var _, out var _, out var name);
return name;
}
/// <exception cref="ArgumentException">
/// Thrown when the format of <paramref cref="tagIndex" /> is invalid
/// </exception>
public static (string, int) ParseTagIndex(string tagIndex)
{
ParseTagIndex(tagIndex, out var tag, out var index);
return (tag, index);
}
/// <exception cref="ArgumentException">
/// Thrown when the format of <paramref cref="stream" /> is invalid
/// </exception>
public static (string, int) ParseTagIndexFromStream(string stream)
{
ParseTagIndexName(stream, out var tag, out var index, out var _);
return (tag, index);
}
public static string CatTag(string tag, int index)
{
var colonIndex = index <= 0 || tag.Length == 0 ? "" : $":{index}";
return $"{tag}{colonIndex}";
}
public static string CatStream((string, int) tagIndex, string name)
{
var tag = CatTag(tagIndex.Item1, tagIndex.Item2);
return tag.Length == 0 ? name : $"{tag}:{name}";
}
}
}
#pragma warning restore IDE0073

11
Packages/com.github.homuler.mediapipe/Runtime/Scripts/Framework/Tool/NameUtil.cs.meta Normal file
View File

@@ -0,0 +1,11 @@
fileFormatVersion: 2
guid: 87bcc70794ac17b298319e0f78cdf904
MonoImporter:
externalObjects: {}
serializedVersion: 2
defaultReferences: []
executionOrder: 0
icon: {instanceID: 0}
userData:
assetBundleName:
assetBundleVariant:

185
Packages/com.github.homuler.mediapipe/Runtime/Scripts/Framework/Tool/ValidateName.cs Normal file
View File

@@ -0,0 +1,185 @@
#pragma warning disable IDE0073
// Copyright 2019 The MediaPipe Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
using System;
using System.Text.RegularExpressions;
namespace Mediapipe
{
/// <summary>
/// translated version of mediapipe/framework/tool/validate_name.cc
/// <summary/>
internal static partial class Internal
{
public const int MaxCollectionItemId = 10000;
}
public static partial class Tool
{
private const string _NameRegex = "[a-z_][a-z0-9_]*";
private const string _NumberRegex = "(0|[1-9][0-9]*)";
private const string _TagRegex = "[A-Z_][A-Z0-9_]*";
private static readonly string _TagAndNameRegex = $"({_TagRegex}:)?{_NameRegex}";
private static readonly string _TagIndexNameRegex = $"({_TagRegex}:({_NumberRegex}:)?)?{_NameRegex}";
private static readonly string _TagIndexRegex = $"({_TagRegex})?(:{_NumberRegex})?";
public static void ValidateName(string name)
{
if (name.Length > 0 && new Regex($"^{_NameRegex}$").IsMatch(name))
{
return;
}
throw new ArgumentException($"Name \"{name}\" does not match \"{_NameRegex}\".");
}
public static void ValidateNumber(string number)
{
if (number.Length > 0 && new Regex($"^{_NumberRegex}$").IsMatch(number))
{
return;
}
throw new ArgumentException($"Number \"{number}\" does not match \"{_NumberRegex}\".");
}
public static void ValidateTag(string tag)
{
if (tag.Length > 0 && new Regex($"^{_TagRegex}$").IsMatch(tag))
{
return;
}
throw new ArgumentException($"Tag \"{tag}\" does not match \"{_TagRegex}\".");
}
public static void ParseTagAndName(string tagAndName, out string tag, out string name)
{
var nameIndex = -1;
var v = tagAndName.Split(':');
try
{
if (v.Length == 1)
{
ValidateName(v[0]);
nameIndex = 0;
}
else if (v.Length == 2)
{
ValidateTag(v[0]);
ValidateName(v[1]);
nameIndex = 1;
}
if (nameIndex == -1)
{
throw new ArgumentException();
}
}
catch (ArgumentException)
{
throw new ArgumentException($"\"tag and name\" is invalid, \"{tagAndName}\" does not match \"{_TagAndNameRegex}\" (examples: \"TAG:name\", \"longer_name\").");
}
tag = nameIndex == 1 ? v[0] : "";
name = v[nameIndex];
}
public static void ParseTagIndexName(string tagIndexName, out string tag, out int index, out string name)
{
var nameIndex = -1;
var theIndex = 0;
var v = tagIndexName.Split(':');
try
{
if (v.Length == 1)
{
ValidateName(v[0]);
theIndex = -1;
nameIndex = 0;
}
else if (v.Length == 2)
{
ValidateTag(v[0]);
ValidateName(v[1]);
nameIndex = 1;
}
else if (v.Length == 3)
{
ValidateTag(v[0]);
ValidateNumber(v[1]);
theIndex = int.TryParse(v[1], out var result) && result <= Internal.MaxCollectionItemId ? result : throw new ArgumentException();
ValidateName(v[2]);
nameIndex = 2;
}
if (nameIndex == -1)
{
throw new ArgumentException();
}
}
catch (ArgumentException)
{
throw new ArgumentException($"TAG:index:name is invalid, \"{tagIndexName}\" does not match \"{_TagIndexNameRegex}\" (examples: \"TAG:name\", \"VIDEO:2:name_b\", \"longer_name\").");
}
tag = nameIndex != 0 ? v[0] : "";
index = theIndex;
name = v[nameIndex];
}
public static void ParseTagIndex(string tagIndex, out string tag, out int index)
{
var theIndex = -1;
var v = tagIndex.Split(':');
try
{
if (v.Length == 1)
{
if (v[0].Length != 0)
{
ValidateTag(v[0]);
}
theIndex = 0;
}
else if (v.Length == 2)
{
if (v[0].Length != 0)
{
ValidateTag(v[0]);
}
ValidateNumber(v[1]);
theIndex = int.TryParse(v[1], out var result) && result <= Internal.MaxCollectionItemId ? result : throw new ArgumentException();
}
if (theIndex == -1)
{
throw new ArgumentException();
}
}
catch (ArgumentException)
{
throw new ArgumentException($"TAG:index is invalid, \"{tagIndex}\" does not match \"{_TagIndexRegex}\" (examples: \"TAG\", \"VIDEO:2\").");
}
tag = v[0];
index = theIndex;
}
}
}
#pragma warning restore IDE0073

11
Packages/com.github.homuler.mediapipe/Runtime/Scripts/Framework/Tool/ValidateName.cs.meta Normal file
View File

@@ -0,0 +1,11 @@
fileFormatVersion: 2
guid: 1421b820df4848b608b207d722d8c016
MonoImporter:
externalObjects: {}
serializedVersion: 2
defaultReferences: []
executionOrder: 0
icon: {instanceID: 0}
userData:
assetBundleName:
assetBundleVariant: