from elems import Element
from mobles import MobleModular

class MobleModularAF(MobleModular):

    def __init__(self, alt, ample, fons, num_cossos):
        super().__init__(alt, ample, num_cossos)
        self.fons = fons
        
    def volum(self):
        return float(self.alt * self.ample * self.fons)

    def espai_lliure(self):
        vocupat = 0.0
        area = self.fons * self.ample / self.ncossos
        for icos in range(self.ncossos):
            for ielem in range(self.nelems(icos)):
                vocupat = vocupat + area * self[icos, ielem].alcada
        return self.volum() - vocupat