Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[jsk_pcl_ros] Cache test_data to ROS home #1424

Merged
merged 1 commit into from Dec 17, 2015
Merged

[jsk_pcl_ros] Cache test_data to ROS home #1424

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
11 changes: 11 additions & 0 deletions jsk_pcl_ros/scripts/install_test_data.py
View file
Open in desktop
Expand Up @@ -3,11 +3,15 @@
import hashlib
import os
import shlex
import shutil
import subprocess

import rospkg


ROS_HOME = os.getenv('ROS_HOME', os.path.expanduser('~/.ros'))


def extract_tgz(filename, cwd):
cmd = 'tar zxf {}'.format(filename)
subprocess.call(shlex.split(cmd), cwd=cwd)
Expand All @@ -34,13 +38,20 @@ def install_test_data(filename, url, md5, compressed_bag=None):
@param compressed_bag: relative path from data_dir to compressed bagfile
"""
rp = rospkg.RosPack()
ros_data_dir = os.path.join(ROS_HOME, 'test_data')
if not os.path.exists(ros_data_dir):
os.mkdir(ros_data_dir)
data_dir = os.path.join(rp.get_path('jsk_pcl_ros'), 'test_data')
cache_file = os.path.join(ros_data_dir, filename)
output = os.path.join(data_dir, filename)
if os.path.exists(cache_file):
shutil.copy(cache_file, output)
if os.path.exists(output) and check_md5sum(output, md5):
print("[jsk_pcl_ros] '{}' is already latest version".format(output))
return
print("[jsk_pcl_ros] Installing '{}'".format(output))
download_from_gdrive(url, output)
shutil.copy(output, cache_file)
extract_tgz(output, cwd=data_dir)
if compressed_bag is not None:
compressed_bag = os.path.join(data_dir, compressed_bag)
Expand Down