Unity OpenCV 识别物体 unity读取csv文件

转载

冷月星 2024-06-27 08:16:04

文章标签 Unity OpenCV 识别物体 Unity CSV 数据泛型 文章分类 计算机视觉人工智能

在上一篇中，保存CSV文件每一行的数据为CSVDemo对象时，总是要用if条件判断当前是哪个字段，然后再给对象赋值。

这样太烦人了，要是在开发的过程中不断地有新的想法，不断地新增修改字段，那真是改死人了。

甚至完全失去了改动的兴趣，于是游戏的创意就大打折扣，后果十分严重，嗯嗯。（旁白：说了这么多，接下来就要宣传你的“产品”了吧？）

不过，大家不用怕，有木头在，木头有个好招，大家看招吧。

1. 临时的字典结构

为了以后更方便，我们在搭建代码的时候，通常都会麻烦一点，但是，一劳永逸的事情，我是不会抗拒的。

为了实现我们的目的，我们需要先用一个字典结构把CSV文件保存起来，如下代码：

1. /// <summary>
2. /// 读取CSV文件
3. /// 结果保存到字典集合，以ID作为Key值，对应每一行的数据，每一行的数据也用字典集合保存。
4. /// </summary>
5. /// <param name="filePath"></param>
6. /// <returns></returns>
7. public static Dictionary<string, Dictionary<string, string>> LoadCsvFile(string filePath)
8. {
9. <string, Dictionary<string, string>> result = new Dictionary<string, Dictionary<string, string>>();
10. 
11. string[] fileData = File.ReadAllLines(filePath);
12. 
13. /* CSV文件的第一行为Key字段，第二行开始是数据。第一个字段一定是ID。 */
14. string[] keys = fileData[0].Split(',');
15. 
16. for (int i = 1; i < fileData.Length; i++)
17. {
18. string[] line = fileData[i].Split(',');
19. /* 以ID为key值，创建一个新的集合，用于保存当前行的数据 */
20. 
21. string ID = line[0];
22. [ID] = new Dictionary<string, string>();
23. 
24. for (int j = 0; j < line.Length; j++)
25. {
26. /* 每一行的数据存储规则：Key字段-Value值 */
27. [ID][keys[j]] = line[j];
28. }
29. }
30. return result;
31. }

实际上大部分代码和上一篇是相同的，主要的改动解释如下：

Dictionary < string , Dictionary < string , string >> 】的形式保存起来

b. 相当于是，以每一行的ID作为key值，value值保存的是每一行的所有数据

c. 每一行的所有数据又由一个Dictionary<string, string>保存，key值是字段名，value值是每一列的数据值

d. 最终，每一行都以这样的形式保存起来

这样保存起来的结构并不是很方便，因为类型都是字符串，不好取值，没关系，这个只是过度的。

2. 强大的反射

你是否曾经有这样的感觉，那些看起来明明一样的代码，仅仅只是类型不一样；那些看起来明明是一样的代码，仅仅只是调用的属性不一样。

总之，那些看起来明明不应该那么重复的代码，它却总是重复了。

大多数开发者都会有这样的感觉，于是，逆天的反射和泛型出现在我们的面前。

没错，木头要用的就是反射+泛型这个强力的组合。

看看优化后的代码：

1. /* 把CSV文件按行存放，每一行的ID作为key值，内容作为value值 */
2. <int, CSVDemo> csvDataDic = new Dictionary<int, CSVDemo>();
3. 
4. /* CSV文件路径 */
5. string filePath = Application.streamingAssetsPath + "/CSVDemo.csv";
6. 
7. /* 从CSV文件读取数据 */
8. <string, Dictionary<string, string>> datasDic = LoadCsvFile(filePath);
9. 
10. /* 遍历每一行数据 */
11. foreach (string ID in datasDic.Keys)
12. {
13. /* CSV的一行数据 */
14. <string, string> datas = datasDic[ID];
15. 
16. /* 读取Csv数据对象的属性 */
17. [] props = typeof(CSVDemo).GetProperties();
18. 
19. /* 使用反射，将CSV文件的数据赋值给CSV数据对象的相应字段，要求CSV文件的字段名和CSV数据对象的字段名完全相同 */
20. = new CSVDemo();
21. foreach (PropertyInfo pi in props)
22. {
23. .SetValue(obj, Convert.ChangeType(datas[pi.Name], pi.PropertyType), null);
24. }
25. 
26. /* 按ID-数据的形式存储 */
27. [obj.ID] = obj;
28. }

主要修改如下：

a. 利用之前写的LoadCsvFile函数获取到字典结构的CSV文件内容

b. 遍历字典结构，取出每一行的内容

c. 利用反射获取CSVDemo类的所有属性

d. 便利CSVDemo类的所有属性，通过每一个属性名字到字典里获取数据内容，然后给CSVDemo对象相应的属性赋值

e. pi.SetValue就是用来给CSVDemo对象的某个属性赋值的，Convert.ChangeType是为了把字符串内容转换为属性对应的类型

最后来测试一下：

1. /* 测试读取ID为1的数据 */
2. = csvDataDic[1];
3. .Log("ID=" + csvDemo1.ID + "，Name=" + csvDemo1.Name);

尝试获取ID为1的那一行数据，输出日志如下：

Unity OpenCV 识别物体 unity读取csv文件_Unity OpenCV 识别物体

旁白：那泛型呢？）

3. 泛型呢？

大家一定还有疑问，每个CSV文件都要写这样一段代码来读取吗？

不，这种重复的事情木头是绝对不会做的，我一做重复的事情就会头晕，于是，泛型救了我。

最终完美的读取并保存CSV文件的函数是这样的：

1. /// <summary>
2. /// 读取CSV文件数据（利用反射）
3. /// </summary>
4. /// <typeparam name="CsvData">CSV数据对象的类型</typeparam>
5. /// <param name="csvFilePath">CSV文件路径</param>
6. /// <param name="csvDatas">用于缓存数据的字典</param>
7. /// <returns>CSV文件所有行内容的数据对象</returns>
8. private Dictionary<int, T_CsvData> LoadCsvData<T_CsvData>(string csvFilePath)
9. {
10. <int, T_CsvData> dic = new Dictionary<int, T_CsvData>();
11. 
12. /* 从CSV文件读取数据 */
13. <string, Dictionary<string, string>> result = LoadCsvFile(csvFilePath);
14. 
15. /* 遍历每一行数据 */
16. foreach (string ID in result.Keys)
17. {
18. /* CSV的一行数据 */
19. <string, string> datas = result[ID];
20. 
21. /* 读取Csv数据对象的属性 */
22. [] props = typeof(T_CsvData).GetProperties();
23. 
24. /* 使用反射，将CSV文件的数据赋值给CSV数据对象的相应字段，要求CSV文件的字段名和CSV数据对象的字段名完全相同 */
25. = Activator.CreateInstance<T_CsvData>();
26. 
27. foreach (PropertyInfo pi in props)
28. {
29. .SetValue(obj, Convert.ChangeType(datas[pi.Name], pi.PropertyType), null);
30. }
31. 
32. /* 按ID-数据的形式存储 */
33. [Convert.ToInt32(ID)] = obj;
34. }
35. return dic;
36. }

基本上把CSVDemo类型改为T_CsvData泛型即可，调用方式如下：

1. /* 把CSV文件按行存放，每一行的ID作为key值，内容作为value值 */
2. <int, CSVDemo> csvDataDic2 = LoadCsvData<CSVDemo>(filePath);
3. 
4. /* 测试读取ID为2的数据 */
5. = csvDataDic2[2];
6. 
7. .Log("ID=" + csvDemo2.ID + "，Name=" + csvDemo2.Name);

输出的日志如下：

Unity OpenCV 识别物体 unity读取csv文件_数据_02