ITECCourse.java

package week6_first_classes.OOPITECCourseManager_Constructor;

import java.util.ArrayList;

/**
 * Stores data about an ITEC course.
 *
 */
public class ITECCourse {

    //Data that an ITECCourse object needs to store
    String name;
    int code;
    ArrayList<String> students;
    int maxStudents;

    //Constructor
    ITECCourse(String courseName, int courseCode, int courseMaxStudents) {
        this.name = courseName;
        this.code = courseCode;
        this.students = new ArrayList<String>();  //Set up the students list
        this.maxStudents = courseMaxStudents;
    }

    void addStudent(String studentName) {
        //No need to create ArrayList, the constructor has already done it
        students.add(studentName);
    }


    void writeCourseInfo() {
        System.out.println("Course Name: " + name);
        System.out.println("Course Code: " + code);
        System.out.println("Students enrolled:");
        for (String student : students) {
            System.out.println(student);
        }
        System.out.println("There are " + getNumberOfStudents() + " students enrolled");
        System.out.println("The max number of students for this course is  " + maxStudents);
    }

    int getNumberOfStudents() {
        return this.students.size();
    }

}