Method augmentUnitData

bwapi/BWAPI/Source/BWAPI/GameUnits.cpp:258–316

    }
  }
  void GameImpl::augmentUnitData()
  {
    //this function modifies the extracted unit data for build unit, loaded units, larva, and interceptors
    for(Unit ui : accessibleUnits)
    {
      UnitImpl *u = static_cast<UnitImpl*>(ui);
      UnitImpl* orderTargetUnit = UnitImpl::BWUnitToBWAPIUnit(u->getOriginalRawData->orderTarget.pUnit);
      UnitImpl* buildUnit       = UnitImpl::BWUnitToBWAPIUnit(u->getOriginalRawData->currentBuildUnit);
      if ( orderTargetUnit && orderTargetUnit->exists() && u->getOrder() == Orders::ConstructingBuilding )
      {
        UnitImpl* j             = orderTargetUnit;
        u->self->buildUnit      = server.getUnitID(j);
        u->self->isConstructing = true;
        u->self->isIdle         = false;
        u->self->buildType      = j->self->type;
        j->self->buildUnit      = server.getUnitID(u);
        j->self->isConstructing = true;
        j->self->isIdle         = false;
        j->self->buildType      = j->self->type;
      }
      else if ( u->getAddon() && !u->getAddon()->isCompleted() )
      {
        UnitImpl* j             = static_cast<UnitImpl*>(u->getAddon());
        u->self->buildUnit      = server.getUnitID(j);
        u->self->isConstructing = true;
        u->self->isIdle         = false;
        u->self->buildType      = j->self->type;
        j->self->buildUnit      = server.getUnitID(u);
        j->self->isConstructing = true;
        j->self->isIdle         = false;
        j->self->buildType      = j->self->type;
      }
      else if ( buildUnit && buildUnit->exists() && u->isTraining() )
      {
        // Apply buildUnit symmetry
        UnitImpl* j             = buildUnit;
        j->self->buildUnit      = server.getUnitID(u);
        j->self->isConstructing = true;
        j->self->isIdle         = false;
        j->self->buildType      = j->self->type;
      }
      if (u->getTransport())
      {
        static_cast<UnitImpl*>(u->getTransport())->loadedUnits.insert(u);
      }

      if ( u->getHatchery() )
      {
        UnitImpl* hatchery = static_cast<UnitImpl*>(u->getHatchery());
        hatchery->connectedUnits.insert(u);
        if (hatchery->connectedUnits.size() >= 3)
          hatchery->self->remainingTrainTime = 0;
      }
      if (u->getCarrier())
      {
        static_cast<UnitImpl*>(u->getCarrier())->connectedUnits.insert(u);
      }
    }