Added segmenter

Author: djthorpe

pkg/whisper/segmenter/segmenter.go

@@ -43,8 +43,8 @@ func NewSegmenter(r io.Reader, dur time.Duration, sample_rate int) (*Segmenter,
 
 	// Sample buffer is duration * sample rate
 	if dur > 0 {
-		segmenter.n = int(dur.Seconds()) * sample_rate
-		segmenter.buf = make([]float32, 0, int(dur.Seconds())*sample_rate)
+		segmenter.n = int(dur.Seconds() * float64(sample_rate))
+		segmenter.buf = make([]float32, 0, segmenter.n)
 	}
 
 	// Open the file
@@ -106,10 +106,10 @@ func (s *Segmenter) Decode(ctx context.Context, fn SegmentFunc) error {
 		// n != 0 and len(buf) >= n we have a segment to process
 		if s.n != 0 && len(s.buf) >= s.n {
 			fn(s.ts, s.buf)
+			// Increment the timestamp
+			s.ts += time.Duration(len(s.buf)) * time.Second / time.Duration(s.sample_rate)
 			// Clear the buffer
 			s.buf = s.buf[:0]
-			// Increment the timestamp
-			s.ts += time.Duration(float64(s.n)/float64(s.sample_rate)) * time.Second
 		}
 
 		// Continue processing

pkg/whisper/segmenter/segmenter_test.go

@@ -20,7 +20,7 @@ func Test_segmenter_001(t *testing.T) {
 	if !assert.NoError(err) {
 		t.SkipNow()
 	}
-	segmenter, err := segmenter.NewSegmenter(f, time.Second, 16000)
+	segmenter, err := segmenter.NewSegmenter(f, 200*time.Millisecond, 16000)
 	if !assert.NoError(err) {
 		t.SkipNow()
 	}