com.thoughtworks.studios.journey.cspmining.Pattern.java Source code

Introduction

Here is the source code for com.thoughtworks.studios.journey.cspmining.Pattern.java

Source

/**
 * This file is part of journey-neo4j-plugin. journey-neo4j-plugin is a neo4j server extension that provides out-of-box action path analysis features on top of the graph database.
 *
 * Copyright 2015 ThoughtWorks, Inc. and Pengchao Wang
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.thoughtworks.studios.journey.cspmining;

import com.fasterxml.jackson.annotation.JsonProperty;

import java.io.Serializable;
import java.util.Collections;
import java.util.List;
import java.util.Set;

public class Pattern implements Serializable {
    private List<String> actions;
    private Set<Long> journeys;

    private long totalJourneys;
    private long totalLeadActionInvolvedJourneys;

    public Pattern(List<String> actions, Set<Long> journeys) {
        this.actions = actions;
        this.journeys = journeys;
    }

    public void setTotalJourneys(long totalJourneys) {
        this.totalJourneys = totalJourneys;
    }

    public void setTotalLeadActionInvolvedJourneys(long totalLeadActionInvolvedJourneys) {
        this.totalLeadActionInvolvedJourneys = totalLeadActionInvolvedJourneys;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o)
            return true;
        if (o == null || getClass() != o.getClass())
            return false;

        Pattern pattern = (Pattern) o;

        if (!actions.equals(pattern.actions))
            return false;
        if (!journeys.equals(pattern.journeys))
            return false;

        return true;
    }

    @Override
    public int hashCode() {
        int result = actions.hashCode();
        result = 31 * result + journeys.hashCode();
        return result;
    }

    @SuppressWarnings("UnusedDeclaration")
    @JsonProperty("absolute_support_base")
    public long getTotalJourneys() {
        return totalJourneys;
    }

    @SuppressWarnings("UnusedDeclaration")
    @JsonProperty("relative_support_base")
    public long getTotalLeadActionInvolvedJourneys() {
        return totalLeadActionInvolvedJourneys;
    }

    @SuppressWarnings("UnusedDeclaration")
    public long getSupport() {
        return journeys.size();
    }

    public List<String> getActions() {
        return Collections.unmodifiableList(actions);
    }

    public Set<Long> getJourneys() {
        return this.journeys;
    }

    public int numberOfLeadActionRepeats() {
        int result = 0;
        for (String action : actions) {
            if (action.equals(actions.get(0))) {
                result++;
            }
        }

        return result;
    }
}