public void setLanguageObject(LanguageObject theLangObj) {
// language object must be cloned here so that the original isn't modified.
// this prevents the original from being modified even if the user cancels out of the builder.
LanguageObject langObj = (theLangObj == null) ? null : (LanguageObject) theLangObj.clone();
savedLangObj = langObj;
setOriginalSql(langObj);
setCurrentSql(langObj);
treeViewer.setLanguageObject(langObj);
treeViewer.selectRoot();
treeViewer.expandAll();
// Defect 22003 - needed a context for the original expression
// Providing a Lable at the top of the dialog.
String labelText =
Util.getString(PREFIX + "initialExpression")
+ //$NON-NLS-1$
CoreStringUtil.Constants.DBL_SPACE
+ CoreStringUtil.Constants.DBL_SPACE
+ Util.getString(PREFIX + "undefined"); // $NON-NLS-1$
if (savedLangObj != null) {
String loString = savedLangObj.toString();
if (loString.length() > 50) {
loString = loString.substring(0, 50) + "..."; // $NON-NLS-1$
}
labelText =
Util.getString(PREFIX + "initialExpression")
+ //$NON-NLS-1$
CoreStringUtil.Constants.DBL_SPACE
+ CoreStringUtil.Constants.DBL_SPACE
+ loString;
}
if (originalLanguageObjLabel != null) {
originalLanguageObjLabel.setText(labelText);
originalLanguageObjLabel.getParent().layout();
}
// select root tree node
Display.getDefault()
.asyncExec(
new Runnable() {
public void run() {
treeViewer.selectRoot();
}
});
}
/**
* @see
* org.teiid.designer.transformation.ui.builder.ILanguageObjectEditor#setLanguageObject(org.teiid.query.sql.LanguageObject)
*/
@Override
public void setLanguageObject(LanguageObject theLanguageObject) {
if (theLanguageObject == null) {
clear();
} else {
CoreArgCheck.isTrue(
(theLanguageObject instanceof Constant),
Util.getString(
PREFIX + "invalidLanguageObject", // $NON-NLS-1$
new Object[] {theLanguageObject.getClass().getName()}));
model.setLanguageObject(theLanguageObject);
}
}
private void addUndefinedCriteria(Criteria theCriteria, boolean theAndFlag) {
CoreArgCheck.isNotNull(theCriteria); // should not be calling if null
Object newSelection = StructuredSelection.EMPTY;
if (theCriteria instanceof CompoundCriteria) {
CompoundCriteria criteria = (CompoundCriteria) theCriteria;
criteria.addCriteria((Criteria) null);
refresh(true);
newSelection = contentProvider.getChildAt((criteria.getCriteriaCount() - 1), criteria);
} else if (theCriteria instanceof NotCriteria) {
// the contained Criteria must be a CompoundCriteria
addUndefinedCriteria(((NotCriteria) theCriteria).getCriteria(), theAndFlag);
} else if (theCriteria instanceof PredicateCriteria) {
CompoundCriteria compoundCriteria = new CompoundCriteria();
compoundCriteria.setOperator((theAndFlag) ? CompoundCriteria.AND : CompoundCriteria.OR);
compoundCriteria.addCriteria(theCriteria);
compoundCriteria.addCriteria((Criteria) null);
// modify parent here
LanguageObject parent = (LanguageObject) contentProvider.getParent(theCriteria);
int index = contentProvider.getChildIndex(theCriteria);
if (parent == null) {
setLanguageObject(compoundCriteria);
refresh(true);
// select undefined criteria
newSelection = contentProvider.getChildAt(1, contentProvider.getRoot());
expandToLevel(newSelection, ALL_LEVELS);
} else if ((parent instanceof NotCriteria) || (parent instanceof CompoundCriteria)) {
CompoundCriteria criteria = null;
if (parent instanceof NotCriteria) {
// then NotCriteria's contained criteria must be a CompoundCriteria
criteria = (CompoundCriteria) ((NotCriteria) theCriteria).getCriteria();
} else {
criteria = (CompoundCriteria) parent;
}
List crits = criteria.getCriteria();
crits.set(index, compoundCriteria);
refresh(true);
// select undefined criteria
newSelection = contentProvider.getChildAt(1, compoundCriteria);
} else {
CoreArgCheck.isTrue(
false,
Util.getString(
PREFIX + "unexpectedType", // $NON-NLS-1$
new Object[] {"addUndefinedCriteria", parent.getClass()})); // $NON-NLS-1$
}
}
// select appropriate tree item
final Object selection = newSelection;
Display.getDefault()
.asyncExec(
new Runnable() {
public void run() {
setSelection(new StructuredSelection(selection));
}
});
}
private void modifyLanguageObject(
Object theObject, LanguageObject theNewValue, boolean retainSelection) {
CoreArgCheck.isNotNull(theNewValue); // should not be null when modifying
//
// set the parent object to reflect the change in it's child. parent's can't be undefined
//
Object newSelection = StructuredSelection.EMPTY;
LanguageObject parent = (LanguageObject) contentProvider.getParent(theObject);
Object newValue = theNewValue;
if (parent == null) {
// root language object
setLanguageObject((LanguageObject) newValue);
refresh(true);
newSelection = contentProvider.getRoot();
expandToLevel(newSelection, ALL_LEVELS);
if (newSelection instanceof Function) {
Function function = (Function) newSelection;
if (function.getArgs().length > 0) {
newSelection = contentProvider.getChildAt(0, function);
}
}
} else if (parent instanceof Function) {
// set the arg to new value in parent
int index = contentProvider.getChildIndex(theObject);
Expression[] args = ((Function) parent).getArgs();
args[index] = (Expression) newValue;
refresh(true);
expandToLevel(parent, ALL_LEVELS);
newSelection = contentProvider.getChildAt(index, parent);
if (!retainSelection) {
if (newSelection instanceof Function) {
// if function arg change to be a function, select first function arg
if (contentProvider.getChildCount(newSelection) > 0) {
newSelection = contentProvider.getChildAt(0, newSelection);
}
} else {
// select next sibling function arg if exists
if ((args.length - 1) > index) {
newSelection = contentProvider.getChildAt(index + 1, parent);
} else if (index > 0) {
newSelection = contentProvider.getChildAt(0, parent);
} else {
newSelection = contentProvider.getChildAt(index, parent);
}
}
}
} else if (parent instanceof Criteria) {
// theLangObj must also be a Criteria
// since NotCriteria aren't really edited in the editor (their contained criteria is). Need to
// save the state in order to restore it when modifying
Criteria newCriteria = (Criteria) newValue;
int index = contentProvider.getChildIndex(theObject);
if ((parent instanceof CompoundCriteria) || (parent instanceof NotCriteria)) {
CompoundCriteria compoundCriteria = null;
if (parent instanceof CompoundCriteria) {
compoundCriteria = (CompoundCriteria) parent;
} else {
compoundCriteria = (CompoundCriteria) ((NotCriteria) parent).getCriteria();
}
List criteriaCollection = compoundCriteria.getCriteria();
criteriaCollection.set(index, newCriteria);
refresh(true);
if (retainSelection) {
newSelection = criteriaCollection.get(index);
} else {
// set selection to next sibling criteria or to the first sibling
if ((criteriaCollection.size() - 1) > index) {
newSelection = criteriaCollection.get(index + 1);
} else if (index > 0) {
newSelection = criteriaCollection.get(0);
} else {
newSelection = contentProvider.getChildAt(index + 1, parent);
}
}
} else {
CoreArgCheck.isTrue(
false,
Util.getString(
PREFIX + "unexpectedType", // $NON-NLS-1$
new Object[] {
"modifyLanguageObject", //$NON-NLS-1$
parent.getClass().getName()
}));
}
expandToLevel(parent, ALL_LEVELS);
}
// select next node
setSelection(
(newSelection == null) ? StructuredSelection.EMPTY : new StructuredSelection(newSelection));
}