org.mitre.mpf.interop.JsonTrackOutputObject.java Source code

Introduction

Here is the source code for org.mitre.mpf.interop.JsonTrackOutputObject.java

Source

/******************************************************************************
 * NOTICE                                                                     *
 *                                                                            *
 * This software (or technical data) was produced for the U.S. Government     *
 * under contract, and is subject to the Rights in Data-General Clause        *
 * 52.227-14, Alt. IV (DEC 2007).                                             *
 *                                                                            *
 * Copyright 2016 The MITRE Corporation. All Rights Reserved.                 *
 ******************************************************************************/

/******************************************************************************
 * Copyright 2016 The MITRE Corporation                                       *
 *                                                                            *
 * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0                              *
 *                                                                            *
 * Unless required by applicable law or agreed to in writing, software        *
 * distributed under the License is distributed on an "AS IS" BASIS,          *
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   *
 * See the License for the specific language governing permissions and        *
 * limitations under the License.                                             *
 ******************************************************************************/

package org.mitre.mpf.interop;

import com.fasterxml.jackson.annotation.*;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;

import java.util.Objects;
import java.util.SortedSet;
import java.util.TreeSet;

@JsonTypeName("TrackOutputObject")
public class JsonTrackOutputObject implements Comparable<JsonTrackOutputObject> {

    @JsonProperty("id")
    @JsonPropertyDescription("The unique identifier for this track.")
    private String id;

    public String getId() {
        return id;
    }

    // MPF R0.6.0 backwards compatibility with MPF R0.5.0 (upgrade path)
    @JsonProperty("startOffset")
    public void setStartOffset(int startOffset) {
        startOffsetFrame = startOffset;
    }

    public int getStartOffset() {
        return startOffsetFrame;
    }

    @JsonProperty("startOffsetFrame")
    @JsonPropertyDescription("The offset in the medium where the track starts.")
    private int startOffsetFrame;

    public int getStartOffsetFrame() {
        return startOffsetFrame;
    }

    // MPF R0.6.0 backwards compatibility with MPF R0.5.0 (upgrade path)
    @JsonProperty("stopOffset")
    public void setStopOffset(int stopOffset) {
        stopOffsetFrame = stopOffset;
    }

    public int getStopOffset() {
        return stopOffsetFrame;
    }

    @JsonProperty("stopOffsetFrame")
    @JsonPropertyDescription("The offset in the medium where the track ends.")
    private int stopOffsetFrame;

    public int getStopOffsetFrame() {
        return stopOffsetFrame;
    }

    @JsonProperty("startOffsetTime")
    @JsonPropertyDescription("The offset in the medium where the track starts, in milliseconds.")
    private int startOffsetTime;

    public int getStartOffsetTime() {
        return startOffsetTime;
    }

    @JsonProperty("stopOffsetTime")
    @JsonPropertyDescription("The offset in the medium where the track ends, in milliseconds.")
    private int stopOffsetTime;

    public int getStopOffsetTime() {
        return stopOffsetTime;
    }

    @JsonProperty("type")
    @JsonPropertyDescription("The type of object associated with this track.")
    private String type;

    public String getType() {
        return type;
    }

    @JsonProperty("source")
    @JsonPropertyDescription("The set of pipeline actions which produced this track.")
    private String source;

    public String getSource() {
        return source;
    }

    @JsonProperty("exemplar")
    @JsonPropertyDescription("The detection which best represents this track.")
    private JsonDetectionOutputObject exemplar;

    public JsonDetectionOutputObject getExemplar() {
        return exemplar;
    }

    public void setExemplar(JsonDetectionOutputObject exemplar) {
        this.exemplar = exemplar;
    }

    @JsonProperty("detections")
    @JsonPropertyDescription("The collection of detections included in this track.")
    private SortedSet<JsonDetectionOutputObject> detections;

    public SortedSet<JsonDetectionOutputObject> getDetections() {
        return detections;
    }

    public JsonTrackOutputObject(String id, int startOffsetFrame, int stopOffsetFrame, int startOffsetTime,
            int stopOffsetTime, String type, String source) {
        this.id = id;
        this.startOffsetFrame = startOffsetFrame;
        this.stopOffsetFrame = stopOffsetFrame;
        this.startOffsetTime = startOffsetTime;
        this.stopOffsetTime = stopOffsetTime;
        this.type = type;
        this.source = source;
        this.detections = new TreeSet<>();
    }

    public JsonTrackOutputObject() {
    }

    @JsonCreator
    public static JsonTrackOutputObject factory(@JsonProperty("id") String id,
            @JsonProperty("startOffsetFrame") int startOffsetFrame,
            @JsonProperty("stopOffsetFrame") int stopOffsetFrame,
            @JsonProperty("startOffsetTime") int startOffsetTime,
            @JsonProperty("stopOffsetTime") int stopOffsetTime, @JsonProperty("type") String type,
            @JsonProperty("source") String source, @JsonProperty("exemplar") JsonDetectionOutputObject exemplar,
            @JsonProperty("detections") SortedSet<JsonDetectionOutputObject> detections) {
        JsonTrackOutputObject trackOutputObject = new JsonTrackOutputObject(id, startOffsetFrame, stopOffsetFrame,
                startOffsetTime, stopOffsetTime, type, source);
        trackOutputObject.exemplar = exemplar;
        if (detections != null) {
            trackOutputObject.detections.addAll(detections);
        }
        return trackOutputObject;
    }

    public int hashCode() {
        return id.hashCode();
    }

    public boolean equals(Object other) {
        if (other == null || !(other instanceof JsonTrackOutputObject)) {
            return false;
        } else {
            JsonTrackOutputObject casted = (JsonTrackOutputObject) other;
            return id == casted.id && startOffsetFrame == casted.startOffsetFrame
                    && stopOffsetFrame == casted.stopOffsetFrame && startOffsetTime == casted.startOffsetTime
                    && stopOffsetTime == casted.stopOffsetTime && StringUtils.equals(type, casted.type)
                    && StringUtils.equals(source, casted.source);
        }
    }

    @Override
    public int compareTo(JsonTrackOutputObject other) {
        int result = 0;
        if (other == null) {
            return 0;
        } else if ((result = Integer.compare(startOffsetFrame, other.startOffsetFrame)) != 0
                || (result = Integer.compare(stopOffsetFrame, other.stopOffsetFrame)) != 0
                || (result = Integer.compare(startOffsetTime, other.startOffsetTime)) != 0
                || (result = Integer.compare(stopOffsetTime, other.stopOffsetTime)) != 0
                || (result = ObjectUtils.compare(type, other.type, false)) != 0
                || (result = ObjectUtils.compare(source, other.source, false)) != 0
                || (result = ObjectUtils.compare(id, other.id, false)) != 0) {
            return result;
        } else {
            return 0;
        }
    }
}