384 lines
47 KiB
HTML
384 lines
47 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.dnalib.dnalib — 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.dnalib.dnalib</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.dnalib.dnalib</h1><div class="highlight"><pre>
|
|
<span></span><span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">List</span><span class="p">,</span> <span class="n">Optional</span><span class="p">,</span> <span class="n">Tuple</span>
|
|
|
|
<span class="kn">from</span> <span class="nn">dna</span> <span class="kn">import</span> <span class="n">BinaryStreamReader</span> <span class="k">as</span> <span class="n">DNAReader</span>
|
|
<span class="kn">from</span> <span class="nn">dna</span> <span class="kn">import</span> <span class="n">DataLayer_All</span><span class="p">,</span> <span class="n">FileStream</span><span class="p">,</span> <span class="n">Status</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">..model</span> <span class="kn">import</span> <span class="n">UV</span><span class="p">,</span> <span class="n">BlendShape</span><span class="p">,</span> <span class="n">Joint</span><span class="p">,</span> <span class="n">Layout</span><span class="p">,</span> <span class="n">Point3</span>
|
|
<span class="kn">from</span> <span class="nn">.behavior</span> <span class="kn">import</span> <span class="n">Behavior</span>
|
|
<span class="kn">from</span> <span class="nn">.geometry</span> <span class="kn">import</span> <span class="n">Geometry</span>
|
|
<span class="kn">from</span> <span class="nn">.layer</span> <span class="kn">import</span> <span class="n">Layer</span>
|
|
|
|
|
|
<div class="viewcode-block" id="DNA"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA">[docs]</a><span class="k">class</span> <span class="nc">DNA</span><span class="p">(</span><span class="n">Behavior</span><span class="p">,</span> <span class="n">Geometry</span><span class="p">):</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> A class used for accessing data in DNA file.</span>
|
|
|
|
<span class="sd"> @type dna_path: str</span>
|
|
<span class="sd"> @param dna_path: The path of the DNA file</span>
|
|
|
|
<span class="sd"> @type layers: Optional[List[Layer]]</span>
|
|
<span class="sd"> @param layers: List of parts of DNA to be loaded. If noting is passed, whole DNA is going to be loaded. Same as</span>
|
|
<span class="sd"> passing Layer.all.</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_path</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">layers</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">List</span><span class="p">[</span><span class="n">Layer</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">path</span> <span class="o">=</span> <span class="n">dna_path</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">reader</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_reader</span><span class="p">(</span><span class="n">dna_path</span><span class="p">)</span>
|
|
<span class="n">layers</span> <span class="o">=</span> <span class="n">layers</span> <span class="ow">or</span> <span class="p">[</span><span class="n">Layer</span><span class="o">.</span><span class="n">all</span><span class="p">]</span>
|
|
<span class="n">Behavior</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">reader</span><span class="p">,</span> <span class="n">layers</span><span class="p">)</span>
|
|
<span class="n">Geometry</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">reader</span><span class="p">,</span> <span class="n">layers</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">read</span><span class="p">()</span>
|
|
|
|
<div class="viewcode-block" id="DNA.create_reader"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.create_reader">[docs]</a> <span class="k">def</span> <span class="nf">create_reader</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">dna_path</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="n">DNAReader</span><span class="p">:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Creates a stream reader needed for reading values from the DNA file.</span>
|
|
|
|
<span class="sd"> @type dna_path: str</span>
|
|
<span class="sd"> @param dna_path: The path of the DNA file</span>
|
|
|
|
<span class="sd"> @rtype: DNA</span>
|
|
<span class="sd"> @returns: The reader needed for reading values from the DNA file</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">stream</span> <span class="o">=</span> <span class="n">FileStream</span><span class="p">(</span>
|
|
<span class="n">dna_path</span><span class="p">,</span> <span class="n">FileStream</span><span class="o">.</span><span class="n">AccessMode_Read</span><span class="p">,</span> <span class="n">FileStream</span><span class="o">.</span><span class="n">OpenMode_Binary</span>
|
|
<span class="p">)</span>
|
|
|
|
<span class="n">reader</span> <span class="o">=</span> <span class="n">DNAReader</span><span class="p">(</span><span class="n">stream</span><span class="p">,</span> <span class="n">DataLayer_All</span><span class="p">)</span>
|
|
<span class="n">reader</span><span class="o">.</span><span class="n">read</span><span class="p">()</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">Status</span><span class="o">.</span><span class="n">isOk</span><span class="p">():</span>
|
|
<span class="n">status</span> <span class="o">=</span> <span class="n">Status</span><span class="o">.</span><span class="n">get</span><span class="p">()</span>
|
|
<span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Error loading DNA: </span><span class="si">{</span><span class="n">status</span><span class="o">.</span><span class="n">message</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">reader</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.is_read"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.is_read">[docs]</a> <span class="k">def</span> <span class="nf">is_read</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="k">return</span> <span class="n">Behavior</span><span class="o">.</span><span class="n">is_read</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="ow">and</span> <span class="n">Geometry</span><span class="o">.</span><span class="n">is_read</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.read"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.read">[docs]</a> <span class="k">def</span> <span class="nf">read</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="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">is_read</span><span class="p">():</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">start_read</span><span class="p">()</span>
|
|
<span class="n">Behavior</span><span class="o">.</span><span class="n">read</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
|
|
<span class="n">Geometry</span><span class="o">.</span><span class="n">read</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.read_all_neutral_joints"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.read_all_neutral_joints">[docs]</a> <span class="k">def</span> <span class="nf">read_all_neutral_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">Joint</span><span class="p">]:</span>
|
|
<span class="n">joints</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">i</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">get_joint_count</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">get_joint_name</span><span class="p">(</span><span class="n">i</span><span class="p">)</span>
|
|
<span class="n">translation</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_neutral_joint_translation</span><span class="p">(</span><span class="n">i</span><span class="p">)</span>
|
|
<span class="n">orientation</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_neutral_joint_rotation</span><span class="p">(</span><span class="n">i</span><span class="p">)</span>
|
|
<span class="n">parent_name</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_joint_name</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">get_joint_parent_index</span><span class="p">(</span><span class="n">i</span><span class="p">))</span>
|
|
|
|
<span class="n">joint</span> <span class="o">=</span> <span class="n">Joint</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">translation</span><span class="o">=</span><span class="n">translation</span><span class="p">,</span>
|
|
<span class="n">orientation</span><span class="o">=</span><span class="n">orientation</span><span class="p">,</span>
|
|
<span class="n">parent_name</span><span class="o">=</span><span class="n">parent_name</span><span class="p">,</span>
|
|
<span class="p">)</span>
|
|
|
|
<span class="n">joints</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">joint</span><span class="p">)</span>
|
|
|
|
<span class="k">return</span> <span class="n">joints</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_all_skin_weights_joint_indices_for_mesh"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_all_skin_weights_joint_indices_for_mesh">[docs]</a> <span class="k">def</span> <span class="nf">get_all_skin_weights_joint_indices_for_mesh</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]]:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">skin_weights</span><span class="o">.</span><span class="n">joint_indices</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_blend_shape_target_deltas_with_vertex_id"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_blend_shape_target_deltas_with_vertex_id">[docs]</a> <span class="k">def</span> <span class="nf">get_blend_shape_target_deltas_with_vertex_id</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span> <span class="n">blend_shape_target_index</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="n">Tuple</span><span class="p">[</span><span class="nb">int</span><span class="p">,</span> <span class="n">Point3</span><span class="p">]]:</span>
|
|
<span class="n">blend_shape</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">blend_shapes</span><span class="p">[</span>
|
|
<span class="n">blend_shape_target_index</span>
|
|
<span class="p">]</span>
|
|
<span class="n">indices</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">blend_shape</span><span class="o">.</span><span class="n">deltas</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
|
|
|
<span class="n">deltas</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Point3</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="n">indices</span><span class="p">:</span>
|
|
<span class="n">deltas</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">blend_shape</span><span class="o">.</span><span class="n">deltas</span><span class="p">[</span><span class="n">i</span><span class="p">])</span>
|
|
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">deltas</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="p">[]</span>
|
|
|
|
<span class="k">return</span> <span class="nb">list</span><span class="p">(</span><span class="nb">zip</span><span class="p">(</span><span class="n">indices</span><span class="p">,</span> <span class="n">deltas</span><span class="p">))</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_all_skin_weights_values_for_mesh"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_all_skin_weights_values_for_mesh">[docs]</a> <span class="k">def</span> <span class="nf">get_all_skin_weights_values_for_mesh</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">List</span><span class="p">[</span><span class="nb">float</span><span class="p">]]:</span>
|
|
<span class="n">skin_weight_values</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="n">mesh</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span>
|
|
<span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">mesh</span><span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">positions</span><span class="p">)):</span>
|
|
<span class="n">skin_weight_values</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">mesh</span><span class="o">.</span><span class="n">skin_weights</span><span class="o">.</span><span class="n">values</span><span class="p">[</span><span class="n">i</span><span class="p">])</span>
|
|
|
|
<span class="k">return</span> <span class="n">skin_weight_values</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_skin_weight_matrix_for_mesh"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_skin_weight_matrix_for_mesh">[docs]</a> <span class="k">def</span> <span class="nf">get_skin_weight_matrix_for_mesh</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">List</span><span class="p">[</span><span class="n">Tuple</span><span class="p">[</span><span class="nb">int</span><span class="p">,</span> <span class="nb">float</span><span class="p">]]]:</span>
|
|
<span class="n">vertex_position_count</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">positions</span><span class="p">)</span>
|
|
|
|
<span class="n">joint_indices</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_all_skin_weights_joint_indices_for_mesh</span><span class="p">(</span><span class="n">mesh_index</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">joint_indices</span><span class="p">)</span> <span class="o">!=</span> <span class="n">vertex_position_count</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span>
|
|
<span class="s2">"Number of joint indices and vertex count don't match!"</span>
|
|
<span class="p">)</span>
|
|
|
|
<span class="n">skin_weight_values</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_all_skin_weights_values_for_mesh</span><span class="p">(</span><span class="n">mesh_index</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">skin_weight_values</span><span class="p">)</span> <span class="o">!=</span> <span class="n">vertex_position_count</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span>
|
|
<span class="s2">"Number of skin weight values and vertex count don't match!"</span>
|
|
<span class="p">)</span>
|
|
<span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">joint_indices</span><span class="p">)</span> <span class="o">!=</span> <span class="nb">len</span><span class="p">(</span><span class="n">skin_weight_values</span><span class="p">):</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span>
|
|
<span class="s2">"Number of skin weight values and joint indices count don't match for vertex!"</span>
|
|
<span class="p">)</span>
|
|
|
|
<span class="n">weight_matrix</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">indices</span><span class="p">,</span> <span class="n">values</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span><span class="n">joint_indices</span><span class="p">,</span> <span class="n">skin_weight_values</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">indices</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span>
|
|
<span class="s2">"JointIndexArray for vertex can't be less than one!"</span>
|
|
<span class="p">)</span>
|
|
<span class="n">vertex_weights</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">joint_index</span><span class="p">,</span> <span class="n">skin_weight_value</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span><span class="n">indices</span><span class="p">,</span> <span class="n">values</span><span class="p">):</span>
|
|
<span class="n">vertex_weights</span><span class="o">.</span><span class="n">append</span><span class="p">((</span><span class="n">joint_index</span><span class="p">,</span> <span class="n">skin_weight_value</span><span class="p">))</span>
|
|
<span class="n">weight_matrix</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">vertex_weights</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">weight_matrix</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_vertex_texture_coordinates_for_mesh"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_vertex_texture_coordinates_for_mesh">[docs]</a> <span class="k">def</span> <span class="nf">get_vertex_texture_coordinates_for_mesh</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">UV</span><span class="p">]:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">texture_coordinates</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_vertex_positions_for_mesh_index"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_vertex_positions_for_mesh_index">[docs]</a> <span class="k">def</span> <span class="nf">get_vertex_positions_for_mesh_index</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">Point3</span><span class="p">]:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">positions</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_vertex_layout_positions_for_mesh_index"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_vertex_layout_positions_for_mesh_index">[docs]</a> <span class="k">def</span> <span class="nf">get_vertex_layout_positions_for_mesh_index</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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">int</span><span class="p">]:</span>
|
|
<span class="k">return</span> <span class="p">[</span>
|
|
<span class="n">item</span><span class="o">.</span><span class="n">position_index</span>
|
|
<span class="k">for</span> <span class="n">item</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">layouts</span>
|
|
<span class="p">]</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_faces"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_faces">[docs]</a> <span class="k">def</span> <span class="nf">get_faces</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]]:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">face_vertex_layouts</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_polygon_faces_and_connects"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_polygon_faces_and_connects">[docs]</a> <span class="k">def</span> <span class="nf">get_polygon_faces_and_connects</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="p">,</span>
|
|
<span class="n">mesh_index</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
|
|
<span class="n">dna_faces</span><span class="p">:</span> <span class="n">List</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="kc">None</span><span class="p">,</span>
|
|
<span class="n">dna_vertex_layout_positions</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="kc">None</span><span class="p">,</span>
|
|
<span class="p">)</span> <span class="o">-></span> <span class="n">Tuple</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="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]]:</span>
|
|
<span class="k">if</span> <span class="n">mesh_index</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="kc">None</span> <span class="ow">in</span> <span class="p">(</span><span class="n">dna_faces</span><span class="p">,</span> <span class="n">dna_vertex_layout_positions</span><span class="p">):</span>
|
|
<span class="k">raise</span> <span class="n">DNAViewerError</span><span class="p">(</span>
|
|
<span class="s2">"get_polygon_faces_and_connects -> Must provide either mesh_index or dna_faces and dna_vertex_layout_positions"</span>
|
|
<span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">dna_faces</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="n">dna_faces</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_faces</span><span class="p">(</span><span class="n">mesh_index</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">dna_vertex_layout_positions</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="n">dna_vertex_layout_positions</span> <span class="o">=</span> <span class="p">(</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">get_vertex_layout_positions_for_mesh_index</span><span class="p">(</span><span class="n">mesh_index</span><span class="p">)</span>
|
|
<span class="p">)</span>
|
|
|
|
<span class="n">polygon_faces</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="n">polygon_connects</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">vertices_layout_index_array</span> <span class="ow">in</span> <span class="n">dna_faces</span><span class="p">:</span>
|
|
<span class="n">polygon_faces</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">vertices_layout_index_array</span><span class="p">))</span>
|
|
<span class="k">for</span> <span class="n">vertex_layout_index_array</span> <span class="ow">in</span> <span class="n">vertices_layout_index_array</span><span class="p">:</span>
|
|
<span class="n">polygon_connects</span><span class="o">.</span><span class="n">append</span><span class="p">(</span>
|
|
<span class="n">dna_vertex_layout_positions</span><span class="p">[</span><span class="n">vertex_layout_index_array</span><span class="p">]</span>
|
|
<span class="p">)</span>
|
|
|
|
<span class="k">return</span> <span class="n">polygon_faces</span><span class="p">,</span> <span class="n">polygon_connects</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_layouts_for_mesh_index"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_layouts_for_mesh_index">[docs]</a> <span class="k">def</span> <span class="nf">get_layouts_for_mesh_index</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">Layout</span><span class="p">]:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">layouts</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_texture_coordinate_index"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_texture_coordinate_index">[docs]</a> <span class="k">def</span> <span class="nf">get_texture_coordinate_index</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span> <span class="n">layout_id</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-></span> <span class="nb">int</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="p">(</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span>
|
|
<span class="o">.</span><span class="n">topology</span><span class="o">.</span><span class="n">layouts</span><span class="p">[</span><span class="n">layout_id</span><span class="p">]</span>
|
|
<span class="o">.</span><span class="n">texture_coordinate_index</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.has_blend_shapes"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.has_blend_shapes">[docs]</a> <span class="k">def</span> <span class="nf">has_blend_shapes</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-></span> <span class="nb">bool</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="p">(</span>
|
|
<span class="nb">len</span><span class="p">([</span><span class="n">bs</span><span class="o">.</span><span class="n">channel</span> <span class="k">for</span> <span class="n">bs</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">blend_shapes</span><span class="p">])</span>
|
|
<span class="o">></span> <span class="mi">0</span>
|
|
<span class="p">)</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_lowest_lod_containing_meshes"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_lowest_lod_containing_meshes">[docs]</a> <span class="k">def</span> <span class="nf">get_lowest_lod_containing_meshes</span><span class="p">(</span>
|
|
<span class="bp">self</span><span class="p">,</span> <span class="n">mesh_indices</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="p">)</span> <span class="o">-></span> <span class="n">Optional</span><span class="p">[</span><span class="nb">int</span><span class="p">]:</span>
|
|
<span class="n">unique_mesh_indices</span> <span class="o">=</span> <span class="nb">set</span><span class="p">(</span><span class="n">mesh_indices</span><span class="p">)</span>
|
|
<span class="k">for</span> <span class="n">lod</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">get_lod_count</span><span class="p">()):</span>
|
|
<span class="k">if</span> <span class="nb">any</span><span class="p">(</span><span class="nb">list</span><span class="p">(</span><span class="n">unique_mesh_indices</span> <span class="o">&</span> <span class="nb">set</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">get_mesh_indices_for_lod</span><span class="p">(</span><span class="n">lod</span><span class="p">)))):</span>
|
|
<span class="k">return</span> <span class="n">lod</span>
|
|
<span class="k">return</span> <span class="kc">None</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_meshes_by_lods"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_meshes_by_lods">[docs]</a> <span class="k">def</span> <span class="nf">get_meshes_by_lods</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_indices</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="n">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]]:</span>
|
|
<span class="n">result_list</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">lod</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">get_lod_count</span><span class="p">()):</span>
|
|
<span class="n">temp</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">mesh_indices</span><span class="p">)</span> <span class="o">&</span> <span class="nb">set</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">get_mesh_indices_for_lod</span><span class="p">(</span><span class="n">lod</span><span class="p">)))</span>
|
|
<span class="n">result_list</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">temp</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">result_list</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_all_meshes_grouped_by_lod"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_all_meshes_grouped_by_lod">[docs]</a> <span class="k">def</span> <span class="nf">get_all_meshes_grouped_by_lod</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">List</span><span class="p">[</span><span class="nb">int</span><span class="p">]]:</span>
|
|
<span class="sd">"""</span>
|
|
<span class="sd"> Gets the list of list of mesh indices grouped by the lod number.</span>
|
|
|
|
<span class="sd"> @type dna: DNA</span>
|
|
<span class="sd"> @param dna: Instance of DNA.</span>
|
|
|
|
<span class="sd"> @rtype: List[List[int]]</span>
|
|
<span class="sd"> @returns: The list of list of mesh indices grouped by the lod number</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">result</span><span class="p">:</span> <span class="n">List</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">lod</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">get_lod_count</span><span class="p">()):</span>
|
|
<span class="n">mesh_indices</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="bp">self</span><span class="o">.</span><span class="n">get_mesh_indices_for_lod</span><span class="p">(</span><span class="n">lod</span><span class="p">):</span>
|
|
<span class="n">mesh_indices</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">mesh_index</span><span class="p">)</span>
|
|
<span class="n">result</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">mesh_indices</span><span class="p">)</span>
|
|
|
|
<span class="k">return</span> <span class="n">result</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_blend_shapes"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_blend_shapes">[docs]</a> <span class="k">def</span> <span class="nf">get_blend_shapes</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mesh_index</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="n">BlendShape</span><span class="p">]:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">geometry_meshes</span><span class="p">[</span><span class="n">mesh_index</span><span class="p">]</span><span class="o">.</span><span class="n">blend_shapes</span></div>
|
|
|
|
<div class="viewcode-block" id="DNA.get_mesh_id_from_mesh_name"><a class="viewcode-back" href="../../../dna_viewer.dnalib.html#dna_viewer.dnalib.dnalib.DNA.get_mesh_id_from_mesh_name">[docs]</a> <span class="k">def</span> <span class="nf">get_mesh_id_from_mesh_name</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="n">Optional</span><span class="p">[</span><span class="nb">int</span><span class="p">]:</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">meshes_mapping</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">mesh_name</span><span class="p">,</span> <span class="kc">None</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> |