concrete.validation.CommunicationValidator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of concrete-validation Show documentation
Validation library for use with Concrete-Java
There is a newer version: 4.15.0
/*
 * Copyright 2012-2014 Johns Hopkins University HLTCOE. All rights reserved.
 * See LICENSE in the project root directory.
 */

package concrete.validation;

import java.util.Iterator;
import java.util.List;
import java.util.concurrent.Future;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import edu.jhu.hlt.concrete.Communication;
import edu.jhu.hlt.concrete.EntityMentionSet;
import edu.jhu.hlt.concrete.Section;
import edu.jhu.hlt.concrete.Sentence;
import edu.jhu.hlt.concrete.Tokenization;
import edu.jhu.hlt.concrete.concurrent.ConcurrentCommunicationLoader;
import edu.jhu.hlt.concrete.validation.ValidatableEntityMentionSet;
import edu.jhu.hlt.concrete.validation.ValidatableSection;
import edu.jhu.hlt.concrete.validation.ValidatableSentence;
import edu.jhu.hlt.concrete.validation.ValidatableTokenization;

/**
 * @author max
 *
 */
public class CommunicationValidator {
  
  private static final Logger logger = LoggerFactory.getLogger(CommunicationValidator.class);
  
  private Communication comm;
  
  /**
   * 
   */
  public CommunicationValidator(Communication comm) {
    this.comm = comm;
  }
  
  public boolean validate() {
    boolean valid = true;
    // for (Section sect : this.comm.getSectionList()) {
    Iterator sectIter = this.comm.getSectionListIterator();
    while (valid && sectIter != null && sectIter.hasNext()) {
      Section s = sectIter.next();
      valid = new ValidatableSection(s).validate(this.comm);
      
      Iterator sentIter = s.getSentenceListIterator();
      while (valid && sentIter != null && sentIter.hasNext()) {
        Sentence st = sentIter.next();
        valid = new ValidatableSentence(st).validate(this.comm);
        Tokenization tok = st.getTokenization();
        valid = new ValidatableTokenization(tok).validate(this.comm);
      }
    }
    
    if (this.comm.isSetEntityMentionSetList()) {
      Iterator emsIter = this.comm.getEntityMentionSetListIterator();
      while (valid && emsIter.hasNext()) {
        EntityMentionSet ems = emsIter.next();
        valid = new ValidatableEntityMentionSet(ems).validate(this.comm);
      }
    }
    
    return valid;  
  }

  /**
   * @param args
   * @throws Exception 
   */
  public static void main(String[] args) throws Exception {
    // TODO Auto-generated method stub
    if (args.length != 1) {
      System.out.println("This program takes a text file containing a communication per line and validates each of them.");
      System.out.println("Usage: " + CommunicationValidator.class.getName() + " ");
      System.exit(1);
    }
    
    try (ConcurrentCommunicationLoader ccl = new ConcurrentCommunicationLoader();) {
      List> commList = ccl.bulkLoad(args[0]);
      for (Future c : commList) {
        Communication comm = c.get();
        String result = new CommunicationValidator(comm).validate() ? "VALID." : "INVALID.";
        logger.info("Communication: " + comm.getId() + " is " + result);
      }
    }
  }
}