source: MotivateDuringTherapy/src/main/java/tudelft/mentalhealth/motivatepersisting/Category.java

package tudelft.mentalhealth.motivatepersisting;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * The catogory (main content type) of a possible Statement.
 */
enum Category {
        EMPATHY(Subcategory.NEGATIVE), //
        MOTIVATION(Subcategory.COMPLIMENT, Subcategory.GOONLIKETHIS, Subcategory.ITSGOINGWELL, Subcategory.HOLDON), //
        GIVEPERSPECTIVE, //
        TOGETHER, //
        FUTURE(Subcategory.PCL), //
        NOTEPCL(Subcategory.STABLE, Subcategory.DROPPING, Subcategory.RISING), //
        NOTETRUST(Subcategory.HIGH, Subcategory.LOW);

        private List<Subcategory> allowedSubcategories;

        Category(Subcategory... subcategories) {
                this.allowedSubcategories = Arrays.asList(subcategories);
        }

        /**
         * 
         * @param sub
         * @return true iff sub is an allowed subcategory for this category.
         */
        public boolean isSubcategoryAllowed(Subcategory sub) {
                return allowedSubcategories.contains(sub);
        }

        /**
         * @return the possible {@link Subcategory}s of this type. Returns empty list if
         *         this category has no subcategories.
         */
        public List<Subcategory> getSubCategories() {
                return Collections.unmodifiableList(allowedSubcategories);
        }

}