From c762050e7fc2c733210f8cd2b147e6084af0afe1 Mon Sep 17 00:00:00 2001
From: Clifford Wolf <clifford@clifford.at>
Date: Fri, 25 Jul 2014 14:23:10 +0200
Subject: Added RTLIL::SigSpec is_chunk()/as_chunk() API

---
 kernel/rtlil.cc | 17 +++++++++++++++++
 kernel/rtlil.h  |  3 +++
 2 files changed, 20 insertions(+)

diff --git a/kernel/rtlil.cc b/kernel/rtlil.cc
index 6bb395ec..83524d79 100644
--- a/kernel/rtlil.cc
+++ b/kernel/rtlil.cc
@@ -2007,6 +2007,14 @@ bool RTLIL::SigSpec::is_wire() const
 	return SIZE(chunks_) == 1 && chunks_[0].wire && chunks_[0].wire->width == width_;
 }
 
+bool RTLIL::SigSpec::is_chunk() const
+{
+	cover("kernel.rtlil.sigspec.is_chunk");
+
+	pack();
+	return SIZE(chunks_) == 1;
+}
+
 bool RTLIL::SigSpec::is_fully_const() const
 {
 	cover("kernel.rtlil.sigspec.is_fully_const");
@@ -2121,6 +2129,15 @@ RTLIL::Wire *RTLIL::SigSpec::as_wire() const
 	return chunks_[0].wire;
 }
 
+RTLIL::SigChunk RTLIL::SigSpec::as_chunk() const
+{
+	cover("kernel.rtlil.sigspec.as_chunk");
+
+	pack();
+	assert(is_chunk());
+	return chunks_[0];
+}
+
 bool RTLIL::SigSpec::match(std::string pattern) const
 {
 	cover("kernel.rtlil.sigspec.match");
diff --git a/kernel/rtlil.h b/kernel/rtlil.h
index a4b7e849..59db099f 100644
--- a/kernel/rtlil.h
+++ b/kernel/rtlil.h
@@ -577,6 +577,8 @@ public:
 	inline bool operator !=(const RTLIL::SigSpec &other) const { return !(*this == other); }
 
 	bool is_wire() const;
+	bool is_chunk() const;
+
 	bool is_fully_const() const;
 	bool is_fully_def() const;
 	bool is_fully_undef() const;
@@ -587,6 +589,7 @@ public:
 	std::string as_string() const;
 	RTLIL::Const as_const() const;
 	RTLIL::Wire *as_wire() const;
+	RTLIL::SigChunk as_chunk() const;
 
 	bool match(std::string pattern) const;
 
-- 
cgit v1.2.3