"""
Floor entity class
"""


import pygame


class Floor(pygame.sprite.Sprite):
    def __init__(self, mainGameClass):
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.Surface((mainGameClass.getScreenWidth(),
                                    mainGameClass.getFloorHeight()*101))
        self.image.fill((255, 229, 180))
        self.rect = self.image.get_rect()
        self.rect.center = (mainGameClass.getScreenWidth()/2,
                            mainGameClass.getScreenHeight() +
                            mainGameClass.getFloorHeight()*49.5)