ohif-viewer/Packages/lesiontracker/compatibility/measurementManagerDAL.js

var measurementManagerDAL = (function() {
    var trialPatientLocations = [];

    // Returns trialPatientLocations array
    function getPatientLocations() {
        return trialPatientLocations;
    }

    function getLocationName(locationUID) {
        for (var i = 0; i < trialPatientLocations.length; i++) {
            var locationObject = trialPatientLocations[i];
            if (locationObject.uid === locationUID) {
                return locationObject.location.location;
            }
        }
        return "";
    }

    // Adds new location to trialPatientLocations array
    function addPatientLocation(location) {
        var locationUID = uuid.v4();
        var locationObject = {
            uid: locationUID,
            location: location
        };
        trialPatientLocations.push(locationObject);
        return locationUID;
    }

    // Add timepoint data to Measurements collection
    function addTimepointData(lesionData) {
        var timepoints = Timepoints.find().fetch();
        var timepointsObject = {};

        for (var i = 0; i < timepoints.length; i++) {
            var timepointId = timepoints[i].timepointID;
            var lesionTimepointId = lesionData.timepointID;

            var timepointObject;
            if (timepointId === lesionTimepointId) {
                // Add real mesurement
                timepointObject = {
                    longestDiameter: lesionData.measurementText,
                    imageId: lesionData.imageId,
                };
            } else {
                // Add null measurement
                timepointObject = {
                    longestDiameter: "",
                    imageId: ""
                };
            }
            timepointsObject[timepointId] = timepointObject;
        }

        var lesionDataObject = {
            lesionUID: uuid.v4(),
            number: Measurements.find().count() + 1,
            lesionNumber: lesionData.lesionNumber,
            isTarget: typeof lesionData.isTarget !== 'undefined'?lesionData.isTarget:true,
            locationUID: lesionData.locationUID,
            location: getLocationName(lesionData.locationUID),
            timepoints: timepointsObject
        };
        Measurements.insert(lesionDataObject);
    }

    // Update timepoint data in Measurements collection
    function updateTimepointData(lesionData) {
        if(typeof lesionData.isTarget === 'undefined') {
            lesionData.isTarget = true;
        }
        // Find the specific lesion to be updated
        var measurement = Measurements.find({
            lesionNumber: lesionData.lesionNumber,
            isTarget: lesionData.isTarget
        }).fetch()[0];

        // If no such lesion exists, stop here
        if (!measurement) {
            return;
        }

        // Update this specific lesion at the given timepoint
        var timepointID = lesionData.timepointID;

        // Update timepoints from lesion data
        var timepoints = measurement.timepoints;
        timepoints[timepointID].longestDiameter = lesionData.measurementText;
        timepoints[timepointID].imageId = lesionData.imageId;

        Measurements.update({
            lesionNumber: lesionData.lesionNumber,
            isTarget: lesionData.isTarget
        }, {
            $set: {
                timepoints: timepoints
            }
        });
    }

    // Check timepointData is found in Measurements collection
    function timepointDataIsFound(lesionData) {
        var timepointData = Measurements.findOne({
            lesionNumber: lesionData.lesionNumber,
            isTarget: typeof lesionData.isTarget!== 'undefined'?lesionData.isTarget:true
        });
        if (timepointData) {
            return true;
        }
        return false;
    }

    // Adds new timepoint item to tiemepoints array
    function addLesionData(lesionData) {
        if (timepointDataIsFound(lesionData)) {
            // Update data
            updateTimepointData(lesionData);
        } else {
            // Insert data
            addTimepointData(lesionData);
        }
    }

    // Returns new lesion number according to timepointID
    function getNewLesionNumber(timepointID, isTarget) {

        if (typeof isTarget === 'undefined') {
            isTarget = true;
        }
        // Get all current lesion measurements
        var measurements = Measurements.find({isTarget: isTarget}).fetch();

        // If no measurements exist yet, start at 1
        if (!measurements.length) {
            return 1;
        }

        // If measurements exist, find the last lesion number
        // from the given timepoint
        var lesionNumberCounter = 0;
        for (var i = 0; i < measurements.length; i++) {
            var measurement = measurements[i];
            var timepoints = measurement.timepoints;

            if (timepoints[timepointID].longestDiameter === '') {
                return measurement.lesionNumber;
            } else {
                lesionNumberCounter = lesionNumberCounter + 1;
            }
        }
        return lesionNumberCounter + 1;
    }

    // If lesion number is added for any timepoint, returns lesion locationUID
    function isLesionNumberAdded(lesionData) {
        if (!timepointDataIsFound(lesionData)) {
            return;
        }

        if(typeof isTarget === undefined) {
            isTarget = true;
        }
        var measurement = Measurements.find({
            lesionNumber: lesionData.lesionNumber,
            isTarget: lesionData.isTarget
        }).fetch()[0];

        return measurement.locationUID;
    }


    return {
        addNewLocation: addPatientLocation,
        getLocations: getPatientLocations,
        addLesionData: addLesionData,
        getNewLesionNumber: getNewLesionNumber,
        isLesionNumberAdded: isLesionNumberAdded,
        updateTimepointData: updateTimepointData,
        getLocationName: getLocationName
    };
})();