ATCS/src/com/gpl/rpg/atcontentstudio/model/saves/SavedGamesSet.java

package com.gpl.rpg.atcontentstudio.model.saves;

import java.awt.Image;
import java.io.File;
import java.io.IOException;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.Vector;

import javax.swing.tree.TreeNode;

import com.gpl.rpg.atcontentstudio.Notification;
import com.gpl.rpg.atcontentstudio.model.GameSource.Type;
import com.gpl.rpg.atcontentstudio.model.Project;
import com.gpl.rpg.atcontentstudio.model.ProjectTreeNode;
import com.gpl.rpg.atcontentstudio.model.gamedata.GameDataSet;
import com.gpl.rpg.atcontentstudio.ui.DefaultIcons;

public class SavedGamesSet implements ProjectTreeNode, Serializable {

	private static final long serialVersionUID = -6565834239789184087L;

	public Vector<SavedGame> saves; //For simulations.

	public Project parent;
	
	public SavedGamesSet(Project parent) {
		this.parent = parent;
		saves = new Vector<SavedGame>();
	}
	
	public void refreshTransients() {
		for (SavedGame save : saves) {
			try {
				save.refreshTransients(this);
			} catch (IOException e) {
				Notification.addError(e.getMessage());
			}
		}
	}
	
	public void addSave(File f) {
		try {
			ProjectTreeNode higherEmptyParent = this;
			while (higherEmptyParent != null) {
				if (higherEmptyParent.getParent() != null && ((ProjectTreeNode)higherEmptyParent.getParent()).isEmpty()) higherEmptyParent = (ProjectTreeNode)higherEmptyParent.getParent();
				else break;
			}
			if (higherEmptyParent == this && !this.isEmpty()) higherEmptyParent = null;
			SavedGame node = new SavedGame(this, f); 
			saves.add(node);
			if (higherEmptyParent != null) higherEmptyParent.notifyCreated();
			else node.notifyCreated();
		} catch (IOException e) {
			Notification.addError(e.getMessage());
		}
	}
	
	public SavedGame getSave(File f) {
		for (SavedGame save : saves) {
			if (save.savedFile.equals(f)) return save;
		}
		return null;
	}
		
	@Override
	public Enumeration<? extends ProjectTreeNode> children() {
		return saves.elements();
	}

	@Override
	public boolean getAllowsChildren() {
		return true;
	}

	@Override
	public TreeNode getChildAt(int arg0) {
		return saves.elementAt(arg0);
	}

	@Override
	public int getChildCount() {
		return saves.size();
	}

	@Override
	public int getIndex(TreeNode arg0) {
		return saves.indexOf(arg0);
	}

	@Override
	public TreeNode getParent() {
		return parent;
	}

	@Override
	public boolean isLeaf() {
		return false;
	}

	@Override
	public void childrenAdded(List<ProjectTreeNode> path) {
		path.add(0, this);
		parent.childrenAdded(path);
	}
	@Override
	public void childrenChanged(List<ProjectTreeNode> path) {
		path.add(0, this);
		parent.childrenChanged(path);
	}
	@Override
	public void childrenRemoved(List<ProjectTreeNode> path) {
		if (path.size() == 1 && this.getChildCount() == 1) {
			childrenRemoved(new ArrayList<ProjectTreeNode>());
		} else {
			path.add(0, this);
			parent.childrenRemoved(path);
		}
	}
	@Override
	public void notifyCreated() {
		childrenAdded(new ArrayList<ProjectTreeNode>());
		for (SavedGame s : saves) {
			s.notifyCreated();
		}
	}
	@Override
	public String getDesc() {
		return (needsSaving() ? "*" : "")+"Saved games";
	}

	@Override
	public Project getProject() {
		return parent.getProject();
	}
	

	@Override
	public Image getIcon() {
		return getOpenIcon();
	}
	@Override
	public Image getClosedIcon() {
		return DefaultIcons.getSavClosedIcon();
	}
	@Override
	public Image getLeafIcon() {
		return DefaultIcons.getSavClosedIcon();
	}
	@Override
	public Image getOpenIcon() {
		return DefaultIcons.getSavOpenIcon();
	}
	

	@Override
	public GameDataSet getDataSet() {
		return null;
	}
	@Override
	public Type getDataType() {
		return null;
	}
	
	@Override
	public boolean isEmpty() {
		return saves.isEmpty();
	}
	

	@Override
	public boolean needsSaving() {
		for (ProjectTreeNode node : saves) {
			if (node.needsSaving()) return true;
		}
		return false;
	}
}