/*
 * Created on Apr 6, 2005
 */
package impl.swrl;

import java.net.URI;

import org.mindswap.owl.OWLClass;
import org.mindswap.owl.OWLDataProperty;
import org.mindswap.owl.OWLIndividual;
import org.mindswap.owl.OWLObjectProperty;
import org.mindswap.owl.OWLOntology;
import org.mindswap.owl.OWLValue;
import org.mindswap.owl.vocabulary.SWRL;
import org.mindswap.owl.vocabulary.SWRLB;
import org.mindswap.swrl.Atom;
import org.mindswap.swrl.AtomList;
import org.mindswap.swrl.BuiltinAtom;
import org.mindswap.swrl.ClassAtom;
import org.mindswap.swrl.DataPropertyAtom;
import org.mindswap.swrl.DifferentIndividualsAtom;
import org.mindswap.swrl.IndividualPropertyAtom;
import org.mindswap.swrl.SWRLDataObject;
import org.mindswap.swrl.SWRLDataVariable;
import org.mindswap.swrl.SWRLFactory;
import org.mindswap.swrl.SWRLIndividualObject;
import org.mindswap.swrl.SWRLIndividualVariable;
import org.mindswap.swrl.SameIndividualAtom;

/**
 * @author Evren Sirin
 *
 */
public class SWRLFactoryImpl implements SWRLFactory {
    private OWLOntology ont;
    
    public SWRLFactoryImpl( OWLOntology ont ) {
        this.ont = ont;
    }
    
    public OWLOntology getOntology() {
        return ont;
    }
    
    public AtomList createList() {
        return (AtomList) ont.createList(SWRL.AtomListVocabulary).castTo(AtomList.class);
    }

    public AtomList createList(Atom atom) {
        return (AtomList) ont.createList(SWRL.AtomListVocabulary, atom).castTo(AtomList.class);
    }
    
    public SWRLIndividualVariable createIndividualVariable(URI uri) {
        return new SWRLIndividualVariableImpl(ont.createInstance(SWRL.Variable, uri));
    }
    
    public SWRLDataVariable createDataVariable(URI uri) {
        return new SWRLDataVariableImpl(ont.createInstance(SWRL.Variable, uri));
    }
    
    public ClassAtom createAtom(OWLClass c, OWLIndividual arg) {
        return createClassAtom( c, (SWRLIndividualObject) arg.castTo( SWRLIndividualObject.class  ) );
    }

    public ClassAtom createClassAtom(OWLClass c, OWLIndividual arg) {
        return createClassAtom( c, (SWRLIndividualObject) arg.castTo( SWRLIndividualObject.class  ) );
    }

    public ClassAtom createClassAtom(OWLClass c, SWRLIndividualObject arg) {
	    ClassAtom atom = (ClassAtom) ont.createInstance( SWRL.ClassAtom ).castTo( ClassAtom.class );
	    atom.setClassPredicate( c );
	    atom.setArgument1( arg );
	    
        return atom;
    }

    public IndividualPropertyAtom createAtom(OWLObjectProperty p, OWLIndividual arg1, OWLIndividual arg2) {
        return createIndividualPropertyAtom(p, 
            (SWRLIndividualObject) arg1.castTo( SWRLIndividualObject.class ), 
            (SWRLIndividualObject) arg2.castTo( SWRLIndividualObject.class ) );
    }
    
    public IndividualPropertyAtom createIndividualPropertyAtom(OWLObjectProperty p, OWLIndividual arg1, OWLIndividual arg2) {
        return createIndividualPropertyAtom(p, 
            (SWRLIndividualObject) arg1.castTo( SWRLIndividualObject.class ), 
            (SWRLIndividualObject) arg2.castTo( SWRLIndividualObject.class ) );
    }
    
    public IndividualPropertyAtom createIndividualPropertyAtom(OWLObjectProperty p, SWRLIndividualObject arg1, SWRLIndividualObject arg2) {
	    IndividualPropertyAtom atom = (IndividualPropertyAtom) 
			ont.createInstance( SWRL.IndividualPropertyAtom ).castTo( IndividualPropertyAtom.class );
	    atom.setPropertyPredicate( p );
	    atom.setArgument1( arg1 );
	    atom.setArgument2( arg2 );

	    return atom;
    }

    public DataPropertyAtom createAtom(OWLDataProperty p, OWLIndividual arg1, OWLValue arg2) {
        return createDataPropertyAtom(p, 
            (SWRLIndividualObject) arg1.castTo( SWRLIndividualObject.class ), 
            (SWRLDataObject) arg2.castTo( SWRLDataObject.class ) );
    }
    
    public DataPropertyAtom createDataPropertyAtom(OWLDataProperty p, OWLIndividual arg1, OWLValue arg2) {
        return createDataPropertyAtom(p, 
            (SWRLIndividualObject) arg1.castTo( SWRLIndividualObject.class ), 
            (SWRLDataObject) arg2.castTo( SWRLDataObject.class ) );
    }
    
    public DataPropertyAtom createDataPropertyAtom(OWLDataProperty p, SWRLIndividualObject arg1, SWRLDataObject arg2) {
	    DataPropertyAtom atom = (DataPropertyAtom) 
    		ont.createInstance( SWRL.DatavaluedPropertyAtom ).castTo( DataPropertyAtom.class );
	    atom.setPropertyPredicate( p );
	    atom.setArgument1( arg1 );
	    atom.setArgument2( arg2 );
	    
	    return atom;
    }

    public SameIndividualAtom createSameIndividualAtom(OWLIndividual arg1, OWLIndividual arg2) {
        return createSameIndividualAtom(
            (SWRLIndividualObject) arg1.castTo( SWRLIndividualObject.class ), 
            (SWRLIndividualObject) arg2.castTo( SWRLIndividualObject.class ) );            
    }
    
    public SameIndividualAtom createSameIndividualAtom(SWRLIndividualObject arg1, SWRLIndividualObject arg2) {
        SameIndividualAtom atom = (SameIndividualAtom) 
			ont.createInstance( SWRL.SameIndividualAtom ).castTo( SameIndividualAtom.class );
	    atom.setArgument1( arg1 );
	    atom.setArgument2( arg2 );
	    
	    return atom;
    }

    public DifferentIndividualsAtom createDifferentIndividualsAtom(OWLIndividual arg1, OWLIndividual arg2) {
        return createDifferentIndividualsAtom(
            (SWRLIndividualObject) arg1.castTo( SWRLIndividualObject.class ), 
            (SWRLIndividualObject) arg2.castTo( SWRLIndividualObject.class ) );            
    }

    public DifferentIndividualsAtom createDifferentIndividualsAtom(SWRLIndividualObject arg1, SWRLIndividualObject arg2) {
        DifferentIndividualsAtom atom = (DifferentIndividualsAtom) 
			ont.createInstance( SWRL.DifferentIndividualsAtom ).castTo( DifferentIndividualsAtom.class );
	    atom.setArgument1( arg1 );
	    atom.setArgument2( arg2 );
	    
	    return atom;
    }

    public BuiltinAtom createEqual(SWRLDataObject arg1, SWRLDataObject arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.equal );
	    atom.setArgument( 0, arg1 );
	    atom.setArgument( 1, arg2 );
	    
	    return atom;
    }

    public BuiltinAtom createNotEqual(SWRLDataObject arg1, SWRLDataObject arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.notEqual );
	    atom.setArgument( 0, arg1 );
	    atom.setArgument( 1, arg2 );
	    
	    return atom;
    }

    public BuiltinAtom createLessThan(SWRLDataObject arg1, SWRLDataObject arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.lessThan );
	    atom.setArgument( 0, arg1 );
	    atom.setArgument( 1, arg2 );
	    
	    return atom;
    }

    public BuiltinAtom createLessThanOrEqual(SWRLDataObject arg1, SWRLDataObject arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.lessThanOrEqual );
	    atom.setArgument( 0, arg1 );
	    atom.setArgument( 1, arg2 );
	    
	    return atom;
    }

    public BuiltinAtom createGreaterThan(SWRLDataObject arg1, SWRLDataObject arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.greaterThan );
	    atom.setArgument( 0, arg1 );
	    atom.setArgument( 1, arg2 );
	    
	    return atom;
    }

    public BuiltinAtom createGreaterThanOrEqual(SWRLDataObject arg1, SWRLDataObject arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.greaterThanOrEqual );
	    atom.setArgument( 0, arg1 );
	    atom.setArgument( 1, arg2 );
	    
	    return atom;
    }

    public BuiltinAtom createAdd(OWLValue result, OWLValue arg1, OWLValue arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.add );
	    atom.setArgument( 0, (SWRLDataObject) result.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 1, (SWRLDataObject) arg1.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 2, (SWRLDataObject) arg2.castTo( SWRLDataObject.class ) );
	    
	    return atom;        
    }
    
    public BuiltinAtom createSubtract(OWLValue result, OWLValue arg1, OWLValue arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.subtract );
	    atom.setArgument( 0, (SWRLDataObject) result.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 1, (SWRLDataObject) arg1.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 2, (SWRLDataObject) arg2.castTo( SWRLDataObject.class ) );
	    
	    return atom;                
    }
    
    public BuiltinAtom createMultiply(OWLValue result, OWLValue arg1, OWLValue arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.multiply );
	    atom.setArgument( 0, (SWRLDataObject) result.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 1, (SWRLDataObject) arg1.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 2, (SWRLDataObject) arg2.castTo( SWRLDataObject.class ) );
	    
	    return atom;                
    }
    
    public BuiltinAtom createDivide(OWLValue result, OWLValue arg1, OWLValue arg2) {
	    BuiltinAtom atom = (BuiltinAtom) ont.createInstance( SWRL.BuiltinAtom ).castTo( BuiltinAtom.class );
	    atom.setBuiltin( SWRLB.divide );
	    atom.setArgument( 0, (SWRLDataObject) result.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 1, (SWRLDataObject) arg1.castTo( SWRLDataObject.class ) );
	    atom.setArgument( 2, (SWRLDataObject) arg2.castTo( SWRLDataObject.class ) );
	    
	    return atom;                
    }

}