LoyaltyProgram.java

/**
 * Copyright (C) 2009 by Claas Wilke (info@claaswilke.de)
 * 
 * This file is part of the Royal and Loyal Example of Dresden OCL2 for Eclipse.
 * 
 * Dresden OCL2 for Eclipse 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.
 * 
 * Dresden OCL2 for Eclipse 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.
 * 
 * You should have received a copy of the GNU Lesser General Public License along 
 * with Dresden OCL2 for Eclipse. If not, see <http://www.gnu.org/licenses/>.
 */
package org.dresdenocl.examples.royalandloyal;

import org.dresdenocl.ocl2java.types.OclOrderedSet;
import org.dresdenocl.ocl2java.types.OclSet;

/**
 * <p>
 * Represents an implementation of the class {@link LoyaltyProgram} of the
 * Loyals and Royals example.
 * </p>
 * 
 * @author Claas Wilke
 */
public class LoyaltyProgram {

	protected String name;

	protected OclSet<Customer> participants = new OclSet<Customer>();

	protected OclSet<ProgramPartner> partners = new OclSet<ProgramPartner>();

	protected OclOrderedSet<ServiceLevel> levels =
			new OclOrderedSet<ServiceLevel>();

	protected Membership membership;

	public void addService(ProgramPartner aPartner, ServiceLevel aLevel,
			Service aService) {

		aPartner.addDeliveredService(aService);
		this.partners.add(aPartner);

		aLevel.addAvailableService(aService);
		this.levels.add(aLevel);
	}

	public Boolean enroll(Customer aCustomer) {

		return this.participants.add(aCustomer);
	}

	public String getName() {

		return name;
	}

	public void setName(String name) {

		this.name = name;
	}

	public Membership getMembership() {

		return membership;
	}

	public void setMembership(Membership membership) {

		this.membership = membership;

		membership.setProgram(this);
	}

	public OclSet<Customer> getParticipants() {

		return participants;
	}

	public OclSet<ProgramPartner> getPartners() {

		return partners;
	}

	public void addPartner(ProgramPartner aPartner) {

		this.partners.add(aPartner);
	}

	public OclOrderedSet<ServiceLevel> getLevels() {

		return levels;
	}

	public void addLevel(ServiceLevel aLevel) {

		this.levels.add(aLevel);
	}

	@Override
	public String toString() {

		StringBuilder builder = new StringBuilder();
		builder.append("LoyaltyProgram [");
		if (levels != null) {
			builder.append("levels=");
			builder.append("[");
			for (ServiceLevel level : levels) {
				builder.append(level.getName());
				builder.append(", ");
			}
			builder.append("]");
			builder.append(", ");
		}
		if (name != null) {
			builder.append("name=");
			builder.append(name);
			builder.append(", ");
		}
		if (participants != null) {
			builder.append("participants=");
			builder.append("[");
			for (Customer participant : participants) {
				builder.append(participant.getName());
				builder.append(", ");
			}
			builder.append("]");
			builder.append(", ");
		}
		if (partners != null) {
			builder.append("partners=");
			builder.append("[");
			for (ProgramPartner partner : partners) {
				builder.append(partner.getName());
				builder.append(", ");
			}
			builder.append("]");
		}
		builder.append("]");
		return builder.toString();
	}
}