prerna.reactor.database.metaeditor.GetOwlMetamodelReactor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation

SEMOSS

The newest version!

package prerna.reactor.database.metaeditor;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import prerna.engine.api.IDatabaseEngine;
import prerna.masterdatabase.utility.MetamodelVertex;
import prerna.reactor.masterdatabase.util.GenerateMetamodelLayout;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.Constants;
import prerna.util.Utility;

public class GetOwlMetamodelReactor extends AbstractMetaEditorReactor {

	public GetOwlMetamodelReactor() {
		this.keysToGet = new String[]{ReactorKeysEnum.DATABASE.getKey()};
	}
	
	@Override
	public NounMetadata execute() {
		organizeKeys();
		String databaseId = this.keyValue.get(this.keysToGet[0]);
		// we may have the alias
		databaseId = testDatabaseId(databaseId, false);
		IDatabaseEngine database = Utility.getDatabase(databaseId);
		Map metamodelObject = database.getMetamodel();
		Object[] nodes = metamodelObject.get("nodes");
		Object[] relationships = metamodelObject.get("edges");
		Map> concepts = new HashMap>();

		for(Object node : nodes) {
			MetamodelVertex v = (MetamodelVertex) node;
			concepts.put(v.getConceptualName(), new ArrayList(v.getPropSet()));
		}
		List> rels = new ArrayList<>();

		for(Object relation : relationships) {
			rels.add((Map) relation);
		}

		HashMap returnMap = new HashMap();
		returnMap.put(Constants.NODE_PROP, nodes);
		returnMap.put(Constants.RELATION_PROP, relationships);
		Map> positions = GenerateMetamodelLayout.generateOWLMetamodelLayout(concepts, rels);
		returnMap.put(Constants.POSITION_PROP, positions);

		return new NounMetadata(returnMap, PixelDataType.CUSTOM_DATA_STRUCTURE);
	}
	
	

}