class Resistance(Sketchy): def setup(self): width, height = [600, 400] self.liquid = Liquid(width/2, height/2, width, height) self.balls = [] for i in range(0, 10): self.balls.append(Ball(width, height)) self.size(width, height) def update(self): width, height = [600, 400] for i in range(len(self.balls)): if self.liquid.isInside(self.balls[i]): self.balls[i].friction = self.liquid.resist(self.balls[i]) applyForce(self.balls[i], self.balls[i].wind, self.balls[i].gravity, self.balls[i].friction) else: applyForce(self.balls[i], self.balls[i].wind, self.balls[i].gravity) self.balls[i].update(width, height) checkEdges(self.balls[i], width, height) def draw(self, g): g.background(1, 1, 1) self.liquid.draw(g) for i in range(len(self.balls)): self.balls[i].draw(g)