DefaultMPPSProcessing.java

/*
 * *** BEGIN LICENSE BLOCK *****
 *  Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 *  The contents of this file are subject to the Mozilla Public License Version
 *  1.1 (the "License"); you may not use this file except in compliance with
 *  the License. You may obtain a copy of the License at
 *  http://www.mozilla.org/MPL/
 *
 *  Software distributed under the License is distributed on an "AS IS" basis,
 *  WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 *  for the specific language governing rights and limitations under the
 *  License.
 *
 *  The Original Code is part of dcm4che, an implementation of DICOM(TM) in
 *  Java(TM), hosted at https://github.com/gunterze/dcm4che.
 *
 *  The Initial Developer of the Original Code is
 *  Agfa Healthcare.
 *  Portions created by the Initial Developer are Copyright (C) 2015
 *  the Initial Developer. All Rights Reserved.
 *
 *  Contributor(s):
 *  See @authors listed below
 *
 *  Alternatively, the contents of this file may be used under the terms of
 *  either the GNU General Public License Version 2 or later (the "GPL"), or
 *  the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 *  in which case the provisions of the GPL or the LGPL are applicable instead
 *  of those above. If you wish to allow use of your version of this file only
 *  under the terms of either the GPL or the LGPL, and not to allow others to
 *  use your version of this file under the terms of the MPL, indicate your
 *  decision by deleting the provisions above and replace them with the notice
 *  and other provisions required by the GPL or the LGPL. If you do not delete
 *  the provisions above, a recipient may use your version of this file under
 *  the terms of any one of the MPL, the GPL or the LGPL.
 *
 *  ***** END LICENSE BLOCK *****
 */

package org.dcm4chee.archive.mpps.impl;

import org.dcm4che3.data.Attributes;
import org.dcm4che3.data.UID;
import org.dcm4che3.io.SAXTransformer;
import org.dcm4che3.net.ApplicationEntity;
import org.dcm4che3.net.Device;
import org.dcm4che3.net.Status;
import org.dcm4che3.net.TransferCapability;
import org.dcm4che3.net.service.DicomServiceException;
import org.dcm4chee.archive.conf.ArchiveAEExtension;
import org.dcm4chee.archive.entity.MPPS;
import org.dcm4chee.archive.mpps.MPPSContext;
import org.dcm4chee.archive.mpps.MPPSHook;

import javax.inject.Inject;
import javax.xml.transform.Templates;

/**
 * @author Roman K
 */
public class DefaultMPPSProcessing extends MPPSHook {

    @Inject
    private Device device;

    @Inject
    private DefaultMPPSProcessingEJB ejb;


    @Override
    public void coerceAttributes(MPPSContext context, Attributes attributes) throws DicomServiceException {

        // XSLT
        try {
            ApplicationEntity ae = device.getApplicationEntityNotNull(context.getReceivingAET());
            ArchiveAEExtension arcAE = ae.getAEExtensionNotNull(ArchiveAEExtension.class);
            Templates tpl = arcAE.getAttributeCoercionTemplates(
                    UID.ModalityPerformedProcedureStepSOPClass,
                    context.getDimse(), TransferCapability.Role.SCP,
                    context.getSendingAET());
            if (tpl != null) {
                Attributes modified = new Attributes();
                attributes.update(SAXTransformer.transform(attributes, tpl, false, false), modified);
            }
        } catch (Exception e) {
            throw new DicomServiceException(Status.ProcessingFailure, e);
        }
    }


    @Override
    public void onMPPSCreate(MPPSContext context, Attributes attributes) throws DicomServiceException {
        ApplicationEntity ae = device.getApplicationEntityNotNull(context.getReceivingAET());
        ejb.createPerformedProcedureStep(ae, context.getMppsSopInstanceUID(), attributes);
    }

    @Override
    public void onMPPSUpdate(MPPSContext context, Attributes attributes) throws DicomServiceException {
        updateOrFinal(context, attributes);
    }

    @Override
    public void onMPPSFinal(MPPSContext context, Attributes attributes) throws DicomServiceException {
        updateOrFinal(context, attributes);
    }

    private void updateOrFinal(MPPSContext context, Attributes attributes) throws DicomServiceException {
        ApplicationEntity ae = device.getApplicationEntityNotNull(context.getReceivingAET());
        ejb.updatePerformedProcedureStep(ae, context.getMppsSopInstanceUID(), attributes);
    }
}