Laboratoire ICAR » Plateforme TXM

package org.txm.scripts.importer.xtz

import java.io.File;

import org.txm.importer.xmltxm.Annotate

import org.txm.importer.xtz.*

import org.txm.Toolbox

import org.txm.core.engines.*

/**

 * Wraps current Annotate class into the import workflow

 * 

 * @author mdecorde

 *

 */

class TTAnnotater extends Annotater {

        public TTAnnotater(ImportModule module) {

                super(module);

        }

        @Override

        public void process() {

                //String model = module.getParameters().get(ImportKeys.TTMODEL);

                boolean cleanDirectories = module.getProject().getCleanAfterBuild();

                String corpusname = module.getProject().getName();

                String lang = module.getProject().getLang();

                def engine = Toolbox.getEngineManager(EngineType.ANNOTATION).getEngine("TreeTagger")

                if (module.isMultiThread()) {

                        def hash = [:]

                        for (File txmFile : inputDirectory.listFiles()) {

                                hash[txmFile.getName()] = lang;

                        }

                        if (engine.processDirectory(inputDirectory, module.getBinaryDirectory(), ["langs":hash])) {

                                isSuccessFul = true;

                        }

                } else {

                        if (engine.processDirectory(inputDirectory, module.getBinaryDirectory(), ["lang":lang])) {

                                isSuccessFul = true;

                        }

                }

                if (cleanDirectories) {

                        new File(module.getBinaryDirectory(), "treetagger").deleteDir()

                        new File(module.getBinaryDirectory(), "annotations").deleteDir()

                }

                if (!isSuccessFul) reason = "TreeTagger annotation failed."

        }

        @Override

        public void cancel() {

                //a.setCancelNow();

        }

}