RelationshipViewer.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2012 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.financials.relation;
 
import java.util.List;
import org.turro.financials.entity.Document;
import org.turro.financials.entity.DocumentDefinition;
import org.turro.financials.entity.DocumentRelation;
import org.turro.financials.entity.DocumentWorkflow;
import org.turro.relationship.Graph;
import org.turro.relationship.Node;
import org.turro.relationship.Relationship;
import org.zkoss.zk.ui.event.Event;
import org.zkoss.zk.ui.event.Events;
import org.zkoss.zul.Hbox;
 
public class RelationshipViewer extends Hbox {
 
  private Document document;
  private Graph graph;
 
  public Document getDocument() {
    return document;
  }
 
  public void setDocument(Document document) {
    this.document = document;
  }
 
  public void notifyChange() {
    Events.postEvent(new Event(Events.ON_CHANGE, this));
    updateGraph();
  }
 
  public void updateGraph() {
    getChildren().clear();
    if(document != null && document.getContract() != null) {
      for(DocumentDefinition dd : document.getContract().getContractDefinition().getWorkflow()) {
        appendChild(new RelationshipColumn(dd, getRelationMode(dd)));
      }
    }
    generateGraph();
  }
  
  private void generateGraph() {
    graph = new Graph();
    Node center = new Node("/document/" + document.getId(), document);
    addToGraph(center);
    addDescendants(document);
    addAncestors(document);
    graph.generateGraph();
  }
 
  private void addDescendants(Document document) {
    for(DocumentRelation dr : document.getDescendants()) {
      Node start = new Node("/document/" + dr.getAncestor().getId(), dr.getAncestor()),
          end = new Node("/document/" + dr.getDescendant().getId(), dr.getDescendant());
      Relationship relationship = new Relationship("/document-relation/" + dr.getId(), dr, start, end);
      if(graph.addRelationship(relationship)) {
        addToGraph(start);
        addToGraph(end);
        addDescendants(dr.getDescendant());
        addAncestors(dr.getDescendant());
      }
    }
  }
 
  private void addAncestors(Document document) {
    for(DocumentRelation dr : document.getAncestors()) {
      Node start = new Node("/document/" + dr.getAncestor().getId(), dr.getAncestor()),
           end = new Node("/document/" + dr.getDescendant().getId(), dr.getDescendant());
      Relationship relationship = new Relationship("/document-relation/" + dr.getId(), dr, start, end);
      if(graph.addRelationship(relationship)) {
        addToGraph(start);
        addToGraph(end);
        addDescendants(dr.getAncestor());
        addAncestors(dr.getAncestor());
      }
    }
  }
  
  private void addToGraph(Node node) {
    Document doc = (Document) node.getData();
    if(doc.isBlank()) return;
    if(graph.addNode(node)) {
      List l = getChildren();
      for(int i = 0; i < l.size(); i++) {
        RelationshipColumn rc = (RelationshipColumn) l.get(i);
        if(rc.getDocumentDefinition().getId() == doc.getDocumentDefinition().getId()) {
          rc.appendChild(new DocumentBox(doc));
          node.setX(i);
          break;
        }
      }
    }
  }
 
  private DocumentRelationMode getRelationMode(DocumentDefinition docDef) {
    for(DocumentWorkflow dw : document.getContract().getContractDefinition().getDocumentWorkflows()) {
      if(dw.getAncestor().getId() == docDef.getId() &&
              dw.getDescendant().getId() == document.getDocumentDefinition().getId()) {
        return DocumentRelationMode.RELATE_TO_ANCESTORS;
      } else if(dw.getDescendant().getId() == docDef.getId() &&
              dw.getAncestor().getId() == document.getDocumentDefinition().getId()) {
        return DocumentRelationMode.RELATE_TO_DESCENDANTS;
      }
    }
    return DocumentRelationMode.RELATE_TO_NONE;
  }
 
}