package org.openmrs.module.appointmentscheduling;

import org.openmrs.Encounter;

import java.util.LinkedHashSet;
import java.util.Set;

public class MEncounter extends Encounter {

    private Set<Appointment> appointments;

    public MEncounter() {
        super();
    }

    public  Set<Appointment> getAppointments(){
        if(appointments == null){
            appointments = new LinkedHashSet<Appointment>();
        }
        return appointments;
    }

    public void setAppointments(Set<Appointment> appointments) {
        this.appointments = appointments;
    }

    public void addAppointment(Appointment appointment){
        if(appointment != null){
            appointment.setEncounter(this);
            getAppointments().add(appointment);
        }
    }

    public void removeAppointment(Appointment appointment){
        if(appointments != null){
            appointments.remove(appointment);
        }
    }
}