package org.opentdc.wtt.file;

import java.util.ArrayList;
import java.util.List;

import org.opentdc.wtt.CompanyModel;

public class WttCompany {
	private CompanyModel companyModel;
	private ArrayList<WttProject> projects;

	public WttCompany() {
	}
	
	public CompanyModel getCompanyModel() {
		return companyModel;
	}
	
	public void setCompanyModel(CompanyModel companyModel) {
		this.companyModel = companyModel;
	}
	
	public List<WttProject> getProjects() {
		return projects;
	}
	
	public void setProjects(ArrayList<WttProject> projects) {
		this.projects = projects;
	}
	
	public void addProject(WttProject p) {
		this.projects.add(p);
	}
	
	public boolean removeProject(WttProject p) {
		return this.projects.remove(p);
	}
}

@@ -241,16 +241,58 @@
 any() %7B%0A
+%09%09projects = new ArrayList%3CWttProject%3E();%0A
 %09%7D%0A%09%0A%09pu
@@ -765,10 +765,183 @@
 (p);%0A%09%7D%0A
+%09%0A%09public void setTitle(String title) %7B%0A%09%09companyModel.setTitle(title);%0A%09%7D%0A%0A%09public void setDescription(String description) %7B%0A%09%09companyModel.setDescription(description);%0A%09%7D%0A
 %7D%0A

initialize projects, add setters for Title and Description