filemanager.java

P2P模拟器P2Psim的程序源码

package graph;

import java.io.FileOutputStream;
import java.io.FileInputStream;
import java.io.ObjectOutputStream;
import java.io.ObjectInputStream;

import org.jgraph.graph.DefaultGraphModel;

import org.jgraph.graph.DefaultGraphCell;

public class FileManager
{
	public FileManager()
	{}
		
	public DefaultGraphModel open(String filename)
	{
		try
		{
			FileInputStream fis = new FileInputStream(filename);
			ObjectInputStream ois = new ObjectInputStream(fis);
	
			DefaultGraphModel model = ((DefaultGraphModel) ois.readObject());
	
			ois.close();		
			fis.close();
			
			return model;
			
		}
		catch(Exception e)
		{
			e.printStackTrace();
			return null;
		}		
	}

	public void save(String filename, DefaultGraphModel model)
	{
		for (int i=0; i<model.getRootCount(); i++)
		{
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("Position");
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("Last_Impulse");
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("Current_Impulse");
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("GEM-TEMPORARY-DATA");
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("Temperature");
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("Mass_Index");
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("Relatives");
			((DefaultGraphCell)model.getRootAt(i)).getAttributes().remove("Skew_Gauge");
			
			//for (int j=0; j<((DefaultGraphCell)model.getRootAt(i)).getAttributes().values().size(); j++)
			//{
				
				//System.out.print(((DefaultGraphCell)model.getRootAt(i)).getAttributes().keySet().toArray()[j]);
				//System.out.print(" -> ");
				//System.out.println(((DefaultGraphCell)model.getRootAt(i)).getAttributes().values().toArray()[j]);
				
				
			//}			
			
		}
		
		try
		{
			FileOutputStream fos = new FileOutputStream(filename, false);
			ObjectOutputStream oos = new ObjectOutputStream(fos);
		
			oos.writeObject(model);
		
			oos.close();
			fos.close();
		}
		catch(Exception e)
		{
			System.out.println("There was a problem");
			//e.printStackTrace();
			//System.exit(-1);
		}
	}

}