AbstractTreeModel

answered 2013-05-22 09:56:23 +0800

fiendnet
8 ● 1

public class OrganizationAbstractTreeModel extends AbstractTreeModel<Object> implements TreeSelectableModel  {

private static final long serialVersionUID = -5513180500300189445L;
private OrganizationRelateManager organizationRelateManager = ServiceLocator.getOrganizationRelateManager();
private String startDate;
private String startTime;
private String finishDate;
private String finishTime;
private Boolean hasDistributetable; 
private Distributedetail distributeDetail;  
private Organizationmaster oneRowParentOrganization;

public OrganizationAbstractTreeModel(Organizationdetail rootNode,String startDate,String startTime,String finishDate,String finishTime,Boolean hasDistributetable,Boolean loadOneRow,Distributedetail distributeDetail) {
    super(rootNode);
    this.startDate=startDate;
    this.startTime=startTime;
    this.finishDate=finishDate;
    this.finishTime=finishTime;
    this.hasDistributetable=hasDistributetable;     
    this.distributeDetail=distributeDetail;
    this.oneRowParentOrganization= loadOneRow ? rootNode.getOrganizationmaster():null;
}

@Override
public Object getChild(Object parentNode, int index) {
    List<Object> nodes;
    Organizationdetail organizationDetail;
    nodes= new ArrayList<Object>();
    if (parentNode instanceof Organizationdetail){
        organizationDetail=(Organizationdetail)parentNode;
        if (distributeDetail!=null &&(( organizationDetail.getHasdistributetable() && oneRowParentOrganization.getId()==organizationDetail.getOrganizationmaster().getId()) || !organizationDetail.getHasdistributetable() ))           
            nodes.addAll(organizationRelateManager.getOrganizationPostDAO().listByMaster(organizationDetail.getOrganizationmaster(),distributeDetail.getStartdate(),distributeDetail.getStarttime(),distributeDetail.getFinishdate(),distributeDetail.getFinishtime()));            
        nodes.addAll(organizationRelateManager.getOrganizationRelateDAO().listByParent(organizationDetail.getOrganizationmaster(),startDate,startTime,finishDate,finishTime,hasDistributetable,oneRowParentOrganization));
    }
    return nodes.get(index);
}

@Override
public int getChildCount(Object parentNode) {
    List<Object> nodes;
    Organizationdetail organizationDetail;
    nodes= new ArrayList<Object>();
    if (parentNode instanceof Organizationdetail){
        organizationDetail=(Organizationdetail)parentNode;
        if (distributeDetail!=null &&(( organizationDetail.getHasdistributetable() && oneRowParentOrganization.getId()==organizationDetail.getOrganizationmaster().getId()) || !organizationDetail.getHasdistributetable() ))           
            nodes.addAll(organizationRelateManager.getOrganizationPostDAO().listByMaster(organizationDetail.getOrganizationmaster(),distributeDetail.getStartdate(),distributeDetail.getStarttime(),distributeDetail.getFinishdate(),distributeDetail.getFinishtime()));            
        nodes.addAll(organizationRelateManager.getOrganizationRelateDAO().listByParent(organizationDetail.getOrganizationmaster(),startDate,startTime,finishDate,finishTime,hasDistributetable,oneRowParentOrganization));
    }
    return nodes.size();
}

@Override
public boolean isLeaf(Object currentNode) {
    Organizationdetail organizationDetail;
    List<Object> childs;    
    childs= new ArrayList<Object>();
    if (currentNode instanceof Organizationdetail){
        organizationDetail=(Organizationdetail)currentNode;
        if (distributeDetail!=null &&(( organizationDetail.getHasdistributetable() && oneRowParentOrganization.getId()==organizationDetail.getOrganizationmaster().getId()) || !organizationDetail.getHasdistributetable() ))           
            childs.addAll(organizationRelateManager.getOrganizationPostDAO().listByMaster(organizationDetail.getOrganizationmaster(),distributeDetail.getStartdate(),distributeDetail.getStarttime(),distributeDetail.getFinishdate(),distributeDetail.getFinishtime()));           
        childs.addAll(organizationRelateManager.getOrganizationRelateDAO().listByParent(organizationDetail.getOrganizationmaster(),startDate,startTime,finishDate,finishTime,hasDistributetable,oneRowParentOrganization));
    }
    return childs.size()==0;
}