567 lines
69 KiB
HTML
567 lines
69 KiB
HTML
<!DOCTYPE html>
|
|
<html class="writer-html5" lang="en" >
|
|
<head>
|
|
<meta charset="utf-8" />
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
|
<title>dna_viewer.builder.builder — DNA Calibration documentation</title>
|
|
<link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" />
|
|
<link rel="stylesheet" href="../../../_static/css/theme.css" type="text/css" />
|
|
<link rel="stylesheet" href="../../../_static/graphviz.css" type="text/css" />
|
|
<!--[if lt IE 9]>
|
|
<script src="../../../_static/js/html5shiv.min.js"></script>
|
|
<![endif]-->
|
|
|
|
<script data-url_root="../../../" id="documentation_options" src="../../../_static/documentation_options.js"></script>
|
|
<script src="../../../_static/jquery.js"></script>
|
|
<script src="../../../_static/underscore.js"></script>
|
|
<script src="../../../_static/doctools.js"></script>
|
|
<script src="../../../_static/js/theme.js"></script>
|
|
<link rel="index" title="Index" href="../../../genindex.html" />
|
|
<link rel="search" title="Search" href="../../../search.html" />
|
|
</head>
|
|
|
|
<body class="wy-body-for-nav">
|
|
<div class="wy-grid-for-nav">
|
|
<nav data-toggle="wy-nav-shift" class="wy-nav-side">
|
|
<div class="wy-side-scroll">
|
|
<div class="wy-side-nav-search" >
|
|
<a href="../../../index.html" class="icon icon-home"> DNA Calibration
|
|
</a>
|
|
<div role="search">
|
|
<form id="rtd-search-form" class="wy-form" action="../../../search.html" method="get">
|
|
<input type="text" name="q" placeholder="Search docs" />
|
|
<input type="hidden" name="check_keywords" value="yes" />
|
|
<input type="hidden" name="area" value="default" />
|
|
</form>
|
|
</div>
|
|
</div><div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="Navigation menu">
|
|
<p class="caption" role="heading"><span class="caption-text">Contents</span></p>
|
|
<ul>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../index_.html">MetaHuman DNA Calibration</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../index_.html#overview">Overview</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../index_.html#examples">Examples</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../index_.html#notes">Notes</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../index_.html#license">License</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../index_.html#indices-and-tables">Indices and tables</a></li>
|
|
</ul>
|
|
<p class="caption" role="heading"><span class="caption-text">Repository Organization</span></p>
|
|
<ul>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../repository_organization_md.html">Repository Organization</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../repository_organization_md.html#folder-structure">Folder structure</a></li>
|
|
</ul>
|
|
<p class="caption" role="heading"><span class="caption-text">DNACalib</span></p>
|
|
<ul>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dnacalib_md.html">DNACalib</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dnacalib_api_md.html">API overview</a></li>
|
|
</ul>
|
|
<p class="caption" role="heading"><span class="caption-text">DNAViewer</span></p>
|
|
<ul>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna_viewer_md.html">DNAViewer</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna_viewer_maya_md.html">Usage in Maya</a></li>
|
|
</ul>
|
|
<p class="caption" role="heading"><span class="caption-text">DNAViewer API</span></p>
|
|
<ul>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna_viewer_api_md.html">Environment Setup</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna_viewer_api_md.html#dna">DNA</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna_viewer_api_build_meshes_md.html">Mesh Utilities</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna_viewer_api_build_rig_md.html">Build rig (<code class="docutils literal notranslate"><span class="pre">build_rig</span></code>)</a></li>
|
|
</ul>
|
|
<p class="caption" role="heading"><span class="caption-text">Technical documentation</span></p>
|
|
<ul>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna.html">DNA</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dnacalib.html">DNACalib</a></li>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../dna_viewer_modules.html">DNAViewer</a></li>
|
|
</ul>
|
|
<p class="caption" role="heading"><span class="caption-text">FAQ</span></p>
|
|
<ul>
|
|
<li class="toctree-l1"><a class="reference internal" href="../../../faq_md.html">Frequently Asked Questions (FAQ)</a></li>
|
|
</ul>
|
|
|
|
</div>
|
|
</div>
|
|
</nav>
|
|
|
|
<section data-toggle="wy-nav-shift" class="wy-nav-content-wrap"><nav class="wy-nav-top" aria-label="Mobile navigation menu" >
|
|
<i data-toggle="wy-nav-top" class="fa fa-bars"></i>
|
|
<a href="../../../index.html">DNA Calibration</a>
|
|
</nav>
|
|
|
|
<div class="wy-nav-content">
|
|
<div class="rst-content">
|
|
<div role="navigation" aria-label="Page navigation">
|
|
<ul class="wy-breadcrumbs">
|
|
<li><a href="../../../index.html" class="icon icon-home"></a> »</li>
|
|
<li><a href="../../index.html">Module code</a> »</li>
|
|
<li>dna_viewer.builder.builder</li>
|
|
<li class="wy-breadcrumbs-aside">
|
|
</li>
|
|
</ul>
|
|
<hr/>
|
|
</div>
|
|
<div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
|
|
<div itemprop="articleBody">
|
|
|
|
<h1>Source code for dna_viewer.builder.builder</h1><div class="highlight"><pre>
|
|
<span></span><span class="kn">import</span> <span class="nn">logging</span>
|
|
<span class="kn">import</span> <span class="nn">traceback</span>
|
|
<span class="kn">from</span> <span class="nn">dataclasses</span> <span class="kn">import</span> <span class="n">dataclass</span><span class="p">,</span> <span class="n">field</span>
|
|
<span class="kn">from</span> <span class="nn">pathlib</span> <span class="kn">import</span> <span class="n">Path</span>
|
|
<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Dict</span><span class="p">,</span> <span class="n">List</span><span class="p">,</span> <span class="n">Optional</span>
|
|
|
|
<span class="kn">from</span> <span class="nn">maya</span> <span class="kn">import</span> <span class="n">cmds</span><span class="p">,</span> <span class="n">mel</span>
|
|
|
|
<span class="kn">from</span> <span class="nn">..builder.maya.util</span> <span class="kn">import</span> <span class="n">Maya</span>
|
|
<span class="kn">from</span> <span class="nn">..common</span> <span class="kn">import</span> <span class="n">DNAViewerError</span>
|
|
<span class="kn">from</span> <span class="nn">..dnalib.dnalib</span> <span class="kn">import</span> <span class="n">DNA</span>
|
|
<span class="kn">from</span> <span class="nn">..model</span> <span class="kn">import</span> <span class="n">Joint</span> <span class="k">as</span> <span class="n">JointModel</span>
|
|
<span class="kn">from</span> <span class="nn">.config</span> <span class="kn">import</span> <span class="n">AngleUnit</span><span class="p">,</span> <span class="n">Config</span><span class="p">,</span> <span class="n">LinearUnit</span>
|
|
<span class="kn">from</span> <span class="nn">.joint</span> <span class="kn">import</span> <span class="n">Joint</span> <span class="k">as</span> <span class="n">JointBuilder</span>
|
|
<span class="kn">from</span> <span class="nn">.mesh</span> <span class="kn">import</span> <span class="n">Mesh</span>
|
|
|
|
|
|
<div class="viewcode-block" id="BuildResult"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.BuildResult">[docs]</a><span class="nd">@dataclass</span>
|
|
<span class="k">class</span> <span class="nc">BuildResult</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> A class used for returning data after finishing the build process</span>
|
|
|
|
<span class="sd"> Attributes</span>
|
|
<span class="sd"> ----------</span>
|
|
<span class="sd"> @type meshes_per_lod: Dict[int, List[str]]</span>
|
|
<span class="sd"> @param meshes_per_lod: The list of mesh names created group by LOD number</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">meshes_per_lod</span><span class="p">:</span> <span class="n">Dict</span><span class="p">[</span><span class="nb">int</span><span class="p">,</span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="n">field</span><span class="p">(</span><span class="n">default_factory</span><span class="o">=</span><span class="nb">dict</span><span class="p">)</span>
|
|
|
|
<div class="viewcode-block" id="BuildResult.get_all_meshes"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.BuildResult.get_all_meshes">[docs]</a> <span class="k">def</span> <span class="nf">get_all_meshes</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Flatten meshes to single list.</span>
|
|
|
|
<span class="sd"> @rtype: List[str]</span>
|
|
<span class="sd"> @returns: The list of all mesh names.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">all_meshes</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">meshes_per_lod</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">meshes_per_lod</span><span class="o">.</span><span class="n">values</span><span class="p">():</span>
|
|
<span class="n">all_meshes</span><span class="o">.</span><span class="n">extend</span><span class="p">(</span><span class="n">meshes_per_lod</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">all_meshes</span></div></div>
|
|
|
|
|
|
<div class="viewcode-block" id="Builder"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder">[docs]</a><span class="k">class</span> <span class="nc">Builder</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> A builder class used for building the character</span>
|
|
|
|
<span class="sd"> Attributes</span>
|
|
<span class="sd"> ----------</span>
|
|
<span class="sd"> @type config: Config</span>
|
|
<span class="sd"> @param config: The configuration options used for building the character</span>
|
|
|
|
<span class="sd"> @type dna: DNA</span>
|
|
<span class="sd"> @param dna: The DNA object read from the DNA file</span>
|
|
|
|
<span class="sd"> @type meshes: Dict[int, List[str]]</span>
|
|
<span class="sd"> @param meshes: A list of meshes created grouped by lod</span>
|
|
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">dna</span><span class="p">:</span> <span class="n">DNA</span><span class="p">,</span> <span class="n">config</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Config</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">config</span> <span class="o">=</span> <span class="n">config</span> <span class="ow">or</span> <span class="n">Config</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">dna</span> <span class="o">=</span> <span class="n">dna</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">meshes</span><span class="p">:</span> <span class="n">Dict</span><span class="p">[</span><span class="nb">int</span><span class="p">,</span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">all_loaded_meshes</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
|
|
<span class="k">def</span> <span class="nf">_build</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="nb">bool</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">new_scene</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">set_filtered_meshes</span><span class="p">()</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">all_loaded_meshes</span><span class="p">:</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">error</span><span class="p">(</span><span class="s2">"No mashes has been loaded."</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="kc">False</span>
|
|
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">create_groups</span><span class="p">()</span>
|
|
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">set_units</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_joints</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">build_meshes</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_ctrl_attributes_on_root_joint</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_animated_map_attributes_on_root_joint</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_key_frames</span><span class="p">()</span>
|
|
<span class="k">return</span> <span class="kc">True</span>
|
|
|
|
<div class="viewcode-block" id="Builder.build"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.build">[docs]</a> <span class="k">def</span> <span class="nf">build</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">BuildResult</span><span class="p">:</span>
|
|
<span class="sd">"""Builds the character"""</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">meshes</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="n">filename</span> <span class="o">=</span> <span class="n">Path</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">path</span><span class="p">)</span><span class="o">.</span><span class="n">stem</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"******************************"</span><span class="p">)</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">filename</span><span class="si">}</span><span class="s2"> started building"</span><span class="p">)</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"******************************"</span><span class="p">)</span>
|
|
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_build</span><span class="p">()</span>
|
|
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">filename</span><span class="si">}</span><span class="s2"> built successfully!"</span><span class="p">)</span>
|
|
|
|
<span class="k">except</span> <span class="n">DNAViewerError</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="n">traceback</span><span class="o">.</span><span class="n">print_exc</span><span class="p">()</span>
|
|
<span class="k">raise</span> <span class="n">e</span>
|
|
<span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="n">traceback</span><span class="o">.</span><span class="n">print_exc</span><span class="p">()</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">error</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Unhandled exception, </span><span class="si">{</span><span class="n">e</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Scene creation failed! Reason: </span><span class="si">{</span><span class="n">e</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> <span class="kn">from</span> <span class="nn">e</span>
|
|
<span class="k">return</span> <span class="n">BuildResult</span><span class="p">(</span><span class="n">meshes_per_lod</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">meshes</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.new_scene"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.new_scene">[docs]</a> <span class="k">def</span> <span class="nf">new_scene</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">file</span><span class="p">(</span><span class="n">new</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">force</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.add_mesh_to_display_layer"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.add_mesh_to_display_layer">[docs]</a> <span class="k">def</span> <span class="nf">add_mesh_to_display_layer</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">lod</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Add the mesh with the given name to an already created display layer.</span>
|
|
|
|
<span class="sd"> @type mesh_name: str</span>
|
|
<span class="sd"> @param mesh_name: The name of the mesh that should be added to a display layer.</span>
|
|
|
|
<span class="sd"> @type lod: int</span>
|
|
<span class="sd"> @param lod: The lod value, this is needed for determining the name of the display layer that the mesh should be added to.</span>
|
|
<span class="sd"> """</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">create_display_layers</span><span class="p">:</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">editDisplayLayerMembers</span><span class="p">(</span>
|
|
<span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">top_level_group</span><span class="si">}</span><span class="s2">_lod</span><span class="si">{</span><span class="n">lod</span><span class="si">}</span><span class="s2">_layer"</span><span class="p">,</span> <span class="n">mesh_name</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<span class="k">def</span> <span class="nf">_add_joints</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">List</span><span class="p">[</span><span class="n">JointModel</span><span class="p">]:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Reads and adds the joints to the scene, also returns a list model objects of joints that were added.</span>
|
|
|
|
<span class="sd"> @rtype: List[JointModel]</span>
|
|
<span class="sd"> @returns: The list containing model objects representing the joints that were added to the scene.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">joints</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">JointModel</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">read_all_neutral_joints</span><span class="p">()</span>
|
|
<span class="n">builder</span> <span class="o">=</span> <span class="n">JointBuilder</span><span class="p">(</span>
|
|
<span class="n">joints</span><span class="p">,</span>
|
|
<span class="p">)</span>
|
|
<span class="n">builder</span><span class="o">.</span><span class="n">process</span><span class="p">()</span>
|
|
<span class="k">return</span> <span class="n">joints</span>
|
|
|
|
<div class="viewcode-block" id="Builder.add_joints"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.add_joints">[docs]</a> <span class="k">def</span> <span class="nf">add_joints</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Starts adding the joints the character, if the character configuration options have add_joints set to False,</span>
|
|
<span class="sd"> this step will be skipped.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">add_joints</span><span class="p">:</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"adding joints to character..."</span><span class="p">)</span>
|
|
<span class="n">joints</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_add_joints</span><span class="p">()</span>
|
|
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">group_by_lod</span> <span class="ow">and</span> <span class="n">joints</span><span class="p">:</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">parent</span><span class="p">(</span><span class="n">joints</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_top_level_group</span><span class="p">())</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.create_groups"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.create_groups">[docs]</a> <span class="k">def</span> <span class="nf">create_groups</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Creates a Maya transform which will hold the character, if the character configuration options have</span>
|
|
<span class="sd"> create_character_node set to False, this step will be skipped.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">group_by_lod</span><span class="p">:</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"building character node..."</span><span class="p">)</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="n">world</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">empty</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_top_level_group</span><span class="p">())</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">group</span><span class="p">(</span>
|
|
<span class="n">parent</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_top_level_group</span><span class="p">(),</span>
|
|
<span class="n">empty</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
|
|
<span class="n">name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_geometry_group</span><span class="p">(),</span>
|
|
<span class="p">)</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">group</span><span class="p">(</span>
|
|
<span class="n">parent</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_top_level_group</span><span class="p">(),</span>
|
|
<span class="n">empty</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
|
|
<span class="n">name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_rig_group</span><span class="p">(),</span>
|
|
<span class="p">)</span>
|
|
<span class="k">for</span> <span class="n">lod</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_display_layers</span><span class="p">():</span>
|
|
<span class="n">name</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">top_level_group</span><span class="si">}</span><span class="s2">_lod</span><span class="si">{</span><span class="n">lod</span><span class="si">}</span><span class="s2">_layer"</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">cmds</span><span class="o">.</span><span class="n">objExists</span><span class="p">(</span><span class="n">name</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">group_by_lod</span><span class="p">:</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">group</span><span class="p">(</span>
|
|
<span class="n">parent</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_geometry_group</span><span class="p">(),</span>
|
|
<span class="n">empty</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
|
|
<span class="n">name</span><span class="o">=</span><span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">top_level_group</span><span class="si">}</span><span class="s2">_lod</span><span class="si">{</span><span class="n">lod</span><span class="si">}</span><span class="s2">_grp"</span><span class="p">,</span>
|
|
<span class="p">)</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">select</span><span class="p">(</span>
|
|
<span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">top_level_group</span><span class="si">}</span><span class="s2">_lod</span><span class="si">{</span><span class="n">lod</span><span class="si">}</span><span class="s2">_grp"</span><span class="p">,</span>
|
|
<span class="n">replace</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
|
|
<span class="p">)</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">create_display_layers</span><span class="p">:</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">createDisplayLayer</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="n">name</span><span class="p">,</span> <span class="n">noRecurse</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.attach_mesh_to_lod"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.attach_mesh_to_lod">[docs]</a> <span class="k">def</span> <span class="nf">attach_mesh_to_lod</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">lod</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Attaches the mesh called mesh_name to a given lod.</span>
|
|
|
|
<span class="sd"> @type mesh_name: str</span>
|
|
<span class="sd"> @param mesh_name: The mesh that needs to be attached to a lod holder object.</span>
|
|
|
|
<span class="sd"> @type lod: str</span>
|
|
<span class="sd"> @param lod: The name of the mesh that should be added to a display layer.</span>
|
|
<span class="sd"> """</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">group_by_lod</span><span class="p">:</span>
|
|
<span class="n">parent_node</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_top_level_group</span><span class="p">()</span><span class="si">}</span><span class="s2">|</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_geometry_group</span><span class="p">()</span><span class="si">}</span><span class="s2">|</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">top_level_group</span><span class="si">}</span><span class="s2">_lod</span><span class="si">{</span><span class="n">lod</span><span class="si">}</span><span class="s2">_grp"</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">parent</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">get_mesh_node_fullpath_on_root</span><span class="p">(</span><span class="n">mesh_name</span><span class="o">=</span><span class="n">mesh_name</span><span class="p">),</span> <span class="n">parent_node</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_mesh_node_fullpath_on_root"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_mesh_node_fullpath_on_root">[docs]</a> <span class="k">def</span> <span class="nf">get_mesh_node_fullpath_on_root</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Gets the full path in the scene of a mesh.</span>
|
|
|
|
<span class="sd"> @type mesh_name: str</span>
|
|
<span class="sd"> @param mesh_name: The mesh thats path is needed.</span>
|
|
|
|
<span class="sd"> @rtype: str</span>
|
|
<span class="sd"> @returns: The full path of the mesh object in the scene</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">return</span> <span class="nb">str</span><span class="p">(</span><span class="n">Maya</span><span class="o">.</span><span class="n">get_element</span><span class="p">(</span><span class="sa">f</span><span class="s2">"|</span><span class="si">{</span><span class="n">mesh_name</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span><span class="o">.</span><span class="n">fullPathName</span><span class="p">())</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.add_ctrl_attributes_on_root_joint"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.add_ctrl_attributes_on_root_joint">[docs]</a> <span class="k">def</span> <span class="nf">add_ctrl_attributes_on_root_joint</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Adds and sets the raw gui control attributes on root joint.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">add_ctrl_attributes_on_root_joint</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">add_joints</span><span class="p">:</span>
|
|
<span class="n">gui_control_names</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_raw_control_names</span><span class="p">()</span>
|
|
<span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">gui_control_names</span><span class="p">:</span>
|
|
<span class="n">ctrl_and_attr_names</span> <span class="o">=</span> <span class="n">name</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s2">"."</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_attribute</span><span class="p">(</span>
|
|
<span class="n">control_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">facial_root_joint_name</span><span class="p">,</span>
|
|
<span class="n">long_name</span><span class="o">=</span><span class="n">ctrl_and_attr_names</span><span class="p">[</span><span class="mi">1</span><span class="p">],</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.add_animated_map_attributes_on_root_joint"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.add_animated_map_attributes_on_root_joint">[docs]</a> <span class="k">def</span> <span class="nf">add_animated_map_attributes_on_root_joint</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Adds and sets the animated map attributes on root joint.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">if</span> <span class="p">(</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">add_animated_map_attributes_on_root_joint</span>
|
|
<span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">add_joints</span>
|
|
<span class="p">):</span>
|
|
<span class="n">names</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_animated_map_names</span><span class="p">()</span>
|
|
<span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
|
<span class="n">long_name</span> <span class="o">=</span> <span class="n">name</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="s2">"."</span><span class="p">,</span> <span class="s2">"_"</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_attribute</span><span class="p">(</span>
|
|
<span class="n">control_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">facial_root_joint_name</span><span class="p">,</span> <span class="n">long_name</span><span class="o">=</span><span class="n">long_name</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.add_attribute"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.add_attribute">[docs]</a> <span class="k">def</span> <span class="nf">add_attribute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">control_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">long_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Adds attributes wrapper for internal usage.</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">addAttr</span><span class="p">(</span>
|
|
<span class="n">control_name</span><span class="p">,</span>
|
|
<span class="n">longName</span><span class="o">=</span><span class="n">long_name</span><span class="p">,</span>
|
|
<span class="n">keyable</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
|
|
<span class="n">attributeType</span><span class="o">=</span><span class="s2">"float"</span><span class="p">,</span>
|
|
<span class="n">minValue</span><span class="o">=</span><span class="mf">0.0</span><span class="p">,</span>
|
|
<span class="n">maxValue</span><span class="o">=</span><span class="mf">1.0</span><span class="p">,</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.add_key_frames"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.add_key_frames">[docs]</a> <span class="k">def</span> <span class="nf">add_key_frames</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Adds a starting key frame to the facial root joint if joints are added and the add_key_frames option is set</span>
|
|
<span class="sd"> to True.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">add_key_frames</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">add_joints</span><span class="p">:</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"setting keyframe on the root joint..."</span><span class="p">)</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">currentTime</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">cmds</span><span class="o">.</span><span class="n">objExists</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">facial_root_joint_name</span><span class="p">):</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">select</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">facial_root_joint_name</span><span class="p">,</span> <span class="n">replace</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">setKeyframe</span><span class="p">(</span><span class="n">inTangentType</span><span class="o">=</span><span class="s2">"linear"</span><span class="p">,</span> <span class="n">outTangentType</span><span class="o">=</span><span class="s2">"linear"</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.set_filtered_meshes"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.set_filtered_meshes">[docs]</a> <span class="k">def</span> <span class="nf">set_filtered_meshes</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">all_loaded_meshes</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_filtered_meshes</span><span class="p">()</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_mesh_indices_filter"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_mesh_indices_filter">[docs]</a> <span class="k">def</span> <span class="nf">get_mesh_indices_filter</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]:</span>
|
|
<span class="n">indices</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">index</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_mesh_count</span><span class="p">()):</span>
|
|
<span class="n">mesh_name</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_mesh_name</span><span class="p">(</span><span class="n">index</span><span class="p">)</span>
|
|
<span class="k">for</span> <span class="n">cur_filter</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">mesh_filter</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="n">cur_filter</span> <span class="ow">in</span> <span class="n">mesh_name</span><span class="p">:</span>
|
|
<span class="n">indices</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">index</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">indices</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_filtered_meshes"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_filtered_meshes">[docs]</a> <span class="k">def</span> <span class="nf">get_filtered_meshes</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]:</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">mesh_filter</span> <span class="ow">and</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">lod_filter</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">meshes</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">meshes</span>
|
|
<span class="k">return</span> <span class="nb">list</span><span class="p">(</span><span class="nb">range</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_mesh_count</span><span class="p">()))</span>
|
|
|
|
<span class="n">meshes</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="n">meshes_by_lod</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_all_meshes_grouped_by_lod</span><span class="p">()</span>
|
|
<span class="n">all_meshes</span> <span class="o">=</span> <span class="p">[</span><span class="n">mesh_index</span> <span class="k">for</span> <span class="n">meshes</span> <span class="ow">in</span> <span class="n">meshes_by_lod</span> <span class="k">for</span> <span class="n">mesh_index</span> <span class="ow">in</span> <span class="n">meshes</span><span class="p">]</span>
|
|
<span class="n">mesh_indices_filter</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_mesh_indices_filter</span><span class="p">()</span>
|
|
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">lod_filter</span><span class="p">:</span>
|
|
<span class="k">for</span> <span class="n">lod</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">lod_filter</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="mi">0</span> <span class="o"><=</span> <span class="n">lod</span> <span class="o"><</span> <span class="nb">len</span><span class="p">(</span><span class="n">meshes_by_lod</span><span class="p">):</span>
|
|
<span class="n">meshes</span><span class="o">.</span><span class="n">extend</span><span class="p">(</span><span class="n">meshes_by_lod</span><span class="p">[</span><span class="n">lod</span><span class="p">])</span>
|
|
<span class="k">if</span> <span class="n">mesh_indices_filter</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">meshes</span><span class="p">)</span> <span class="o">&</span> <span class="nb">set</span><span class="p">(</span><span class="n">mesh_indices_filter</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="n">meshes</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">mesh_filter</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">all_meshes</span><span class="p">)</span> <span class="o">&</span> <span class="nb">set</span><span class="p">(</span><span class="n">mesh_indices_filter</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="n">all_meshes</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.build_meshes"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.build_meshes">[docs]</a> <span class="k">def</span> <span class="nf">build_meshes</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Builds the meshes. If specified in the config they get parented to a created</span>
|
|
<span class="sd"> character node transform, otherwise the meshes get put to the root level of the scene.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"adding character meshes..."</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">meshes</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">for</span> <span class="n">lod</span><span class="p">,</span> <span class="n">meshes_per_lod</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_meshes_by_lods</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">all_loaded_meshes</span><span class="p">)</span>
|
|
<span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">meshes</span><span class="p">[</span><span class="n">lod</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">build_meshes_by_lod</span><span class="p">(</span>
|
|
<span class="n">lod</span><span class="o">=</span><span class="n">lod</span><span class="p">,</span> <span class="n">meshes_per_lod</span><span class="o">=</span><span class="n">meshes_per_lod</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.build_meshes_by_lod"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.build_meshes_by_lod">[docs]</a> <span class="k">def</span> <span class="nf">build_meshes_by_lod</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">lod</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span> <span class="n">meshes_per_lod</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">])</span> <span class="o">-></span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Builds the meshes from the provided mesh ids and then attaches them to a given lod if specified in the</span>
|
|
<span class="sd"> character configuration.</span>
|
|
|
|
<span class="sd"> @type lod: int</span>
|
|
<span class="sd"> @param lod: The lod number representing the display layer the meshes to the display layer.</span>
|
|
|
|
<span class="sd"> @type meshes_per_lod: List[int]</span>
|
|
<span class="sd"> @param meshes_per_lod: List of mesh indices that are being built.</span>
|
|
|
|
<span class="sd"> @rtype: List[MObject]</span>
|
|
<span class="sd"> @returns: The list of maya objects that represent the meshes added to the scene.</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">meshes</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">mesh_index</span> <span class="ow">in</span> <span class="n">meshes_per_lod</span><span class="p">:</span>
|
|
<span class="n">builder</span> <span class="o">=</span> <span class="n">Mesh</span><span class="p">(</span>
|
|
<span class="n">config</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="p">,</span>
|
|
<span class="n">dna</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="p">,</span>
|
|
<span class="n">mesh_index</span><span class="o">=</span><span class="n">mesh_index</span><span class="p">,</span>
|
|
<span class="p">)</span>
|
|
<span class="n">builder</span><span class="o">.</span><span class="n">build</span><span class="p">()</span>
|
|
|
|
<span class="n">mesh_name</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_mesh_name</span><span class="p">(</span><span class="n">index</span><span class="o">=</span><span class="n">mesh_index</span><span class="p">)</span>
|
|
<span class="n">meshes</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">mesh_name</span><span class="p">)</span>
|
|
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_mesh_to_display_layer</span><span class="p">(</span><span class="n">mesh_name</span><span class="p">,</span> <span class="n">lod</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">attach_mesh_to_lod</span><span class="p">(</span><span class="n">mesh_name</span><span class="p">,</span> <span class="n">lod</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">default_lambert_shader</span><span class="p">(</span><span class="n">mesh_name</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">meshes</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.default_lambert_shader"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.default_lambert_shader">[docs]</a> <span class="k">def</span> <span class="nf">default_lambert_shader</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">group_by_lod</span><span class="p">:</span>
|
|
<span class="n">names</span> <span class="o">=</span> <span class="n">cmds</span><span class="o">.</span><span class="n">ls</span><span class="p">(</span><span class="sa">f</span><span class="s2">"*|</span><span class="si">{</span><span class="n">mesh_name</span><span class="si">}</span><span class="s2">"</span><span class="p">,</span> <span class="n">l</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
|
|
<span class="k">for</span> <span class="n">item</span> <span class="ow">in</span> <span class="n">names</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="n">item</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="sa">f</span><span class="s2">"|</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get_top_level_group</span><span class="p">()</span><span class="si">}</span><span class="s2">"</span><span class="p">):</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">select</span><span class="p">(</span><span class="n">item</span><span class="p">,</span> <span class="n">r</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
|
|
<span class="k">break</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">select</span><span class="p">(</span><span class="n">mesh_name</span><span class="p">,</span> <span class="n">r</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
|
|
|
|
<span class="n">mel</span><span class="o">.</span><span class="n">eval</span><span class="p">(</span><span class="s2">"sets -e -forceElement initialShadingGroup"</span><span class="p">)</span>
|
|
|
|
<span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="n">logging</span><span class="o">.</span><span class="n">error</span><span class="p">(</span>
|
|
<span class="sa">f</span><span class="s2">"Couldn't set lambert shader for mesh </span><span class="si">{</span><span class="n">mesh_name</span><span class="si">}</span><span class="s2">. Reason: </span><span class="si">{</span><span class="n">e</span><span class="si">}</span><span class="s2">"</span>
|
|
<span class="p">)</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span><span class="n">e</span><span class="p">)</span> <span class="kn">from</span> <span class="nn">e</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.set_units"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.set_units">[docs]</a> <span class="k">def</span> <span class="nf">set_units</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="sd">"""Sets the translation and rotation units of the scene from @config"""</span>
|
|
|
|
<span class="n">linear_unit</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_linear_unit</span><span class="p">()</span>
|
|
<span class="n">angle_unit</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_angle_unit</span><span class="p">()</span>
|
|
|
|
<span class="n">cmds</span><span class="o">.</span><span class="n">currentUnit</span><span class="p">(</span><span class="n">linear</span><span class="o">=</span><span class="n">linear_unit</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="n">angle</span><span class="o">=</span><span class="n">angle_unit</span><span class="o">.</span><span class="n">name</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_linear_unit"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_linear_unit">[docs]</a> <span class="k">def</span> <span class="nf">get_linear_unit</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">LinearUnit</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_linear_unit_from_int</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_translation_unit</span><span class="p">())</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_angle_unit"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_angle_unit">[docs]</a> <span class="k">def</span> <span class="nf">get_angle_unit</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">AngleUnit</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_angle_unit_from_int</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_rotation_unit</span><span class="p">())</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_linear_unit_from_int"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_linear_unit_from_int">[docs]</a> <span class="k">def</span> <span class="nf">get_linear_unit_from_int</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-></span> <span class="n">LinearUnit</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Returns an enum from an int value.</span>
|
|
<span class="sd"> 0 -> cm</span>
|
|
<span class="sd"> 1 -> m</span>
|
|
|
|
<span class="sd"> @type value: int</span>
|
|
<span class="sd"> @param value: The value that the enum is mapped to.</span>
|
|
|
|
<span class="sd"> @rtype: LinearUnit</span>
|
|
<span class="sd"> @returns: LinearUnit.cm or LinearUnit.m</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">if</span> <span class="n">value</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="n">LinearUnit</span><span class="o">.</span><span class="n">cm</span>
|
|
<span class="k">if</span> <span class="n">value</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="n">LinearUnit</span><span class="o">.</span><span class="n">m</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Unknown linear unit set in DNA file! value </span><span class="si">{</span><span class="n">value</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_angle_unit_from_int"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_angle_unit_from_int">[docs]</a> <span class="k">def</span> <span class="nf">get_angle_unit_from_int</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-></span> <span class="n">AngleUnit</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Returns an enum from an int value.</span>
|
|
<span class="sd"> 0 -> degree</span>
|
|
<span class="sd"> 1 -> radian</span>
|
|
|
|
<span class="sd"> @type value: int</span>
|
|
<span class="sd"> @param value: The value that the enum is mapped to.</span>
|
|
|
|
<span class="sd"> @rtype: AngleUnit</span>
|
|
<span class="sd"> @returns: AngleUnit.degree or AngleUnit.radian</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="k">if</span> <span class="n">value</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="n">AngleUnit</span><span class="o">.</span><span class="n">degree</span>
|
|
<span class="k">if</span> <span class="n">value</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="n">AngleUnit</span><span class="o">.</span><span class="n">radian</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Unknown angle unit set in DNA file! value </span><span class="si">{</span><span class="n">value</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="Builder.get_display_layers"><a class="viewcode-back" href="../../../dna_viewer.builder.html#dna_viewer.builder.builder.Builder.get_display_layers">[docs]</a> <span class="k">def</span> <span class="nf">get_display_layers</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-></span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]:</span>
|
|
<span class="sd">"""Gets a lod id list that need to be created for the meshes from @config"""</span>
|
|
<span class="n">meshes</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">idx</span><span class="p">,</span> <span class="n">meshes_per_lod</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">dna</span><span class="o">.</span><span class="n">get_meshes_by_lods</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">all_loaded_meshes</span><span class="p">)</span>
|
|
<span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">meshes_per_lod</span><span class="p">:</span>
|
|
<span class="n">meshes</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">idx</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">meshes</span><span class="p">))</span></div></div>
|
|
</pre></div>
|
|
|
|
</div>
|
|
</div>
|
|
<footer>
|
|
|
|
<hr/>
|
|
|
|
<div role="contentinfo">
|
|
<p>© Copyright 2023, Epic Games.</p>
|
|
</div>
|
|
|
|
Built with <a href="https://www.sphinx-doc.org/">Sphinx</a> using a
|
|
<a href="https://github.com/readthedocs/sphinx_rtd_theme">theme</a>
|
|
provided by <a href="https://readthedocs.org">Read the Docs</a>.
|
|
|
|
|
|
</footer>
|
|
</div>
|
|
</div>
|
|
</section>
|
|
</div>
|
|
<script>
|
|
jQuery(function () {
|
|
SphinxRtdTheme.Navigation.enable(true);
|
|
});
|
|
</script>
|
|
|
|
</body>
|
|
</html> |