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.kb.manipulator;
020
021import java.util.SortedSet;
022import java.util.TreeSet;
023
024import org.dllearner.kb.extraction.Node;
025import org.dllearner.utilities.JamonMonitorLogger;
026import org.dllearner.utilities.datastructures.RDFNodeTuple;
027
028import org.apache.jena.rdf.model.impl.ResourceImpl;
029
030public class PredicateReplacementRule extends Rule{
031        
032        String oldPredicate;
033        String newPredicate;
034
035        public PredicateReplacementRule(Months month, String oldPredicate, String newPredicate) {
036                super(month);
037                this.oldPredicate = oldPredicate;
038                this.newPredicate = newPredicate;
039        }
040        
041        
042        @Override
043        public  SortedSet<RDFNodeTuple> applyRule(Node subject, SortedSet<RDFNodeTuple> tuples){
044                SortedSet<RDFNodeTuple> keep = new TreeSet<>();
045                for (RDFNodeTuple tuple : tuples) {
046                        if(tuple.aPartContains(oldPredicate)){
047                                tuple.a = new ResourceImpl(newPredicate);
048                                logJamon();
049                        }
050                        keep.add(tuple);
051                }
052                return  keep;
053        }
054
055        @Override
056        public void logJamon(){
057                JamonMonitorLogger.increaseCount(PredicateReplacementRule.class, "replacedPredicates");
058        }
059
060        
061        
062}