Patient.java

/*******************************************************************************
 * Copyright (c) 2014 Weasis Team.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Nicolas Roduit - initial API and implementation
 *******************************************************************************/
package org.weasis.dicom.qr.manisfest;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Date;
import java.util.List;
import java.util.Locale;

import org.dcm4che3.data.Tag;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.weasis.dicom.qr.manisfest.xml.TagUtil;
import org.weasis.dicom.qr.manisfest.xml.XmlDescription;
import org.weasis.dicom.util.DateUtil;

public class Patient implements XmlDescription {

    private static final Logger LOGGER = LoggerFactory.getLogger(Patient.class);

    private final String patientID;
    private String issuerOfPatientID = null;
    private String patientName = null;
    private String patientBirthDate = null;
    private String patientBirthTime = null;
    private String patientSex = null;
    private final List<Study> studiesList;

    public Patient(String patientID) {
        this(patientID, null);
    }

    public Patient(String patientID, String issuerOfPatientID) {
        if (patientID == null) {
            throw new IllegalArgumentException("PaientID cannot be null!"); //$NON-NLS-1$
        }
        this.patientID = patientID;
        this.issuerOfPatientID = issuerOfPatientID;
        studiesList = new ArrayList<>();
    }

    public boolean hasSameUniqueID(String patientID, String issuerOfPatientID) {
        if (this.patientID.equals(patientID)) {
            if ((this.issuerOfPatientID == null && issuerOfPatientID == null)
                || (this.issuerOfPatientID != null && this.issuerOfPatientID.equals(issuerOfPatientID))) {
                return true;
            }
        }
        return false;
    }

    public String getPatientID() {
        return patientID;
    }

    public String getPatientName() {
        return patientName;
    }

    public List<Study> getStudies() {
        return studiesList;
    }

    public boolean isEmpty() {
        for (Study s : studiesList) {
            if (!s.isEmpty()) {
                return false;
            }
        }
        return true;
    }

    public String getPatientBirthTime() {
        return patientBirthTime;
    }

    public void setPatientBirthTime(String patientBirthTime) {
        this.patientBirthTime = patientBirthTime;
    }

    public String getPatientBirthDate() {
        return patientBirthDate;
    }

    public void setPatientBirthDate(String patientBirthDate) {
        this.patientBirthDate = patientBirthDate;
    }

    public String getPatientSex() {
        return patientSex;
    }

    public void setPatientSex(String patientSex) {
        if (patientSex == null) {
            this.patientSex = null;
        } else {
            String val = patientSex.toUpperCase(Locale.getDefault());
            this.patientSex = val.startsWith("M") ? "M" : val.startsWith("F") ? "F" : "O"; //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$ //$NON-NLS-4$ //$NON-NLS-5$
        }
    }

    public void setPatientName(String patientName) {
        this.patientName = patientName == null ? "" : patientName.replace("^", " ").trim(); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
    }

    public void addStudy(Study study) {
        if (!studiesList.contains(study)) {
            studiesList.add(study);
        }
    }

    /**
     *
     * @return
     */
    @Override
    public String toXml() {
        StringBuilder result = new StringBuilder();
        if (patientID != null && patientName != null) {
            result.append("\n<"); //$NON-NLS-1$
            result.append(TagUtil.Level.PATIENT);
            result.append(" "); //$NON-NLS-1$

            TagUtil.addXmlAttribute(Tag.PatientID, patientID, result);
            TagUtil.addXmlAttribute(Tag.IssuerOfPatientID, issuerOfPatientID, result);
            TagUtil.addXmlAttribute(Tag.PatientName, patientName, result);
            TagUtil.addXmlAttribute(Tag.PatientBirthDate, patientBirthDate, result);
            TagUtil.addXmlAttribute(Tag.PatientBirthTime, patientBirthTime, result);
            TagUtil.addXmlAttribute(Tag.PatientSex, patientSex, result);
            result.append(">"); //$NON-NLS-1$

            Collections.sort(studiesList, getStudyComparator());
            for (Study s : studiesList) {
                result.append(s.toXml());
            }
            result.append("\n</"); //$NON-NLS-1$
            result.append(TagUtil.Level.PATIENT);
            result.append(">"); //$NON-NLS-1$
        }

        String ptXml = result.toString();
        LOGGER.debug("Patient toXml [{}]", ptXml); //$NON-NLS-1$
        return ptXml;
    }

    private static Comparator<Study> getStudyComparator() {
        return new Comparator<Study>() {

            @Override
            public int compare(Study o1, Study o2) {
                Date date1 = DateUtil.getDicomDate(o1.getStudyDate());
                Date date2 = DateUtil.getDicomDate(o2.getStudyDate());
                if (date1 != null && date2 != null) {
                    // inverse time
                    int rep = date2.compareTo(date1);
                    if (rep == 0) {
                        Date time1 = DateUtil.getDicomTime(o1.getStudyTime());
                        Date time2 = DateUtil.getDicomTime(o2.getStudyTime());
                        if (time1 != null && time2 != null) {
                            // inverse time
                            return time2.compareTo(time1);
                        }
                    } else {
                        return rep;
                    }
                }
                if (date1 == null && date2 == null) {
                    return o1.getStudyInstanceUID().compareTo(o2.getStudyInstanceUID());
                } else {
                    if (date1 == null) {
                        return 1;
                    }
                    if (date2 == null) {
                        return -1;
                    }
                }
                return 0;
            }
        };
    }

    public Study getStudy(String uid) {
        for (Study s : studiesList) {
            if (s.getStudyInstanceUID().equals(uid)) {
                return s;
            }
        }
        return null;
    }

}