1
2
3
4
5
6
7 package com.vikash.firsttool.Diagram;
8
9 import com.vikash.firsttool.UI.*;
10 import com.vikash.firsttool.ProfileImpl.*;
11 import org.jgraph.graph.GraphModel;
12 import org.jgraph.graph.GraphLayoutCache;
13 import org.jgraph.graph.DefaultCellViewFactory;
14
15
16 import javax.swing.tree.*;
17 /***
18 *
19 * @author studajb
20 */
21 public class TreatmentEffectDiagram extends ToolGraph{
22
23 /*** Creates a new instance of TreatmentEffectDiagram */
24 public TreatmentEffectDiagram() {
25 super(ToolModel.getInstance(false),"TreatmentEffectDiagram");
26
27 }
28 public boolean isValidElement(DefaultMutableTreeNode cell){
29 if(cell instanceof UnwantedIncidentGraphCell || cell instanceof TreatmentEffectGraphCell || cell instanceof TreatmentGraphCell
30 || cell instanceof RiskGraphCell )
31 return true;
32 else
33 return false;
34 }
35 public String toString(){
36 String name="TreatmentEffect Diagram";
37 return name;
38 }
39 }