aboutsummaryrefslogtreecommitdiffstats
path: root/src/gr-fadingui/python
diff options
context:
space:
mode:
authorNao Pross <np@0hm.ch>2021-12-11 16:05:41 +0100
committerNao Pross <np@0hm.ch>2021-12-11 16:05:41 +0100
commit7381a5dcf266c532a490192ed229ac002d35f8b1 (patch)
tree3567a9bd3cada6fde773cca1472745798da68ac5 /src/gr-fadingui/python
parentDelete old stuff (diff)
downloadFading-7381a5dcf266c532a490192ed229ac002d35f8b1.tar.gz
Fading-7381a5dcf266c532a490192ed229ac002d35f8b1.zip
Create new QPSK simulation
Diffstat (limited to 'src/gr-fadingui/python')
-rw-r--r--src/gr-fadingui/python/logger.py2
-rw-r--r--src/gr-fadingui/python/phasecorrection.py34
2 files changed, 24 insertions, 12 deletions
diff --git a/src/gr-fadingui/python/logger.py b/src/gr-fadingui/python/logger.py
index a189aeb..a101b9e 100644
--- a/src/gr-fadingui/python/logger.py
+++ b/src/gr-fadingui/python/logger.py
@@ -10,5 +10,5 @@ def get_logger(module):
log.addHandler(stdout_handler)
stdout_handler.setFormatter(formatter)
- log.setLevel(logging.DEBUG)
+ log.setLevel(logging.WARNING)
return log
diff --git a/src/gr-fadingui/python/phasecorrection.py b/src/gr-fadingui/python/phasecorrection.py
index 75ca280..cd3fa37 100644
--- a/src/gr-fadingui/python/phasecorrection.py
+++ b/src/gr-fadingui/python/phasecorrection.py
@@ -20,7 +20,7 @@ class phasecorrection(gr.sync_block):
The correlation peak tags are NOT propagated, and instead replaced with a
frame_start tag.
"""
- def __init__(self):
+ def __init__(self, frame_len):
gr.sync_block.__init__(
self,
name='Phase and Frequency Correction',
@@ -31,6 +31,9 @@ class phasecorrection(gr.sync_block):
# tags should not be propagated, we then output our own tags
self.set_tag_propagation_policy(gr.TPP_DONT)
+ # save frame length
+ self.frame_len = frame_len
+
# because we do block processing, we need to keep track of the last tag
# of the previous block to correct the first values of the next block
self.last = None
@@ -54,6 +57,9 @@ class phasecorrection(gr.sync_block):
# compute number of samples between tags
nsamples = end.offset - start.offset
+ # debugging, see last lines of self.work()
+ self.lastnsamples = nsamples
+
# unpack pmt values into start and end phase
sphase = pmt.to_python(start.value)
ephase = pmt.to_python(end.value)
@@ -75,6 +81,7 @@ class phasecorrection(gr.sync_block):
def work(self, input_items, output_items):
counter = self.nitems_written(0)
+ log.debug(f"Counter is at {counter} (nitems_written)")
# nicer aliases
inp = input_items[0]
@@ -82,7 +89,8 @@ class phasecorrection(gr.sync_block):
# read phase tags
is_phase = lambda tag: pmt.to_python(tag.key) == "phase_est"
- tags = list(filter(is_phase, self.get_tags_in_window(0, 0, len(inp))))
+ tags = list(filter(is_phase, self.get_tags_in_range( \
+ 0, counter, counter + len(inp))))
if not tags:
log.warning(f"There were no tags in {len(inp)} samples!")
@@ -113,11 +121,11 @@ class phasecorrection(gr.sync_block):
if self.last and nfront else np.zeros(nfront)
# debugging
- if self.lastnback + self.nfront != self.lastnsamples:
- log.warn("Something went wrong during block processing!\n" \
+ if self.lastnback + nfront != self.lastnsamples:
+ log.warn("Something went wrong during block processing! " \
f"Last block finished with nback = {self.lastnback} samples, " \
- f"current block starts with nstart = {self.nstart}, but their sum" \
- f"is not {self.latnsamples} = size of last chunk")
+ f"current block starts with nfront = {nfront}, but their sum " \
+ f"is not {self.lastnsamples} = size of last chunk")
self.lastnback = nback
# compute correction
@@ -125,13 +133,17 @@ class phasecorrection(gr.sync_block):
length = len(correction)
# write outputs
- out[:length] = inp[:length] * correction
+ out[:] = inp * correction
# save last tag for next call
self.last = tags[-1]
- # add tags
- for tag in tags:
- self.add_item_tag(0, tag.offset, pmt.intern("frame_start"), pmt.PMT_T)
+ # add frame_start tags
+ is_start = lambda tag: pmt.to_python(tag.key) == "corr_start"
+ start_tags = list(filter(is_start, \
+ self.get_tags_in_range(0, counter, counter + len(inp))))
+ for tag in start_tags:
+ self.add_item_tag(0, tag.offset +1, pmt.intern("frame_start"), \
+ pmt.from_long(self.frame_len))
- return len(out) \ No newline at end of file
+ return len(out)