newgrf_class_func.h

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * 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.
 * 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.
 * 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 <https://www.gnu.org/licenses/old-licenses/gpl-2.0>.
 */

 
#include "newgrf_class.h"
 
#include "table/strings.h"

template <typename Tspec, typename Tindex>
void NewGRFClass<Tspec, Tindex>::Reset()
{
  NewGRFClass::classes.clear();
  NewGRFClass::classes.shrink_to_fit();
 
  NewGRFClass::InsertDefaults();
}

template <typename Tspec, typename Tindex>
Tindex NewGRFClass<Tspec, Tindex>::Allocate(uint32_t global_id)
{
  auto found = std::ranges::find(NewGRFClass::classes, global_id, &NewGRFClass::global_id);
 
  /* Id is already allocated, so reuse it. */
  if (found != std::end(NewGRFClass::classes)) return found->Index();
 
  /* More slots available, allocate a slot to the global id. */
  if (NewGRFClass::classes.size() < Tindex::End().base()) {
    auto it = NewGRFClass::classes.emplace(std::end(NewGRFClass::classes), global_id, STR_EMPTY);
    it->index = static_cast<Tindex>(std::distance(std::begin(NewGRFClass::classes), it));
    return it->Index();
  }
 
  GrfMsg(2, "ClassAllocate: already allocated {} classes, using default", Tindex::End().base());
  return static_cast<Tindex>(0);
}

template <typename Tspec, typename Tindex>
void NewGRFClass<Tspec, Tindex>::Insert(Tspec *spec)
{
  auto it = this->spec.insert(std::end(this->spec), spec);
  uint16_t index = static_cast<uint16_t>(std::distance(std::begin(this->spec), it));
  if (spec != nullptr) (*it)->index = index;
 
  if (this->IsUIAvailable(index)) this->ui_count++;
}

template <typename Tspec, typename Tindex>
void NewGRFClass<Tspec, Tindex>::Assign(Tspec *spec)
{
  assert(spec->class_index.base() < NewGRFClass::classes.size());
  Get(spec->class_index)->Insert(spec);
}

template <typename Tspec, typename Tindex>
NewGRFClass<Tspec, Tindex> *NewGRFClass<Tspec, Tindex>::Get(Tindex class_index)
{
  assert(class_index.base() < NewGRFClass::classes.size());
  return &NewGRFClass::classes[class_index.base()];
}

template <typename Tspec, typename Tindex>
uint NewGRFClass<Tspec, Tindex>::GetClassCount()
{
  return static_cast<uint>(NewGRFClass::classes.size());
}

template <typename Tspec, typename Tindex>
uint NewGRFClass<Tspec, Tindex>::GetUIClassCount()
{
  return std::ranges::count_if(NewGRFClass::classes, [](const auto &cls) { return cls.GetUISpecCount() > 0; });
}

template <typename Tspec, typename Tindex>
const Tspec *NewGRFClass<Tspec, Tindex>::GetSpec(uint index) const
{
  /* If the custom spec isn't defined any more, then the GRF file probably was not loaded. */
  return index < this->GetSpecCount() ? this->spec[index] : nullptr;
}

template <typename Tspec, typename Tindex>
const Tspec *NewGRFClass<Tspec, Tindex>::GetByGrf(uint32_t grfid, uint16_t local_id)
{
  for (const auto &cls : NewGRFClass::classes) {
    for (const auto &spec : cls.spec) {
      if (spec == nullptr) continue;
      if (spec->grf_prop.local_id != local_id) continue;
      if (spec->grf_prop.grfid == grfid) return spec;
    }
  }
 
  return nullptr;
}