scriptconsolevicinitytab.c 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. class ScriptConsoleVicinityTab : ScriptConsoleTabBase
  2. {
  3. protected static float m_Distance = 30;
  4. protected static Object m_Selected = null;
  5. protected TextListboxWidget m_VicinityListbox;
  6. protected ButtonWidget m_WatchItem;
  7. protected EditBoxWidget m_DistanceWidget;
  8. protected ref array<Object> m_VicinityItems = new array<Object>;
  9. protected ref ContextMenu m_ContextMenu;
  10. void ScriptConsoleVicinityTab(Widget root, ScriptConsole console, Widget button, ScriptConsoleTabBase parent = null)
  11. {
  12. m_VicinityListbox = TextListboxWidget.Cast(root.FindAnyWidget("VicinityListbox"));
  13. m_WatchItem = ButtonWidget.Cast(root.FindAnyWidget("WatchItemButton"));
  14. m_DistanceWidget = EditBoxWidget.Cast(root.FindAnyWidget("DistanceInput"));
  15. m_DistanceWidget.SetText(m_Distance.ToString());
  16. m_ContextMenu = new ContextMenu();
  17. m_ContextMenu.Init(root.FindAnyWidget("VicinityContextMenu"), true);
  18. m_ContextMenu.SetSize(1, 1);
  19. m_ContextMenu.Show(0, 0);
  20. RenderList();
  21. }
  22. override void OnSelected()
  23. {
  24. UpdateSelected();
  25. }
  26. void RenderList()
  27. {
  28. float radius = m_DistanceWidget.GetText().ToFloat();
  29. if (!radius)
  30. return;
  31. array<Object> objects = new array<Object>();
  32. array<CargoBase> cargo = new array<CargoBase>();
  33. vector playerPos = GetGame().GetCurrentCameraPosition();
  34. GetGame().GetObjectsAtPosition3D(playerPos, radius, objects,cargo);
  35. m_VicinityListbox.ClearItems();
  36. m_VicinityItems.Clear();
  37. if (m_Selected && objects.Find(m_Selected) == -1)
  38. {
  39. objects.Insert(m_Selected);
  40. }
  41. map<int, Object> objectsMap = new map<int, Object>();
  42. array<int> distances = new array<int>();
  43. foreach (int x, Object obj:objects)
  44. {
  45. int dist = vector.Distance(playerPos, obj.GetPosition()) * 1000;
  46. while (objectsMap.Contains(dist))//hacking for when multiple objects are at the same distance from the player, otherwise sorting using map will fail
  47. dist++;
  48. objectsMap.Insert(dist, obj);
  49. distances.Insert(dist);
  50. }
  51. distances.Sort();
  52. foreach (int dst:distances)
  53. {
  54. Object o = objectsMap.Get(dst);
  55. if (!o)
  56. continue;
  57. int i = 0;
  58. if (o.Type() != Object)
  59. {
  60. m_VicinityItems.Insert(o);//insert at the end
  61. int indx = m_VicinityListbox.AddItem(o.GetType(),null,0,-1);
  62. float distanceMeters = dst / 1000;
  63. m_VicinityListbox.SetItem(indx, distanceMeters.ToString(),null,1);
  64. i++;
  65. if (m_Selected == o)
  66. {
  67. m_VicinityListbox.SelectRow(indx);
  68. }
  69. }
  70. }
  71. UpdateSelected();
  72. }
  73. void UpdateSelected()
  74. {
  75. int row = m_VicinityListbox.GetSelectedRow();
  76. m_Selected = null;
  77. if (m_VicinityItems.IsValidIndex(row))
  78. {
  79. m_Selected = m_VicinityItems.Get(row);
  80. }
  81. EntityAI entity = EntityAI.Cast(m_Selected);
  82. if (entity)
  83. {
  84. m_ContextMenu.BuildContextMenu(entity, m_VicinityListbox, this);
  85. m_ContextMenu.SetSize(1, 1);
  86. m_ContextMenu.Show(0, 0);
  87. }
  88. else
  89. {
  90. m_ContextMenu.Hide();
  91. }
  92. }
  93. override bool OnClick(Widget w, int x, int y, int button)
  94. {
  95. super.OnClick(w,x,y,button);
  96. UpdateSelected();
  97. int row = m_VicinityListbox.GetSelectedRow();
  98. if (w == m_WatchItem)
  99. {
  100. if (m_VicinityItems.IsValidIndex(row))
  101. {
  102. if (m_Selected)
  103. {
  104. PluginDeveloper.SetDeveloperItemClientEx(m_Selected);
  105. return true;
  106. }
  107. }
  108. }
  109. return false;
  110. }
  111. override bool OnChange(Widget w, int x, int y, bool finished)
  112. {
  113. super.OnChange(w, x, y, finished);
  114. UpdateSelected();
  115. if (w == m_DistanceWidget)
  116. {
  117. RenderList();
  118. m_Distance = m_DistanceWidget.GetText().ToFloat();
  119. }
  120. return false;
  121. }
  122. void OnSelectAction(EntityAI ent, int actionId)
  123. {
  124. #ifdef DIAG_DEVELOPER
  125. PlayerBase player = PlayerBase.Cast( GetGame().GetPlayer() );
  126. player.GetActionManager().OnInstantAction(ActionDebug,new Param2<EntityAI,int>(ent,actionId));
  127. #endif
  128. }
  129. }