# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

import platform

from spack import *


class Mujoco(Package):
    """MuJoCo is a physics engine that aims to facilitate research and
    development in robotics, biomechanics, graphics and animation, and
    other areas where fast and accurate simulation is needed. """

    homepage = "https://mujoco.org/"

    mujoco_releases = {
        '2.1.0': {
            'Linux-x86_64':  'a436ca2f4144c38b837205635bbd60ffe1162d5b44c87df22232795978d7d012',
            'Darwin-x86_64':  '50226f859d9d3742fa57e1a0a92d656197ec5786f75bfa50ae00eb80fae25e90',
        }
    }

    for ver, packages in mujoco_releases.items():
        key = "{0}-{1}".format(platform.system(), platform.machine())
        pkg_sha256 = packages.get(key)
        if pkg_sha256:
            version(ver, sha256=pkg_sha256)

    def url_for_version(self, version):

        url = "https://mujoco.org/download/mujoco{0}-{1}-x86_64.tar.gz"

        system_map = {
            'Linux': 'linux',
            'Darwin': 'macos',
        }

        return url.format(version.joined, system_map[platform.system()])

    def install(self, spec, prefix):
        copy_tree('.', prefix)

    def setup_run_environment(self, env):
        env.prepend_path('CPATH', prefix.include)
        env.prepend_path('LD_LIBRARY_PATH', prefix.bin)
        if platform.system() == 'Darwin':
            env.prepend_path('DYLD_LIBRARY_PATH', prefix.bin)