X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/blast.git/blobdiff_plain/cfe8322b50c9ef08ffbc3e52b0849bca4cd1d0bf..9ea814d76cc52e24b80be4a6ea78cca24a9a4915:/AbstractBlock.cpp?ds=sidebyside

diff --git a/AbstractBlock.cpp b/AbstractBlock.cpp
index c71b2e0..66c8f37 100644
--- a/AbstractBlock.cpp
+++ b/AbstractBlock.cpp
@@ -51,11 +51,12 @@ bool AbstractBlock::isTopGroupBlock() {
 bool AbstractBlock::isSourceBlock() {
   return false;
 }
-
+/* NB: a generator is a block that has no data inputs
+ * and has at least one data output.
+ */
 bool AbstractBlock::isGeneratorBlock() {
-  foreach(AbstractInterface* iface, inputs) {
-    if (iface->getPurpose() == AbstractInterface::Data) return false;
-  }
+  if (getDataInputs().size() > 0) return false;
+  
   return true;
 }
 
@@ -126,42 +127,58 @@ void AbstractBlock::defineBlockParam(BlockParameter *param)
   param->setValue(value);  
 }
 
-QList<AbstractInterface *> AbstractBlock::getInterfaces() {
+QList<AbstractInterface *> AbstractBlock::getInterfaces(int direction, int purpose) {
   QList<AbstractInterface *> list;
-  list.append(inputs);
-  list.append(outputs);
-  list.append(bidirs);
+  bool selIn = false;
+  bool selOut = false;
+  bool selInOut = false;
+  
+  if (direction == AbstractInterface::AnyDirection) {
+    selIn = true;
+    selOut = true;
+    selInOut = true;
+  }
+  else if (direction == AbstractInterface::Input) {
+    selIn = true;    
+  }
+  else if (direction == AbstractInterface::Output) {
+    selOut = true;    
+  }
+  else if (direction == AbstractInterface::InOut) {
+    selInOut = true;    
+  }
+  if (selIn) {
+    foreach(AbstractInterface* iface, inputs) {
+      if ((iface->getPurpose() == purpose) || (purpose == AbstractInterface::AnyPurpose)) list.append(iface);      
+    }
+  }
+  if (selOut) {
+    foreach(AbstractInterface* iface, outputs) {
+      if ((iface->getPurpose() == purpose) || (purpose == AbstractInterface::AnyPurpose)) list.append(iface);      
+    }
+  }
+  if (selInOut) {
+    foreach(AbstractInterface* iface, bidirs) {
+      if ((iface->getPurpose() == purpose) || (purpose == AbstractInterface::AnyPurpose)) list.append(iface);      
+    }
+  }
   return list;
 }
 
 QList<AbstractInterface *> AbstractBlock::getDataInputs() {
-  QList<AbstractInterface *> list;
-  foreach(AbstractInterface* iface, inputs) {
-    if (iface->getPurpose() == AbstractInterface::Data) {
-      list.append(iface);
-    }
-  }
-  return list;
+  return getInterfaces(AbstractInterface::Input, AbstractInterface::Data);  
+}
+
+QList<AbstractInterface *> AbstractBlock::getDataOutputs() {
+  return getInterfaces(AbstractInterface::Output, AbstractInterface::Data);  
 }
 
 QList<AbstractInterface *> AbstractBlock::getControlInputs() {
-  QList<AbstractInterface *> list;
-  foreach(AbstractInterface* iface, inputs) {
-    if (iface->getPurpose() == AbstractInterface::Control) {
-      list.append(iface);
-    }
-  }
-  return list;
+  return getInterfaces(AbstractInterface::Input, AbstractInterface::Control);  
 }
 
 QList<AbstractInterface *> AbstractBlock::getControlOutputs() {
-  QList<AbstractInterface *> list;
-  foreach(AbstractInterface* iface, outputs) {
-    if (iface->getPurpose() == AbstractInterface::Control) {
-      list.append(iface);
-    }
-  }
-  return list;
+  return getInterfaces(AbstractInterface::Output, AbstractInterface::Control);    
 }
 
 AbstractInterface* AbstractBlock::getIfaceFromName(QString name) {