/* -*-Java-*- */

/* 
 * Copyright (C) 2006,2010  Anders Nilsson <anders.nilsson@control.lth.se>
 *
 * This file is part of OntologyCompiler.
 *
 * OntologyCompiler is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.

 * OntologyCompiler is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 
 * You should have received a copy of the GNU General Public License
 * along with OntologyCompiler.  If not, see <http://www.gnu.org/licenses/>.
 */

import java.util.ArrayList;

aspect Types {
    syn lazy Declaration Declaration.getSuperClass() = null;
	eq ClassDeclaration.getSuperClass() {
		return goStart().getSuperClass(name());
	}

	syn Declaration ASTNode.getSuperClass(String name) = null;
	eq Start.getSuperClass(String name) =
		getSpecification().getSuperClass(name);
	eq Specification.getSuperClass(String name) {
		Declaration decl = getOntoDeclaration().getSuperClass(name);
		if (decl == null) {
			decl = findDecl("Thing");
			if (decl == null) {
				decl = new ClassDeclaration();
				Clazz c = new Clazz();
				c.addAttribute(new IRI(new Value("Thing")));
				decl.addElement(c);
				getOntoDeclaration().addElement(decl);
			}
		}
		return decl;
	}
	eq ComplexElement.getSuperClass(String name) {
		for (Element e : getElements()) {
			Declaration d = e.getSuperClass(name);
			if (d != null) {
				return d;
			}
		}
		return null;
	}
	eq SubClassOf.getSuperClass(String name) {
		if (getElement(0).name().equals(name)) {
			return goStart().findDecl(getElement(1).name());
		}
		return null;
	}

	syn Declaration ASTNode.findDecl(String name) = null;
	eq Start.findDecl(String name) =
		getSpecification().findDecl(name);
	eq Specification.findDecl(String name) = getOntoDeclaration().findDecl(name);
	eq ComplexElement.findDecl(String name) {
		for (Element e : getElements()) {
			Declaration d = e.findDecl(name);
			if (d != null) {
				return d;
			}
		}
		return null;
	}
	eq ClassDeclaration.findDecl(String name) {
		if (name.equals(name())) {
			return this;
		}
		return null;
	}

    syn lazy OClass OClass.getSuperClass();
    eq OwlClassDecl.getSuperClass() {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i) instanceof RdfsSubClassOf) {
				RdfsSubClassOf e = (RdfsSubClassOf) getElement(i);
				if (e.getElement(0) instanceof OClass) {
					return ((OClass) e.getElement(0)).decl();
				}
			}
		}
		// No super class found, return an owl Thing instead.
		return new OwlClassDecl(new List(), new List(), "Thing");
    }
    eq OwlClassUse.getSuperClass() {
		return decl().getSuperClass();
    }

    syn lazy OwlClassDecl OClass.decl();
    eq OwlClassDecl.decl() = this;
    eq OwlClassUse.decl() {
		ComplexElement top = getTopElement();
		for (int i=0; i<top.getNumElement(); i++) {
// 			System.out.println(getId()+" == "+((ComplexElement) top.getElement(i)).getId());
			if ((top.getElement(i) instanceof ComplexElement) &&
				getId().equals(((ComplexElement) top.getElement(i)).getId())) {
				return (OwlClassDecl) top.getElement(i);
			}
		}
		// OK, then let's see if this class is declared by w3.org
		String id = getId();
		if (id.endsWith("Thing")) {
			return new OwlClassDecl(new List(), new List(), "Thing");
		}
		// No decl found, so let's alert user. If the reason for a
		// missing decl is that it is a top level class which Protege
		// for some unknown reason no longer explicitly declares on
		// top level, the right thing to do would be to create a class
		// declaration here. And that we do.
        System.out.println("* Couldn't find decl for: "+getId());
		OwlClassDecl nClass = new OwlClassDecl(new List(), new List(), id);
		top.addElement(nClass);
		return nClass;
    }

    syn lazy Properties OwlClassDecl.getProperties() {
		List l = new List();
		getTopElement().collectProperties(l);
// 		collectProperties(l);
// 		return new Properties(l);
		Properties p = new Properties();
		for (int i=0; i<l.getNumChild(); i++) {
			OwlProperty op = (OwlProperty) l.getChild(i);
			if (op.domainIncludes(this)) {
				p.addProperty(op);
			} else {
// 				System.out.println(op);
			}
		}
		return p;
    }

	syn lazy ArrayList OClass.getSubClasses();
	eq OwlClassUse.getSubClasses() = decl().getSubClasses();
	eq OwlClassDecl.getSubClasses() {
		return goStart().getSubClasses(this,new ArrayList());
	}

	syn ArrayList ASTNode.getSubClasses(OwlClassDecl klass, ArrayList l) {
		for (int i=0; i<getNumChild(); i++) {
			getChild(i).getSubClasses(klass,l);
		}
		return l;
	}
	eq OwlClassDecl.getSubClasses(OwlClassDecl klass, ArrayList l) {
		if (getSuperClass() == klass) {
			l.add(this);
		}
		return l;
	}
}

aspect Restrictions {
  
	Restrictions OwlClassDecl.restrictions;
	Restrictions OwlClassDecl.ownRestrictions;
  
	syn lazy Restrictions OwlClassDecl.getRestrictions() {
		if (restrictions == null) {
			List l = new List();
			collectRestrictions(l);
			restrictions = new Restrictions(l);
		}
		return restrictions;
	}

	void ASTNode.collectRestrictions(List l) {
		for (int i=0; i<getNumChild(); i++) {
			getChild(i).collectRestrictions(l);
		}
	}

	void OwlClassDecl.collectRestrictions(List l) {
		if (!getSuperClass().getId().equals("Thing")) {
			getSuperClass().decl().collectRestrictions(l);
		}
		super.collectRestrictions(l);
	}
  
	void OwlRestriction.collectRestrictions(List l) {
		if (allValuesFrom()) {
			l.add(this);
		}
	}
	
	syn int OwlClassDecl.getNumRestriction() = getRestrictions().getNumOwlRestriction();

	syn lazy Restrictions OwlClassDecl.getOwnRestrictions() {
		if (ownRestrictions == null) {
			List l = new List();
			collectOwnRestrictions(l);
			ownRestrictions = new Restrictions(l);
		}
		return ownRestrictions;
	}

	void ASTNode.collectOwnRestrictions(List l) {
		for (int i=0; i<getNumChild(); i++) {
			getChild(i).collectOwnRestrictions(l);
		}
	}

	void OwlRestriction.collectOwnRestrictions(List l) {
		if (allValuesFrom()) {
			l.add(this);
		}
	}
	
	syn int OwlClassDecl.getNumOwnRestriction() = 
		getOwnRestrictions().getNumOwlRestriction();
}

aspect Properties {
    void ASTNode.collectProperties(List l) {
		for (int i=0; i<getNumChild(); i++) {
			getChild(i).collectProperties(l);
		}
    }

    void OwlProperty.collectProperties(List l) {
		l.add(this);
    }

    boolean ASTNode.domainIncludes(OClass clazz) {
		return false;
    }

    boolean OwlObjectProperty.domainIncludes(OClass clazz) {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i) instanceof RdfsDomain) {
				return ((RdfsDomain) getElement(i)).domainIncludes(clazz);
			}
		}
		return false;
    }

    boolean OwlFunctionalProperty.domainIncludes(OClass clazz) {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i) instanceof RdfsDomain) {
				return ((RdfsDomain) getElement(i)).domainIncludes(clazz);
			}
		}
		return false;
    }

    boolean OwlDatatypeProperty.domainIncludes(OClass clazz) {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i) instanceof RdfsDomain) {
				return ((RdfsDomain) getElement(i)).domainIncludes(clazz);
			}
		}
		return false;
    }

    boolean RdfsDomain.domainIncludes(OClass clazz) {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i).domainIncludes(clazz)) {
				return true;
			}
		}
		return false;
    }

    boolean OwlUnionOf.domainIncludes(OClass clazz) {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i).domainIncludes(clazz)) {
				return true;
			}
		}
		return false;
    }

    boolean OClass.domainIncludes(OClass clazz) {
		if (!getId().equals("_Unknown_")) {
			return getId().equals(clazz.getId());
		}
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i).domainIncludes(clazz)) {
				return true;
			}
		}
		return false;	
    }

    syn lazy RdfsRange Element.range() = null;
    
    eq OwlObjectProperty.range() {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i) instanceof RdfsRange) {
				return (RdfsRange) getElement(i);
			}
		}
		return null;
    }
    eq OwlFunctionalProperty.range() {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i) instanceof RdfsRange) {
				return (RdfsRange) getElement(i);
			}
		}
		return null;
    }
    eq OwlDatatypeProperty.range() {
		for (int i=0; i<getNumElement(); i++) {
			if (getElement(i) instanceof RdfsRange) {
				return (RdfsRange) getElement(i);
			}
		}
		return null;
    }
}