Total.java

package org.softlang.company.features;

import org.softlang.company.features.visitor.*;
import org.softlang.company.model.*;

public class Total implements ReturningVisitor<Double> {

    public Double visit(Company o) {
        double total = 0;
        for (Department d : o.getDepts())
            total += d.accept(this);
        return total;
    }

    public Double visit(Department o) {
        double total = 0;
        total += o.getManager().accept(this);
        for (Subunit s : o.getSubunits())
            total += s.accept(this);
        return total;
    }

    public Double visit(Employee o) {
        return o.getSalary();
    }

}