com.imsweb.algorithms.AlgorithmsUtils Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2014 Information Management Services, Inc.
 */
package com.imsweb.algorithms;

import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang3.Range;
import org.apache.commons.lang3.StringUtils;

public final class AlgorithmsUtils {

    /**
     * Expands the provided string of sites into a list of sites, individual elements or ranges should be comma separated.
     * 
     * Supports ranges (C120-C129) and optional last digits (C12 means C120-C129; C1 means C100-C199).
     * 

     * Method is not case sensitive (c123 will be translated into C123).
     * 

     * Created on Dec 19, 2011 by depryf
     * @param sites string of sites
     * @return the list of expanded sites, null if no site has been provided
     */
    public static List expandSites(String sites) {
        if (sites == null || sites.trim().isEmpty())
            return null;

        List result = new ArrayList<>();
        sites = sites.replace("\\s+", "").toUpperCase(); // don't deal with spaces and/or case

        for (String elem : sites.split(",|\\s")) {
            if (elem.isEmpty())
                continue;

            // range
            if (elem.contains("-")) {
                String[] parts = elem.split("\\-");
                String left = parts[0];
                if (left.matches("\\d+"))
                    left = "C" + elem;
                String right = parts[1];
                if (right.matches("\\d+"))
                    right = "C" + elem;

                if (left.matches("[A-Z](\\d){0,3}") && right.matches("[A-Z](\\d){0,3}")) {
                    String leftPrefix = elem.substring(0, 1);
                    String rightPrefix = elem.substring(0, 1);
                    if (leftPrefix.equals(rightPrefix)) {
                        String leftIndex = left.substring(1);
                        String rightIndex = right.substring(1);

                        int start = Integer.parseInt(StringUtils.rightPad(leftIndex, 3, "0"));
                        int end = Integer.parseInt(StringUtils.rightPad(rightIndex, 3, "9"));
                        for (int i = start; i <= end; i++)
                            result.add(leftPrefix + StringUtils.leftPad(String.valueOf(i), 3, "0"));
                    }
                }
            }
            else {
                if (elem.matches("\\d+"))
                    elem = "C" + elem;
                if (elem.matches("[A-Z](\\d){0,3}")) {
                    String prefix = elem.substring(0, 1);
                    String index = elem.substring(1);

                    int start = Integer.parseInt(StringUtils.rightPad(index, 3, "0"));
                    int end = Integer.parseInt(StringUtils.rightPad(index, 3, "9"));
                    for (int i = start; i <= end; i++)
                        result.add(prefix + StringUtils.leftPad(String.valueOf(i), 3, "0"));
                }
            }
        }

        return result;
    }

    /**
     * Expands the provided primary sites as either individual integer codes, or as ranges.
     * @param toExpand sites to expand
     * @return expanded sites
     */
    public static List