com.cognite.client.servicesV1.parser.ThreeDNodeParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cdf-sdk-java Show documentation
Java SDK for reading and writing from/to CDF resources.
There is a newer version: 2.3.3
package com.cognite.client.servicesV1.parser;

import com.cognite.client.dto.ThreeDNode;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

public class ThreeDNodeParser {

    private static final Logger LOG = LoggerFactory.getLogger(ThreeDNodeParser.class);

    static final ObjectMapper objectMapper = new ObjectMapper();

    public static List parseThreeDNodesToList(String json) throws JsonProcessingException {
        List list = new ArrayList<>();
        JsonNode root = objectMapper.readTree(json);
        ThreeDNode.Builder builder = ThreeDNode.newBuilder();

        if (root.path("items").isArray()) {
            for (JsonNode node : root.path("items")) {
                extractNodes(builder, node);
                list.add(builder.build());
                builder.clear();
            }
        }else if (root.isObject()) {
            extractNodes(builder, root);
            list.add(builder.build());
        }
        return list;
    }

    public static ThreeDNode parseThreeDNodes(String json) throws JsonProcessingException {
        JsonNode root = objectMapper.readTree(json);
        ThreeDNode.Builder builder = ThreeDNode.newBuilder();

        if (root.isObject()) {
            extractNodes(builder, root);
        }

        return builder.build();
    }

    private static void extractNodes(ThreeDNode.Builder builder, JsonNode node) {
        if (node.path("id").isIntegralNumber()) {
            builder.setId(node.get("id").longValue());
        }
        if (node.path("treeIndex").isIntegralNumber()) {
            builder.setTreeIndex(node.get("treeIndex").longValue());
        }
        if (node.path("parentId").isIntegralNumber()) {
            builder.setParentId(node.get("parentId").longValue());
        }
        if (node.path("depth").isIntegralNumber()) {
            builder.setDepth(node.get("depth").longValue());
        }
        if (node.path("name").isTextual()) {
            builder.setName(node.get("name").textValue());
        }
        if (node.path("subtreeSize").isIntegralNumber()) {
            builder.setSubtreeSize(node.get("subtreeSize").longValue());
        }

        if (node.path("boundingBox").isObject()) {
            Iterator> it = node.path("boundingBox").fields();
            ThreeDNode.BoundingBox.Builder builderBound = ThreeDNode.BoundingBox.newBuilder();
            while (it.hasNext()) {
                Map.Entry entry = it.next();
                if (entry.getKey().equals("max")) {
                    entry.getValue().forEach(value-> builderBound.addMax(value.doubleValue()));
                } else if (entry.getKey().equals("min")) {
                    entry.getValue().forEach(value-> builderBound.addMin(value.doubleValue()));
                } else {
                    LOG.warn("BoundingBox PROPERTY {} IS NOT MAPPED IN OBJECT ThreeDNode ", entry.getKey());
                }
            }
            builder.setBoundingBox(builderBound.build());
        }

        if (node.path("properties").isObject()) {
            ThreeDNode.Properties.Builder builderProps = ThreeDNode.Properties.newBuilder();
            Iterator> it = node.path("properties").fields();

            while (it.hasNext()) {
                ThreeDNode.Categories.Builder builderCat = ThreeDNode.Categories.newBuilder();
                Map.Entry entry = it.next();
                JsonNode jsonNode = entry.getValue();
                Iterator itFileNames = jsonNode.fieldNames();
                while(itFileNames.hasNext()) {
                    String fileName = itFileNames.next();
                    JsonNode value = jsonNode.get(fileName);
                    builderCat.putValues(fileName, value.textValue());
                }
                builderCat.setName(entry.getKey());
                builderProps.addCategories(builderCat.build());
            }
            builder.setProperties(builderProps.build());
        }
    }
}