io.github.qudtlib.tools.contribute.QudtContribution_ucumCode Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qudtlib-tools Show documentation
There is a newer version: 6.8.0
package io.github.qudtlib.tools.contribute;

import io.github.qudtlib.Qudt;
import io.github.qudtlib.model.FactorUnits;
import io.github.qudtlib.model.Unit;
import io.github.qudtlib.vocab.QUDT;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Optional;
import java.util.function.Consumer;

public class QudtContribution_ucumCode {
    public static void main(String[] args) throws Exception {
        QudtEntityGenerator entityGenerator = new QudtEntityGenerator(false);

        Consumer symbolAndUcumGenerator =
                new Consumer() {
                    @Override
                    public void accept(Tool tool) {
                        List allUnits = new ArrayList<>(Qudt.allUnits());
                        allUnits.stream()
                                .sorted(Comparator.comparing(Unit::getIri))
                                .filter(u -> u.getUcumCode().isEmpty())
                                .forEach(
                                        u -> {
                                            final String ucumCode = getUcumCode(u).orElse(null);
                                            FactorUnits factorUnits = u.getFactorUnits();
                                            try {
                                                if (ucumCode != null) {
                                                    tool.addDerivedUnit(
                                                            factorUnits,
                                                            ud -> ud.ucumCode(ucumCode));
                                                } else {
                                                    System.err.println(
                                                            String.format(
                                                                    "Cannot add ucum code for %s: one of the constituent units has no ucum code",
                                                                    u.getIri()));
                                                }
                                            } catch (Exception e) {
                                                System.err.println(
                                                        String.format(
                                                                "Cannot add ucum code for %s",
                                                                u.getIri()));
                                            }
                                        });
                    }
                };

        for (int i = 0; i < 5; i++) {
            entityGenerator.unitOfWork(symbolAndUcumGenerator);
        }
        entityGenerator.writeRdf(s -> s.getPredicate().equals(QUDT.ucumCode));
    }

    private static Optional getUcumCode(Unit u) {
        if (u.hasFactorUnits()) {
            // System.err.println(String.format("Cannot add symbol for %s: unit has no factor
            // units", u.getIri()));
            return u.getFactorUnits().getUcumCode();
        }
        if (u.isScaled()) {
            String baseUnitSymbol =
                    u.getScalingOf()
                            .map(base -> Qudt.unitRequired(base.getIri()))
                            .map(base -> base.getUcumCode().orElse(null))
                            .orElse(null);
            if (baseUnitSymbol != null) {
                String prefixSymbol =
                        u.getPrefix().map(p -> p.getUcumCode().orElse(null)).orElse(null);
                if (prefixSymbol != null) {
                    return Optional.ofNullable(prefixSymbol + baseUnitSymbol);
                }
            }
        }
        return Optional.empty();
    }
}