public LinkCountView(MSystem system) {
super("Association", "# Links", Color.red);
fSystem = system;
Collection<MAssociation> associations = fSystem.model().associations();
fAssociations = associations.toArray(new MAssociation[0]);
Arrays.sort(fAssociations);
setNames(fAssociations);
fSystem.addChangeListener(this);
update();
}
private void update() {
int[] values = new int[fAssociations.length];
MSystemState systemState = fSystem.state();
for (int i = 0; i < fAssociations.length; i++) {
values[i] = systemState.linksOfAssociation((MAssociation) fAssociations[i]).size();
}
setValues(values);
}
/** Detaches the view from its model. */
public void detachModel() {
fSystem.removeChangeListener(this);
}
/**
* Create a new link between the (source or target) domain and the corr object
*
* @throws org.tzi.use.uml.sys.MSystemException
*/
public MLink createMLink(
Context ctx, MObject corrObject, MClass domainClass, MObject domainObject)
throws MSystemException {
MLink link = null;
MAssociation assoc = null;
String roleName = "";
MClass corrClass = ctx.model().getClass(corrObject.cls().name());
Set tmpClasses = new HashSet();
if (domainObject.name().equals(fSourceObject.getText()))
if (fSourceRole != null) roleName = fSourceRole.getText();
else roleName = fSourceObject.getClass().getName().toLowerCase();
else if (fTargetRole != null) roleName = fTargetRole.getText();
else roleName = fTargetObject.getClass().getName().toLowerCase();
tmpClasses.add(corrClass);
tmpClasses.add(domainClass);
Set<MAssociation> assocs = ctx.model().getAssociationsBetweenClasses(tmpClasses);
if (assocs.size() == 0) {
ModelFactory mf = new ModelFactory();
List<VarDecl> emptyQualifiers = Collections.emptyList();
try {
assoc = mf.createAssociation(domainClass.name() + "_" + corrClass.name());
assoc.addAssociationEnd(
mf.createAssociationEnd(
domainClass,
roleName,
new MMultiplicity(1, 1),
MAggregationKind.NONE,
false,
emptyQualifiers));
assoc.addAssociationEnd(
mf.createAssociationEnd(
corrClass,
"cp" + "_" + assoc.name(),
MMultiplicity.ZERO_MANY,
MAggregationKind.NONE,
false,
emptyQualifiers));
ctx.model().addAssociation(assoc);
MSystem system = new MSystem(ctx.model());
MSystemState temp = system.state();
// copy object, link from ctx.systemState -> temp
// Objects
for (Iterator iter = ctx.systemState().allObjects().iterator(); iter.hasNext(); ) {
MObject obj = (MObject) iter.next();
temp.createObject(obj.cls(), obj.name());
}
// Links
for (Iterator iter = ctx.systemState().allLinks().iterator(); iter.hasNext(); ) {
MLink obj = (MLink) iter.next();
temp.insertLink(obj);
}
ctx.setfCurrentSystemState(system.state());
} catch (MInvalidModelException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
} else {
Iterator it = assocs.iterator();
if (!it.hasNext()) throw new MSystemException();
assoc = (MAssociation) it.next();
}
List tmpList = new ArrayList<MObject>();
tmpList.add(domainObject);
tmpList.add(corrObject);
if (!ctx.systemState().hasLink(assoc, tmpList, null)) {
link = ctx.systemState().createLink(assoc, tmpList, null);
} else {
link = ctx.systemState().linkBetweenObjects(assoc, tmpList).iterator().next();
}
return link;
}