�
��Wc @ s� d d l m Z d d l Z d d l Z d d l Z d d l m Z e d k r-d d l m
Z
d d4 d � � YZ e d � Z d e _
d
e _ d e _ d e _ d
e _ d e _ d e _ d e _ e d � Z e d � Z e d � Z e d � Z e d � Z e d � Z g e _ d � Z e e _ n
d d l Te d k rXd e _ d e _ n d � Z! d d5 d � � YZ" d d6 d � � YZ# d d7 d � � YZ$ d! d8 d" � � YZ% d# d9 d$ � � YZ& d% e' d& e( d' � Z) g d( � Z* d) e j+ f d* � � YZ, d+ e j+ f d, � � YZ- d- e j+ f d. � � YZ. d/ e j+ f d0 � � YZ/ d1 e j+ f d2 � � YZ0 e d k r|d3 GHe j1 � n d S(: i����( t divisionN( t ndarrayt __main__( t vectort Mockc B s# e Z d � Z d � Z d � Z RS( c O s | | _ d | _ d S( Ni ( t namet called( t selfR t argst kwargs( ( s physutil.pyt __init__ s c O sJ | | _ | | _ x" | D] } t | | | | � q W| j d 7_ | S( Ni ( R R t setattrR ( R R R R ( ( s physutil.pyt __call__ s
c C s
d | _ d S( Ni ( R ( R ( ( s physutil.pyt reset# s ( t __name__t
__module__R
R R
( ( ( s physutil.pyR s t colort redt greent bluet yellowt oranget cyant magentat whitet arrowt labelt pointst curvet gdisplayt gcurvec C s t j j | � d S( N( R t plotst append( t pos( ( s physutil.pyt mockPlot7 s ( t *i2 c C sZ t | � t k s$ t | � t k r+ | j St | � t k rV t | j | j | j � Sd S( N( t typet boxt pyramidt sizet sphereR t radius( t obj( ( s physutil.pyt obj_sizeX s $t MotionMapc B st e Z d Z d d e j e e d d d � e e d d d � e d d d � e j d � Z e d d d � d � Z
RS( s�
This class assists students in constructing motion maps
using either arrows (measuring a quantity) or "breadcrumbs"
(with timestamps).
R i i c
C s� | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _ |
| _ | | _ | | _
| | _ y | j | j | _ Wn- t
k
r� }
d GHd GHd GH|
GH|
� n Xd | _ d S( Ns **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************i ( R* t tft
numMarkerst
markerTypet markerScalet markerColort labelMarkerOrdert labelMarkerOffsett
timeOffsett dropTimet arrowOffsett
labelColort intervalt TypeErrort curMarker( R R* R- R. R/ R0 R1 R2 R3 R5 R4 R6 R7 t err( ( s physutil.pyR
e s*
c C s� y�| | j | j k r�| j d 7_ | j d k rj t d | j j | j d | j | d | j � n? | j d k r� t d | j j d d | j | d | j � n | j
t k r t d | j d
d � | j
} t d | j j | d d t | � d
d d d t d | j � } n | j t k r�t d | j j t d | j d
d � | j d t | j � d d d t d | j � q�n Wn- t k
r�} d GHd GHd GH| GH| � n Xd S( Ni R R! t axisR t breadcrumbsR' i
i g �?t texts t=t st heightR% s **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************( R8 R: R/ R R* R! R6 R0 R1 R R5 t FalseR R4 R t strR7 R2 R3 R9 ( R t tt quantityt epsilont droptimeTextR; ( ( s physutil.pyt update� s( E_( R R t __doc__R R t TrueR RA R R
RG ( ( ( s physutil.pyR, ^ s 0't
MotionMapNc B st e Z d Z d d e j e e d d d � e e d d d � e d d d � e j d � Z e d d d � d � Z
RS( s�
This class assists students in constructing motion maps
using either arrows (measuring a quantity) or "breadcrumbs"
(with timestamps).
R i i c
C s� | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _ |
| _ | | _ | | _
| | _ y | j | j | _ Wn- t
k
r� }
d GHd GHd GH|
GH|
� n Xd | _ d S( Ns **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************i ( R* t dtt numStepsR/ R0 R1 R2 R3 R4 R5 R6 R7 R8 R9 R: ( R R* RK RL R/ R0 R1 R2 R3 R5 R4 R6 R7 R; ( ( s physutil.pyR
� s*
c C s� y�| j | j } | | k r�| j d 7_ | j d k rp t d | j j | j d | j | d | j � n? | j d k r� t d | j j d d | j | d | j � n | j
t k r#t d | j d
d � | j
} t d | j j | d d t | � d
d d d t d | j � } n | j t k r�t d | j j t d | j d
d � | j d t | j � d d d t d | j � q�n Wn- t k
r�} d GHd GHd GH| GH| � n Xd S( Ni R R! R< R R= R' i
i g �?R> s t=R? R@ R% s **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************( R8 R: R/ R R* R! R6 R0 R1 R R5 RA R R4 R RB R7 R2 R3 R9 ( R RC RD t thresholdRE RF R; ( ( s physutil.pyRG � s* E_( R R RH R R RI R RA R R
RG ( ( ( s physutil.pyRJ � s 0't PhysAxisc B sk e Z d Z d e d d d � d d d d e j e j d � Z d � Z d d d d d d � Z
d � Z RS(
sP
This class assists students in creating dynamic axes for their models.
t xi i t downc C s� y�g | _ g | _ | | _ | | _ t | j j j | j j j | j j j � | _ | | _
| | _ | d k r{ | n t d d d � | _ | d k r� | n t | � j | _ | d k r� | n4 t t | � j d d t | � j d � | j j | _ | | _ |
| _ | d k r=t d d | j d � | _ n� | d k rht d d | j d � | _ nV | d
k r�t d | j d d � | _ n+ | d k r�t d
| j d d � | _ n | j � Wn- t k
r�} d GHd GHd GH| GH| � n Xd S( Nt yi i i i����RP g��������t upg�������?t leftg��������t rightg�������?s **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************( t intervalMarkerst intervalLabelst labelTextR* R R! RO RQ t zt lastPost numLabelst axisTypeR< t NoneR+ t lengtht startPost axisColorR7 t
labelShiftt _PhysAxis__reorientR9 ( R R* RZ R[ R< R^ R] t labelst labelOrientationR_ R7 R; ( ( s physutil.pyR
s8 0 '$L c C s y� | j j | j k r� | j j | j } xI t t | j � � D]2 } | j | j | 7_ | j | j | 7_ qA Wg | j j D] } | | ^ q� | j _ t | j j j | j j j
| j j j � | _ n Wn- t k
r} d GHd GHd GH| GH| � n Xd S( Ns **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************(
R* R! RY t ranget lenRU RV t axisCurveR RO RQ RX R9 ( R t difft iRO R; ( ( s physutil.pyRG / s )7c C sf y2| d k r | n | j | _ | d k r3 | n | j | _ | d k rQ | n | j | _ | d k ro | n | j | _ | d k r� t d d | j d � | _ n� | d k r� t d d | j d � | _ nV | d k r� t d | j d d � | _ n+ | d k r't d | j d d � | _ n | j � Wn- t k
ra} d
GHd GHd GH| GH| � n Xd S(
NRP i g��������RR g�������?RS g��������RT g�������?s **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************(
R\ R< R^ R] Rb RW R R` Ra R9 ( R R< R^ R] Rb Rc R; ( ( s physutil.pyt reorientB s( c C s� t | j � d k r t n t } | j | j | j } | j | j d | j } d } x(| | j k r�| j | | } | j d k r� | j | } n, | j
d k r� d | j } n
d | j } | r
| | j | _
| | j | j | _
t | � | j | _ ni | j j t d | d | j d d � � | j j t d | | j d t | � d
t d d d | j � � | d } q\ W| r�| j | g | j _
n$ t d | j | g d | j � | _ d S(
Ni i RQ s %.2fR! R R' i R> R% R@ i ( Re RU RI RA R^ R] R< RZ RW R\ R[ RQ RO R! R` RV RB R> R R R_ R R7 Rf R ( R t updatingt finalR8 Rh t intervalPosRW ( ( s physutil.pyt
__reorient\ s. !
8N( R R RH R R\ R R R R
RG Ri Ra ( ( ( s physutil.pyRN s ) t PhysTimerc B s, e Z d Z d e e j d � Z d � Z RS( sL
This class assists students in creating an onscreen timer display.
i
c C s� y� | | _ | | _ | t k rT t d t | | d � d d d t d | � | _ n0 t d t | | d � d d d t d | � | _ Wn- t k
r� } d GHd GHd
GH| GH| � n Xd S( NR! i R> s 00:00:00.00R% R@ t 00E01s **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************( t
useScientifict
timerColorRA R R t
timerLabelR9 ( R RO RQ t fontsizeRp Rq R; ( ( s physutil.pyR
� s 34c C s� y� | j r d | | j _ n� t | d � } t | d d � } t | d � } t t d | d � � } | d k r� d } | d } n d | | | | f | j _ Wn- t k
r� } d GHd GHd
GH| GH| � n Xd S( Ns %.4Ei i< id i i s %02d:%02d:%02d.%02ds **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************( Rp Rr R> t intt roundR9 ( R RC t hourst minst secst fracR; ( ( s physutil.pyRG � s"
( R R RH RA R R R
RG ( ( ( s physutil.pyRn � s t PhysGraphc B sh e Z d Z e j e j e j e j e j e j e j
e j g Z d d d d e j d � Z d � Z RS( sU
This class assists students in creating graphs with advanced functionality.
i c
C s� y� t d d d d d | d | d | d | � | _ | | _ g | _ xA t | � D]3 } | j j t d t j | t t j � � � qR WWn- t
k
r� } d
GHd GHd GH| GH| � n Xd S(
NRO i� RQ i^ t titlet xtitlet ytitlet
backgroundR s **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************( t grapht graphDisplayt numPlotst graphsRd R R Rz t graphColorsRe R9 ( R R� R{ t xlabelt ylabelt backgroundColorRh R; ( ( s physutil.pyR
� s 0 5c G s� yf t | � | j k r' t d � � n x; t t | � � D]' } | j | j d | | | f � q: WWn- t k
r� } d GHd GHd GH| GH| � n Xd S( Ns\ ERROR: Number of dependent parameters given does not match numPlots given at initialization!R! s **********TYPE ERROR**********sq Please check that you are not passing in a variable of the wrong type (e.g. a scalar as a vector, or vice-versa)!s ******************************( Re R� t ExceptionRd R� t plotR9 ( R t independentt
dependentsRh R; ( ( s physutil.pyR� � s )N( R R RH R R R R R R R R R R� R\ R
R� ( ( ( s physutil.pyRz � s
i i c
C s d g | } x t | � D] } g | | <q Wt j j d k rt | d � �� } t j | � } x! t d | � D] } | j � qq W| r� | d k r� | j � n x^ | D]V } xM t | � D]? }
| r� | |
j t | |
� � q� | |
j | |
� q� Wq� WWd QXn� t j j d k r�t | d d ��� } t j | � } x! t d | � D] } | j � qTWt
r�| d k r�| j � n x^ | D]V } xM t | � D]? }
| r�| |
j t | |
� � q�| |
j | |
� q�Wq�WWd QXn t j j d � t
d � | S( Ni i t rbi t newlinet s! You need to use python 2* or 3*
i ( Rd t syst version_infot majort opent csvt readert nextR t floatt ignoreHeadert stderrt writet exit( t filenamet colst IgnoreHeadert startrowt NumericDatat dataRh t csvfilet csvdatat rowt c( ( s physutil.pyt readcsv� s<
&
&
c
C s� g } t } t j j d k r0 t | d � } nD t j j d k rZ t d d d d �} n t j j d � t d � t | t � r� | j
} | j � } n t | � d k r� | j
� d St } d
} d
} t | d
t t f � r t } t | d
� } t | � } n t } t | � } d } | r�xT t d t | � � D]: } t | | � | k r?t j j d � | j
� d Sq?Wn t | � d
k r�t | � | k r�t j j d � q�t } n t j | d
d d d d t j �} | r�| j | � n xy t d
| � D]h }
g } | d k rTx@ t d
| � D] } | j | | |
� q2Wn | j | |
� | j | � q
W| j
� d S( Ni t wbi s pythonTest.csvt wR� R� s! You need to use python 2* or 3*
i i s/ All lists in datalist must be the same length
sD Header length did not match the number of columns, ignoring header.
t delimitert ,t quotechart |t quoting( RA R� R� R� R� R� R� R� t
isinstancet
numpy_ndarrayt Tt tolistRe t closet listt tupleRI Rd R� t writert
QUOTE_MINIMALt writerowR (
R� t datalistt headerR� t useheadert isLofLt
ListLengtht numListst Lidxt
DataWriterR� t thisrowt col( ( s physutil.pyt writecsv sZ
$ t
TestMotionMapc B s# e Z d � Z d � Z d � Z RS( c C s� t d � | _ t d d d � | j _ d | _ d | _ t d d d � | _ d | _ t d d d � | _ t j
� t j
� t j
� t
| j | j | j d d d d d
t j d t d | j d
| j �| _ d S( NR* i i
i i i R/ R R0 R1 R5 R4 R6 ( R R* R R! R- R. R4 R0 R6 R R
R R R, R R RI t map( R ( ( s physutil.pyt setUpS s
c C s' | j | j | j j � | j | j | j j � | j | j | j j � | j d | j j � | j | j | j j � | j t j | j j � | j t
d d d � | j j � | j t | j j
� | j | j j | j | j � | j | j j d � | j t
d d d � | j j � d S( NR i i ( t assertEqualR* R� R- R. R/ R0 R R R1 R R4 RI R5 R8 R: R6 ( R ( ( s physutil.pyt test_initd s " c C s d | j _ | j j d � | j t j d � | j t j d � | j t j d � | j j d d d �| j t j d � | j t j d � | j t j d � | j | j j d � | j t j | j j | j
� | j t j d � | j t j t j
� | j t j d � d S( Ni i i RD i i t 2( R� R: RG R� R R R R R! R* R6 R< R R R> ( R ( ( s physutil.pyt test_updater s ( R R R� R� R� ( ( ( s physutil.pyR� R s t TestMotionMapNc B s# e Z d � Z d � Z d � Z RS( c C s� t d � | _ t d d d � | j _ d | _ d | _ t d d d � | _ d | _ t d d d � | _ t j
� t j
� t j
� t
| j | j | j d d d d d t j d
t d | j d | j �| _ d S(
NR* i i i i R/ R R0 R1 R5 R4 R6 ( R R* R R! RK RL R4 R0 R6 R R
R R RJ R R RI R� ( R ( ( s physutil.pyR� � s
c C s | j | j | j j � | j | j | j j � | j | j | j j � | j d | j j � | j | j | j j � | j t j | j j � | j t
d d d � | j j � | j t | j j
� | j | j j d � | j t
d d d � | j j � d S( NR i i ( R� R* R� RK RL R/ R0 R R R1 R R4 RI R5 R: R6 ( R ( ( s physutil.pyR� � s "c C s d | j _ | j j d � | j t j d � | j t j d � | j t j d � | j j d d d �| j t j d � | j t j d � | j t j d � | j | j j d � | j t j | j j | j
� | j t j d � | j t j t j
� | j t j d � d S( Ni i i RD i i R� ( R� R: RG R� R R R R R! R* R6 R< R R R> ( R ( ( s physutil.pyR� � s ( R R R� R� R� ( ( ( s physutil.pyR� � s t TestPhysAxisc B s, e Z d � Z d � Z d � Z d � Z RS( c C s� t d � | _ t d d d � | j _ d | _ t d d d � | _ t d d d � | _ d | _ d d d d d
g | _ d g | _ d | _
d | _ t j
� t | j | j d
| j
d | j d | j d | j d | j d | j �| _ d S( NR* i i i i
t at bR� t dt et arbitraryRS R[ R< R^ R] Rb Rc ( R R* R R! RZ R< R^ R] Rb t wrongLabelsR[ Rc R R
RN t physAxis( R ( ( s physutil.pyR� � s
$c C s� | j | j j | j � | j | j j | j � | j | j j | j j � | j | j j | j � | j | j j | j � | j | j j | j � | j | j j
| j
� | j | j j | j � | j | j j t
d | j d d � � | j t | j j � | j � | j t | j j � | j � | j
| j | j } | j | j j d j | � | j | j j d j | | j j � | j | j j d j d � | j t j d � d S( Ng��������i i����R� i ( R� R� RW Rb R* RY R! RZ R< R] R^ R[ R` R Re RU RV R> R R ( R Rl ( ( s physutil.pyR� � s ) ' c C sJ t d d d � } t d d d � } d d d d d g } | j j d | d | d
d d | d d
� | j | j j | � | j | j j | � | j | j j d � | j | j j t d d d � � | | } | j | j j d j | � | j | j j
d j | | j j � | j | j j
d j d � | j t j
d � d S( Ni i t ft gt hRh t jR< R^ R] Rb Rc RT g�������?i����( R R� Ri R� R< R^ R] R` RU R! RV R> R R ( R t newAxisR^ t otherLabelsRl ( ( s physutil.pyt
test_reorient� s
"
' c C s� t | j j d j j | j j d j j | j j d j j � } t | j j d j j | j j d j j | j j d j j � } t | j j j d j | j j j d j | j j j d j � } | j j � | j
| | j j d j � | j
| | j j d j � | j
| | j j j d � | j j j t d d d � | j j _ | j j � | j | | j j d j � | j | | j j d j � | j | | j j j d � d S( Ni����i i (
R R� RU R! RO RQ RX RV Rf RG R� R* t assertNotEqual( R t startMarkerPost
startLabelPost
startCurvePos( ( s physutil.pyR� � s$
(
( R R R� R� R� R� ( ( ( s physutil.pyR� � s t
TestPhysGraphc B s# e Z d � Z d � Z d � Z RS( c C s t d d � | _ d S( NR� i ( Rz t physGraph( R ( ( s physutil.pyR� s c C sh | j | j j j d � | j | j j j d � | j | j j d � | j t | j j � d � d S( Ni� i^ i ( R� R� R� RO RQ R� Re R� ( R ( ( s physutil.pyR� s c
C s | j t | j j t d d d � t d d d � � | j j t d d d � t d d d � t d d d � t d d d � t d d d � t d d d � � | j t | j j d j � d � | j | j j d j d t d d d � t d d d � f � d S( Ni i i i i i i����( t assertRaisesR� R� R� R R� Re R� R ( R ( ( s physutil.pyt test_plot s 4g#( R R R� R� R� ( ( ( s physutil.pyR� s t
TestPhysTimerc B s# e Z d � Z d � Z d � Z RS( c C s t d d � | _ d S( Ni ( Rn t timer( R ( ( s physutil.pyR� s c C sB | j | j j j d � | j | j j j t d d d � � d S( Ns 00:00:00.00i i ( t assertEqualsR� Rr R> R! R ( R ( ( s physutil.pyR� s c C sb | j j d � | j | j j j d � t | j _ | j j d � | j | j j j d � d S( Ng����L��@s 01:05:23.65s
3.9237E+03( R� RG R� Rr R> RI Rp ( R ( ( s physutil.pyR� s
( R R R� R� R� ( ( ( s physutil.pyR� s s Beginning unit tests!( ( ( ( ( ( (2 t
__future__R t unittestR� R� t numpyR R� R t vpythonR R R R R R R R R R R R R R R R R R R" R� t sceneRO RQ R+ R, RJ RN Rn Rz RA RI R� R� t TestCaseR� R� R� R� R� t main( ( ( s physutil.pyt <module> s\
OS�02% I20S 