TypeMapping.java

/*
 * Copyright 2015
 * Ubiquitous Knowledge Processing (UKP) Lab
 * Technische Universität Darmstadt
 *
 * Licensed 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 de.tudarmstadt.ukp.dkpro.core.io.brat.internal.model;

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

import org.apache.uima.cas.Type;
import org.apache.uima.cas.TypeSystem;

public class TypeMapping
{
    private final List<MappingParam> parsedMappings;
    private final Map<String, Type> brat2UimaMappingCache;
    private final Map<String, String> uima2BratMappingCache;

    public TypeMapping(String... aMappings)
    {
        parsedMappings = new ArrayList<>();
        for (String m : aMappings) {
            parsedMappings.add(MappingParam.parse(m));
        }

        brat2UimaMappingCache = new HashMap<>();
        uima2BratMappingCache = new HashMap<>();
    }
    
    private String apply(String aType)
    {
        String type = aType;
        for (MappingParam m : parsedMappings) {
            if (m.matches(aType)) {
                type = m.apply();
                break;
            }
        }
        return type;
    }
    
    public Type getUimaType(TypeSystem aTs, BratAnnotation aAnno)
    {
        Type t = brat2UimaMappingCache.get(aAnno.getType());
        
        if (t == null) {
            // brat doesn't like dots in name names, so we had replaced them with dashes. Now revert.
            String type = apply(aAnno.getType().replace("-", "."));
            t = aTs.getType(type);
            
            // if the lookup didn't work with replacing the dashes, try without, e.g. because the
            // brat name *really* contains dashes and we only resolve them through mapping
            if (t == null) {
                type = apply(aAnno.getType());
                t = aTs.getType(type);
            }
            
            brat2UimaMappingCache.put(aAnno.getType(), t);
        }

        if (t == null) {
            throw new IllegalStateException("Unable to find appropriate UIMA type for brat type ["
                    + aAnno.getType() + "]");
        }

        return t;
    }

    public String getBratType(Type aType)
    {
        String bratType = uima2BratMappingCache.get(aType.getName());
        
        if (bratType == null) {
            String uimaType = aType.getName();
            
            for (MappingParam m : parsedMappings) {
                if (m.matches(aType.getName())) {
                    uimaType = m.apply();
                    break;
                }
            }
            
            // brat doesn't like dots in name names, so we had replaced them with dashes.
            bratType = uimaType.replace(".", "-");
            uima2BratMappingCache.put(uimaType, bratType);
        }
        
        return bratType;
    }
}