Yuval Tassa
6419534bad
PiperOrigin-RevId: 866478839 Change-Id: Id21a6da0f98454c8fa39ea5af8a5e213d6eae497
117 lines
3.5 KiB
C++
117 lines
3.5 KiB
C++
// Copyright 2021 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.
|
|
|
|
#ifndef MUJOCO_SRC_XML_XML_BASE_H_
|
|
#define MUJOCO_SRC_XML_XML_BASE_H_
|
|
|
|
#include <cstdlib>
|
|
#include <string>
|
|
|
|
#include <mujoco/mjmodel.h>
|
|
#include <mujoco/mjspec.h>
|
|
#include "xml/xml_util.h"
|
|
#include "tinyxml2.h"
|
|
|
|
|
|
// keyword maps (defined in implementation files)
|
|
extern const int joint_sz;
|
|
extern const int bodysleep_sz;
|
|
extern const int projection_sz;
|
|
extern const int camlight_sz;
|
|
extern const int lighttype_sz;
|
|
extern const int integrator_sz;
|
|
extern const int collision_sz;
|
|
extern const int cone_sz;
|
|
extern const int jac_sz;
|
|
extern const int solver_sz;
|
|
extern const int equality_sz;
|
|
extern const int texture_sz;
|
|
extern const int colorspace_sz;
|
|
extern const int builtin_sz;
|
|
extern const int mark_sz;
|
|
extern const int dyn_sz;
|
|
extern const int gain_sz;
|
|
extern const int bias_sz;
|
|
extern const int interp_sz;
|
|
extern const int stage_sz;
|
|
extern const int datatype_sz;
|
|
extern const int camout_sz;
|
|
extern const int reduce_sz;
|
|
extern const mjMap angle_map[];
|
|
extern const mjMap enable_map[];
|
|
extern const mjMap bool_map[];
|
|
extern const mjMap fluid_map[];
|
|
extern const mjMap TFAuto_map[];
|
|
extern const mjMap joint_map[];
|
|
extern const mjMap bodysleep_map[];
|
|
extern const mjMap geom_map[];
|
|
extern const mjMap projection_map[];
|
|
extern const mjMap camlight_map[];
|
|
extern const mjMap lighttype_map[];
|
|
extern const mjMap integrator_map[];
|
|
extern const mjMap collision_map[];
|
|
extern const mjMap impedance_map[];
|
|
extern const mjMap reference_map[];
|
|
extern const mjMap cone_map[];
|
|
extern const mjMap jac_map[];
|
|
extern const mjMap solver_map[];
|
|
extern const mjMap equality_map[];
|
|
extern const mjMap texture_map[];
|
|
extern const mjMap colorspace_map[];
|
|
extern const mjMap texrole_map[];
|
|
extern const mjMap builtin_map[];
|
|
extern const mjMap mark_map[];
|
|
extern const mjMap dyn_map[];
|
|
extern const mjMap gain_map[];
|
|
extern const mjMap bias_map[];
|
|
extern const mjMap interp_map[];
|
|
extern const mjMap stage_map[];
|
|
extern const mjMap datatype_map[];
|
|
extern const mjMap condata_map[];
|
|
extern const mjMap raydata_map[];
|
|
extern const mjMap camout_map[];
|
|
extern const mjMap reduce_map[];
|
|
extern const mjMap meshtype_map[];
|
|
extern const mjMap meshinertia_map[];
|
|
extern const mjMap flexself_map[];
|
|
extern const mjMap elastic2d_map[];
|
|
|
|
|
|
//---------------------------------- Base XML class ------------------------------------------------
|
|
|
|
class mjXBase : public mjXUtil {
|
|
public:
|
|
mjXBase();
|
|
virtual ~mjXBase() = default;
|
|
|
|
// parse: implemented in derived parser classes
|
|
virtual void Parse(tinyxml2::XMLElement* root, const mjVFS* vfs = nullptr) {};
|
|
|
|
// write: implemented in derived writer class
|
|
virtual std::string Write(char *error, std::size_t error_sz) {
|
|
return "";
|
|
};
|
|
|
|
// set the model allocated externally
|
|
virtual void SetModel(mjSpec*, const mjModel* = nullptr);
|
|
|
|
// read alternative orientation specification
|
|
static int ReadAlternative(tinyxml2::XMLElement* elem, mjsOrientation& alt);
|
|
|
|
protected:
|
|
mjSpec* spec; // internally-allocated model
|
|
};
|
|
|
|
#endif // MUJOCO_SRC_XML_XML_BASE_H_
|