001/**
002 * Copyright (C) 2007 - 2016, Jens Lehmann
003 *
004 * This file is part of DL-Learner.
005 *
006 * DL-Learner is free software; you can redistribute it and/or modify
007 * it under the terms of the GNU General Public License as published by
008 * the Free Software Foundation; either version 3 of the License, or
009 * (at your option) any later version.
010 *
011 * DL-Learner is distributed in the hope that it will be useful,
012 * but WITHOUT ANY WARRANTY; without even the implied warranty of
013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
014 * GNU General Public License for more details.
015 *
016 * You should have received a copy of the GNU General Public License
017 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
018 */
019package org.dllearner.utilities.owl;
020
021import java.util.HashMap;
022
023import org.semanticweb.owlapi.model.OWLEntity;
024
025public class VariablesMapping extends HashMap<OWLEntity, String>{
026
027        private int classCnt = 0;
028        private int propCnt = 0;
029        private int indCnt = 0;
030        
031        public String getVariable(OWLEntity entity){
032                String var = get(entity);
033                if(var == null){
034                        if(entity.isOWLClass()){
035                                var = "?cls" + classCnt++;
036                        } else if(entity.isOWLObjectProperty() || entity.isOWLDataProperty()){
037                                var = "?p" + propCnt++;
038                        } else if(entity.isOWLNamedIndividual()){
039                                var = "?s" + indCnt++;
040                        } 
041                        put(entity, var);
042                }
043                return var;
044        }
045        
046        public String newIndividualVariable(){
047                return "?s" + indCnt++;
048        }
049        
050        public String newPropertyVariable(){
051                return "?p" + propCnt++;
052        }
053        
054        public void reset(){
055                clear();
056                classCnt = 0;
057                propCnt = 0;
058                indCnt = 0;
059        }
060}