Laboratoire ICAR » Plateforme TXM

// Copyright © 2019 ENS de Lyon, CNRS, University of Franche-Comté

// Licensed under the terms of the GNU General Public License (http://www.gnu.org/licenses)

// @author mdecorde

// STANDARD DECLARATIONS

package org.txm.macro.tiger.exploit

import groovy.transform.Field

import org.txm.searchengine.core.SearchEnginesManager

import org.txm.searchengine.cqp.corpus.*

import org.txm.searchengine.ts.TIGERSearchEngine

def scriptName = this.class.getSimpleName()

def selection = []

for (def s : corpusViewSelections) {

        if (s instanceof CQPCorpus) selection << s

        else if (s instanceof Partition) selection.addAll(s.getParts())

}

if (selection.size() == 0) {

        println "** $scriptName: please select a Corpus or a Partition in the Corpus view: "+corpusViewSelections

        return false

} else {

        for (def c : selection) c.compute(false)

}

@Field @Option(name="tiger_query", usage="A Full TIGERSearch query", widget="Text", required=true, def="[]")

                String tiger_query

@Field @Option(name="count_sub_matches", usage="A Full TIGERSearch query", widget="Boolean", required=true, def="true")

                def count_sub_matches

@Field @Option(name="debug", usage="Show internal variable content", widget="StringArray", metaVar="OFF        ON        ALL        REALLY ALL", required=true, def="OFF")

                debug

if (!ParametersDialog.open(this)) return        

if (debug == "OFF") debug = 0; else if (debug == "ON") debug = 1; else if (debug == "ALL") debug = 2 else if (debug == "REALLY ALL") debug = 3

TIGERSearchEngine tse = SearchEnginesManager.getTIGERSearchEngine()

println "\t"+selection.join("\t")

print "F"

def results = new LinkedHashMap()

for (def corpus : selection) {

def root = corpus.getRootCorpusBuild();

File buildDirectory = new File(root.getProjectDirectory(), "tiger");

        if (!tse.hasIndexes(corpus)) {

                println "Warning: skipping $corpus: no TIGERSearch indexes found."

                continue;

        }

        def tcorpus = tse.getTSCorpus(corpus);

        def sentences_min_max = tse.getSentMinMax(corpus);

        def mresult = tcorpus.manager.processQuery(tiger_query, sentences_min_max[0], sentences_min_max[1], 9999999);

        int size = 0;

        if (count_sub_matches) {

                size = mresult.submatchSize();

        } else {

                size = mresult.size();

        }

        results[corpus] = size

        print "\t"+size

}

println ""

println "Done."

return results