Author: mdryakhlenkov Date: 2007-07-05 05:43:16 -0400 (Thu, 05 Jul 2007) New Revision: 2296 Modified: trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/model/OrmShape.java Log: EXIN-366: Adding elements on the diagram by double-click on fields of classes which have additional information in mapping files. Modified: trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/model/OrmShape.java =================================================================== --- trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/model/OrmShape.java 2007-07-05 09:42:13 UTC (rev 2295) +++ trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/veditor/editors/model/OrmShape.java 2007-07-05 09:43:16 UTC (rev 2296) @@ -153,13 +153,13 @@ } } - public Shape getChild(Object ormElement) { + public Shape getChild(Column ormElement) { Shape shape = null; Iterator iter = getChildren().iterator(); while (iter.hasNext()) { Shape child = (Shape)iter.next(); Object childElement = child.getOrmElement(); - if (ormElement == childElement) { + if (childElement instanceof Column && ormElement.getName().equals(((Column)childElement).getName())) { return child; } }