Changed GetCurrentModel to return the model instead of the tuple

This commit is contained in:
CoudronJerome
2023-03-29 19:17:42 +02:00
parent be8885a508
commit a44532b94a
3 changed files with 9 additions and 7 deletions

View File

@@ -41,10 +41,9 @@ public class ModelList : ScriptableObject
/// </summary>
/// <param name="modelIndex">ModelIndex of the model</param>
/// <returns>Model associated with this index, null if no model was found</returns>
public ModelTuple GetCurrentModel()
public NNModel GetCurrentModel()
{
//if(currentModelIndex < models.Find)
return models.Find(x => x.model == models[currentModelIndex].model);
return models.Find(x => x.model == models[currentModelIndex].model)?.model;
}
/// <summary>

View File

@@ -35,14 +35,17 @@ public class ModelListTest
}
}
}
/// <summary>
/// Check if current model can be correctly gotten as current via GetCurrentModel
/// </summary>
[Test]
public void TestGetCurrentModel()
{
System.Random random = new System.Random();
ModelIndex value = (ModelIndex)random.Next(modelList.models.Count);
modelList.SetCurrentModel(value);
Assert.AreEqual(value, modelList.GetCurrentModel().index);
Assert.AreEqual(modelList.models[modelList.currentModelIndex].model, modelList.GetCurrentModel());
// Check if empty model fails gracefully (returns null)
Assert.IsNull(ScriptableObject.CreateInstance<ModelList>().GetCurrentModel());

View File

@@ -210,7 +210,7 @@ namespace Mediapipe.Unity.Tutorial
// check if model exists at path
//var model = ModelLoader.Load(Resources.Load<NNModel>("Models/Fingerspelling/model_A-L"));
worker = modelList.GetCurrentModel().model.CreateWorker();
worker = modelList.GetCurrentModel().CreateWorker();
StartCoroutine(SignRecognitionCoroutine());
StartCoroutine(MediapipeCoroutine());
@@ -223,7 +223,7 @@ namespace Mediapipe.Unity.Tutorial
worker?.Dispose();
// Add a new worker for the new model
worker = modelList.GetCurrentModel().model.CreateWorker();
worker = modelList.GetCurrentModel().CreateWorker();
}
/// <summary>