public class CDPersistencyBehavior extends DefaultPersistencyBehavior {

	public CDPersistencyBehavior(DiagramEditor diagramEditor) {
		super(diagramEditor);
	}

	@Override
	public Diagram loadDiagram(URI uri) {

		ClassObject textObject = new ClassObject("TestAdd");

		if (uri != null) {

			// final TransactionalEditingDomain editingDomain = diagramEditor
			// .getEditingDomain();

			// Create the diagram and its file
			Diagram diagram = Graphiti.getPeCreateService().createDiagram(
					"CDDiagramType", "Testing", true); //$NON-NLS-1$

			IDiagramTypeProvider dtp = GraphitiUi.getExtensionManager()
					.createDiagramTypeProvider(diagram,
							"demo.cd.diagram.CDDiagramTypeProvider"); //$NON-NLS-1$

			IFeatureProvider fp = dtp.getFeatureProvider();

			AddContext ac = new AddContext();
			ac.setTargetContainer(diagram);
			ac.setNewObject(textObject);
			ac.setLocation(10, 10);
			ac.setSize(50, 50);

			PictogramElement pe = fp.addIfPossible(ac);

			CDPatternFeatureProvider cpfp = (CDPatternFeatureProvider) fp;
			CDIndependenceSolver cdis = cpfp.getCDIndependenceSolver();

			cdis.printMap();
			cdis.printClasses();

			return diagram;
		}
		return null;

		// return super.loadDiagram(uri);
	}

	@Override
	public void saveDiagram(IProgressMonitor monitor) {
		super.saveDiagram(monitor);
	}

}