MegaPointCache.cs 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434
  1. using UnityEngine;
  2. using System.Collections.Generic;
  3. using System.IO;
  4. [System.Serializable]
  5. public class MegaPCVert
  6. {
  7. public int[] indices;
  8. public Vector3[] points;
  9. }
  10. public enum MegaInterpMethod
  11. {
  12. None,
  13. Linear,
  14. Bez,
  15. }
  16. public enum MegaBlendAnimMode
  17. {
  18. Replace,
  19. Additive,
  20. }
  21. [AddComponentMenu("Modifiers/Point Cache")]
  22. public class MegaPointCache : MegaModifier
  23. {
  24. public float time = 0.0f;
  25. public bool animated = false;
  26. public float speed = 1.0f;
  27. public float maxtime = 1.0f;
  28. public MegaRepeatMode LoopMode = MegaRepeatMode.PingPong;
  29. public MegaInterpMethod interpMethod = MegaInterpMethod.Linear;
  30. public MegaPCVert[] Verts;
  31. public float weight = 1.0f;
  32. public bool framedelay = true;
  33. public bool ignoreFirst = false;
  34. public MegaBlendAnimMode blendMode = MegaBlendAnimMode.Additive; // local space
  35. int numPoints; // Number of points per sample
  36. float startFrame; // Corresponds to the UI value of the same name.
  37. float sampleRate; // Corresponds to the UI value of the same name.
  38. int numSamples; // Defines how many samples are stored in the file.
  39. float t;
  40. float alpha = 0.0f;
  41. float dalpha = 0.0f;
  42. int sindex;
  43. int sindex1;
  44. public bool showmapping = false;
  45. public float mappingSize = 0.001f;
  46. public int mapStart = 0;
  47. public int mapEnd = 0;
  48. public bool havemapping = false;
  49. public float scl = 1.0f;
  50. public bool flipyz = false;
  51. public bool negx = false;
  52. public bool negz = false;
  53. public float adjustscl = 1.0f;
  54. public Vector3 adjustoff = Vector3.zero;
  55. public override string ModName() { return "Point Cache"; }
  56. public override string GetHelpURL() { return "?page_id=1335"; }
  57. void LinearAbs(MegaModifiers mc, int start, int end)
  58. {
  59. for ( int i = start; i < end; i++ )
  60. {
  61. Vector3 p = Verts[i].points[sindex];
  62. Vector3 p1 = Verts[i].points[sindex1];
  63. p.x = p.x + ((p1.x - p.x) * dalpha);
  64. p.y = p.y + ((p1.y - p.y) * dalpha);
  65. p.z = p.z + ((p1.z - p.z) * dalpha);
  66. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  67. sverts[Verts[i].indices[v]] = p;
  68. }
  69. }
  70. void LinearAbsWeighted(MegaModifiers mc, int start, int end)
  71. {
  72. for ( int i = start; i < end; i++ )
  73. {
  74. Vector3 p = Verts[i].points[sindex];
  75. Vector3 p1 = Verts[i].points[sindex1];
  76. p.x = p.x + ((p1.x - p.x) * dalpha);
  77. p.y = p.y + ((p1.y - p.y) * dalpha);
  78. p.z = p.z + ((p1.z - p.z) * dalpha);
  79. float w = mc.selection[Verts[i].indices[0]]; //[wc];
  80. p1 = verts[Verts[i].indices[0]];
  81. p = p1 + ((p - p1) * w);
  82. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  83. sverts[Verts[i].indices[v]] = p;
  84. }
  85. }
  86. void LinearRel(MegaModifiers mc, int start, int end)
  87. {
  88. for ( int i = start; i < end; i++ )
  89. {
  90. int ix = Verts[i].indices[0];
  91. Vector3 basep = mc.verts[ix];
  92. Vector3 p = Verts[i].points[sindex];
  93. Vector3 p1 = Verts[i].points[sindex1];
  94. p.x += (((p1.x - p.x) * dalpha) - basep.x); // * weight; //mc.verts[ix].x;
  95. p.y += (((p1.y - p.y) * dalpha) - basep.y); // * weight; //mc.verts[ix].y;
  96. p.z += (((p1.z - p.z) * dalpha) - basep.z); // * weight; //mc.verts[ix].z;
  97. p1 = verts[Verts[i].indices[0]];
  98. p.x = p1.x + (p.x * weight);
  99. p.y = p1.y + (p.y * weight);
  100. p.z = p1.z + (p.z * weight);
  101. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  102. {
  103. int idx = Verts[i].indices[v];
  104. sverts[idx] = p;
  105. }
  106. }
  107. }
  108. void LinearRelWeighted(MegaModifiers mc, int start, int end)
  109. {
  110. for ( int i = start; i < end; i++ )
  111. {
  112. int ix = Verts[i].indices[0];
  113. Vector3 basep = verts[ix];
  114. Vector3 p = Verts[i].points[sindex];
  115. Vector3 p1 = Verts[i].points[sindex1];
  116. p.x += (((p1.x - p.x) * dalpha) - basep.x); // * weight; //mc.verts[ix].x;
  117. p.y += (((p1.y - p.y) * dalpha) - basep.y); // * weight; //mc.verts[ix].y;
  118. p.z += (((p1.z - p.z) * dalpha) - basep.z); // * weight; //mc.verts[ix].z;
  119. float w = mc.selection[Verts[i].indices[0]] * weight; //[wc];
  120. p1 = verts[Verts[i].indices[0]];
  121. p.x = p1.x + (p.x * w);
  122. p.y = p1.y + (p.y * w);
  123. p.z = p1.z + (p.z * w);
  124. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  125. {
  126. int idx = Verts[i].indices[v];
  127. sverts[idx] = p;
  128. }
  129. }
  130. }
  131. void NoInterpAbs(MegaModifiers mc, int start, int end)
  132. {
  133. for ( int i = start; i < end; i++ )
  134. {
  135. Vector3 p = Verts[i].points[sindex];
  136. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  137. sverts[Verts[i].indices[v]] = p;
  138. }
  139. }
  140. void NoInterpAbsWeighted(MegaModifiers mc, int start, int end)
  141. {
  142. for ( int i = start; i < end; i++ )
  143. {
  144. Vector3 p = Verts[i].points[sindex];
  145. float w = mc.selection[Verts[i].indices[0]] * weight; //[wc];
  146. Vector3 p1 = verts[Verts[i].indices[0]];
  147. p = p1 + ((p - p1) * w);
  148. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  149. sverts[Verts[i].indices[v]] = p;
  150. }
  151. }
  152. void NoInterpRel(MegaModifiers mc, int start, int end)
  153. {
  154. for ( int i = start; i < end; i++ )
  155. {
  156. int ix = Verts[i].indices[0];
  157. Vector3 p = Verts[i].points[sindex] - verts[ix];
  158. Vector3 p1 = verts[Verts[i].indices[0]];
  159. p.x = p1.x + (p.x * weight);
  160. p.y = p1.y + (p.y * weight);
  161. p.z = p1.z + (p.z * weight);
  162. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  163. {
  164. int idx = Verts[i].indices[v];
  165. sverts[idx] = p;
  166. }
  167. }
  168. }
  169. void NoInterpRelWeighted(MegaModifiers mc, int start, int end)
  170. {
  171. for ( int i = start; i < end; i++ )
  172. {
  173. int ix = Verts[i].indices[0];
  174. Vector3 p = Verts[i].points[sindex] - verts[ix];
  175. float w = mc.selection[Verts[i].indices[0]] * weight; //[wc];
  176. Vector3 p1 = verts[Verts[i].indices[0]];
  177. p = p1 + ((p - p1) * w);
  178. for ( int v = 0; v < Verts[i].indices.Length; v++ )
  179. {
  180. int idx = Verts[i].indices[v];
  181. sverts[idx] = p;
  182. }
  183. }
  184. }
  185. bool skipframe = true;
  186. // TODO: Option to lerp or even bez, depends on how many samples
  187. public override void Modify(MegaModifiers mc)
  188. {
  189. if ( Verts != null )
  190. {
  191. switch ( interpMethod )
  192. {
  193. case MegaInterpMethod.Linear:
  194. switch ( blendMode )
  195. {
  196. case MegaBlendAnimMode.Additive: LinearRel(mc, 0, Verts.Length); break;
  197. case MegaBlendAnimMode.Replace: LinearAbs(mc, 0, Verts.Length); break;
  198. }
  199. break;
  200. case MegaInterpMethod.Bez:
  201. switch ( blendMode )
  202. {
  203. case MegaBlendAnimMode.Additive: LinearRel(mc, 0, Verts.Length); break;
  204. case MegaBlendAnimMode.Replace: LinearAbs(mc, 0, Verts.Length); break;
  205. }
  206. break;
  207. case MegaInterpMethod.None:
  208. switch ( blendMode )
  209. {
  210. case MegaBlendAnimMode.Additive: NoInterpRel(mc, 0, Verts.Length); break;
  211. case MegaBlendAnimMode.Replace: NoInterpAbs(mc, 0, Verts.Length); break;
  212. }
  213. break;
  214. }
  215. }
  216. else
  217. {
  218. for ( int i = 0; i < verts.Length; i++ )
  219. sverts[i] = verts[i];
  220. }
  221. }
  222. public void ModifyInstance(MegaModifiers mc, float itime)
  223. {
  224. if ( Verts != null )
  225. {
  226. switch ( LoopMode )
  227. {
  228. case MegaRepeatMode.Loop: t = Mathf.Repeat(itime, maxtime); break;
  229. case MegaRepeatMode.PingPong: t = Mathf.PingPong(itime, maxtime); break;
  230. case MegaRepeatMode.Clamp: t = Mathf.Clamp(itime, 0.0f, maxtime); break;
  231. }
  232. alpha = t / maxtime;
  233. float val = 0.0f;
  234. if ( ignoreFirst )
  235. val = 1.0f + (float)(Verts[0].points.Length - 2) * alpha;
  236. else
  237. val = (float)(Verts[0].points.Length - 1) * alpha;
  238. sindex = (int)val;
  239. dalpha = val - sindex;
  240. if ( sindex == Verts[0].points.Length - 1 )
  241. {
  242. sindex1 = sindex;
  243. dalpha = 0.0f;
  244. }
  245. else
  246. sindex1 = sindex + 1;
  247. switch ( interpMethod )
  248. {
  249. case MegaInterpMethod.Linear:
  250. switch ( blendMode )
  251. {
  252. case MegaBlendAnimMode.Additive: LinearRel(mc, 0, Verts.Length); break;
  253. case MegaBlendAnimMode.Replace: LinearAbs(mc, 0, Verts.Length); break;
  254. }
  255. break;
  256. case MegaInterpMethod.Bez:
  257. switch ( blendMode )
  258. {
  259. case MegaBlendAnimMode.Additive: LinearRel(mc, 0, Verts.Length); break;
  260. case MegaBlendAnimMode.Replace: LinearAbs(mc, 0, Verts.Length); break;
  261. }
  262. break;
  263. case MegaInterpMethod.None:
  264. switch ( blendMode )
  265. {
  266. case MegaBlendAnimMode.Additive: NoInterpRel(mc, 0, Verts.Length); break;
  267. case MegaBlendAnimMode.Replace: NoInterpAbs(mc, 0, Verts.Length); break;
  268. }
  269. break;
  270. }
  271. }
  272. else
  273. {
  274. for ( int i = 0; i < verts.Length; i++ )
  275. sverts[i] = verts[i];
  276. }
  277. }
  278. public void SetAnim(float _t)
  279. {
  280. time = _t;
  281. t = _t;
  282. skipframe = true;
  283. }
  284. public override bool ModLateUpdate(MegaModContext mc)
  285. {
  286. if ( !Prepare(mc) )
  287. return false;
  288. if ( animated ) //&& !lateanimupdate )
  289. {
  290. if ( framedelay && skipframe )
  291. skipframe = false;
  292. else
  293. {
  294. if ( Application.isPlaying )
  295. time += Time.deltaTime * speed;
  296. }
  297. }
  298. switch ( LoopMode )
  299. {
  300. case MegaRepeatMode.Loop: t = Mathf.Repeat(time, maxtime); break;
  301. case MegaRepeatMode.PingPong: t = Mathf.PingPong(time, maxtime); break;
  302. case MegaRepeatMode.Clamp: t = Mathf.Clamp(time, 0.0f, maxtime); break;
  303. }
  304. alpha = t / maxtime;
  305. float val = (float)(Verts[0].points.Length - 1) * alpha;
  306. sindex = (int)val;
  307. dalpha = val - sindex;
  308. if ( sindex == Verts[0].points.Length - 1 )
  309. {
  310. sindex1 = sindex;
  311. dalpha = 0.0f;
  312. }
  313. else
  314. sindex1 = sindex + 1;
  315. return true;
  316. }
  317. public override bool Prepare(MegaModContext mc)
  318. {
  319. if ( Verts != null && Verts.Length > 0 && Verts[0].indices != null && Verts[0].indices.Length > 0 )
  320. return true;
  321. return false;
  322. }
  323. public override void DoWork(MegaModifiers mc, int index, int start, int end, int cores)
  324. {
  325. ModifyCompressedMT(mc, index, cores);
  326. }
  327. public void ModifyCompressedMT(MegaModifiers mc, int tindex, int cores)
  328. {
  329. if ( Verts != null )
  330. {
  331. int step = Verts.Length / cores;
  332. int startvert = (tindex * step);
  333. int endvert = startvert + step;
  334. if ( tindex == cores - 1 )
  335. endvert = Verts.Length;
  336. switch ( interpMethod )
  337. {
  338. case MegaInterpMethod.Linear:
  339. switch ( blendMode )
  340. {
  341. case MegaBlendAnimMode.Additive: LinearRel(mc, startvert, endvert); break;
  342. case MegaBlendAnimMode.Replace: LinearAbs(mc, startvert, endvert); break;
  343. }
  344. break;
  345. case MegaInterpMethod.Bez:
  346. switch ( blendMode )
  347. {
  348. case MegaBlendAnimMode.Additive: LinearRel(mc, startvert, endvert); break;
  349. case MegaBlendAnimMode.Replace: LinearAbs(mc, startvert, endvert); break;
  350. }
  351. break;
  352. case MegaInterpMethod.None:
  353. switch ( blendMode )
  354. {
  355. case MegaBlendAnimMode.Additive: NoInterpRel(mc, startvert, endvert); break;
  356. case MegaBlendAnimMode.Replace: NoInterpAbs(mc, startvert, endvert); break;
  357. }
  358. break;
  359. }
  360. }
  361. }
  362. }