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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of algorithms Show documentation
Java implementation of cancer-related algorithms (NHIA, NAPIIA, Survival Time, etc...)
The newest version!
/*
 * Copyright (C) 2020 Information Management Services, Inc.
 */
package com.imsweb.algorithms;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static com.imsweb.algorithms.Algorithms.FIELD_CENSUS_2000;
import static com.imsweb.algorithms.Algorithms.FIELD_CENSUS_2010;
import static com.imsweb.algorithms.Algorithms.FIELD_COUNTY_AT_DX_ANALYSIS;
import static com.imsweb.algorithms.Algorithms.FIELD_STATE_DX;

public abstract class AbstractAlgorithm implements Algorithm {

    protected String _id;
    protected String _name;
    protected String _version;
    protected String _url;

    protected List> _params;

    protected List _inputFields;
    protected List _outputFields;

    protected Map> _unknownValues;

    protected AbstractAlgorithm(String id, String name, String version) {
        if (id == null)
            throw new IllegalStateException("ID is required");
        if (name == null)
            throw new IllegalStateException("Name is required");
        if (version == null)
            throw new IllegalStateException("Version is required (use N/A if your algorithm doesn't support a version)");

        _id = id;
        _name = name;
        _version = version;

        _params = new ArrayList<>();

        _inputFields = new ArrayList<>();
        _outputFields = new ArrayList<>();

        _unknownValues = new HashMap<>();
    }

    @Override
    public String getId() {
        return _id;
    }

    @Override
    public String getName() {
        return _name;
    }

    @Override
    public String getVersion() {
        return _version;
    }

    @Override
    public String getDocumentationUrl() {
        return _url;
    }

    @Override
    public List> getParameters() {
        return _params;
    }

    @Override
    public List getInputFields() {
        return _inputFields;
    }

    @Override
    public List getOutputFields() {
        return _outputFields;
    }

    @Override
    public Map> getUnknownValues() {
        return _unknownValues;
    }

    protected StateCountyTractInputDto createStateCountyTractInputDto(Map inputTumor) {
        StateCountyTractInputDto inputDto = new StateCountyTractInputDto();

        inputDto.setAddressAtDxState((String)inputTumor.get(FIELD_STATE_DX));
        inputDto.setCountyAtDxAnalysis((String)inputTumor.get(FIELD_COUNTY_AT_DX_ANALYSIS));
        inputDto.setCensusTract2000((String)inputTumor.get(FIELD_CENSUS_2000));
        inputDto.setCensusTract2010((String)inputTumor.get(FIELD_CENSUS_2010));

        return inputDto;
    }
}