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_subgraph", usage="A Full TIGERSearch query", widget="Boolean", required=true, def="true")

		def count_subgraph

@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 tse

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_subgraph) {

		size = mresult.submatchSize();

	} else {

		size = mresult.size();

	}

	results[corpus] = size

	print "\t"+size

}

println ""

println "Done."

return results