From f69708ec3ca83d019a50396a5947abb24bee15d3 Mon Sep 17 00:00:00 2001
From: Michael Dickens
Date: Tue, 14 Feb 2012 11:08:53 -0800
Subject: uhd: fix source block xml file generation

---
 gr-uhd/grc/gen_uhd_usrp_blocks.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'gr-uhd/grc')

diff --git a/gr-uhd/grc/gen_uhd_usrp_blocks.py b/gr-uhd/grc/gen_uhd_usrp_blocks.py
index 67343541c..25b0a4c46 100644
--- a/gr-uhd/grc/gen_uhd_usrp_blocks.py
+++ b/gr-uhd/grc/gen_uhd_usrp_blocks.py
@@ -411,10 +411,10 @@ max_num_channels = max_num_mboards*4
 if __name__ == '__main__':
 	import sys
 	for file in sys.argv[1:]:
-		if 'source' in file:
+		if file.endswith ('source.xml'):
 			sourk = 'source'
 			direction = 'out'
-		elif 'sink' in file:
+		elif file.endswith ('sink.xml'):
 			sourk = 'sink'
 			direction = 'in'
 		else: raise Exception, 'is %s a source or sink?'%file
-- 
cgit