containerwithelectricmanager.c

Go to the documentation of this file.

class ContainerWithElectricManager : ContainerWithCargoAndAttachments
{
 override void UpdateInterval()
 {
 super.UpdateInterval();
 
 if (m_Entity)
 ElectricityIcon();
 }
 
 override void SetEntity(EntityAI entity, bool immedUpdate = true)
 {
 super.SetEntity(entity, immedUpdate);
 
 ElectricityIcon();
 }
 
 void ElectricityIcon()
 {
 if (GetRootWidget().FindAnyWidget("electricity"))
 {
 bool hasElectricityIcon = m_Entity.GetCompEM().HasElectricityIcon();
 GetRootWidget().FindAnyWidget("electricity").Show(hasElectricityIcon);
 }
 
 if (GetRootWidget().FindAnyWidget("electricity_can_work"))
 {
 bool electricityCanWorkIcon = m_Entity.GetCompEM().CanWork() && !m_Entity.IsRuined();
 GetRootWidget().FindAnyWidget("electricity_can_work").Show(electricityCanWorkIcon);
 }
 }
}