mirror of
http://CODE.RHODECODE.COM/u/O/O/O
synced 2024-09-20 13:16:42 -04:00
137 lines
11 KiB
Plaintext
137 lines
11 KiB
Plaintext
|
<EFBFBD><EFBFBD>i m p o r t n u m p y a s n p
|
|||
|
|
i m p o r t p l o t l y . g r a p h _ o b j e c t s a s g o
|
|||
|
|
f r o m m a t h i m p o r t *
|
|||
|
|
i m p o r t m p m a t h
|
|||
|
|
f r o m i p y w i d g e t s i m p o r t i n t e r a c t , w i d g e t s , T e x t , L a y o u t
|
|||
|
|
f r o m p l o t l y . s u b p l o t s i m p o r t m a k e _ s u b p l o t s
|
|||
|
|
|
|||
|
|
# I n p u t f i e l d f o r f o r m u l a
|
|||
|
|
f o r m u l a = w i d g e t s . T e x t (
|
|||
|
|
v a l u e = ' ( 1 - c o s ( ( ( 4 ) / 2 ) * x ) ) / 2 ' ,
|
|||
|
|
l a y o u t = w i d g e t s . L a y o u t ( w i d t h = ' 1 0 0 % ' ) ,
|
|||
|
|
d e s c r i p t i o n = ' <00>&<26><15>&A<16>&N<14>&O<16>&% <1D>&<26>$<24>&<26><><EFBFBD>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26><><EFBFBD>&<26>$<24>&% <1D>&O<16>&N<14>&A<16>&<26><15>&'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
R A N G E _ F R O M _ S L I D E R = w i d g e t s . F l o a t S l i d e r (
|
|||
|
|
m i n = - 4 * ( 4 * a t a n ( 1 ) ) ,
|
|||
|
|
m a x = 4 * ( 4 * a t a n ( 1 ) ) ,
|
|||
|
|
v a l u e = - 2 * ( 4 * a t a n ( 1 ) ) ,
|
|||
|
|
s t e p = ( 4 * a t a n ( 1 ) ) / 4 ,
|
|||
|
|
l a y o u t = w i d g e t s . L a y o u t ( w i d t h = ' 1 0 0 % ' ) ,
|
|||
|
|
r e a d o u t _ f o r m a t = ' . 2 5 6 f ' ,
|
|||
|
|
d e s c r i p t i o n = ' <00>&O<16>&<26>$<24>&% <1D>&<26><><EFBFBD>&<26>%<25>&<26><15><15>&df<15>&N <00>&<26><15>&% <1D>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&% <1D>&<26><15>&N <00>&df<15>&<26><15><15>&<26>%<25>&<26><><EFBFBD>&% <1D>&<26>$<24>&O<16>&'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
R A N G E _ T O _ S L I D E R = w i d g e t s . F l o a t S l i d e r (
|
|||
|
|
m i n = - 4 * ( 4 * a t a n ( 1 ) ) ,
|
|||
|
|
m a x = 4 * ( 4 * a t a n ( 1 ) ) ,
|
|||
|
|
v a l u e = 2 * ( 4 * a t a n ( 1 ) ) ,
|
|||
|
|
s t e p = ( 4 * a t a n ( 1 ) ) / 4 ,
|
|||
|
|
l a y o u t = w i d g e t s . L a y o u t ( w i d t h = ' 1 0 0 % ' ) ,
|
|||
|
|
r e a d o u t _ f o r m a t = ' . 2 5 6 f ' ,
|
|||
|
|
d e s c r i p t i o n = ' <00>&<26>$<24>&$'<27>&<26>%<25>&<26><15><15>&df<15>&N <00>&<26><15>&% <1D>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&% <1D>&<26><15>&N <00>&df<15>&<26><15><15>&<26>%<25>&$'<27>&<26>$<24>&'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
N _ s l i d e r = w i d g e t s . I n t S l i d e r (
|
|||
|
|
m i n = 1 ,
|
|||
|
|
m a x = 1 6 ,
|
|||
|
|
v a l u e = 8 ,
|
|||
|
|
l a y o u t = w i d g e t s . L a y o u t ( w i d t h = ' 1 0 0 % ' ) ,
|
|||
|
|
r e a d o u t _ f o r m a t = ' . 2 5 6 f ' ,
|
|||
|
|
d e s c r i p t i o n = ' <00>&% <1D>&<26><15><15>&<26><15>&<26>$<24>&<26><07>&<26>%<25>&% <1D>&<26><15><15>&<26>&<26>&O<16>&N<14>&N <00>&<26>%<25>&$'<27>&N <00>&<26><><EFBFBD>&<26>$<24>&km<14>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&km<14>&<26>$<24>&<26><><EFBFBD>&N <00>&$'<27>&<26>%<25>&N <00>&N<14>&O<16>&<26>&<26>&<26><15><15>&% <1D>&<26>%<25>&<26><07>&<26>$<24>&<26><15>&<26><15><15>&% <1D>&'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
O _ R E D I L S _ 1 _ W _ O _ W _ 1 _ S L I D E R _ O = w i d g e t s . F l o a t S l i d e r (
|
|||
|
|
m i n = - 2 ,
|
|||
|
|
m a x = 2 ,
|
|||
|
|
v a l u e = 1 ,
|
|||
|
|
s t e p = 1 / 2 5 6 ,
|
|||
|
|
l a y o u t = w i d g e t s . L a y o u t ( w i d t h = ' 1 0 0 % ' ) ,
|
|||
|
|
r e a d o u t _ f o r m a t = ' . 2 5 6 f ' ,
|
|||
|
|
d e s c r i p t i o n = ' <00>&<26> <00>&<26>%<25>&<26><15><15>&A<16>&<26>&<26>&<26><15>&<26><><EFBFBD>&% <1D>&<26><15>&<26><15>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26><15>&<26><15>&% <1D>&<26><><EFBFBD>&<26><15>&<26>&<26>&A<16>&<26><15><15>&<26>%<25>&<26> <00>&'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
O _ R E D I L S _ 2 _ W _ O _ W _ 2 _ S L I D E R _ O = w i d g e t s . F l o a t S l i d e r (
|
|||
|
|
m i n = - 2 ,
|
|||
|
|
m a x = 2 ,
|
|||
|
|
v a l u e = 1 ,
|
|||
|
|
s t e p = 1 / 2 5 6 ,
|
|||
|
|
l a y o u t = w i d g e t s . L a y o u t ( w i d t h = ' 1 0 0 % ' ) ,
|
|||
|
|
r e a d o u t _ f o r m a t = ' . 2 5 6 f ' ,
|
|||
|
|
d e s c r i p t i o n = ' <00>&<26><><EFBFBD>&<26>%<25>&<26><15><15>&A<16>&<26>&<26>&<26><15>&<26><><EFBFBD>&% <1D>&<26><15>&<26><15>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26>%<25>&<26><15>&<26><15>&% <1D>&<26><><EFBFBD>&<26><15>&<26>&<26>&A<16>&<26><15><15>&<26>%<25>&<26><><EFBFBD>&'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
d e f c l a m p ( x ) :
|
|||
|
|
r e t u r n m a x ( m i n ( 1 , x ) , - 1 )
|
|||
|
|
|
|||
|
|
d e f c o m p u t e ( f o r m u l a , x , v a r i a b l e 1 , v a r i a b l e 2 ) :
|
|||
|
|
f u n c _ d i c t = { f n : e v a l ( f ' l a m b d a * a r g s : m p m a t h . { f n } ( * a r g s ) ' ) f o r f n i n d i r ( m p m a t h ) }
|
|||
|
|
r e t u r n f l o a t ( e v a l ( f o r m u l a , { ' x ' : x , ' O _ 1 _ W _ O _ W _ 1 _ O ' : v a r i a b l e 1 , ' O _ 2 _ W _ O _ W _ 2 _ O ' : v a r i a b l e 2 , ' c l a m p ' : c l a m p , * * f u n c _ d i c t } ) )
|
|||
|
|
|
|||
|
|
d e f p l o t ( f o r m u l a = ' 1 . 4 7 9 5 / O _ 2 _ W _ O _ W _ 2 _ O - ( ( x ) / ( 4 * a t a n ( 1 ) * O _ 1 _ W _ O _ W _ 1 _ O / 2 * O _ 2 _ W _ O _ W _ 2 _ O ) ) * 1 . 4 7 9 5 ' , R A N G E _ F R O M = 0 * ( 4 * a t a n ( 1 ) ) , R A N G E _ T O = 4 * ( 4 * a t a n ( 1 ) ) , N = 8 , O _ 1 _ W _ O _ W _ 1 _ O = 1 , O _ 2 _ W _ O _ W _ 2 _ O = 1 ) :
|
|||
|
|
n u m _ p o i n t s = 1 + 2 * * N
|
|||
|
|
x _ v a l s = n p . l i n s p a c e ( R A N G E _ F R O M , R A N G E _ T O , n u m _ p o i n t s )
|
|||
|
|
|
|||
|
|
k a p p a _ v a l s = n p . a r r a y ( [ c o m p u t e ( f o r m u l a , x _ v a l , O _ 1 _ W _ O _ W _ 1 _ O , O _ 2 _ W _ O _ W _ 2 _ O ) f o r x _ v a l i n x _ v a l s ] )
|
|||
|
|
t h e t a _ v a l s = n p . c u m s u m ( k a p p a _ v a l s ) * ( x _ v a l s [ 1 ] - x _ v a l s [ 0 ] ) i f n u m _ p o i n t s > 1 e l s e n p . a r r a y ( [ 0 ] )
|
|||
|
|
x _ c o o r d s _ = n p . c u m s u m ( n p . c o s ( t h e t a _ v a l s ) ) * ( x _ v a l s [ 1 ] - x _ v a l s [ 0 ] ) i f n u m _ p o i n t s > 1 e l s e n p . a r r a y ( [ 0 ] )
|
|||
|
|
y _ c o o r d s _ = n p . c u m s u m ( n p . s i n ( t h e t a _ v a l s ) ) * ( x _ v a l s [ 1 ] - x _ v a l s [ 0 ] ) i f n u m _ p o i n t s > 1 e l s e n p . a r r a y ( [ 0 ] )
|
|||
|
|
|
|||
|
|
i f x _ c o o r d s _ [ 0 ] ! = 0 o r y _ c o o r d s _ [ 0 ] ! = 0 :
|
|||
|
|
x _ c o o r d s = n p . i n s e r t ( x _ c o o r d s _ , 0 , 0 )
|
|||
|
|
y _ c o o r d s = n p . i n s e r t ( y _ c o o r d s _ , 0 , 0 )
|
|||
|
|
e l s e :
|
|||
|
|
x _ c o o r d s = x _ c o o r d s _
|
|||
|
|
y _ c o o r d s = y _ c o o r d s _
|
|||
|
|
|
|||
|
|
y _ v a l s = [ c o m p u t e ( f o r m u l a , x _ v a l , O _ 1 _ W _ O _ W _ 1 _ O , O _ 2 _ W _ O _ W _ 2 _ O ) f o r x _ v a l i n x _ v a l s ]
|
|||
|
|
|
|||
|
|
# D e t e r m i n e m a x i m u m r a n g e f o r c o n s i s t e n c y a c r o s s a l l c h a r t s
|
|||
|
|
m i n _ x _ r a n g e = m i n ( m a p ( m i n , [ x _ c o o r d s , x _ v a l s ] ) )
|
|||
|
|
m i n _ y _ r a n g e = m i n ( m a p ( m i n , [ y _ c o o r d s , y _ v a l s ] ) )
|
|||
|
|
m a x _ x _ r a n g e = m a x ( m a p ( m a x , [ x _ c o o r d s , x _ v a l s ] ) )
|
|||
|
|
m a x _ y _ r a n g e = m a x ( m a p ( m a x , [ y _ c o o r d s , y _ v a l s ] ) )
|
|||
|
|
|
|||
|
|
# C r e a t e a n d d i s p l a y t h e f i r s t p l o t
|
|||
|
|
f i g 1 = g o . F i g u r e ( )
|
|||
|
|
f i g 1 . a d d _ t r a c e ( g o . S c a t t e r ( x = x _ c o o r d s , y = y _ c o o r d s , m o d e = ' l i n e s ' , l i n e = d i c t ( c o l o r = ' # C E C E C E ' ) , n a m e = ' ' , h o v e r t e m p l a t e = ' X : % { x : . 2 5 6 f } ' + ' < b r > Y : % { y : . 2 5 6 f } ' ) )
|
|||
|
|
f i g 1 . u p d a t e _ l a y o u t (
|
|||
|
|
a u t o s i z e = T r u e ,
|
|||
|
|
m a r g i n = d i c t (
|
|||
|
|
l = 0 , # l e f t m a r g i n
|
|||
|
|
r = 0 , # r i g h t m a r g i n
|
|||
|
|
b = 0 , # b o t t o m m a r g i n
|
|||
|
|
t = 0 , # t o p m a r g i n
|
|||
|
|
p a d = 0 # p a d d i n g
|
|||
|
|
) ,
|
|||
|
|
h e i g h t = 5 1 2 ,
|
|||
|
|
x a x i s = d i c t ( s c a l e a n c h o r = ' y ' , s c a l e r a t i o = 1 , g r i d c o l o r = ' # C E C E C E ' , z e r o l i n e = T r u e , z e r o l i n e c o l o r = ' # C E C E C E ' , t i c k f o n t = d i c t ( c o l o r = ' # 9 C 9 C 9 C ' , s i z e = 1 6 ) ) ,
|
|||
|
|
y a x i s = d i c t ( g r i d c o l o r = ' # C E C E C E ' , z e r o l i n e = T r u e , z e r o l i n e c o l o r = ' # C E C E C E ' , t i c k f o n t = d i c t ( c o l o r = ' # 9 C 9 C 9 C ' , s i z e = 1 6 ) ) ,
|
|||
|
|
h o v e r l a b e l = d i c t ( b g c o l o r = " # F F F F F F " , f o n t _ c o l o r = ' # 9 C 9 C 9 C ' , b o r d e r c o l o r = " # C E C E C E " , f o n t _ s i z e = 1 6 ) , p l o t _ b g c o l o r = ' # F F F F F F '
|
|||
|
|
)
|
|||
|
|
f i g 1 . u p d a t e _ x a x e s ( r a n g e = [ m i n _ x _ r a n g e , m a x _ x _ r a n g e ] )
|
|||
|
|
f i g 1 . u p d a t e _ y a x e s ( r a n g e = [ m i n _ y _ r a n g e , m a x _ y _ r a n g e ] )
|
|||
|
|
f i g 1 . s h o w ( )
|
|||
|
|
|
|||
|
|
# C r e a t e a n d d i s p l a y t h e s e c o n d p l o t
|
|||
|
|
f i g 2 = g o . F i g u r e ( )
|
|||
|
|
f i g 2 . a d d _ t r a c e ( g o . S c a t t e r ( x = x _ v a l s , y = y _ v a l s , m o d e = ' l i n e s ' , l i n e = d i c t ( c o l o r = ' # C E C E C E ' ) , n a m e = ' ' , h o v e r t e m p l a t e = ' X : % { x : . 2 5 6 f } ' + ' < b r > Y : % { y : . 2 5 6 f } ' ) )
|
|||
|
|
f i g 2 . u p d a t e _ l a y o u t (
|
|||
|
|
a u t o s i z e = T r u e ,
|
|||
|
|
m a r g i n = d i c t (
|
|||
|
|
l = 0 , # l e f t m a r g i n
|
|||
|
|
r = 0 , # r i g h t m a r g i n
|
|||
|
|
b = 0 , # b o t t o m m a r g i n
|
|||
|
|
t = 0 , # t o p m a r g i n
|
|||
|
|
p a d = 0 # p a d d i n g
|
|||
|
|
) ,
|
|||
|
|
h e i g h t = 5 1 2 ,
|
|||
|
|
x a x i s = d i c t ( s c a l e a n c h o r = ' y ' , s c a l e r a t i o = 1 , g r i d c o l o r = ' # C E C E C E ' , z e r o l i n e = T r u e , z e r o l i n e c o l o r = ' # C E C E C E ' , t i c k f o n t = d i c t ( c o l o r = ' # 9 C 9 C 9 C ' , s i z e = 1 6 ) ) ,
|
|||
|
|
y a x i s = d i c t ( g r i d c o l o r = ' # C E C E C E ' , z e r o l i n e = T r u e , z e r o l i n e c o l o r = ' # C E C E C E ' , t i c k f o n t = d i c t ( c o l o r = ' # 9 C 9 C 9 C ' , s i z e = 1 6 ) ) ,
|
|||
|
|
h o v e r l a b e l = d i c t ( b g c o l o r = " # F F F F F F " , f o n t _ c o l o r = ' # 9 C 9 C 9 C ' , b o r d e r c o l o r = " # C E C E C E " , f o n t _ s i z e = 1 6 ) , p l o t _ b g c o l o r = ' # F F F F F F '
|
|||
|
|
)
|
|||
|
|
f i g 2 . u p d a t e _ x a x e s ( r a n g e = [ m i n _ x _ r a n g e , m a x _ x _ r a n g e ] )
|
|||
|
|
f i g 2 . u p d a t e _ y a x e s ( r a n g e = [ m i n _ y _ r a n g e , m a x _ y _ r a n g e ] )
|
|||
|
|
f i g 2 . s h o w ( )
|
|||
|
|
|
|||
|
|
i n t e r a c t ( p l o t , f o r m u l a = f o r m u l a , R A N G E _ F R O M = R A N G E _ F R O M _ S L I D E R , R A N G E _ T O = R A N G E _ T O _ S L I D E R , N = N _ s l i d e r , O _ 1 _ W _ O _ W _ 1 _ O = O _ R E D I L S _ 1 _ W _ O _ W _ 1 _ S L I D E R _ O , O _ 2 _ W _ O _ W _ 2 _ O = O _ R E D I L S _ 2 _ W _ O _ W _ 2 _ S L I D E R _ O ) ;
|