ó
ׯXc @ s d d l Z d d l Z d d l Z d d l Td Z d Z d Z d Z d Z d Z d Z
d
Z d Z d Z
d
Z d Z d S( iÿÿÿÿN( t *c C s4 t d } | j | _ | j | _ | j | _ | S( Ng ( t vec3t xt yt z( t vec4dt ret( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt
vec4ToVec3 s
c C sF t d } t | j | _ t | j | _ t | j | _ | S( Ni ( R t intR R R ( R R ( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt vec4ToVec3i s
c C s8 d t | j t | j t | j t | j f S( Ns
[%d,%d,%d,%d]( R R R R t t( R ( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt
vec4intString s c C sZ t d } | d | _ | | | _ | d | _ | | | | _ t | } | | f S( Ng i ( t vec4R R R R
R ( t rest yFactort fourdFactort
loadTimeScaleR t ret3d( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt initRes s
c C sª x, t | t t j k r. | j d q Wxt t d t t j D]Z } t j | j | k rH | d t | k rH d | | <| | d <t j | d SqH W| S( Ni i ( t lent syst argvt appendt ranget lower( t namet defaultt paramUsedt iter( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt getParam& s /c C so t } xO t d t t j D]5 } | | d k r d | t j | f GHt } q q W| rk t d n d S( Ni i s Error: param %d '%s' not used!( t FalseR R R R t Truet exit( R t errR ( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt checkUnusedParam/ s
c
C s t d } | j | | _ | j | | _ | j | | _ | j | | _ | j | | _ t d d | } | j | | _ t d } t d d d | } t d } | j | j | _ | j | j | _ | j | j | _ | j | j | _ | | | | | f S( Ng g ð?g @( R R R R R
(
t gs4dt gs4dDefot
autoBordert repeatStartDistR t
loadOffsett loadScalet
defoOffsett defoScalet
defoFactor( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt initOffsets; s c C sM | j t } | j d | t | | d | d | ~ | j | d S( Ns
%s_vel.unit offsett scale( t createt Grid4Vec4t loadt interpolateGrid4dVect multConst( t velgridt prefixt
defoSolverR) R* R+ t ipol_vel( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt loadVelRescaled[ s
c
C s |
d k rL d |
| f GHt | d d | d t | d d | d n | s^ | s^ | rt | j | d t n | r | j d | | f d t n | rÀ | j d | | f d t n | ræ t | d | | f d d
n | r | j d | | f d t n d S( Ni s( Output sdf smoothing active %d, bord %d i g g 8Àt notimings %s_%04d.pngs %s_%04d.bobj.gzs %s_%04d.ppmg ð?s %s_%04d.uni( t simpleBlurSpecialt
createMeshR t
screenshott savet projectPpmFull(
R R
t meshCntt gridt meshoutt guit writePngst triMesht projPpmt writeUnit sdfSmootht
sdfSmoothBordt
meshPrefix( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt writeResultsl s c C sú | | } | | } | j } | j } | j } | j } t | | | | d k rx d | | | | GHt d n d | | | | }
|
| } |
| } |
| }
|
| } | | } | | j | | j } |
| j | | j } d | | } d } | d | k sf| d | k sf| d | k sf| d | k sf| d | k sf| d | k rÂd t | d t | d t | d t | d
| | | f GHd GHt d n t | t | d | | } t | | | } | S( Ngñhãµøä>s Invalid coords? %f i g ð?gü©ñÒMbP?g s w2uvw s tri t s , uvw %f,%f,%f s Invalid UVW!( R R t absR t strt clamp01R ( t pt r1t r2t r3t r13t r23t mat mct mbt mdt mdett inat inbt inct indt pr3t triut trivt triwt epst uvw( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt world2uvw s8
`J
c C s$ | d k r d S| d k r d S| S( Ng g ð?( ( t v( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyRN © s
c C so | j } | j } | j } d } } } } } } }
t | | | } | d k rÀ t d d d } t d d d }
t d d d } t | | |
| } | j } | j } | j } n| d k r5t d d d } t d d d }
t d d d } t | | |
| } | j } | j } | j } n!| d k rªt d d d } t d d d }
t d d d } t | | |
| } | j } | j } | j } n¬| | k r7| | k r7t d d d } t d d d }
t d d d d d d } t | | |
| } | j } | j } | j }
n | | k rÄ| | k rÄt d d d } t d d d }
t d d d d d d } t | | |
| } | j } | j } | j }
n | | k rQ| | k rQt d d d } t d d d }
t d d d d d d } t | | |
| } | j } | j } | j }
n d GH| | | | | | |
f S( Ng g à?g ð?g @s This should never happen...???( R R R R Rd ( Rc t uRe t wt w0t w1t w2t w_uvt w_vwt w_uwt w_uvwRO RP RQ RR ( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt weights2dim® sn ( R t ost shutilt mantaR R R R R R" R, R8 RJ Rd RN Ro ( ( ( s< /Users/sinithue/temp/ofbpuc/mantaflowgit/scenes/ofHelpers.pyt <module> s
" 