Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00012 #include "stdafx.h"
00013 #include "command_func.h"
00014 #include "group.h"
00015 #include "autoreplace_base.h"
00016 #include "core/pool_func.hpp"
00017
00019 EngineRenewPool _enginerenew_pool("EngineRenew");
00020 INSTANTIATE_POOL_METHODS(EngineRenew)
00021
00022
00026 static EngineRenew *GetEngineReplacement(EngineRenewList erl, EngineID engine, GroupID group)
00027 {
00028 EngineRenew *er = (EngineRenew *)erl;
00029
00030 while (er != NULL) {
00031 if (er->from == engine && er->group_id == group) return er;
00032 er = er->next;
00033 }
00034 return NULL;
00035 }
00036
00042 void RemoveAllEngineReplacement(EngineRenewList *erl)
00043 {
00044 EngineRenew *er = (EngineRenew *)(*erl);
00045 EngineRenew *next;
00046
00047 while (er != NULL) {
00048 next = er->next;
00049 delete er;
00050 er = next;
00051 }
00052 *erl = NULL;
00053 }
00054
00064 EngineID EngineReplacement(EngineRenewList erl, EngineID engine, GroupID group, bool *replace_when_old)
00065 {
00066 const EngineRenew *er = GetEngineReplacement(erl, engine, group);
00067 if (er == NULL && (group == DEFAULT_GROUP || (Group::IsValidID(group) && !Group::Get(group)->replace_protection))) {
00068
00069 er = GetEngineReplacement(erl, engine, ALL_GROUP);
00070 }
00071 if (replace_when_old != NULL) *replace_when_old = er == NULL ? false : er->replace_when_old;
00072 return er == NULL ? INVALID_ENGINE : er->to;
00073 }
00074
00085 CommandCost AddEngineReplacement(EngineRenewList *erl, EngineID old_engine, EngineID new_engine, GroupID group, bool replace_when_old, DoCommandFlag flags)
00086 {
00087
00088 EngineRenew *er = GetEngineReplacement(*erl, old_engine, group);
00089 if (er != NULL) {
00090 if (flags & DC_EXEC) {
00091 er->to = new_engine;
00092 er->replace_when_old = replace_when_old;
00093 }
00094 return CommandCost();
00095 }
00096
00097 if (!EngineRenew::CanAllocateItem()) return CMD_ERROR;
00098
00099 if (flags & DC_EXEC) {
00100 er = new EngineRenew(old_engine, new_engine);
00101 er->group_id = group;
00102 er->replace_when_old = replace_when_old;
00103
00104
00105 er->next = (EngineRenew *)(*erl);
00106 *erl = (EngineRenewList)er;
00107 }
00108
00109 return CommandCost();
00110 }
00111
00120 CommandCost RemoveEngineReplacement(EngineRenewList *erl, EngineID engine, GroupID group, DoCommandFlag flags)
00121 {
00122 EngineRenew *er = (EngineRenew *)(*erl);
00123 EngineRenew *prev = NULL;
00124
00125 while (er != NULL) {
00126 if (er->from == engine && er->group_id == group) {
00127 if (flags & DC_EXEC) {
00128 if (prev == NULL) {
00129
00130 *erl = (EngineRenewList)er->next;
00131 } else {
00132
00133 prev->next = er->next;
00134 }
00135 delete er;
00136 }
00137 return CommandCost();
00138 }
00139 prev = er;
00140 er = er->next;
00141 }
00142
00143 return CMD_ERROR;
00144 }