IdentifierNamer.java

Go to the documentation of this file.

package edu.tum.cs.srldb;

import java.util.HashSet;
import java.util.Map;

import edu.tum.cs.srldb.datadict.DDAttribute;
import edu.tum.cs.srldb.datadict.DDRelation;
import edu.tum.cs.srldb.datadict.DataDictionary;
import edu.tum.cs.util.datastruct.Map2D;

public class IdentifierNamer {
        protected Map2D<String, String, String> identifiers;
        protected Map2D<String, String, Integer> counts;
        protected HashSet<String> reservedWords;
        
        public IdentifierNamer(DataDictionary dd) {
                identifiers = new Map2D<String, String, String>();
                counts = new Map2D<String, String, Integer>();
                reservedWords = new HashSet<String>();
                // init reserved words: identifiers shouldn't coincide with predicate names
                for(DDAttribute attrib : dd.getAttributes()) {
                        reservedWords.add(Database.stdPredicateName(attrib.getName()));                         
                }
                for(DDRelation rel : dd.getRelations()) {
                        reservedWords.add(Database.stdPredicateName(rel.getName()));
                }
        }
        
        protected boolean isAvailable(String context, String id) {
                if(reservedWords.contains(id))
                        return false;
                Map<String, String> submap = identifiers.getSubmap(context);
                if(submap == null)
                        return true;
                return !submap.containsValue(id);
        }
        
        public String getShortIdentifier(String category, String name) {
                return getIdentifier(category, name.toLowerCase(), true, false);                        
        }

        public String getCountedShortIdentifier(String category, String name) {
                return getIdentifier(category, name.toLowerCase(), true, true);                 
        }
        
        public String getLongIdentifier(String category, String name) {
                return getIdentifier(category, name, false, false);                     
        }

        public String getCountedLongIdentifier(String category, String name) {
                return getIdentifier(category, name, false, true);                      
        }
        
        public String getIdentifier(String category, String name, boolean shortName, boolean counted) {
                String id = identifiers.get(category, name);
                if(id != null) {
                        if(!counted)
                                return id;
                        Integer count = counts.get(category, name);
                        if(count == null) {
                                count = new Integer(0);
                                counts.put(count);
                        }
                        count++;
                        if(count == 1)
                                return id; 
                        else
                                return id + count;
                }
                String idProposal = name;
                for(int i = shortName ? 1 : name.length(); true; i++) {
                        idProposal = i > name.length() ? idProposal + "_" : name.substring(0, i);
                        if(isAvailable(category, idProposal)) {
                                identifiers.put(category, name, idProposal);
                                return idProposal;
                        }                               
                }
        }
        
        public void resetCounts() {
                counts = new Map2D<String, String, Integer>();
        }
}

---

srldb
Author(s): Dominik Jain, Stefan Waldherr, Moritz Tenorth
autogenerated on Fri Jan 11 09:58:29 2013