import pygame

class Shelf:
    def __init__(self, screen, cell, we, ns, kierunek, rodzaj):
        self.cell = cell
        self.ns = ns
        self.we = we
        self.screen = screen
        self.image = pygame.image.load(r'images/re.png')
        self.image = pygame.transform.scale(self.image, (cell, cell))
        if (kierunek==1):
            self.image = pygame.transform.rotate(self.image, 180)
        elif (kierunek==2):
            self.image = pygame.transform.rotate(self.image, 270)
        elif (kierunek==3):
            self.image = pygame.transform.rotate(self.image, 90)
    def draw(self):
        self.screen.blit(self.image, (self.cell*self.ns, self.cell*self.we))