package com.bbn.nlp.corpora.ere;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;

import java.util.List;

import static com.google.common.base.Preconditions.checkNotNull;

public final class EREDocument {
  private final String docid;
  private final String sourceType;
  private final ImmutableList<EREEntity> entities;
  private final ImmutableList<EREFiller> fillers;
  private final ImmutableList<ERERelation> relations;
  private final ImmutableList<EREEvent> events;

  private EREDocument(final String docid, final String sourceType,
      final List<EREEntity> entities, final List<EREFiller> fillers,
      final List<ERERelation> relations, final List<EREEvent> events) {
    this.docid = checkNotNull(docid);
    this.sourceType = checkNotNull(sourceType);
    this.entities = ImmutableList.copyOf(entities);
    this.fillers = ImmutableList.copyOf(fillers);
    this.relations = ImmutableList.copyOf(relations);
    this.events = ImmutableList.copyOf(events);
  }

  public String getDocId() {
    return docid;
  }

  public String getSourceType() {
    return sourceType;
  }

  public List<EREEntity> getEntities() {
    return entities;
  }

  public List<EREFiller> getFillers() {
    return fillers;
  }

  public List<ERERelation> getRelations() {
    return relations;
  }

  public List<EREEvent> getEvents() {
    return events;
  }

  public static Builder builder(final String docid, final String sourceType) {
    return new Builder(docid, sourceType);
  }

  public static class Builder {
    private final String docid;
    private final String sourceType;
    private final List<EREEntity> entities;
    private final List<EREFiller> fillers;
    private final List<ERERelation> relations;
    private final List<EREEvent> events;

    public Builder(final String docid, final String sourceType) {
      this.docid = checkNotNull(docid);
      this.sourceType = checkNotNull(sourceType);
      this.entities = Lists.newArrayList();
      this.fillers = Lists.newArrayList();
      this.relations = Lists.newArrayList();
      this.events = Lists.newArrayList();
    }

    public EREDocument build() {
      return new EREDocument(docid, sourceType, entities, fillers, relations, events);
    }

    public Builder withEntity(EREEntity e) {
      this.entities.add(e);
      return this;
    }

    public Builder withFiller(EREFiller v) {
      this.fillers.add(v);
      return this;
    }

    public Builder withRelation(ERERelation r) {
      this.relations.add(r);
      return this;
    }

    public Builder withEvent(EREEvent e) {
      this.events.add(e);
      return this;
    }

  }

  public static String formatText(String text) {
    return text.replaceAll("\\n+", "").replace("&", " &amp; ");
  }

}

@@ -2663,124 +2663,6 @@
  %7D%0A%0A
-  public static String formatText(String text) %7B%0A    return text.replaceAll(%22%5C%5Cn+%22, %22%22).replace(%22&%22, %22 &amp; %22);%0A  %7D%0A%0A
 %7D%0A

Remove unused method