var viewer;

const devices = [
    {
        id: "Sensor 1",
        position: {
            x: -22.779729106182415,
            y: 5.431043023608719,
            z: 4.553068469137088,
        },
        type: "combo",
        sensorTypes: ["temperature", "co2"],
        dbId: 1,
    },
    {
        id: "Sensor 2",
        position: {
            x: 0.20752051811882666,
            y: 5.431043023608719,
            z: 4.553068469137088,
        },
        type: "combo",
        sensorTypes: ["temperature", "co2"],
        dbId: 2,
    },
];

var sensorStyleDefinitions = {
    co2: {
        url: "https://d2zqnmauvnpnnm.cloudfront.net/assets-1/images/co2.svg",
        color: 0xffffff,
    },
    temperature: {
        url: "https://d2zqnmauvnpnnm.cloudfront.net/assets-1/images/thermometer.svg",
        color: 0xffffff,
    },
    default: {
        url: "https://d2zqnmauvnpnnm.cloudfront.net/assets-1/images/circle.svg",
        color: 0xffffff,
    },
};

// Initialize sensor values
let sensorVals = [];
for (let i = 0; i < devices.length; i++) {
    sensorVals[i] = Math.random();
}

function launchViewer(urn) {
    var options = {
        env: 'AutodeskProduction',
        getAccessToken: getForgeToken
    };

    Autodesk.Viewing.Initializer(options, () => {
        viewer = new Autodesk.Viewing.GuiViewer3D(document.getElementById('forgeViewer'));
        viewer.start();
        var documentId = 'urn:' + urn;
        Autodesk.Viewing.Document.load(documentId, onDocumentLoadSuccess, onDocumentLoadFailure);



        //test 
        for (let i = 0; i < urns.length; i++) {
            Autodesk.Viewing.Document.load(urns[i]["urn"], async (doc) => {
                let viewables = doc.getRoot().getDefaultGeometry();
                let model = await viewer.loadDocumentNode(doc, viewables, {
                    preserveView: false,
                    keepCurrentModels: true,
                    placementTransform: (new THREE.Matrix4()).setPosition(urns[i]["xform"]),
                    keepCurrentModels: true,
                    globalOffset: {
                        x: 0,
                        y: 0,
                        z: 0
                    }
                });

                await viewer.waitForLoadDone(); //!<<< Wait for loading materials, properties and geometries for this model (URN)
            });
        }

        loadHeatmaps(viewer.getAllModels()[0]); //!<<< equals to viewer.model



    });
}

function onDocumentLoadSuccess(doc) {
    var viewables = doc.getRoot().getDefaultGeometry();
    viewer.loadDocumentNode(doc, viewables).then(i => {
        // documented loaded, any action?
    });

}

/**
 * Autodesk.Viewing.Document.load() failuire callback.
 */
function onDocumentLoadFailure(viewerErrorCode) {
    console.error("onDocumentLoadFailure() - errorCode:" + viewerErrorCode);
}

function getForgeToken(callback) {
    fetch('http://localhost:3604/api/forge/oauth/token').then(res => {
        res.json().then(data => {
            callback(data.dictionary.access_token, data.dictionary.expires_in);
        });
    });
    //callback("eyJhbGciOiJSUzI1NiIsImtpZCI6IlU3c0dGRldUTzlBekNhSzBqZURRM2dQZXBURVdWN2VhIn0.eyJzY29wZSI6WyJkYXRhOndyaXRlIiwiZGF0YTpyZWFkIiwiYnVja2V0OnJlYWQiLCJidWNrZXQ6dXBkYXRlIiwiYnVja2V0OmNyZWF0ZSJdLCJjbGllbnRfaWQiOiJUQTNocXNGZnpRYk5PVVhLcGxkS1VLU2V3NFNKMjF3NSIsImF1ZCI6Imh0dHBzOi8vYXV0b2Rlc2suY29tL2F1ZC9hand0ZXhwNjAiLCJqdGkiOiJiemxzWE5qWElvZ2R1UjUzTUJkdlhrTTNTT01qeVB1bHJrMmdTVWJudGNTeDg1b01kRG1xejg3Z05jenJkRzhpIiwiZXhwIjoxNjY4MTgzMDM2fQ.VU3qLwTJ9nlXnomKCdk4y5UcgszGEO_zlvE7w5mWWajeBMwKLo-zw7LJEqUEajRksvssppR9SbVsjLSx-vDVc3DRhCo3jYTWKPT1T3wQrlkOSqLeIrAdnKdBDNBWKgrGJt_xcmXc3dZ3XNKf9L_F6Ex808rUlo6cem1mcPpKl1jCBDqKu1mAX7aDtZ65TTQZbGGhbG4HdnET-d1i5w4LunGN11UAHhDUW3n0SWWIBL27PiiUQONZttajhD5st6IngYLcjr93BYVyJmDF7-wm4WZlHSw2OnXIfbJcFXEd83uVv_Rej4UXjzZ0e6kHwzc2nvGvKSIFu3Nt7CabdR8CkA", 3599);
}

async function loadHeatmaps(model) {

    const dataVizExtn = await viewer.loadExtension("Autodesk.DataVisualization");

    // Given a model loaded from Forge
    const structureInfo = new Autodesk.DataVisualization.Core.ModelStructureInfo(model);

    const devices = [
        {
            id: "Oficina 6",
            name: "Oficina-",
            position: { x: 22.475382737884104, y: 7.4884431474006163, z: 3.0 },
            sensorTypes: ["temperature", "humidity"]
        }
    ];

    var offset = Autodesk.viewer.model.getGlobalOffset();
    removeOffset(devices[0], offset)

    // Generates `SurfaceShadingData` after assigning each device to a room.

    const shadingData = await Autodesk.structureInfo.generateSurfaceShadingData(devices);

    // Use the resulting shading data to generate heatmap from.
    await dataVizExtn.setupSurfaceShading(model, shadingData);

    // Register color stops for the heatmap. Along with the normalized sensor value
    // in the range of [0.0, 1.0], `renderSurfaceShading` will interpolate the final
    // heatmap color based on these specified colors.
    const sensorColors = [0x0000ff, 0x00ff00, 0xffff00, 0xff0000];

    // Set heatmap colors for temperature
    const sensorType = "temperature";
    dataVizExtn.registerSurfaceShadingColors(sensorType, sensorColors);

    // Function that provides sensor value in the range of [0.0, 1.0]

    function getSensorValue(surfaceShadingPoint, sensorType) {
        // The `SurfaceShadingPoint.id` property matches one of the identifiers passed
        // to `generateSurfaceShadingData` function. In our case above, this will either
        // be "cafeteria-entrace-01" or "cafeteria-exit-01".
        const deviceId = surfaceShadingPoint.id;

        // Read the sensor data, along with its possible value range
        let sensorValue = readSensorValue(deviceId, sensorType);
        const maxSensorValue = getMaxSensorValue(sensorType);
        const minSensorValue = getMinSensorValue(sensorType);

        // Normalize sensor value to [0, 1.0]
        sensorValue = (sensorValue - minSensorValue) / (maxSensorValue - minSensorValue);
        return clamp(sensorValue, 0.0, 1.0);
    }

    // This value can also be a room instead of a floor
    const floorName = "01 - Entry Level";
    dataVizExtn.renderSurfaceShading(floorName, sensorType, getSensorValue);
}