cmake_minimum_required(VERSION 3.5)
project(armw)

# Need to do different things for different ROS versions
if ("$ENV{ROS_VERSION}" STREQUAL "1")
    find_package(catkin REQUIRED COMPONENTS rospy std_msgs dynamic_reconfigure)

    # Run setup.py
    catkin_python_setup()

    # Finish package generation
    catkin_package(CATKIN_DEPENDS rospy)
elseif ("$ENV{ROS_VERSION}" STREQUAL "2")
    find_package(ament_cmake REQUIRED)
    find_package(ament_cmake_python REQUIRED)
    find_package(rclpy REQUIRED)

    # Install python package
    ament_python_install_package(armw PACKAGE_DIR armw)

    ament_package()
else ()
    message("A ROS Version must be specified")
endif ()
