/hSddlZddlZddlmZ ddlZdZedn#e$rdZedYnwxYwGddZe d kr d Z d Z ee Z e e ed ede e ddddede e dede e dede  ejdn #e$rYnwxYwe dSdS)N)DeadbandControllerTz:gpiod library loaded successfully - GPIO functions enabledFz5Warning: gpiod not available, GPIO functions disabledcteZdZdZddZdZdZdZdZddZ dd Z d Z d Z dd Z d ZddZdZdZdS)GPIOControllerz3Handles all GPIO operations for the CSEM TriggerboxNc|pd|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_t$|_d|_t+|_|jdd|jdd|dS) Nc&td|S)Nz[GPIO] printmsgs 0/home/morepork/Documents/CSEM/gpio_controller.pyz)GPIOController.__init__..su_s__?U?U F) rising_ms falling_ms)status_callback tx_status_pin frequency_pin gpio26_pinpps_pin gpio_state gpio26_state pps_state transmittingfrequency_output_thread gpio26_threadpps_monitoring_threadpps_monitoring gpio_chiptx_status_linefrequency_line gpio26_linepps_lineGPIO_AVAILABLEgpio_available pps_callbackrdeadband_controllerset_duty_deadbandsset_polarity_deadbands _init_gpio)selfrs r __init__zGPIOController.__init__s.V3U3U   !"'+$!%)"#"" ,!$6#7#7   33aA3NNN  77!PQ7RRR rc <|js|ddS tjd|_|j|j|_|jdtj |j d|j|j |_ |j dtj |j d|j|j |_|jdtj |j d|j|j|_|jd tj|d |jd |j d |j d |jdS#t$$r+}|d|dd|_Yd}~dSd}~wwxYw)zInitialize GPIO pinsz*GPIO not available - PPS will be simulatedN gpiochip4 tx_status)consumertyperfrequency_output gpio26_output pps_inputz+GPIO initialized successfully - TX Status: z , Frequency: z , GPIO26: z, PPS: zFailed to initialize GPIO: z - PPS will be simulatedF)r)rgpiodChipr#get_linerr$requestLINE_REQ_DIR_OUT set_valuerr%rr&rr'LINE_REQ_DIR_IN Exceptionr/es r r.zGPIOController._init_gpio8sP"   !M N N N F ("Z 44DN#'."9"9$:L"M"MD    ' '5CY ' Z Z Z   ) )! , , ,#'."9"9$:L"M"MD    ' '1C%J` ' a a a   ) )! , , , $~66tGGD    $ $oEDZ $ [ [ [   & &q ) ) )!N33DLAADM M ! !;U=R ! S S S  "utOa"u"uptqC"u"uOSO^"u"ugkgs"u"u v v v v v ( ( (  !Zq!Z!Z!Z [ [ ["'D        (sGG&& H0 HHc*|jrdS||_d|_tj|jd|_|j|jr&|jr| d|j dS| ddS)z Start monitoring PPS signal on GPIO 17 Args: pps_callback: Function to call with PPS state changes (True/False) NT)targetdaemonz$Started REAL PPS monitoring on GPIO z5Started SIMULATED PPS monitoring (GPIO not available)) r"r* threadingThread_pps_monitoring_loopr!startr)r'rr)r/r*s r start_pps_monitoringz#GPIOController.start_pps_monitoringZs    F("&/%5,& & & " "((***   [4= [  !V !V!V W W W W W  !Y Z Z Z Z Zrcd|_|jr4|jr|jd|ddS)zStop PPS monitoringF?timeoutzStopped PPS monitoringN)r"r!is_alivejoinrr/s r stop_pps_monitoringz"GPIOController.stop_pps_monitoringss`#  % 9$*D*M*M*O*O 9  & + +C + 8 8 8 566666rcXd}|jr|jr|jr t|j}||kr%||_|jr|||}n#t$r#}|d|Yd}~dSd}~wwxYwtj }t|dzdk}||krA||_|jr|||}|d|rdndtj d |jdSdS) z.PPS monitoring loop running in separate threadFzPPS monitoring error: NrzPPS (simulated): HIGHLOW{Gz?) r"r)r'bool get_valuerr*r@rtimeintsleep)r/last_pps_statecurrent_pps_staterB current_time simulated_ppss r rHz#GPIOController._pps_monitoring_loop|s! " ct} c (,T]-D-D-F-F(G(G%)N::):,A --.?@@@): (()E!)E)EFFFEEEEE $y{{ !$\!2!2Q!61 <  N22%2DN(9))-888%2N(()a}=_VVZ_)a)abbb Jt   9!     sAA,, B6BBc n |dkrtdd_j|r|dnd}|r|dnd |r|dnd fd} fd }jr}jrv jd |r |dd j d nQ#t$r"}d |Yd}~n*d}~wwxYw|r |ddtj j |dz|fd_jtj j||fd_jj}jrdnd}d|d|dzd|dd|ddd|ddd|ddd|ddd dS) z Start GPIO transmission at specified frequency Args: frequency: Transmission frequency in Hz indicator_callbacks: Dict with 'onoff', 'duty', 'polarity' callback functions rzFrequency must be positiveTonoffNdutypolaritycFrj|dSdSN)r+apply_duty_deadband)stateoriginal_duty_callbackr/s r duty_with_deadbandz=GPIOController.start_transmission..duty_with_deadbands8% \(<.polarity_with_deadbands8) d(@@Hbccccc d drTX Status pin z set HIGHzFailed to set TX status HIGH: zTX Status (simulated): HIGHrT)rDargsrE REAL GPIO SIMULATEDzStarted z. - Status pin HIGH, Frequency pin toggling at z Hz, GPIO26 toggling at z HzzDeadbands: Duty(duty_rising_ms.1f/duty_falling_msms), Polarity(polarity_rising_mspolarity_falling_msms)) ValueErrorrr+ reset_stategetr)r$r>rrr@rFrG_frequency_output_looprrI _gpio26_loopr get_deadband_info) r/ frequencyindicator_callbacksoriginal_onoff_callbackrjrnrB deadband_infotransmission_moderirms ` @@r start_transmissionz!GPIOController.start_transmissions >>9:: :   ,,...GZ"c"5"9"9'"B"B"B_cDW!a!4!8!8!@!@!@]aL_%i%8%<%OKKK _(9__iruviv__QZ___ ` ` ` b >N0ObbbVcduVvbbbKXYmKnbbbuBCXuYbbbb c c c c csAC%% D/D  Dcd|_|j|jr4|jr|jd|jr4|jr|jd|jr|jr |j d|rd|vr|dd| d|j dn/#t$r"}| d|Yd }~nd }~wwxYw|j r |j dd|_|rd |vr|d d| d |jdn/#t$r"}| d |Yd }~nd }~wwxYw|jr |j dd|_|rd |vr|d d| d|jdnV#t$r"}| d|Yd }~n/d }~wwxYwn&|r$|D]}|r |d|jrdnd}| d|dd S)zStop GPIO transmissionFrLrMrrbrpz set LOWzFailed to set TX status LOW: NrczFrequency pin z$Failed to set frequency output LOW: rdz GPIO26 pin zFailed to set GPIO26 LOW: rrrszStopped z - All GPIO pins set LOW)rr+r}rrOrPr r)r$r>rrr@r%rrr&rrvalues)r/rrBcallbackrs r stop_transmissionz GPIOController.stop_transmissions"  ,,...  ' ;D,H,Q,Q,S,S ;  ( - -c - : : :   1$"4"="="?"? 1   # #C # 0 0 0  " (" NN'11!444*>>(()Pt)P)P)PQQQQ KKK(()Ia)I)IJJJJJJJJK K# ( 3 : : < <((H( +/+>OKKK S(9SSSTTTTTsK'AC77 D#DD#.AF F1F,,F1r@rrZr\)r/rindicator_callback half_periodrBs r rzGPIOController._gpio26_loops$Io  $" :t'7 :,0,=(=D%$..D4E/Lqq1MMM)>**4+<=== (()E!)E)EFFFEEEEE )-(9$9!%:&&t'8999 J{ # # #! $ $ $ $ $AA B )BB cd|z }|jr|jr{|jrt |j |_|j|jrdnd|r||jnO#t $r#}|d|Yd}~dSd}~wwxYw|j |_|r||jtj||jdSdS)z0Frequency output loop running in separate threadrLrorzFrequency output error: N) rr)r%rr>r@rrZr\)r/toggle_frequencyrrrBs r rz%GPIOController._frequency_output_loop.s,,  $" 8t': 8*./&9DO'11t2M!!ANNN)<**4?;;; (()GA)G)GHHHEEEEE '+o"5%8&&t777 J{ # # #! $ $ $ $ $rc ||L|j}||n|d}||n|d}|j||||L|j}||n|d}||n|d}|j|||j}|d|ddd|ddd |ddd|ddd dS) a Configure deadband timing for GPIO signals Args: duty_rising_ms: Duty signal rising edge deadband (milliseconds) duty_falling_ms: Duty signal falling edge deadband (milliseconds) polarity_rising_ms: Polarity signal rising edge deadband (milliseconds) polarity_falling_ms: Polarity signal falling edge deadband (milliseconds) NrtrwryrzzDeadbands updated: Duty(rurvrxr{)r+rr,r-r) r/rtrwryrzcurrentrisingfallingrs r configure_deadbandsz"GPIOController.configure_deadbandsDs  %)D.@@BBG'5'A^^wO_G`F)8)Doo'RcJdG  $ 7 7 H H H  )-@-L.@@BBG+=+I''wWkOlF-@-L))RYZoRpG  $ ; ;FG L L L0BBDD  j FV8Wjjj^kl}^~jjjS`auSvjjj}JK`}ajjjj k k k k krc4|jS)z"Get current deadband configuration)r+rrQs r get_deadband_statusz"GPIOController.get_deadband_status^s'99;;;rstandardc |dkr|jddddnn|dkr|jddd d nI|d kr|jddddn$|d kr|jd d d d |j}|d |d|ddd|ddd|ddd|ddd dS)z'Set deadbands using predefined profilescsem_optimizedgMb?g~jth?g~jt?{Gzt?) duty_rising duty_fallingpolarity_risingpolarity_falling fast_responseMbP?gMb`?r no_deadbandgz Profile 'z' loaded: Duty(rtrurvrwrxryrzr{N)r+rrr)r/profilers r set_custom_deadbandsz#GPIOController.set_custom_deadbandsbs & & &  $ 8 8!" %!& 9      ' '  $ 8 8!" %!& 9      " "  $ 8 8!" %!& 9      % %  $ 8 8 #!$ 9   0BBDD  sssO_A`sssgtvGhHsss\ij~\sssFSTiFjssss t t t t trc&d|z }|dkrd}n |dkrd}nd}|dz}t||}|j|dz|dz|j|dz|dz|d |d |dzd d d S)z>Automatically adjust deadbands based on transmission frequencyg? rrorrWg?izAuto-adjusted deadbands for z Hz: rumsN)minr+r,r-r)r/rperiod base_deadband max_deadbandactual_deadbands r adjust_deadbands_for_frequencyz-GPIOController.adjust_deadbands_for_frequencysy r>>!MM ]]!MM!M| m\::  33Od4JO^bLbccc  77$8NP_bfPfggg jIjjO^bLbjjjjkkkkkrcHd|_||j|jr|jrd |jd|jn/#t$r"}| d|Yd}~nd}~wwxYw|j rd |j d|j n/#t$r"}| d|Yd}~nd}~wwxYw|j rd |j d|j n/#t$r"}| d|Yd}~nd}~wwxYw|j rJ |j n/#t$r"}| d|Yd}~nd}~wwxYw|j rL |j dS#t$r#}| d|Yd}~dSd}~wwxYwdS) zClean up GPIO resourcesFrzTX status cleanup error: Nz Frequency output cleanup error: zGPIO26 cleanup error: zPPS cleanup error: zChip close error: )rrRr+r}r)r$r>releaser@rr%r&r'r#closerAs r cleanupzGPIOController.cleanups!   """  ,,...   D" JJ'11!444'//1111 JJJ(()HQ)H)HIIIIIIIIJ" QQ'11!444'//1111 QQQ(()OA)O)OPPPPPPPPQ GG$..q111$,,.... GGG(()E!)E)EFFFFFFFFG} DDM))++++ DDD(()Bq)B)BCCCCCCCCD > ? ?$$&&&&& ? ? ?$$%=!%=%=>>>>>>>>> ? ? ?sx3A99 B%B  B%03C$$ D.D  D3E E;E66E;F G *GG G22 H<HHrf)NNNN)r)__name__ __module__ __qualname____doc__r0r.rJrRrHrrrrrrrrrrr rr s/==&&&&P ( ( (D[[[2777   DAcAcAcAcF5U5U5U5Un$$$,$$$,HLHLkkkk4<<< t t t tDlll**?*?*?*?*?rr__main__c*td|dS)Nz[TEST] rr s r test_callbackrs ooorc2td|rdnddS)Nz[PPS] rUrVr)rhs r pps_test_callbackrs( 31vvE3344444r)rz Testing deadband configuration:zCurrent deadbands: r)rtrwryrzzAfter custom config:rzAfter fast profile:rLz"After frequency adaptive (0.5 Hz):)rFrZ deadbandsrr9r(r ImportErrorrrrrgpiorJrrrrr\KeyboardInterruptrrrr rsD ((((((CLLLN E FGGGGCCCN E ABBBBBCv?v?v?v?v?v?v?v?t  z555 >- 8 8 8D /000 E -... E  8 8 : :;;; AqUWmnooo E $":":"<"<=== o... E !9!9!;!;<<< '',,, E .0H0H0J0JKKK  1       LLNNNNNGs"77<E EE