tests/org.eclipse.gmf.tests/src/org/eclipse/gmf/tests/Utils.java

/*
 * Copyright (c) 2005, 2008 Borland Software Corporation
 * 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Artem Tikhomirov (Borland) - initial API and implementation
 */
package org.eclipse.gmf.tests;

import java.util.Calendar;

import org.eclipse.core.runtime.IStatus;
import org.eclipse.emf.codegen.ecore.genmodel.GenClass;
import org.eclipse.emf.codegen.ecore.genmodel.GenClassifier;
import org.eclipse.emf.codegen.ecore.genmodel.GenModel;
import org.eclipse.emf.common.util.TreeIterator;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.gmf.internal.bridge.genmodel.BasicGenModelAccess;
import org.eclipse.swt.widgets.Display;

/**
 * @author artem
 *
 */
public class Utils {

	/**
	 * FIXME use DummyGenModel instead of BasicGenModelAccess
	 * Create in-memory genmodel for provided (in-memory) domain model
	 * @param aModel source model
	 * @return initilized genModel, ready to run code generation
	 */
	public static GenModel createGenModel(EPackage aModel) {
		BasicGenModelAccess gmAccess = new BasicGenModelAccess(aModel);
		IStatus s = gmAccess.createDummy();
		assert s.isOK();
		GenModel genModel = gmAccess.model();
		// not sure I need these
		String pluginID = Utils.createUniquePluginID();
        genModel.setModelPluginID(pluginID);
        genModel.setModelDirectory("/" + pluginID + "/src/");
        genModel.setEditDirectory(genModel.getModelDirectory());
        return genModel;
	}

	public static GenClass findGenClass(GenModel genModel, String className) {
		for (TreeIterator<EObject> it = genModel.eAllContents(); it.hasNext();) {
			EObject next = it.next();
			if (next instanceof GenClassifier == false) {
				continue;
			}
			if (next instanceof GenClass) {
				GenClass rc = (GenClass) next;
				if (rc.getName().equals(className)) {
					return rc;
				}
			}
			it.prune();
		}
		return null;
	}

	/**
	 * Tests need class matching using names as it's not always ok 
	 * to compare ecore classes as done in 
	 * {@link org.eclipse.gmf.internal.bridge.genmodel.GenModelMatcher#findGenClass(EClass)} 
	 */
	public static GenClass findGenClass(GenModel genModel, EClass domainClass) {
		return findGenClass(genModel, domainClass.getName());
	}

	public static String createUniquePluginID() {
		return String.format("sample.t%1$tH-%1$tM-%1$tS.%1$tL", Calendar.getInstance());
	}

	/**
	 * @return false if timeout broke the loop
	 */
	public static boolean dispatchDisplayMessages(boolean[] condition, int timeoutSeconds) {
		assert Display.getCurrent() != null;
		final long start = System.currentTimeMillis();
		final long deltaMillis = timeoutSeconds * 1000; 
		do {
			while (Display.getCurrent().readAndDispatch()) {
				;
			}
		} while (condition[0] && (System.currentTimeMillis() - start) < deltaMillis);
		return !condition[0];
	}

	/**
	 * @return false if message re-dispatch was broken by timeout
	 */
	public static boolean dispatchDisplayMessages(int timeoutSeconds) {
		final long start = System.currentTimeMillis();
		final long deltaMillis = timeoutSeconds  * 1000; 
		while (Display.getCurrent().readAndDispatch()) {
			if ((System.currentTimeMillis() - start) > deltaMillis) {
				return false;
			}
		}
		return true;
	}
}