autoreplace.cpp

Go to the documentation of this file.
00001 /* $Id$ */
00002 
00003 /*
00004  * This file is part of OpenTTD.
00005  * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
00006  * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
00007  * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
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; // Empty list
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     /* We didn't find anything useful in the vehicle's own group so we will try ALL_GROUP */
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   /* Check if the old vehicle is already in the list */
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     /* Insert before the first element */
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) { // First element
00129           /* The second becomes the new first element */
00130           *erl = (EngineRenewList)er->next;
00131         } else {
00132           /* Cut this element out */
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 }