Laboratoire ICAR » Plateforme TXM

/**

 * A partition on a corpus. The partition is composed of {@link Part}s

 * 

 *

 * @author mdecorde

 * @author Jean-Philippe Magué

 * 

 */

public class Partition extends org.txm.objects.Partition {

	int totalsize = -1;

	/** The property, if not null the partition will forge its own queries */

	protected StructuralUnitProperty pProperty;

	/**

	 * The parts property values to use, may be null.

	 */

	protected List<String> pValues;

	/**

	 * The parts queries

	 */

	protected List<String> pQueries;

	/**

	 * The parts names -> do the parts order

	 */

	protected List<String> pPartNames;

	/**

	 * 

	 * @param parent

	 */

		super(parent);

	}

	/**

	 * 

	 * @param parametersNodePath

	 */

		super(parametersNodePath);

	}

	/**

	 * Instantiates a new partition.

	 *

	 * @param name the name

	 * @param queries the queries

	 * @param partnames the partnames

	 */

	public boolean _compute_with_lists() throws Exception {

		if (this.userName == null || this.userName.length() == 0) {

			this.userName = "no_name";

		}

		Log.fine(NLS.bind(SearchEngineCoreMessages.info_creatingNewPartition, this.getParent(), this.userName));

		long start = System.currentTimeMillis();

		for (int i = 0; i < pQueries.size(); i++) {

			String queryS = pQueries.get(i);

			String partName = String.valueOf(i);

				partName = pPartNames.get(i);

			}

			if (partName.trim().length() == 0) {

		}

		long end = System.currentTimeMillis();

		Log.fine(NLS.bind(SearchEngineCoreMessages.info_partitionCreatedInXMs, this.userName, (end - start)));

		return pQueries.size() > 0;

	}

	/**

	 * Creates a new partition on a corpus given a structure, a property of this

	 * structure and a list of values this property can take

	 * @param structure the structure

	 * @param property the property

	 * @param values the values, raw values special chars will be backslashed

	 */

		CQPCorpus corpus = this.getParent();

		if (this.userName == null || this.userName.length() == 0) {

			this.userName = corpus.getName() + "_" + pProperty.getFullName(); //$NON-NLS-1$

		}

		if (pValues == null) {

			pValues = pProperty.getValues(corpus);

		}

		// sort parts

		HashMap<String, String> infos = this.getParent().getSAttributesInfos().get(pProperty.getFullName());

		String type = "String"; //$NON-NLS-1$

		if (infos != null) {

			type = infos.get("type"); //$NON-NLS-1$

		}

		else if (type.equals("Integer")) { //$NON-NLS-1$

			Collections.sort(pValues, new Comparator<String>() {

				@Override

				public int compare(String arg0, String arg1) {

					int i0 = Integer.parseInt(arg0);

					int i1 = Integer.parseInt(arg1);

					return i0 - i1;

				}

		}

		else if (type.equals("Date")) { //$NON-NLS-1$

			String format = infos.get("inputFormat"); //$NON-NLS-1$

			final DateFormat formater = new SimpleDateFormat(format);

			Collections.sort(pValues, new Comparator<String>() {

				@Override

				public int compare(String arg0, String arg1) {

					try {

						Date i0 = formater.parse(arg0);

						Date i1 = formater.parse(arg1);

						return i0.compareTo(i1);

						return arg0.compareTo(arg1);

					}

				}

		}

			final List<String> sortedValues = Arrays.asList(type.split("\\|")); //$NON-NLS-1$

			Collections.sort(pValues, new Comparator<String>() {

				@Override

				public int compare(String arg0, String arg1) {

					int i0 = sortedValues.indexOf(arg0);

					int i1 = sortedValues.indexOf(arg1);

				}

		}

		pQueries = new ArrayList<>();

		pPartNames = new ArrayList<>();

		for (String value : pValues) {

			pPartNames.add(partName);

			CQLQuery query = new QueryPart(pProperty.getStructuralUnit(), pProperty, CQLQuery.addBackSlash(value)); // second option is faster

			pQueries.add(query.getQueryString());

		}

		return _compute_with_lists();

	}

	@Override

	protected boolean _compute() throws Exception {

		// Parts already created

		if (this.getParts().size() > 0) {

			if (monitor != null) monitor.beginTask("Building parts...", this.getParts().size());

			for (int i = 0; i < parts.size(); i++) {

				parts.get(i).compute();

				if (monitor != null) monitor.worked(1);

		}

		else {

			Log.finest(NLS.bind(SearchEngineCoreMessages.info_creatingNewPartition, this.getParent(), this.userName));

			if (this.pProperty != null) {

				_compute_with_property();

			}

		totalsize = -1; // refresh total size at next call of getTotalSize

		return true;

	}

	 * @see org.txm.objects.TxmObject#load()

	 */

	protected boolean _load(Element e) {

		// partition already computed

		if (this.hasBeenComputedOnce) {

			return true;

		}

		if (e != null && this.getParent() != null) {

			NodeList partList = e.getChildNodes();

			for (int i = 0; i < partList.getLength(); i++) {

					}

				}

			}

			return true;

		}

		return false;

	}

	@Override

	public boolean canCompute() {

		return this.getParent() != null && (this.pProperty != null || (this.pQueries != null && this.pQueries.size() > 0));

	}

	@Override

	public void clean() {

		// TODO Auto-generated method stub

	}

	@Override

	public int compareTo(TXMResult o) {

		// Compare node weights

		int output = super.compareTo(o);

		// Compare simple names if weights are equal

			output = this.getSimpleName().compareToIgnoreCase(o.getSimpleName());

		}

		return output;

	}

	//

	//

	@Override

	public String getDetails() {

		return this.getName();

	}

	@Override

	public String getSimpleName() {

		return this.userName;

	}

	@Override

	public String getName() {

		String name = this.userName;

		int partsCount = this.getPartsCount();

			name += NLS.bind(" ({0})", partsCount);

		}

	public String getComputingStartMessage() {

		return TXMCoreMessages.bind("Création de la partition {0} dans le corpus {1}...", this.getSimpleName(), this.getCorpus().getName());

	}

	@Override

	public String getComputingDoneMessage() {

		return TXMCoreMessages.bind(TXMCoreMessages.P0BuiltParts, this.getPartsCount());

	public int getPartsCount() {

		return this.getParts().size();

	}

	/**

	 * Gets the name of the parts.

	 * 

	public List<String> getPartNames() {

		if (this.hasBeenComputedOnce()) {

			List<Part> parts = this.getParts();

			for (Subcorpus part : this.getParts()) {

				partNames.add(part.getName());

			}

			return partNames;

			return this.pPartNames;

		}

	}

	/**

	 * Gets the children parts.

	 * 

	public List<Part> getParts() {

		return (List<Part>) this.getChildren(Part.class);

	}

	/**

	 * Gets the part sizes as int array.

	 * 

	 */

	public int[] getPartSizes() throws CqiClientException {

		int[] partSizes = new int[this.getParts().size()];

			Part p = this.getParts().get(i);

			partSizes[i] = p.getSize();

		}

		return partSizes;

	}

	/**

	 * Gets the property.

	 *

	public Property getProperty() {

		return pProperty;

	}

	/**

	 * Gets the structure.

	 *

		if (pProperty == null) return null;

		return pProperty.getStructuralUnit();

	}

	/**

	 * Gets the total of all part sizes.

	 * @return

	 * @throws CqiClientException

	 */

	public int getTotalSize() throws CqiClientException {

		if (totalsize == -1) {

			totalsize = 0;

			for (Part p : getParts()) {

		}

		return totalsize;

	}

	/**

	 * Run a query on all the parts of this partition.

	 * 

	public List<QueryResult> query(CQLQuery query, String name) throws CqiClientException {

		Log.finest(TXMCoreMessages.queryingPartitionP0 + this.userName);

		List<Part> parts = getParts();

		for (Subcorpus part : parts) {

			results.add(part.query(query, part.getName() + "_" + name, false)); //$NON-NLS-1$

		}

		return results;

	}

	//

	public CQPCorpus getParent() {

		return (CQPCorpus) super.getParent();

	}

	/**

	 * Remove the specified <code>Part</code> from the partition.

	 * @param p

	 * @throws CqiClientException

	 */

	public void removePart(Part p) throws CqiClientException {

		p.delete();

	}

	@Override

	public String toString() {

		if (!this.userName.isEmpty()) {

		}

		return super.toString();

	}

	@Override

	public boolean toTxt(File output, String encoding, String colseparator, String txtseparator) throws Exception {

		IOUtils.write(output, this.dumpPreferences());

		return output.exists();

	}

	public boolean setParameters(String name, StructuralUnitProperty property, List<String> values) {

		this.pProperty = property;

		this.pValues = values;

		this.pPartNames = null;

		return true;

	}

	public boolean setParameters(String name, List<String> queries, List<String> names) {

		this.pProperty = null;

		this.pValues = null;

		this.pPartNames = names;

		return true;

	}

	@Override

	public boolean setParameters(TXMParameters parameters) {

		this.userName = parameters.getString(TXMPreferences.USER_NAME);

		String tmp = parameters.getString(CQPPreferences.PART_NAMES);

		if (tmp != null) {

			this.pPartNames = Arrays.asList(tmp.split("\t"));

			this.pPartNames = null;

		}

		try {

			this.pProperty = StructuralUnitProperty.stringToStructuralUnitProperty(getParent(), parameters.getString(TXMPreferences.STRUCTURAL_UNIT_PROPERTY));

			// TODO Auto-generated catch block

			e.printStackTrace();

			this.pProperty = null;

		}

		tmp = parameters.getString(CQPPreferences.QUERIES);

		if (tmp != null) {

			this.pQueries = Arrays.asList(tmp.split("\t"));

			this.pQueries = null;

		}

		tmp = parameters.getString(CQPPreferences.VALUES);

		if (tmp != null) {

			this.pValues = Arrays.asList(tmp.split("\t"));

			this.pValues = null;

		}

		return true;

	}

	@Override

	public boolean loadParameters() {

		String tmp = null;

		try {

			tmp = this.getStringParameterValue(TBXPreferences.STRUCTURAL_UNIT_PROPERTY);

				this.pProperty = StructuralUnitProperty.stringToStructuralUnitProperty(this.getParent(), tmp);

			}

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		tmp = this.getStringParameterValue(CQPPreferences.PART_NAMES);

		if (tmp != null) {

			this.pPartNames = Arrays.asList(tmp.split("\t"));

			this.pPartNames = null;

		}

		tmp = this.getStringParameterValue(TXMPreferences.QUERIES);

		if (tmp != null) {

			this.pQueries = Arrays.asList(tmp.split("\t"));

			this.pQueries = null;

		}

		tmp = this.getStringParameterValue(TXMPreferences.VALUES);

		if (tmp != null) {

			this.pValues = Arrays.asList(tmp.split("\t"));

			this.pValues = null;

		}

		return true;

	}

	@Override

	public boolean saveParameters() {

			this.saveParameter(TXMPreferences.STRUCTURAL_UNIT_PROPERTY, this.pProperty.getName());

		}

			this.saveParameter(CQPPreferences.PART_NAMES, StringUtils.join(pPartNames, "\t"));

		}

			this.saveParameter(TXMPreferences.QUERIES, StringUtils.join(pQueries, "\t"));

		}

			this.saveParameter(TBXPreferences.VALUES, StringUtils.join(pValues, "\t"));

		}

		return true;

	}

	/**

	 * Get the safe values (special char regexp safe).

	 *

	 * @return the safe values

	 * @throws CqiClientException the cqi client exception

	 */

		List<String> values = property.getValues(corpus);

			values.set(i, CQLQuery.addBackSlash(values.get(i)));

		}

		return values;

	}

	@Override

	public String getResultType() {

		return "Partition";

	}

	/**

	 * Gets the parent Partition of the specified result if exists.

	 * @param result

	 * @return the parent Partition if exists otherwise null

	 */

	synchronized public static Partition getFirstParentPartition(TXMResult result) {

		return (Partition) result.getFirstParent(Partition.class);

	}

	@Override

	public CQPCorpus getCorpus() {

		return (CQPCorpus) parent;

	}