Java tutorial
/* ***** 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) 2011 * 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.dcm4che.tool.storescp; import java.io.File; import java.io.IOException; import java.util.Properties; import java.util.ResourceBundle; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.ScheduledExecutorService; import org.apache.commons.cli.CommandLine; import org.apache.commons.cli.OptionBuilder; import org.apache.commons.cli.Options; import org.apache.commons.cli.ParseException; import org.dcm4che.data.Attributes; import org.dcm4che.data.Tag; import org.dcm4che.data.VR; import org.dcm4che.io.DicomInputStream; import org.dcm4che.io.DicomInputStream.IncludeBulkData; import org.dcm4che.io.DicomOutputStream; import org.dcm4che.net.ApplicationEntity; import org.dcm4che.net.Association; import org.dcm4che.net.Connection; import org.dcm4che.net.Device; import org.dcm4che.net.PDVInputStream; import org.dcm4che.net.Status; import org.dcm4che.net.TransferCapability; import org.dcm4che.net.pdu.PresentationContext; import org.dcm4che.net.service.BasicCEchoSCP; import org.dcm4che.net.service.BasicCStoreSCP; import org.dcm4che.net.service.DicomServiceException; import org.dcm4che.net.service.DicomServiceRegistry; import org.dcm4che.tool.common.CLIUtils; import org.dcm4che.util.AttributesFormat; import org.dcm4che.util.SafeClose; import org.dcm4che.util.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; /** * @author Gunter Zeilinger <gunterze@gmail.com> * */ public class StoreSCP { private static final Logger LOG = LoggerFactory.getLogger(StoreSCP.class); private static ResourceBundle rb = ResourceBundle.getBundle("org.dcm4che.tool.storescp.messages"); private static final String PART_EXT = ".part"; private final Device device = new Device("storescp"); private final ApplicationEntity ae = new ApplicationEntity("*"); private final Connection conn = new Connection(); private File storageDir; private AttributesFormat filePathFormat; private int status; private final BasicCStoreSCP cstoreSCP = new BasicCStoreSCP("*") { @Override protected void store(Association as, PresentationContext pc, Attributes rq, PDVInputStream data, Attributes rsp) throws IOException { rsp.setInt(Tag.Status, VR.US, status); if (storageDir == null) return; String cuid = rq.getString(Tag.AffectedSOPClassUID); String iuid = rq.getString(Tag.AffectedSOPInstanceUID); String tsuid = pc.getTransferSyntax(); File file = new File(storageDir, iuid + PART_EXT); try { storeTo(as, as.createFileMetaInformation(iuid, cuid, tsuid), data, file); renameTo(as, file, new File(storageDir, filePathFormat == null ? iuid : filePathFormat.format(parse(file)))); } catch (Exception e) { deleteFile(as, file); throw new DicomServiceException(Status.ProcessingFailure, e); } } }; public StoreSCP() throws IOException { device.setDimseRQHandler(createServiceRegistry()); device.addConnection(conn); device.addApplicationEntity(ae); ae.setAssociationAcceptor(true); ae.addConnection(conn); } private void storeTo(Association as, Attributes fmi, PDVInputStream data, File file) throws IOException { LOG.info("{}: M-WRITE {}", as, file); file.getParentFile().mkdirs(); DicomOutputStream out = new DicomOutputStream(file); try { out.writeFileMetaInformation(fmi); data.copyTo(out); } finally { SafeClose.close(out); } } private static void renameTo(Association as, File from, File dest) throws IOException { LOG.info("{}: M-RENAME {}", new Object[] { as, from, dest }); dest.getParentFile().mkdirs(); if (!from.renameTo(dest)) throw new IOException("Failed to rename " + from + " to " + dest); } private static Attributes parse(File file) throws IOException { DicomInputStream in = new DicomInputStream(file); try { in.setIncludeBulkData(IncludeBulkData.NO); return in.readDataset(-1, Tag.PixelData); } finally { SafeClose.close(in); } } private static void deleteFile(Association as, File file) { if (file.delete()) LOG.info("{}: M-DELETE {}", as, file); else LOG.warn("{}: M-DELETE {} failed!", as, file); } private DicomServiceRegistry createServiceRegistry() { DicomServiceRegistry serviceRegistry = new DicomServiceRegistry(); serviceRegistry.addDicomService(new BasicCEchoSCP()); serviceRegistry.addDicomService(cstoreSCP); return serviceRegistry; } public void setStorageDirectory(File storageDir) { if (storageDir != null) storageDir.mkdirs(); this.storageDir = storageDir; } public void setStorageFilePathFormat(String pattern) { this.filePathFormat = new AttributesFormat(pattern); } public void setStatus(int status) { this.status = status; } private static CommandLine parseComandLine(String[] args) throws ParseException { Options opts = new Options(); CLIUtils.addBindServerOption(opts); CLIUtils.addAEOptions(opts); CLIUtils.addCommonOptions(opts); addStatusOption(opts); addStorageDirectoryOptions(opts); addTransferCapabilityOptions(opts); return CLIUtils.parseComandLine(args, opts, rb, StoreSCP.class); } @SuppressWarnings("static-access") private static void addStatusOption(Options opts) { opts.addOption(OptionBuilder.hasArg().withArgName("code").withDescription(rb.getString("status")) .withLongOpt("status").create(null)); } @SuppressWarnings("static-access") private static void addStorageDirectoryOptions(Options opts) { opts.addOption(null, "ignore", false, rb.getString("ignore")); opts.addOption(OptionBuilder.hasArg().withArgName("path").withDescription(rb.getString("directory")) .withLongOpt("directory").create(null)); opts.addOption(OptionBuilder.hasArg().withArgName("pattern").withDescription(rb.getString("filepath")) .withLongOpt("filepath").create(null)); } @SuppressWarnings("static-access") private static void addTransferCapabilityOptions(Options opts) { opts.addOption(null, "accept-unknown", false, rb.getString("accept-unknown")); opts.addOption(OptionBuilder.hasArg().withArgName("file|url").withDescription(rb.getString("sop-classes")) .withLongOpt("sop-classes").create(null)); } public static void main(String[] args) { try { CommandLine cl = parseComandLine(args); StoreSCP main = new StoreSCP(); CLIUtils.configureBindServer(main.conn, main.ae, cl); CLIUtils.configure(main.conn, cl); main.setStatus(CLIUtils.getIntOption(cl, "status", 0)); configureTransferCapability(main.ae, cl); configureStorageDirectory(main, cl); ExecutorService executorService = Executors.newCachedThreadPool(); ScheduledExecutorService scheduledExecutorService = Executors.newSingleThreadScheduledExecutor(); main.device.setScheduledExecutor(scheduledExecutorService); main.device.setExecutor(executorService); main.device.bindConnections(); } catch (ParseException e) { System.err.println("storescp: " + e.getMessage()); System.err.println(rb.getString("try")); System.exit(2); } catch (Exception e) { System.err.println("storescp: " + e.getMessage()); e.printStackTrace(); System.exit(2); } } private static void configureStorageDirectory(StoreSCP main, CommandLine cl) { if (!cl.hasOption("ignore")) { main.setStorageDirectory(new File(cl.getOptionValue("directory", "."))); if (cl.hasOption("filepath")) main.setStorageFilePathFormat(cl.getOptionValue("filepath")); } } private static void configureTransferCapability(ApplicationEntity ae, CommandLine cl) throws IOException { if (cl.hasOption("accept-unknown")) { ae.addTransferCapability(new TransferCapability(null, "*", TransferCapability.Role.SCP, "*")); } else { Properties p = CLIUtils .loadProperties(cl.getOptionValue("sop-classes", "resource:sop-classes.properties"), null); for (String cuid : p.stringPropertyNames()) { String ts = p.getProperty(cuid); ae.addTransferCapability( ts.equals("*") ? new TransferCapability(null, cuid, TransferCapability.Role.SCP, "*") : new TransferCapability(null, cuid, TransferCapability.Role.SCP, StringUtils.split(ts, ','))); } } } }