82 lines
2.1 KiB
Plaintext
82 lines
2.1 KiB
Plaintext
|
#!/usr/bin/env ruby
|
||
|
|
|
||
|
|
require 'rubygems'
|
||
|
|
require 'tmpdir'
|
||
|
|
|
||
|
|
begin
|
||
|
|
require 'zip'
|
||
|
|
rescue LoadError
|
||
|
|
puts "Requires the rubyzip gem. To install:"
|
||
|
|
puts " gem install rubyzip"
|
||
|
|
exit
|
||
|
|
end
|
||
|
|
|
||
|
|
#################################################
|
||
|
|
# Display help information
|
||
|
|
def help
|
||
|
|
puts "gzprop -- Create a model zip for upload to thepropshop.org\n"
|
||
|
|
puts "\n"
|
||
|
|
puts "'gzprop' <model_sdf_file>\n"
|
||
|
|
puts "\n"
|
||
|
|
puts "Package an existing model SDF, materials, and meshes\n"
|
||
|
|
puts "into a zip suitable for upload to thepropshop.org. A \n"
|
||
|
|
puts "working instanstance of Gazebo >4.0 is required.\n"
|
||
|
|
puts "\n"
|
||
|
|
puts "Example Usage:\n"
|
||
|
|
puts " $ gzprop ~/.gazebo/models/ground_plane/model.sdf\n"
|
||
|
|
puts "\n"
|
||
|
|
end
|
||
|
|
|
||
|
|
#################################################
|
||
|
|
# Main
|
||
|
|
|
||
|
|
# Make sure there are command line arguments.
|
||
|
|
if ARGV.empty?
|
||
|
|
help()
|
||
|
|
abort("")
|
||
|
|
end
|
||
|
|
|
||
|
|
# Get the model file name
|
||
|
|
modelFile = File.absolute_path(ARGV[0])
|
||
|
|
|
||
|
|
modelParts = modelFile.split('/')
|
||
|
|
modelDir = modelParts[0, modelParts.size() - 1].join('/')
|
||
|
|
|
||
|
|
# Make sure there is a model.sdf file in the model directory
|
||
|
|
abort("Missing model file [#{modelFile}]") unless File.exist?(modelFile)
|
||
|
|
|
||
|
|
# Get the name of the model (last part of the directory name)
|
||
|
|
modelName = modelParts[-2]
|
||
|
|
|
||
|
|
## Make a temporary directory
|
||
|
|
Dir.mktmpdir { |dir|
|
||
|
|
|
||
|
|
metaDir = File.join(dir, "meta")
|
||
|
|
|
||
|
|
# Copy the model information into a temp directory
|
||
|
|
FileUtils.cp_r Dir[modelDir + '/*'], dir
|
||
|
|
|
||
|
|
# Create the meta directory
|
||
|
|
Dir.mkdir(metaDir)
|
||
|
|
|
||
|
|
# Create a set of images
|
||
|
|
`gzserver -s libModelPropShop.so worlds/blank.world --propshop-save "#{metaDir}" --propshop-model "#{modelFile}"`
|
||
|
|
|
||
|
|
# Create a zip file for the model
|
||
|
|
zipFilename = File.join(Dir.pwd, modelName + '.zip')
|
||
|
|
if !File.exists?(zipFilename)
|
||
|
|
Zip::File.open(zipFilename, Zip::File::CREATE) { |zipfile|
|
||
|
|
Dir[File.join(dir, '**', '**')].each do |file|
|
||
|
|
zipfile.add(file.sub(dir + '/', ''), file)
|
||
|
|
end
|
||
|
|
}
|
||
|
|
else
|
||
|
|
puts "Error: #{zipFilename} already exists. Please move or rename."
|
||
|
|
exit
|
||
|
|
end
|
||
|
|
}
|
||
|
|
|
||
|
|
# Completion message
|
||
|
|
puts "Propshop model created.\n"
|
||
|
|
puts "Upload [#{modelName}.zip] to thepropshop.org"
|