// Copyright 2014 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import 'package:flutter/animation.dart';
import 'package:flutter/foundation.dart';
import 'package:flutter_test/flutter_test.dart';

import '../common.dart';

const int _kNumIterationsList = 2 << 10;
const int _kNumIterationsHashed = 2 << 15;
const List<int> callbackCounts = <int>[1, 10, 100, 500];

class TestAnimationController extends AnimationController {
  TestAnimationController() : super(vsync: const TestVSync());

  @override
  void notifyListeners() => super.notifyListeners();
}

Future<void> execute() async {
  assert(false, "Don't run benchmarks in debug mode! Use 'flutter run --release'.");

  final printer = BenchmarkResultPrinter();

  void runNotifyListenersLoopWithObserverList(
    int totalIterations, {
    bool failRemoval = false,
    bool addResult = true,
  }) {
    final suffix = failRemoval ? 'removalFail' : 'removalSuccess';
    final name = 'notifyListeners:ObserverList:$suffix';

    void miss() {}

    for (final int callbackCount in callbackCounts) {
      final int iterations = totalIterations ~/ callbackCount;

      final observerList = ObserverList<VoidCallback>();
      for (var i = 0; i < callbackCount; ++i) {
        observerList.add(switch (failRemoval) {
          false => () {
            final VoidCallback first = (observerList.iterator..moveNext()).current;

            observerList.remove(first);
            observerList.add(first);
          },
          true => () => observerList.remove(miss),
        });
      }

      final watch = Stopwatch()..start();

      for (var i = 0; i < iterations; ++i) {
        final List<VoidCallback> list = observerList.toList(growable: false);
        for (final cb in list) {
          if (observerList.contains(cb)) {
            cb();
          }
        }
      }

      watch.stop();

      if (addResult) {
        printer.addResult(
          description: '$name ($callbackCount callbacks)',
          value: watch.elapsedMicroseconds / iterations,
          unit: 'µs per iteration',
          name: '$name$callbackCount',
        );
      }
    }
  }

  void runNotifyListenersLoopWithHashedObserverList(int totalIterations, {bool addResult = true}) {
    const name = 'notifyListeners:HashedObserverList';

    for (final int callbackCount in callbackCounts) {
      final int iterations = totalIterations ~/ callbackCount;

      final observerList = HashedObserverList<VoidCallback>();
      for (var i = 0; i < callbackCount; ++i) {
        observerList.add(() {
          final VoidCallback first = (observerList.iterator..moveNext()).current;

          observerList.remove(first);
          observerList.add(first);
        });
      }

      final watch = Stopwatch()..start();

      for (var i = 0; i < iterations; ++i) {
        final List<VoidCallback> list = observerList.toList(growable: false);
        for (final cb in list) {
          if (observerList.contains(cb)) {
            cb();
          }
        }
      }

      watch.stop();

      if (addResult) {
        printer.addResult(
          description: '$name ($callbackCount callbacks)',
          value: watch.elapsedMicroseconds / iterations,
          unit: 'µs per iteration',
          name: '$name$callbackCount',
        );
      }
    }
  }

  void runNotifyListenersLoopWithAnimationController(int totalIterations, {bool addResult = true}) {
    const name = 'notifyListeners:AnimationController';

    for (final int callbackCount in callbackCounts) {
      final int iterations = totalIterations ~/ callbackCount;

      final controller = TestAnimationController();
      for (var i = 0; i < callbackCount; ++i) {
        late final VoidCallback cb;
        cb = () {
          controller.removeListener(cb);
          controller.addListener(cb);
        };
        controller.addListener(cb);
      }

      final watch = Stopwatch()..start();

      for (var i = 0; i < iterations; ++i) {
        controller.notifyListeners();
      }

      watch.stop();

      if (addResult) {
        printer.addResult(
          description: '$name ($callbackCount callbacks)',
          value: watch.elapsedMicroseconds / iterations,
          unit: 'µs per iteration',
          name: '$name$callbackCount',
        );
      }
    }
  }

  runNotifyListenersLoopWithObserverList(_kNumIterationsList);
  runNotifyListenersLoopWithObserverList(_kNumIterationsList, failRemoval: true);
  runNotifyListenersLoopWithHashedObserverList(_kNumIterationsHashed);
  runNotifyListenersLoopWithAnimationController(_kNumIterationsHashed);

  printer.printToStdout();
}