package uk.ac.ox.cs.pagoda.rules;

import org.semanticweb.owlapi.model.OWLOntology;
import uk.ac.ox.cs.pagoda.constraints.BottomStrategy;
import uk.ac.ox.cs.pagoda.constraints.UpperUnaryBottom;
import uk.ac.ox.cs.pagoda.util.disposable.Disposable;
import uk.ac.ox.cs.pagoda.util.disposable.DisposedException;

import java.io.InputStream;

public class DatalogProgram extends Disposable {

    UpperDatalogProgram upperProgram = new UpperDatalogProgram();
    LowerDatalogProgram lowerProgram;
    GeneralProgram program = new GeneralProgram();

    BottomStrategy upperBottom;

    public DatalogProgram(InputStream inputStream) {
        lowerProgram = new LowerDatalogProgram();

        upperProgram.load(inputStream, upperBottom = new UpperUnaryBottom());
        lowerProgram.clone(upperProgram);
        program.clone(upperProgram);

        upperProgram.transform();
        lowerProgram.transform();
        program.transform();

        program.buildDependencyGraph();
        lowerProgram.dependencyGraph = upperProgram.buildDependencyGraph();
    }

    public DatalogProgram(OWLOntology o) {
        lowerProgram = new LowerDatalogProgram();

        upperProgram.load(o, upperBottom = new UpperUnaryBottom());
//		upperProgram.load(o, upperBottom = new UnaryBottom());
        lowerProgram.clone(upperProgram);
        program.clone(upperProgram);
//		program.botStrategy = new UnaryBottom();

        upperProgram.transform();
        lowerProgram.transform();
        program.transform();

        program.buildDependencyGraph();
        lowerProgram.dependencyGraph = upperProgram.buildDependencyGraph();
    }

    public LowerDatalogProgram getLower() {
        if(isDisposed()) throw new DisposedException();
        return lowerProgram;
    }

    public UpperDatalogProgram getUpper() {
        if(isDisposed()) throw new DisposedException();
        return upperProgram;
    }

    public GeneralProgram getGeneral() {
        if(isDisposed()) throw new DisposedException();
        return program;
    }

    public String getAdditionalDataFile() {
        if(isDisposed()) throw new DisposedException();
        return upperProgram.getAdditionalDataFile();
    }

    public BottomStrategy getUpperBottomStrategy() {
        if(isDisposed()) throw new DisposedException();
        return upperBottom;
    }

    @Override
    public void dispose() {
        super.dispose();
        if(upperProgram != null) upperProgram.deleteABoxTurtleFile();
    }
}