Add exporter based on existing MjScene functionality
This commit is contained in:
@@ -0,0 +1,62 @@
|
|||||||
|
// Copyright 2019 DeepMind Technologies Limited
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
using System.IO;
|
||||||
|
using System.Text;
|
||||||
|
using System.Xml;
|
||||||
|
using UnityEditor;
|
||||||
|
using UnityEngine;
|
||||||
|
|
||||||
|
namespace Mujoco
|
||||||
|
{
|
||||||
|
|
||||||
|
// Mujoco scenes exporter window.
|
||||||
|
public class MjExporterEditorWindow : EditorWindow
|
||||||
|
{
|
||||||
|
[MenuItem("Assets/Export MuJoCo Scene")]
|
||||||
|
public static void Apply()
|
||||||
|
{
|
||||||
|
if (Application.isPlaying || MjScene.InstanceExists)
|
||||||
|
{
|
||||||
|
Debug.LogWarning("MJCF exporting only available outside Play mode, and with no MjScene in the hierarchy.");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
string path = EditorUtility.SaveFilePanel("Save MJCF model", "", "model", "xml");
|
||||||
|
if (string.IsNullOrEmpty(path)) return;
|
||||||
|
|
||||||
|
// Due to singleton pattern, this creates a MjScene GameObject we need to remove at the end.
|
||||||
|
var mjcf = MjScene.Instance.CreateScene(skipCompile: true);
|
||||||
|
try
|
||||||
|
{
|
||||||
|
using (var stream = File.Open(path, FileMode.Create))
|
||||||
|
{
|
||||||
|
using (var writer = new XmlTextWriter(stream, new UTF8Encoding(false)))
|
||||||
|
{
|
||||||
|
writer.Formatting = Formatting.Indented;
|
||||||
|
mjcf.WriteContentTo(writer);
|
||||||
|
Debug.Log($"MJCF saved to {path}");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (IOException ex)
|
||||||
|
{
|
||||||
|
Debug.LogWarning("Failed to save Xml to a file: " + ex.ToString());
|
||||||
|
}
|
||||||
|
|
||||||
|
// Removing the added MjScene.
|
||||||
|
DestroyImmediate(MjScene.Instance.gameObject);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
fileFormatVersion: 2
|
||||||
|
guid: 7695f261c02fdc343a09b219c92a6aab
|
||||||
|
MonoImporter:
|
||||||
|
externalObjects: {}
|
||||||
|
serializedVersion: 2
|
||||||
|
defaultReferences: []
|
||||||
|
executionOrder: 0
|
||||||
|
icon: {instanceID: 0}
|
||||||
|
userData:
|
||||||
|
assetBundleName:
|
||||||
|
assetBundleVariant:
|
||||||
Reference in New Issue
Block a user