/*
    * The basecode project
    *
    * Copyright (c) 2007-2019 University of British Columbia
    *
    * 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 ubic.basecode.ontology.jena;
  
  import java.util.Collection;
  import java.util.HashSet;
  import java.util.Set;
  
  import com.hp.hpl.jena.ontology.ObjectProperty;
  import com.hp.hpl.jena.ontology.OntClass;
  import com.hp.hpl.jena.ontology.OntResource;
  import com.hp.hpl.jena.ontology.Restriction;
  import com.hp.hpl.jena.util.iterator.ExtendedIterator;
  import ubic.basecode.ontology.model.OntologyTerm;
  
  /**
   * @author pavlidis
   */
  public class ObjectPropertyImpl extends OntologyPropertyImpl implements ubic.basecode.ontology.model.ObjectProperty {
  
      /**
       *
       */
      private static final long serialVersionUID = 1L;
      private final com.hp.hpl.jena.ontology.ObjectProperty resource;
      private final Set<Restriction> additionalRestrictions;
  
      public ObjectPropertyImpl( ObjectProperty resource, Set<Restriction> additionalRestrictions ) {
          super( resource );
          this.isFunctional = resource.isFunctionalProperty();
          this.resource = resource;
          this.additionalRestrictions = additionalRestrictions;
      }
  
      @Override
      public Collection<OntologyTerm> getRange() {
          ExtendedIterator<? extends OntResource> iterator = resource.listRange();
          Collection<OntologyTerm> result = new HashSet<>();
          while ( iterator.hasNext() ) {
              OntResource r = iterator.next();
              if ( r.isClass() ) {
                  OntClass class1 = r.asClass();
                  result.add( new OntologyTermImpl( class1, additionalRestrictions ) );
              } else {
                  log.warn( "Don't know how to deal with " + r );
              }
          }
          return result;
      }
  }