com.jdiai.visual.RelationsManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jdi-nova Show documentation
JDI Nova
There is a newer version: 1.1.8
package com.jdiai.visual;

import com.epam.jdi.tools.map.MapArray;
import com.google.gson.Gson;
import com.jdiai.JS;
import com.jdiai.jsdriver.JDINovaException;

import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Map;
import java.util.function.BiFunction;

import static com.jdiai.JDI.JDI_STORAGE;
import static java.lang.Math.abs;

public class RelationsManager {
    public static String RELATIONS_STORAGE = JDI_STORAGE + "/relations.json";
    public static BiFunction EQUAL_POSITIONS = (actual, expected)
        -> actual.angle() == expected.angle() && actual.length() == expected.length();
    public static BiFunction ALMOST_SAME_POSITIONS = (actual, expected)
        -> abs(actual.angle() - expected.angle()) < 3 && abs(actual.length() / expected.length()) < 1.05;
    public static BiFunction COMPARE_POSITIONS = ALMOST_SAME_POSITIONS;

    public static File getRelationsStorage() {
        File storage = new File(RELATIONS_STORAGE);
        if (!storage.exists()) {
            try {
                storage.getParentFile().mkdirs();
                storage.createNewFile();
            } catch (IOException ex) {
                throw new JDINovaException(ex, "Failed to create relations storage: " + RELATIONS_STORAGE);
            }
        }
        return storage;
    }
    public static void storeRelations(JS element, MapArray relations) {
        storeRelations(element.getFullName(), relations);
    }
    public static void storeRelations(String name, MapArray relations) {
        try {
            new Gson().toJson(relations, new FileWriter(getRelationsStorage()));
        } catch (IOException ex) {
            throw new JDINovaException(ex, "Failed to store relations: " + relations);
        }
    }
    public static MapArray readRelations(JS element) {
        return readRelations(element.getFullName());
    }
    public static MapArray readRelations(String name) {
        Map allRelations;
        try {
             allRelations = new Gson().fromJson(new FileReader(getRelationsStorage()), Map.class);
        } catch (IOException ex) {
            throw new JDINovaException(ex, "Failed to read relations storage: " + RELATIONS_STORAGE);
        }
        return allRelations != null
            ? (MapArray) allRelations.get(name)
            : null;
    }
}