123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622 |
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Linq;
- using System.Text;
- using UnityEngine;
- using UnityEngine.Rendering;
- using static MeidoPhotoStudio.Plugin.MenuFileUtility;
- using Object = UnityEngine.Object;
- namespace MeidoPhotoStudio.Plugin;
- public static class ModelUtility
- {
- private static GameObject deploymentObject;
- private enum IMode
- {
- None,
- ItemChange,
- TexChange,
- }
- public static GameObject LoadMyRoomModel(MyRoomItem item)
- {
- var data = MyRoomCustom.PlacementData.GetData(item.ID);
- var gameObject = Object.Instantiate(data.GetPrefab());
- if (!gameObject)
- {
- Utility.LogMessage($"Could not load MyRoomCreative model '{item.PrefabName}'");
- return null;
- }
- var final = new GameObject();
- gameObject.transform.SetParent(final.transform, true);
- final.transform.SetParent(GetDeploymentObject().transform, false);
- return final;
- }
- public static GameObject LoadBgModel(string bgName)
- {
- var gameObject = GameMain.Instance.BgMgr.CreateAssetBundle(bgName);
- if (!gameObject)
- gameObject = Resources.Load<GameObject>("BG/" + bgName);
- if (!gameObject)
- gameObject = Resources.Load<GameObject>("BG/2_0/" + bgName);
- if (!gameObject)
- {
- Utility.LogMessage($"Could not load BG model '{bgName}'");
- return null;
- }
- var final = Object.Instantiate(gameObject);
- final.transform.localScale = Vector3.one * 0.1f;
- return final;
- }
- public static GameObject LoadGameModel(string assetName)
- {
- var gameObject = GameMain.Instance.BgMgr.CreateAssetBundle(assetName);
- if (!gameObject)
- gameObject = Resources.Load<GameObject>("Prefab/" + assetName);
- if (!gameObject)
- gameObject = Resources.Load<GameObject>("BG/" + assetName);
- if (!gameObject)
- {
- Utility.LogMessage($"Could not load game model '{assetName}'");
- return null;
- }
- var final = Object.Instantiate(gameObject);
- final.transform.localPosition = Vector3.zero;
- var renderers = final.GetComponentsInChildren<Renderer>();
- foreach (var renderer in renderers)
- if (renderer && renderer.gameObject.name.Contains("castshadow"))
- renderer.shadowCastingMode = ShadowCastingMode.Off;
- var colliders = final.GetComponentsInChildren<Collider>();
- foreach (var collider in colliders)
- if (collider)
- collider.enabled = false;
- if (final.transform.localScale != Vector3.one)
- {
- var parent = new GameObject();
- final.transform.SetParent(parent.transform, true);
- final = parent;
- }
- return final;
- }
- public static GameObject LoadMenuModel(string menuFile) =>
- LoadMenuModel(new ModItem(menuFile));
- public static GameObject LoadMenuModel(ModItem modItem)
- {
- var menu = modItem.IsOfficialMod ? modItem.BaseMenuFile : modItem.MenuFile;
- byte[] modelBuffer;
- try
- {
- modelBuffer = ReadAFileBase(menu);
- }
- catch (Exception e)
- {
- Utility.LogError($"Could not read menu file '{menu}' because {e.Message}\n{e.StackTrace}");
- return null;
- }
- if (!ProcScriptBin(modelBuffer, out var modelInfo) || !InstantiateModel(modelInfo.ModelFile, out var finalModel))
- {
- Utility.LogMessage($"Could not load menu model '{modItem.MenuFile}'");
- return null;
- }
- IEnumerable<Renderer> renderers = GetRenderers(finalModel).ToList();
- foreach (var matChange in modelInfo.MaterialChanges)
- foreach (var renderer in renderers)
- if (matChange.MaterialIndex < renderer.materials.Length)
- renderer.materials[matChange.MaterialIndex] = ImportCM.LoadMaterial(
- matChange.MaterialFile, null, renderer.materials[matChange.MaterialIndex]);
- if (!modItem.IsOfficialMod)
- return finalModel;
- try
- {
- modelBuffer = ReadOfficialMod(modItem.MenuFile);
- }
- catch (Exception e)
- {
- Utility.LogError(
- $"Could not read mod menu file '{modItem.MenuFile}' because {e.Message}\n{e.StackTrace}");
- return null;
- }
- ProcModScriptBin(modelBuffer, finalModel);
- return finalModel;
- }
- private static GameObject GetDeploymentObject()
- {
- if (deploymentObject)
- return deploymentObject;
- if (!(deploymentObject = GameObject.Find("Deployment Object Parent")))
- deploymentObject = new("Deployment Object Parent");
- return deploymentObject;
- }
- private static IEnumerable<Renderer> GetRenderers(GameObject gameObject) =>
- gameObject.transform
- .GetComponentsInChildren<Transform>(true)
- .Select(transform => transform.GetComponent<Renderer>())
- .Where(renderer => renderer && renderer.material).ToList();
- private static GameObject CreateSeed() =>
- Object.Instantiate(Resources.Load<GameObject>("seed"));
- private static bool InstantiateModel(string modelFilename, out GameObject modelParent)
- {
- byte[] buffer;
- modelParent = default;
- try
- {
- buffer = ReadAFileBase(modelFilename);
- }
- catch
- {
- Utility.LogError($"Could not load model file '{modelFilename}'");
- return false;
- }
- using var binaryReader = new BinaryReader(new MemoryStream(buffer), Encoding.UTF8);
- if (binaryReader.ReadString() is not "CM3D2_MESH")
- {
- Utility.LogError($"{modelFilename} is not a model file");
- return false;
- }
- var modelVersion = binaryReader.ReadInt32();
- var modelName = binaryReader.ReadString();
- modelParent = CreateSeed();
- modelParent.layer = 1;
- modelParent.name = "_SM_" + modelName;
- var rootName = binaryReader.ReadString();
- var boneCount = binaryReader.ReadInt32();
- var boneDict = new Dictionary<string, GameObject>();
- var boneList = new List<GameObject>(boneCount);
- GameObject rootBone = null;
- try
- {
- // read bone data
- for (var i = 0; i < boneCount; i++)
- {
- var bone = CreateSeed();
- bone.layer = 1;
- bone.name = binaryReader.ReadString();
- if (binaryReader.ReadByte() is not 0)
- {
- var otherBone = CreateSeed();
- otherBone.name = bone.name + "_SCL_";
- otherBone.transform.parent = bone.transform;
- boneDict[bone.name + "$_SCL_"] = otherBone;
- }
- boneList.Add(bone);
- boneDict[bone.name] = bone;
- if (bone.name == rootName)
- rootBone = bone;
- }
- for (var i = 0; i < boneCount; i++)
- {
- var parentIndex = binaryReader.ReadInt32();
- boneList[i].transform.parent = parentIndex >= 0
- ? boneList[parentIndex].transform
- : modelParent.transform;
- }
- for (var i = 0; i < boneCount; i++)
- {
- var transform = boneList[i].transform;
- transform.localPosition = binaryReader.ReadVector3();
- transform.localRotation = binaryReader.ReadQuaternion();
- if (modelVersion >= 2001 && binaryReader.ReadBoolean())
- transform.localScale = binaryReader.ReadVector3();
- }
- // read mesh data
- var meshRenderer = rootBone.AddComponent<SkinnedMeshRenderer>();
- meshRenderer.updateWhenOffscreen = true;
- meshRenderer.skinnedMotionVectors = false;
- meshRenderer.lightProbeUsage = LightProbeUsage.Off;
- meshRenderer.reflectionProbeUsage = ReflectionProbeUsage.Off;
- meshRenderer.motionVectorGenerationMode = MotionVectorGenerationMode.ForceNoMotion;
- var sharedMesh = meshRenderer.sharedMesh = new();
- var vertCount = binaryReader.ReadInt32();
- var subMeshCount = binaryReader.ReadInt32();
- var meshBoneCount = binaryReader.ReadInt32();
- var meshBones = new Transform[meshBoneCount];
- for (var i = 0; i < meshBoneCount; i++)
- {
- var boneName = binaryReader.ReadString();
- if (!boneDict.ContainsKey(boneName))
- {
- Debug.LogError("nullbone= " + boneName);
- }
- else
- {
- var keyName = boneName + "$_SCL_";
- var bone = boneDict.ContainsKey(keyName) ? boneDict[keyName] : boneDict[boneName];
- meshBones[i] = bone.transform;
- }
- }
- meshRenderer.bones = meshBones;
- var bindPoses = new Matrix4x4[meshBoneCount];
- for (var i = 0; i < meshBoneCount; i++)
- bindPoses[i] = binaryReader.ReadMatrix4x4();
- sharedMesh.bindposes = bindPoses;
- var vertices = new Vector3[vertCount];
- var normals = new Vector3[vertCount];
- var uv = new Vector2[vertCount];
- for (var i = 0; i < vertCount; i++)
- {
- vertices[i] = binaryReader.ReadVector3();
- normals[i] = binaryReader.ReadVector3();
- uv[i] = binaryReader.ReadVector2();
- }
- sharedMesh.vertices = vertices;
- sharedMesh.normals = normals;
- sharedMesh.uv = uv;
- var tangentCount = binaryReader.ReadInt32();
- if (tangentCount > 0)
- {
- var tangents = new Vector4[tangentCount];
- for (var i = 0; i < tangentCount; i++)
- tangents[i] = binaryReader.ReadVector4();
- sharedMesh.tangents = tangents;
- }
- var boneWeights = new BoneWeight[vertCount];
- for (var i = 0; i < vertCount; i++)
- {
- boneWeights[i].boneIndex0 = binaryReader.ReadUInt16();
- boneWeights[i].boneIndex1 = binaryReader.ReadUInt16();
- boneWeights[i].boneIndex2 = binaryReader.ReadUInt16();
- boneWeights[i].boneIndex3 = binaryReader.ReadUInt16();
- boneWeights[i].weight0 = binaryReader.ReadSingle();
- boneWeights[i].weight1 = binaryReader.ReadSingle();
- boneWeights[i].weight2 = binaryReader.ReadSingle();
- boneWeights[i].weight3 = binaryReader.ReadSingle();
- }
- sharedMesh.boneWeights = boneWeights;
- sharedMesh.subMeshCount = subMeshCount;
- for (var i = 0; i < subMeshCount; i++)
- {
- var pointCount = binaryReader.ReadInt32();
- var triangles = new int[pointCount];
- for (var j = 0; j < pointCount; j++)
- triangles[j] = binaryReader.ReadUInt16();
- sharedMesh.SetTriangles(triangles, i);
- }
- // read materials
- var materialCount = binaryReader.ReadInt32();
- var materials = new Material[materialCount];
- for (var i = 0; i < materialCount; i++)
- materials[i] = ImportCM.ReadMaterial(binaryReader);
- meshRenderer.materials = materials;
- modelParent.AddComponent<Animation>();
- return true;
- }
- catch (Exception e)
- {
- Utility.LogError($"Could not load mesh for '{modelFilename}' because {e.Message}\n{e.StackTrace}");
- foreach (var bone in boneList.Where(bone => bone))
- Object.Destroy(bone);
- if (modelParent)
- Object.Destroy(modelParent);
- modelParent = null;
- return false;
- }
- }
- private static bool ProcScriptBin(byte[] menuBuf, out ModelInfo modelInfo)
- {
- modelInfo = null;
- using var binaryReader = new BinaryReader(new MemoryStream(menuBuf), Encoding.UTF8);
- if (binaryReader.ReadString() is not "CM3D2_MENU")
- return false;
- modelInfo = new();
- binaryReader.ReadInt32(); // file version
- binaryReader.ReadString(); // txt path
- binaryReader.ReadString(); // name
- binaryReader.ReadString(); // category
- binaryReader.ReadString(); // description
- binaryReader.ReadInt32(); // idk (as long)
- try
- {
- while (true)
- {
- var numberOfProps = binaryReader.ReadByte();
- var menuPropString = string.Empty;
- if (numberOfProps is 0)
- break;
- for (var i = 0; i < numberOfProps; i++)
- menuPropString = $"{menuPropString}\"{binaryReader.ReadString()}\"";
- if (menuPropString == string.Empty)
- continue;
- var header = UTY.GetStringCom(menuPropString);
- var menuProps = UTY.GetStringList(menuPropString);
- if (header is "end")
- {
- break;
- }
- else if (header is "マテリアル変更")
- {
- var matNo = int.Parse(menuProps[2]);
- var materialFile = menuProps[3];
- modelInfo.MaterialChanges.Add(new MaterialChange(matNo, materialFile));
- }
- else if (header is "additem")
- {
- modelInfo.ModelFile = menuProps[1];
- }
- }
- }
- catch
- {
- return false;
- }
- return true;
- }
- private static void ProcModScriptBin(byte[] cd, GameObject go)
- {
- var matDict = new Dictionary<string, byte[]>();
- string modData;
- using (var binaryReader = new BinaryReader(new MemoryStream(cd), Encoding.UTF8))
- {
- if (binaryReader.ReadString() is not "CM3D2_MOD")
- return;
- binaryReader.ReadInt32();
- binaryReader.ReadString();
- binaryReader.ReadString();
- binaryReader.ReadString();
- binaryReader.ReadString();
- binaryReader.ReadString();
- var serializedMpn = binaryReader.ReadString();
- var mpn = MPN.null_mpn;
- try
- {
- mpn = (MPN)Enum.Parse(typeof(MPN), serializedMpn, true);
- }
- catch
- {
- // Ignored.
- }
- if (mpn is not MPN.null_mpn)
- binaryReader.ReadString();
- modData = binaryReader.ReadString();
- var entryCount = binaryReader.ReadInt32();
- for (var i = 0; i < entryCount; i++)
- {
- var key = binaryReader.ReadString();
- var count = binaryReader.ReadInt32();
- var value = binaryReader.ReadBytes(count);
- matDict.Add(key, value);
- }
- }
- var mode = IMode.None;
- var materialChange = false;
- Material material = null;
- var materialIndex = 0;
- using var stringReader = new StringReader(modData);
- string line;
- List<Renderer> renderers = null;
- while ((line = stringReader.ReadLine()) is not null)
- {
- var data = line.Split(new[] { '\t', ' ' }, StringSplitOptions.RemoveEmptyEntries);
- var header = data[0];
- if (header is "アイテム変更" or "マテリアル変更")
- mode = IMode.ItemChange;
- else if (header is "テクスチャ変更")
- mode = IMode.TexChange;
- if (mode is IMode.ItemChange)
- {
- if (header is "スロット名")
- materialChange = true;
- if (!materialChange)
- continue;
- if (header is "マテリアル番号")
- {
- materialIndex = int.Parse(data[1]);
- renderers ??= GetRenderers(go).ToList();
- foreach (var renderer in renderers)
- if (materialIndex < renderer.materials.Length)
- material = renderer.materials[materialIndex];
- }
- if (!material)
- continue;
- if (header is "テクスチャ設定")
- ChangeTex(materialIndex, data[1], data[2].ToLower());
- else if (header is "色設定")
- material.SetColor(
- data[1],
- new(
- float.Parse(data[2]) / 255f,
- float.Parse(data[3]) / 255f,
- float.Parse(data[4]) / 255f,
- float.Parse(data[5]) / 255f));
- else if (header is "数値設定")
- material.SetFloat(data[1], float.Parse(data[2]));
- }
- else if (mode is IMode.TexChange)
- {
- var matno = int.Parse(data[2]);
- ChangeTex(matno, data[3], data[4].ToLower());
- }
- else if (mode is IMode.None)
- {
- continue;
- }
- else
- {
- throw new ArgumentOutOfRangeException(nameof(mode));
- }
- }
- void ChangeTex(int matno, string prop, string filename)
- {
- var buf = matDict[filename.ToLowerInvariant()];
- var textureResource = new TextureResource(2, 2, TextureFormat.ARGB32, null, buf);
- renderers ??= GetRenderers(go).ToList();
- foreach (var r in renderers)
- {
- r.materials[matno].SetTexture(prop, null);
- var texture2D = textureResource.CreateTexture2D();
- texture2D.name = filename;
- r.materials[matno].SetTexture(prop, texture2D);
- }
- }
- }
- private readonly struct MaterialChange
- {
- public MaterialChange(int materialIndex, string materialFile)
- {
- MaterialIndex = materialIndex;
- MaterialFile = materialFile;
- }
- public int MaterialIndex { get; }
- public string MaterialFile { get; }
- }
- private class ModelInfo
- {
- public List<MaterialChange> MaterialChanges { get; } = new();
- public string ModelFile { get; set; }
- }
- }
|