public TreatmentDto createTreatmentDto(DrugTreatment t) { TreatmentDto treatment = factory.createTreatmentDto(); treatment.setDiagnosis(t.getDiagnosis()); DrugTreatmentType drugTreatment = new DrugTreatmentType(); drugTreatment.setDosage(t.getDosage()); drugTreatment.setName(t.getDrug()); treatment.setDrugTreatment(drugTreatment); return treatment; }