Laboratoire ICAR » Plateforme TXM

package org.txm.scripts.importer.xtz;

import java.io.File;

import java.io.FileFilter;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import org.eclipse.core.runtime.IProgressMonitor;

import org.txm.utils.xml.DomUtils;

import org.txm.metadatas.Metadatas

import org.txm.utils.io.FileCopy;

import org.txm.*

import org.txm.objects.*

import org.w3c.dom.Element

import org.txm.importer.xtz.*

public class XTZImport extends ImportModule {

        public XTZImport(File importParametersFile) {

                super(importParametersFile);

        }

        public XTZImport(Project p) {

                super(p);

        }

        @Override

        public void init(Project p) {

                super.init(p);

                importer = new XTZImporter(this)

                compiler = new XTZCompiler(this)

                annotater = new TTAnnotater(this);

                pager = new XTZPager(this)

        }

        @Override

        protected ArrayList<File> getTXMFilesOrder() {

                //System.out.println("XTZ FILES ORDER");

                if (importer == null) {

                        println "no importer step, using default text order"

                        return super.getTXMFilesOrder();

                }

                Metadatas metadata = importer.getMetadata();

                if (metadata == null) {

                        println "no metadata, using default text order"

                        return super.getTXMFilesOrder();

                }

                File txmDirectory = new File(binaryDirectory, "txm/"+corpusName);

                ArrayList<File> files = new ArrayList<File>(Arrays.asList(txmDirectory.listFiles(new FileFilter() {

                        @Override

                        public boolean accept(File file) {

                                return file.isFile() && file.getName().endsWith(".xml");

                        }

                })));

                final HashMap<File, String> textorder = new HashMap<File, String>();

                for (File f : files) {

                        HashMap<String, String> m = metadata.getTextMetadata(f);

                        if (m != null && m.containsKey("textorder")) textorder[f] = m["textorder"];

                }

                println "Sorting texts using 'textorder' metadata values: "+textorder

                Collections.sort(files, new Comparator<File>() {

                        public int compare(File f1, File f2) {

                                String o1 = textorder[f1];

                                String o2 = textorder[f2];

                                if (o1 == null && o2 == null) {

                                        return f1.getName().compareTo(f2.getName());

                                } else if (o1 == null) {

                                        return 1

                                } else if (o2 == null) {

                                        return -1

                                } else {

                                        int c = o1.compareTo(o2);

                                        if (c == 0) return f1.getName().compareTo(f2.getName());

                                        else return c;

                                }

                        }

                });

                //println files

                return files;

        }

        public void start() throws InterruptedException {

                super.start();

                if (isSuccessful) {

                        //declare a local KR

                        //TODO find out how the annotation plugin may hook the import steps

//                        List<String> krnames = importParameters.getKnowledgeRepositoryNames();

//                        if (krnames.size() == 0) {

//                                importParameters.createKnowledgeRepositoryElement("DEFAULT"); // set a default KR shared by all XTZ corpus

//                        } else if (krnames.size() == 1 && krnames.get(0).equals("DEFAULT")) {

//                                // nothing to do

//                        } else {

//                                println("Corpus is using custom Knowledge repositories: "+importParameters.getKnowledgeRepositoryNames());

//                        }

                        //copy sub directories

                        if (isUpdatingCorpus()) {

                        } else {

                                def dirToCopy = ["xsl", "css", "dtd"]

                                println "--- Copying subdirectories $dirToCopy"

                                for (String dir : dirToCopy) {

                                        File origDirectory = new File(this.sourceDirectory, dir)

                                        if (origDirectory.exists()) {

                                                print "."

                                                File copyDirectory = new File(this.binaryDirectory, dir)

                                                copyDirectory.deleteDir(); // clean before copying

                                                FileCopy.copyFiles(origDirectory, copyDirectory)

                                        }

                                }

                                println ""

                        }

                }

        }

}