#!/usr/bin/env python
#
# Copyright 2004,2010,2012,2013 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# GNU Radio is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
#
# GNU Radio is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with GNU Radio; see the file COPYING.  If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
#

import math

from gnuradio import gr, gr_unittest, analog, blocks

class test_pll_refout(gr_unittest.TestCase):

    def setUp(self):
        self.tb = gr.top_block()

    def tearDown(self):
        self.tb = None

    def test_pll_refout(self):
        expected_result = ((1+0j),
                            (1+6.408735764296125e-10j),
                            (0.9999844431877136+0.005577784031629562j),
                            (0.9998642802238464+0.016474783420562744j),
                            (0.9994739890098572+0.032431427389383316j),
                            (0.9985847473144531+0.05318402871489525j),
                            (0.996917188167572+0.07846084982156754j),
                            (0.9941533207893372+0.10797744989395142j),
                            (0.9899479150772095+0.14143235981464386j),
                            (0.9839394092559814+0.1785029172897339j),
                            (0.9757603406906128+0.2188417762517929j),
                            (0.9650475978851318+0.26207470893859863j),
                            (0.9514514803886414+0.30779871344566345j),
                            (0.9346449971199036+0.35558223724365234j),
                            (0.9143316149711609+0.40496626496315j),
                            (0.8902531862258911+0.4554659426212311j),
                            (0.8621962666511536+0.5065743923187256j),
                            (0.8299974799156189+0.5577671527862549j),
                            (0.7935484647750854+0.6085070967674255j),
                            (0.7527987360954285+0.6582507491111755j),
                            (0.7077582478523254+0.7064547538757324j),
                            (0.6584978699684143+0.7525825500488281j),
                            (0.6051493883132935+0.7961119413375854j),
                            (0.547903835773468+0.8365413546562195j),
                            (0.48700881004333496+0.8733970522880554j),
                            (0.42276495695114136+0.90623939037323j),
                            (0.35552138090133667+0.9346681237220764j),
                            (0.2856702208518982+0.9583280086517334j),
                            (0.21364101767539978+0.976912260055542j),
                            (0.13989387452602386+0.9901664853096008j),
                            (0.06491273641586304+0.9978909492492676j),
                            (-0.01080091018229723+0.9999416470527649j),
                            (-0.08673560619354248+0.9962313771247864j),
                            (-0.16237612068653107+0.9867289662361145j),
                            (-0.23721040785312653+0.9714583158493042j),
                            (-0.3107353150844574+0.95049649477005j),
                            (-0.3824624717235565+0.9239710569381714j),
                            (-0.45192304253578186+0.892056941986084j),
                            (-0.5186731219291687+0.8549726009368896j),
                            (-0.5822963714599609+0.812976598739624j),
                            (-0.6424083709716797+0.7663624882698059j),
                            (-0.6986585855484009+0.7154552340507507j),
                            (-0.7507330775260925+0.6606056690216064j),
                            (-0.7983550429344177+0.6021870970726013j),
                            (-0.841286301612854+0.5405898094177246j),
                            (-0.879327654838562+0.47621726989746094j),
                            (-0.912318229675293+0.4094819128513336j),
                            (-0.9401354789733887+0.340800940990448j),
                            (-0.9626938104629517+0.27059316635131836j),
                            (-0.979943573474884+0.1992751508951187j),
                            (-0.9918696284294128+0.12725839018821716j),
                            (-0.9984893202781677+0.054946307092905045j),
                            (-0.9998509287834167-0.017267409712076187j),
                            (-0.9960314631462097-0.08900183439254761j),
                            (-0.9871346950531006-0.1598907858133316j),
                            (-0.9732890129089355-0.2295832633972168j),
                            (-0.9546451568603516-0.29774588346481323j),
                            (-0.9313743710517883-0.3640628457069397j),
                            (-0.9036663174629211-0.42823725938796997j),
                            (-0.8717266321182251-0.48999255895614624j),
                            (-0.8357754945755005-0.5490713119506836j),
                            (-0.7960456013679504-0.6052366495132446j),
                            (-0.7527803182601929-0.658271849155426j),
                            (-0.706232488155365-0.7079799771308899j),
                            (-0.6566619873046875-0.7541850209236145j),
                            (-0.6043350696563721-0.7967302799224854j),
                            (-0.5495226979255676-0.8354787826538086j),
                            (-0.4924990236759186-0.8703129887580872j),
                            (-0.4335414469242096-0.9011335968971252j),
                            (-0.3729270100593567-0.927860677242279j),
                            (-0.3109343349933624-0.9504314064979553j),
                            (-0.2478405237197876-0.9688008427619934j),
                            (-0.18392162024974823-0.9829409122467041j),
                            (-0.11945075541734695-0.9928401112556458j),
                            (-0.05469784513115883-0.9985029697418213j),
                            (0.010069688782095909-0.9999492764472961j),
                            (0.07459097355604172-0.9972141981124878j),
                            (0.13860897719860077-0.9903472065925598j),
                            (0.2018725872039795-0.979411780834198j),
                            (0.2641367018222809-0.964485228061676j),
                            (0.32516375184059143-0.9456577301025391j),
                            (0.3847236633300781-0.9230318069458008j),
                            (0.44259318709373474-0.8967224955558777j),
                            (0.49855801463127136-0.8668563365936279j),
                            (0.5524120926856995-0.8335711359977722j),
                            (0.6039596796035767-0.7970148921012878j),
                            (0.6530137062072754-0.7573460936546326j),
                            (0.6993972063064575-0.7147331833839417j),
                            (0.7429447770118713-0.6693527102470398j),
                            (0.7835012078285217-0.6213902235031128j),
                            (0.8209227919578552-0.5710391998291016j),
                            (0.8550769090652466-0.5185011625289917j),
                            (0.8858439326286316-0.46398329734802246j),
                            (0.9131162166595459-0.4076994061470032j),
                            (0.936798632144928-0.3498689830303192j),
                            (0.956809401512146-0.2907160222530365j),
                            (0.9730796813964844-0.23046888411045074j),
                            (0.9855544567108154-0.16935895383358002j),
                            (0.9941920042037964-0.10762103646993637j),
                            (0.9989647269248962-0.045491550117731094j)) 

        sampling_freq = 10e3
        freq = sampling_freq / 100

        loop_bw = math.pi/100.0
        maxf = 1
        minf = -1

        src = analog.sig_source_c(sampling_freq, analog.GR_COS_WAVE, freq, 1.0)
        pll = analog.pll_refout_cc(loop_bw, maxf, minf)
        head = blocks.head(gr.sizeof_gr_complex, int (freq))
        dst = blocks.vector_sink_c()

        self.tb.connect(src, pll, head)
        self.tb.connect(head, dst)

        self.tb.run()
        dst_data = dst.data()
        self.assertComplexTuplesAlmostEqual(expected_result, dst_data, 4)

if __name__ == '__main__':
    gr_unittest.run(test_pll_refout, "test_pll_refout.xml")