package org.cpsolver.studentsct.reservation;

import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import org.cpsolver.studentsct.model.AreaClassificationMajor;
import org.cpsolver.studentsct.model.Offering;
import org.cpsolver.studentsct.model.Student;

/**
 * Group restriction. Students are matched based on their academic area.
 * If classifications and/or majors are included, student must match on them as well.  
 * <br>
 * <br>
 * 
 * @author  Tomáš Müller
 * @version StudentSct 1.3 (Student Sectioning)<br>
 *          Copyright (C) 2007 - 2020 Tomáš Müller<br>
 *          <a href="mailto:muller@unitime.org">muller@unitime.org</a><br>
 *          <a href="http://muller.unitime.org">http://muller.unitime.org</a><br>
 * <br>
 *          This library is free software; you can redistribute it and/or modify
 *          it under the terms of the GNU Lesser General Public License as
 *          published by the Free Software Foundation; either version 3 of the
 *          License, or (at your option) any later version. <br>
 * <br>
 *          This library is distributed in the hope that it will be useful, but
 *          WITHOUT ANY WARRANTY; without even the implied warranty of
 *          MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 *          Lesser General Public License for more details. <br>
 * <br>
 *          You should have received a copy of the GNU Lesser General Public
 *          License along with this library; if not see
 *          <a href='http://www.gnu.org/licenses/'>http://www.gnu.org/licenses/</a>.
 */
public class CurriculumRestriction extends Restriction {
    private Set<String> iAcadAreas = new HashSet<String>();
    private Set<String> iClassifications = new HashSet<String>();
    private Set<String> iMajors = new HashSet<String>();
    private Set<String> iMinors = new HashSet<String>();
    private Map<String, Set<String>> iConcentrations = null;
    
    /**
     * Constructor
     * @param id restriction unique id
     * @param offering instructional offering on which the restriction is set
     * @param acadAreas one or more academic areas
     * @param classifications zero or more classifications (classifications must match if not empty)
     * @param majors zero or more majors (majors must match if not empty)
     * @param minors zero or more majors (minors must match if not empty)
     */
    public CurriculumRestriction(long id, Offering offering, Collection<String> acadAreas, Collection<String> classifications, Collection<String> majors, Collection<String> minors) {
        super(id, offering);
        if (acadAreas != null)
            iAcadAreas.addAll(acadAreas);
        if (classifications != null)
            iClassifications.addAll(classifications);
        if (majors != null)
            iMajors.addAll(majors);
        if (minors != null)
            iMinors.addAll(minors);
    }
    
    /**
     * Constructor
     * @param id restriction unique id
     * @param offering instructional offering on which the restriction is set
     * @param acadAreas one or more academic areas
     * @param classifications zero or more classifications (classifications must match if not empty)
     * @param majors zero or more majors (majors must match if not empty)
     */
    public CurriculumRestriction(long id, Offering offering, Collection<String> acadAreas, Collection<String> classifications, Collection<String> majors) {
        super(id, offering);
        if (acadAreas != null)
            iAcadAreas.addAll(acadAreas);
        if (classifications != null)
            iClassifications.addAll(classifications);
        if (majors != null)
            iMajors.addAll(majors);
    }
    
    /**
     * Constructor
     * @param id restriction unique id
     * @param offering instructional offering on which the restriction is set
     * @param acadArea academic area
     * @param classifications zero or more classifications (classifications must match if not empty)
     * @param majors zero or more majors (majors must match if not empty)
     */
    @Deprecated
    public CurriculumRestriction(long id, Offering offering, String acadArea, Collection<String> classifications, Collection<String> majors) {
        super(id, offering);
        iAcadAreas.add(acadArea);
        if (classifications != null)
            iClassifications.addAll(classifications);
        if (majors != null)
            iMajors.addAll(majors);
    }

    /**
     * Academic areas
     * @return selected academic areas
     */
    public Set<String> getAcademicAreas() {
        return iAcadAreas;
    }
    
    /**
     * Academic area
     * @return selected academic area
     */
    @Deprecated
    public String getAcademicArea() {
        if (getAcademicAreas().isEmpty()) return "";
        return getAcademicAreas().iterator().next();
    }
    
    /**
     * Majors
     * @return selected majors
     */
    public Set<String> getMajors() {
        return iMajors;
    }
    
    /**
     * Minors
     * @return selected minors
     */
    public Set<String> getMinors() {
        return iMinors;
    }
    
    /**
     * Academic classifications
     * @return selected academic classifications
     */
    public Set<String> getClassifications() {
        return iClassifications;
    }
    
    /** Concentrations for major */
    public Set<String> getConcentrations(String major) {
        return (iConcentrations == null ? null : iConcentrations.get(major));
    }
    
    /** Add concentration for major */
    public void addConcentration(String major, String concentration) {
        if (iConcentrations == null) iConcentrations = new HashMap<String, Set<String>>();
        Set<String> concentrations = iConcentrations.get(major);
        if (concentrations == null) {
            concentrations = new HashSet<String>();
            iConcentrations.put(major, concentrations);
        }
        concentrations.add(concentration);
    }

    /**
     * Check the area, classifications and majors
     */
    @Override
    public boolean isApplicable(Student student) {
        if (!getMajors().isEmpty() || getMinors().isEmpty())
            for (AreaClassificationMajor acm: student.getAreaClassificationMajors())
                if (getAcademicAreas().contains(acm.getArea()) &&
                    (getClassifications().isEmpty() || getClassifications().contains(acm.getClassification())) &&
                    (getMajors().isEmpty() || getMajors().contains(acm.getMajor()))) {
                    Set<String> conc = getConcentrations(acm.getMajor());
                    if (conc != null && !conc.isEmpty()) {
                        return acm.getConcentration() != null && conc.contains(acm.getConcentration());
                    } else {
                        return true;
                    }
                }
        if (!getMinors().isEmpty())
            for (AreaClassificationMajor acm: student.getAreaClassificationMinors())
                if (getAcademicAreas().contains(acm.getArea()) &&
                    (getClassifications().isEmpty() || getClassifications().contains(acm.getClassification())) &&
                    (getMinors().contains(acm.getMajor())))
                        return true;
        return false;
    }
}