com.mirth.connect.server.attachments.DICOMAttachmentHandler.java Source code

Introduction

Here is the source code for com.mirth.connect.server.attachments.DICOMAttachmentHandler.java

Source

/*
 * Copyright (c) Mirth Corporation. All rights reserved.
 * http://www.mirthcorp.com
 * 
 * The software in this package is published under the terms of the MPL
 * license a copy of which has been included with this distribution in
 * the LICENSE.txt file.
 */

package com.mirth.connect.server.attachments;

import org.apache.commons.codec.binary.StringUtils;
import org.dcm4che2.data.DicomElement;
import org.dcm4che2.data.DicomObject;
import org.dcm4che2.data.Tag;

import com.mirth.connect.donkey.model.message.attachment.Attachment;
import com.mirth.connect.donkey.model.message.attachment.AttachmentHandler;
import com.mirth.connect.donkey.model.message.attachment.AttachmentHandlerProperties;
import com.mirth.connect.donkey.server.channel.Channel;
import com.mirth.connect.donkey.util.Base64Util;
import com.mirth.connect.model.attachments.AttachmentException;
import com.mirth.connect.model.converters.DICOMConverter;
import com.mirth.connect.server.util.UUIDGenerator;

public class DICOMAttachmentHandler extends AttachmentHandler {

    private DicomObject dicomObject;
    private DicomElement dicomElement;
    private int index;

    public DICOMAttachmentHandler() {

    }

    @Override
    public void initialize(String message, Channel channel) throws AttachmentException {
        // Taking a string is much more inefficient than taking in a byte array. 
        // If the user manually sends a message, it will arrive as a base64 encoded string, so we must support Strings for DICOM still.
        // However, DICOM messages that use this initializer should be relatively small in size.
        index = 0;
        try {
            dicomObject = DICOMConverter.byteArrayToDicomObject(StringUtils.getBytesUsAscii(message), true);
            dicomElement = dicomObject.remove(Tag.PixelData);
        } catch (Throwable t) {
            throw new AttachmentException(t);
        }
    }

    @Override
    public void initialize(byte[] bytes, Channel channel) throws AttachmentException {
        index = 0;
        try {
            dicomObject = DICOMConverter.byteArrayToDicomObject(bytes, false);
            dicomElement = dicomObject.remove(Tag.PixelData);
        } catch (Throwable t) {
            throw new AttachmentException(t);
        }
    }

    @Override
    public Attachment nextAttachment() throws AttachmentException {
        try {
            if (dicomElement != null) {
                if (dicomElement.hasItems()) {
                    if (index < dicomElement.countItems()) {
                        return new Attachment(UUIDGenerator.getUUID(), dicomElement.getFragment(index++), "DICOM");
                    }
                } else {
                    Attachment attachment = new Attachment(UUIDGenerator.getUUID(), dicomElement.getBytes(),
                            "DICOM");
                    dicomElement = null;
                    return attachment;
                }
            }

            return null;
        } catch (Throwable t) {
            throw new AttachmentException(t);
        }
    }

    @Override
    public String shutdown() throws AttachmentException {
        try {
            byte[] encodedMessage = Base64Util.encodeBase64(DICOMConverter.dicomObjectToByteArray(dicomObject));
            dicomElement = null;
            dicomObject = null;
            return StringUtils.newStringUsAscii(encodedMessage);
        } catch (Throwable t) {
            throw new AttachmentException(t);
        }

    }

    @Override
    public void setProperties(AttachmentHandlerProperties attachmentProperties) {

    }

}