org.dcm4chee.archive.query.impl.SeriesQuery.java Source code

Java tutorial

Introduction

Here is the source code for org.dcm4chee.archive.query.impl.SeriesQuery.java

Source

/* ***** 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.dcm4chee.archive.query.impl;

import org.dcm4che3.data.Attributes;
import org.dcm4chee.archive.conf.ArchiveDeviceExtension;
import org.dcm4chee.archive.conf.PrivateTag;
import org.dcm4chee.archive.entity.QSeries;
import org.dcm4chee.archive.entity.QSeriesQueryAttributes;
import org.dcm4chee.archive.entity.QStudy;
import org.dcm4chee.archive.entity.QStudyQueryAttributes;
import org.dcm4chee.archive.entity.Series;
import org.dcm4chee.archive.entity.SeriesQueryAttributes;
import org.dcm4chee.archive.entity.StudyQueryAttributes;
import org.dcm4chee.archive.entity.Utils;
import org.dcm4chee.archive.query.QueryContext;
import org.dcm4chee.archive.query.util.QueryBuilder;
import org.dcm4chee.storage.conf.Availability;
import org.hibernate.ScrollableResults;
import org.hibernate.StatelessSession;

import com.mysema.query.BooleanBuilder;
import com.mysema.query.jpa.hibernate.HibernateQuery;
import com.mysema.query.types.Expression;
import com.mysema.query.types.Predicate;

import java.util.Date;

/**
 * @author Gunter Zeilinger <gunterze@gmail.com>
 */
class SeriesQuery extends AbstractQuery<Series> {

    private static final Expression<?>[] SELECT = { QStudy.study.pk, // (0)
            QSeries.series.pk, // (1)
            QSeriesQueryAttributes.seriesQueryAttributes.numberOfInstances, // (2)
            QStudyQueryAttributes.studyQueryAttributes.numberOfInstances, // (3)
            QStudyQueryAttributes.studyQueryAttributes.numberOfSeries, // (4)
            QStudyQueryAttributes.studyQueryAttributes.modalitiesInStudy, // (5)
            QStudyQueryAttributes.studyQueryAttributes.sopClassesInStudy, // (6)
            QStudyQueryAttributes.studyQueryAttributes.numberOfVisibleInstances, // (7)
            QStudyQueryAttributes.studyQueryAttributes.lastUpdateTime, // (8)
            QSeriesQueryAttributes.seriesQueryAttributes.retrieveAETs, // (9)
            QSeriesQueryAttributes.seriesQueryAttributes.availability, // (10)
            QSeriesQueryAttributes.seriesQueryAttributes.numberOfVisibleInstances, // (11)
            QSeriesQueryAttributes.seriesQueryAttributes.lastUpdateTime, // (12)
            QueryBuilder.seriesAttributesBlob.encodedAttributes, // (13)
            QueryBuilder.studyAttributesBlob.encodedAttributes, // (14)
            QueryBuilder.patientAttributesBlob.encodedAttributes // (15)
    };

    private Long studyPk;
    private Attributes studyAttrs;

    public SeriesQuery(QueryContext context, StatelessSession session) {
        super(context, session, QSeries.series);
    }

    @Override
    protected Expression<?>[] select() {
        return SELECT;
    }

    @Override
    protected HibernateQuery applyJoins(HibernateQuery query) {
        query = QueryBuilder.applySeriesLevelJoins(query, context.getKeys(), context.getQueryParam());
        query = QueryBuilder.applyStudyLevelJoins(query, context.getKeys(), context.getQueryParam());
        query = QueryBuilder.applyPatientLevelJoins(query, context.getKeys(), context.getQueryParam());
        return query;
    }

    @Override
    protected Predicate predicate() {
        BooleanBuilder builder = new BooleanBuilder();
        QueryBuilder.addPatientLevelPredicates(builder, context.getPatientIDs(), context.getKeys(),
                context.getQueryParam());
        QueryBuilder.addStudyLevelPredicates(builder, context.getKeys(), context.getQueryParam());
        QueryBuilder.addSeriesLevelPredicates(builder, context.getKeys(), context.getQueryParam());
        return builder;
    }

    @Override
    public Attributes toAttributes(ScrollableResults results, QueryContext context) {
        Long studyPk = results.getLong(0);
        Long seriesPk = results.getLong(1);
        Integer numberOfInstancesI = results.getInteger(2);
        int numberOfSeriesRelatedInstances;
        String retrieveAETs;
        Availability availability;
        int numberOfSeriesVisibleInstances;
        Date seriesLastUpdateTime;
        if (numberOfInstancesI != null) {
            numberOfSeriesRelatedInstances = numberOfInstancesI;
            if (numberOfSeriesRelatedInstances == 0)
                return null;
            retrieveAETs = results.getString(9);
            availability = (Availability) results.get(10);
            numberOfSeriesVisibleInstances = results.getInteger(11);
            seriesLastUpdateTime = results.getDate(12);
        } else {
            SeriesQueryAttributes seriesView = context.getQueryService().createSeriesView(seriesPk,
                    context.getQueryParam());
            numberOfSeriesRelatedInstances = seriesView.getNumberOfInstances();
            if (numberOfSeriesRelatedInstances == 0)
                return null;
            retrieveAETs = seriesView.getRawRetrieveAETs();
            availability = seriesView.getAvailability();
            numberOfSeriesVisibleInstances = seriesView.getNumberOfVisibleInstances();
            seriesLastUpdateTime = seriesView.getLastUpdateTime();
        }

        byte[] seriesAttributes = results.getBinary(13);
        if (!studyPk.equals(this.studyPk)) {
            this.studyAttrs = toStudyAttributes(studyPk, results, context);
            this.studyPk = studyPk;
        }
        Attributes seriesAttrs = new Attributes();
        Utils.decodeAttributes(seriesAttrs, seriesAttributes);
        Attributes attrs = Utils.mergeAndNormalize(studyAttrs, seriesAttrs);
        ArchiveDeviceExtension ade = context.getArchiveAEExtension().getApplicationEntity().getDevice()
                .getDeviceExtension(ArchiveDeviceExtension.class);
        Utils.setSeriesQueryAttributes(attrs, numberOfSeriesRelatedInstances, numberOfSeriesVisibleInstances,
                ade.getPrivateDerivedFields().findSeriesNumberOfVisibleInstancesTag(), seriesLastUpdateTime,
                ade.getPrivateDerivedFields().findSeriesUpdateTimeTag());
        Utils.setRetrieveAET(attrs, retrieveAETs);
        Utils.setAvailability(attrs, availability);

        return attrs;
    }

    private Attributes toStudyAttributes(Long studyPk, ScrollableResults results, QueryContext context) {
        Integer numberOfInstancesI = results.getInteger(3);
        int numberOfStudyRelatedInstances;
        int numberOfStudyRelatedSeries;
        String modalitiesInStudy;
        String sopClassesInStudy;
        int numberOfStudyVisibleInstances;
        Date studyLastUpdateTime;
        if (numberOfInstancesI != null) {
            numberOfStudyRelatedInstances = numberOfInstancesI;
            numberOfStudyRelatedSeries = results.getInteger(4);
            modalitiesInStudy = results.getString(5);
            sopClassesInStudy = results.getString(6);
            numberOfStudyVisibleInstances = results.getInteger(7);
            studyLastUpdateTime = results.getDate(8);
        } else {
            StudyQueryAttributes studyView = context.getQueryService().createStudyView(studyPk,
                    context.getQueryParam());
            numberOfStudyRelatedInstances = studyView.getNumberOfInstances();
            numberOfStudyRelatedSeries = studyView.getNumberOfSeries();
            modalitiesInStudy = studyView.getRawModalitiesInStudy();
            sopClassesInStudy = studyView.getRawSOPClassesInStudy();
            numberOfStudyVisibleInstances = studyView.getNumberOfVisibleInstances();
            studyLastUpdateTime = studyView.getLastUpdateTime();
        }

        byte[] studyByteAttributes = results.getBinary(14);
        byte[] patientByteAttributes = results.getBinary(15);
        Attributes patientAttrs = new Attributes();
        Attributes studyAttrs = new Attributes();
        Utils.decodeAttributes(patientAttrs, patientByteAttributes);
        Utils.decodeAttributes(studyAttrs, studyByteAttributes);
        Attributes attrs = Utils.mergeAndNormalize(patientAttrs, studyAttrs);
        ArchiveDeviceExtension ade = context.getArchiveAEExtension().getApplicationEntity().getDevice()
                .getDeviceExtension(ArchiveDeviceExtension.class);
        Utils.setStudyQueryAttributes(attrs, numberOfStudyRelatedSeries, numberOfStudyRelatedInstances,
                modalitiesInStudy, sopClassesInStudy, numberOfStudyVisibleInstances,
                ade.getPrivateDerivedFields().findStudyNumberOfVisibleInstancesTag(), studyLastUpdateTime,
                ade.getPrivateDerivedFields().findStudyUpdateTimeTag());

        return attrs;
    }
}