org.openprovenance.prov.template.log2prov.Runner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of prov-template-compiler Show documentation
A template system for PROV bundles.
The newest version!
package org.openprovenance.prov.template.log2prov;

import java.io.*;
import java.lang.reflect.InvocationTargetException;

import org.openprovenance.prov.model.ProvFactory;
import org.openprovenance.prov.template.compiler.CompilerUtil;

public class Runner {
    
    final static boolean debug=false;
    
    public Runner (ProvFactory pf) {
        this.pf=pf;
        this.cu=new CompilerUtil(pf);
    }

    public static void main (String [] args) throws IOException, ClassNotFoundException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
        String in=args[0];
        String out=args[1];
        String flag=args[2];
        String extra=(args.length>3)? args[3]: null;
        if (debug) System.out.println("Start main");
        new Runner(org.openprovenance.prov.vanilla.ProvFactory.getFactory()).processLogWithTime(in, out, flag, extra);
        if (debug) System.out.println("End main"); 
    }

    final private ProvFactory pf;
    
    final CompilerUtil cu;
    
    
    public void processLogWithTime(String in, String out, String flag, String extra) throws IOException, ClassNotFoundException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
        long currentTime1=System.currentTimeMillis();
        long currentTime2=processLog(in, out, flag, extra);
        long currentTime3=System.currentTimeMillis();
        if (debug) System.out.println("Expand/merge time " + (currentTime2-currentTime1)); 
        if (debug) System.out.println("Save time         " + (currentTime3-currentTime2));
    }

    public long processLog(String in, String out, String flag, String extra) throws IOException, ClassNotFoundException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
        org.openprovenance.prov.model.IndexedDocument iDoc=new org.openprovenance.prov.model.IndexedDocument(pf,pf.newDocument(),true);


        if (debug) System.out.println("IN: " + in);
        InputStream is= ("-".equals(in))? System.in: new FileInputStream(in);

        DocumentProcessor dp=null;
        RecordProcessor rp=null;
        if ("-merge".equals(flag)) dp=new DocumentMergerProcessor(iDoc);
        if ("-nomerge".equals(flag)) dp=new DocumentConcatenatorProcessor(pf.newDocument());
        if ("-discard".equals(flag)) dp=new DocumentDiscarderProcessor(pf.newDocument());

        if ("-db".equals(flag)) rp=new RecordSQLProcessor(extra);
        if ("-dbInit".equals(flag)) {
            if (rp==null) rp=new RecordSQLProcessor(extra);
            rp.initDb();
        }

        FileBuilder.reader(is,dp,rp);

        is.close();



        long currentTime2=System.currentTimeMillis();
        if (dp!=null) cu.writeDocument(out,dp.getDocument());
        if (rp!=null) {
            if (extra!=null) {
                final PrintWriter pw = new PrintWriter(new FileOutputStream(extra));
                pw.println(rp.getDocument());
                pw.close();
            } else {
                System.out.println(rp.getDocument());
            }
        }
        return currentTime2;
    }
}