save

Author: rainyl

Diff for: .gitmodules

@@ -0,0 +1,3 @@
+[submodule "src"]
+	path = src
+	url = https://github.com/rainyl/dartcv.git

Diff for: example/dart_cli/pubspec.yaml

@@ -16,3 +16,10 @@ dependencies:
 dev_dependencies:
   lints: ^3.0.0
   test: ^1.24.0
+
+dependency_overrides:
+  native_assets_cli:
+    git:
+      url: https://github.com/dart-lang/native.git
+      path: pkgs/native_assets_cli
+      ref: 3fbbf07

Diff for: hook/build.dart

@@ -1,8 +1,5 @@
-// Copyright (c) 2025, the Dart project authors.  Please see the AUTHORS file
-// for details. All rights reserved. Use of this source code is governed by a
-// BSD-style license that can be found in the LICENSE file.
-
-import 'dart:io';
+// Copyright (c) 2025, Rainyl. All rights reserved. Use of this source code is governed by a
+// Apache 2.0 license that can be found in the LICENSE file.
 
 import 'package:dartcv4/src/hook_helpers/run_build.dart';
 import 'package:native_assets_cli/native_assets_cli.dart';

Diff for: lib/src/hook_helpers/parse_user_defines.dart

@@ -0,0 +1,15 @@
+import 'dart:io';
+
+import 'package:logging/logging.dart';
+import 'package:yaml/yaml.dart';
+
+Future<Map<String, Object?>> parseUserDefines(File yamlFile, String pkgName, {Logger? logger}) async {
+  final yaml = loadYaml(await yamlFile.readAsString()) as YamlMap;
+  try {
+    final m = yaml['hooks']['user_defines'][pkgName] as YamlMap;
+    return Map<String, Object?>.from(m);
+  } catch (e) {
+    logger?.warning('No user defines found for $pkgName, error: $e\n');
+    return {};
+  }
+}

Diff for: lib/src/hook_helpers/run_build.dart

@@ -6,21 +6,61 @@ import 'dart:io';
 import 'package:logging/logging.dart';
 import 'package:native_assets_cli/code_assets_builder.dart';
 import 'package:native_toolchain_cmake/native_toolchain_cmake.dart';
+import 'package:yaml/yaml.dart';
+
+import 'parse_user_defines.dart';
 
 Future<void> runBuild(BuildInput input, BuildOutputBuilder output, {List<String>? optionalModules}) async {
-  final packagePath = Directory(await getPackagePath('opencv_dart'));
+  final packagePath = Directory(await getPackagePath('dartcv4'));
   final logger = Logger('')
     ..level = Level.ALL
-    ..onRecord.listen((record) => print(record.message));
+    // ..onRecord.listen((record) => print(record.message));
+    ..onRecord.listen((record) => stderr.write(record.message));
+
+  // final userDefines = input.userDefines;
+  final userDefines = await parseUserDefines(
+    File.fromUri(packagePath.uri.resolve('pubspec.yaml')),
+    'dartcv4',
+    logger: logger,
+  );
+  logger.info(userDefines);
+  // {include_modules: [calib3d, contrib, dnn, features2d, highgui, imgcodecs, imgproc, objdetect, photo, stitching, video, videoio]}
+
+  optionalModules ??= [];
+  if (userDefines.containsKey('include_modules')) {
+    final moduels = userDefines['include_modules'] as YamlList?;
+    if (moduels != null) {
+      optionalModules.addAll(moduels.cast<String>());
+    }
+  }
+
+  final moduleDefines = {
+    'DARTCV_WITH_CALIB3D': optionalModules.contains('calib3d') ? 'ON' : 'OFF',
+    'DARTCV_WITH_CONTRIB': optionalModules.contains('contrib') ? 'ON' : 'OFF',
+    'DARTCV_WITH_DNN': optionalModules.contains('dnn') ? 'ON' : 'OFF',
+    'DARTCV_WITH_FEATURES2D': optionalModules.contains('features2d') ? 'ON' : 'OFF',
+    'DARTCV_WITH_HIGHGUI': optionalModules.contains('highgui') ? 'ON' : 'OFF',
+    // 'DARTCV_WITH_IMGCODECS': optionalModules.contains('imgcodecs')? 'ON' : 'OFF',
+    'DARTCV_WITH_IMGPROC': optionalModules.contains('imgproc') ? 'ON' : 'OFF',
+    'DARTCV_WITH_OBJDETECT': optionalModules.contains('objdetect') ? 'ON' : 'OFF',
+    'DARTCV_WITH_PHOTO': optionalModules.contains('photo') ? 'ON' : 'OFF',
+    'DARTCV_WITH_STITCHING': optionalModules.contains('stitching') ? 'ON' : 'OFF',
+    'DARTCV_WITH_VIDEO': optionalModules.contains('video') ? 'ON' : 'OFF',
+    'DARTCV_WITH_VIDEOIO': optionalModules.contains('videoio') ? 'ON' : 'OFF',
+  };
 
   final builder = CMakeBuilder.create(
     name: input.packageName,
     sourceDir: packagePath.uri.resolve("src"),
     targets: ['install'],
+    buildLocal: true,
     // Waiting for the support of user-defined variables,
     // which can be used to choose the components
     defines: {
+      'DARTCV_ENABLE_INSTALL': 'ON',
+      'DARTCV_WORLD': 'OFF',
       'CMAKE_INSTALL_PREFIX': input.outputDirectory.resolve('install/').toFilePath(),
+      ...moduleDefines,
     },
   );
   await builder.run(input: input, output: output, logger: logger);

Diff for: pubspec.yaml

@@ -12,11 +12,35 @@ dependencies:
   logging: ^1.2.0
   native_assets_cli: ^0.13.0
   native_toolchain_cmake: ^0.0.5
+  yaml: ^3.1.3
 
+dependency_overrides:
+  native_assets_cli:
+    git:
+      url: https://github.com/dart-lang/native.git
+      path: pkgs/native_assets_cli
+      ref: 3fbbf07
 dev_dependencies:
   ffigen: ^16.1.0
   test: ^1.25.2
 
+hooks:
+  user_defines:
+    dartcv4:
+      include_modules:
+        # - calib3d
+        # - contrib
+        - dnn
+        # - features2d
+        # - highgui
+        # - imgcodecs
+        - imgproc
+        # - objdetect
+        # - photo
+        # - stitching
+        # - video
+        # - videoio
+
 topics:
   - opencv
   - image