X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=type_logic.c;h=28c36707b9a789d9bf4dcf5191960661e9079163;hb=f38ec2855ce41154bc1035dd7f1ab9a21f411f0d;hp=2f7923425b08bae786db5e106bdc153864ae1584;hpb=bc5f5a434e148db1843be7b02b95a35d21237702;p=libsigrokdecode.git

diff --git a/type_logic.c b/type_logic.c
index 2f79234..28c3670 100644
--- a/type_logic.c
+++ b/type_logic.c
@@ -38,34 +38,29 @@ PyObject *srd_logic_iternext(PyObject *self)
 	unsigned char probe_samples[SRD_MAX_NUM_PROBES];
 
 	logic = (srd_logic *) self;
-	if (logic->itercnt >= logic->inbuflen / logic->di->unitsize) {
+	if (logic->itercnt >= logic->inbuflen / logic->di->data_unitsize) {
 		/* End iteration loop. */
 		return NULL;
 	}
 
-	/* TODO: use number of probes defined in the PD, in the order the PD
-	 * defined them -- not whatever came in from the driver.
-	 */
 	/* Convert the bit-packed sample to an array of bytes, with only 0x01
 	 * and 0x00 values, so the PD doesn't need to do any bitshifting.
 	 */
-	memcpy(&sample, logic->inbuf + logic->itercnt * logic->di->unitsize,
-			logic->di->unitsize);
-	for (i = 0; i < logic->di->num_probes; i++) {
-		probe_samples[i] = sample & 0x01;
+	memcpy(&sample, logic->inbuf + logic->itercnt * logic->di->data_unitsize,
+			logic->di->data_unitsize);
+	for (i = 0; i < logic->di->dec_num_probes; i++) {
+		probe_samples[logic->di->dec_probemap[i]] = sample & 0x01;
 		sample >>= 1;
 	}
 
-	/* TODO: samplenum should be in the inbuf feed, instead of time/duration.
-	 * fake it for now...
-	 */
 	/* Prepare the next samplenum/sample list in this iteration. */
-	py_samplenum = PyLong_FromUnsignedLongLong(logic->itercnt++);
+	py_samplenum = PyLong_FromUnsignedLongLong(logic->start_samplenum + logic->itercnt);
 	PyList_SetItem(logic->sample, 0, py_samplenum);
 	py_samples = PyBytes_FromStringAndSize((const char *)probe_samples,
-			logic->di->num_probes);
+			logic->di->dec_num_probes);
 	PyList_SetItem(logic->sample, 1, py_samples);
 	Py_INCREF(logic->sample);
+	logic->itercnt++;
 
 	return logic->sample;
 }