trunk/ruta-ep-ide/src/main/java/org/apache/uima/ruta/ide/core/builder/DescriptorManager.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.uima.ruta.ide.core.builder;

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

public class DescriptorManager {

  private static final boolean CONVERT = false;

  private List<String> typeShortNames;

  private List<StringTriple> typeTriples;

  private Collection<String> importedTypeSystems;

  private Map<String, Collection<StringTriple>> features;

  private Collection<String> importedScripts;

  private Collection<String> importedEngines;

  private Collection<String> importedUimafitEngines;

  public DescriptorManager() {
    super();
    typeShortNames = new ArrayList<String>();
    typeTriples = new ArrayList<StringTriple>();
    importedTypeSystems = new HashSet<String>();
    importedScripts = new HashSet<String>();
    importedEngines = new HashSet<String>();
    importedUimafitEngines = new HashSet<String>();
    features = new HashMap<String, Collection<StringTriple>>();
  }

  public void addType(String name, String description, String parent) {
    if (parent == null) {
      parent = "uima.tcas.Annotation";
    }
    typeShortNames.add(name);
    typeTriples.add(new StringTriple(name, description, parent));
  }

  public void addFeature(String type, String name, String description, String parent) {
    Collection<StringTriple> list = features.get(type);
    if (list == null) {
      list = new ArrayList<StringTriple>();
      features.put(type, list);
    }
    list.add(new StringTriple(name, description, parent));
  }

  public void addScript(String name, boolean convert) {
    if (convert) {
      name = name.replaceAll("[.]", "/");
    }
    importedScripts.add(name);
  }

  public void addEngine(String name, boolean convert) {
    if (convert) {
      name = name.replaceAll("[.]", "/");
    }
    importedEngines.add(name);
  }

  public void addUimafitEngine(String name) {
    importedUimafitEngines.add(name);
  }

  public void addTypeSystem(String name, boolean convert) {
    if (convert) {
      name = name.replaceAll("[.]", "/");
    }
    importedTypeSystems.add(name);
  }

  public void addScript(String name) {
    addScript(name, CONVERT);
  }

  public void addEngine(String name) {
    addEngine(name, CONVERT);
  }

  public void addTypeSystem(String name) {
    addTypeSystem(name, CONVERT);
  }

  public List<String> getTypeShortNames() {
    return typeShortNames;
  }

  public List<StringTriple> getTypeTriples() {
    return typeTriples;
  }

  public Map<String, Collection<StringTriple>> getFeatures() {
    return features;
  }

  public Collection<String> getImportedTypeSystems() {
    return importedTypeSystems;
  }

  public Collection<String> getImportedScripts() {
    return importedScripts;
  }

  public Collection<String> getImportedEngines() {
    return importedEngines;
  }

  public Collection<String> getImportedUimafitEngines() {
    return importedUimafitEngines;
  }

}