from toontown.building.SuitPlannerInteriorAI import SuitPlannerInteriorAI


class SuitPlannerCogdoInteriorAI(SuitPlannerInteriorAI):
    def __init__(self, cogdoLayout, bldgLevel, bldgTrack, zone):
        self._cogdoLayout = cogdoLayout
        SuitPlannerInteriorAI.__init__(self, self._cogdoLayout.getNumGameFloors(), bldgLevel, bldgTrack, zone)

    def _genSuitInfos(self, numFloors, bldgLevel, bldgTrack):
        SuitPlannerInteriorAI._genSuitInfos(self, self._cogdoLayout.getNumFloors(), bldgLevel, bldgTrack)