#!/usr/bin/env python
# -*- coding: utf-8 -*-
# 
# Copyright 2017 <+YOU OR YOUR COMPANY+>.
# 
# This 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.
# 
# This software 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 this software; see the file COPYING.  If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
# 

import numpy as np
from gnuradio import gr
import pmt,code


class pilot_quality(gr.sync_block):
    """
    docstring for block pilot_quality
    """
    def __init__(self, debug,samp_rate,fft_len,carrier_freq,gap_width,msg_adr):
        gr.sync_block.__init__(self,
            name="pilot_quality",
            in_sig=[(np.float32,fft_len)],
            out_sig=None)
        #self.carrier_width=1
        self.debug=debug
        self.message_port_register_out(pmt.intern('out'))
        self.carrier_index=int(carrier_freq*fft_len/float(samp_rate))
        self.lowbound_index=int((carrier_freq-gap_width)*fft_len/float(samp_rate))
        self.highbound_index=int((carrier_freq+gap_width)*fft_len/float(samp_rate))

    def work(self, input_items, output_items):
        in0 = input_items[0]
        # <+signal processing here+>
        for i,in_vec in enumerate(in0):
            surrounding=np.mean(in_vec[self.lowbound_index:self.highbound_index])
            carrier=np.mean(in_vec[self.carrier_index-1:self.carrier_index+1])
            #code.interact(local=locals())
            strength=abs(carrier-surrounding)
            if self.debug:
                print("strength: %f,carrier: %f, around:%f"%(strength,carrier,surrounding))
        return len(input_items[0])