Skip to content

Commit

Permalink
add moving average
Browse files Browse the repository at this point in the history
  • Loading branch information
gordinmitya committed Apr 27, 2020
1 parent 95b7880 commit 037f7fb
Show file tree
Hide file tree
Showing 2 changed files with 36 additions and 2 deletions.
9 changes: 7 additions & 2 deletions app/src/main/java/ru/gordinmitya/yuv2buf_demo/MainActivity.kt
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,12 @@ import java.util.concurrent.ExecutorService
import java.util.concurrent.Executors

class MainActivity : AppCompatActivity(), CompositeConverter.Listener {
val movingAverageSize = 64

private lateinit var analysisExecutor: ExecutorService
private lateinit var cameraProviderFuture: ListenableFuture<ProcessCameraProvider>
private lateinit var imageAnalyzer: CompositeConverter
private lateinit var resultAverages: Array<MovingAverage>
private lateinit var resultViews: Array<View>

override fun onCreate(savedInstanceState: Bundle?) {
Expand All @@ -36,12 +38,13 @@ class MainActivity : AppCompatActivity(), CompositeConverter.Listener {
initCamera()
else
requestPermissions()

analysisExecutor = Executors.newSingleThreadExecutor()
val converters = arrayOf(
OpenCVConverter(),
RenderScriptConverter(this)
)
resultAverages = Array(converters.size) { MovingAverage(movingAverageSize) }
resultViews = Array(converters.size) {
return@Array layoutInflater.inflate(R.layout.item_converted, list_results, false)
}
Expand All @@ -61,10 +64,12 @@ class MainActivity : AppCompatActivity(), CompositeConverter.Listener {
override fun onAnalyzed(results: List<ConversionResult>) {
if (!lifecycle.currentState.isAtLeast(Lifecycle.State.STARTED)) return
results.forEachIndexed { index, result ->
val average = resultAverages[index]
average.add(result.time)
resultViews[index].let {
it.image.setImageBitmap(result.image)
it.text_name.text = result.method
it.text_time.text = "${result.time}ms"
it.text_time.text = "${result.time}ms\n$average"
}
}
}
Expand Down
29 changes: 29 additions & 0 deletions app/src/main/java/ru/gordinmitya/yuv2buf_demo/MovingAverage.kt
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
package ru.gordinmitya.yuv2buf_demo

import kotlin.math.min

class MovingAverage(val capacity: Int) {
private var pointer: Int = 0
private val array: LongArray = LongArray(capacity)

fun add(time: Long) {
array[pointer % capacity] = time
pointer += 1
}

fun avg(): Double {
if (pointer == 0) return 0.0
var sum = 0.0
for (l in 0 until size())
sum += array[l]
return sum / size()
}

fun size(): Int {
return min(pointer, capacity)
}

override fun toString(): String {
return String.format("avg of %d = %.2fms", size(), avg())
}
}

0 comments on commit 037f7fb

Please sign in to comment.