463 lines
14 KiB
Dart
463 lines
14 KiB
Dart
import 'dart:async';
|
||
import 'dart:io';
|
||
import 'package:camera/camera.dart';
|
||
import 'package:flutter/material.dart';
|
||
import 'package:flutter/foundation.dart' show kIsWeb;
|
||
import 'package:image_picker/image_picker.dart';
|
||
import 'package:gal/gal.dart';
|
||
import 'package:path_provider/path_provider.dart';
|
||
import 'package:path/path.dart' as path;
|
||
import 'package:permission_handler/permission_handler.dart';
|
||
import 'package:camera_app/services/yolo_service.dart';
|
||
import 'package:camera_app/screens/result_screen.dart';
|
||
|
||
class CameraScreen extends StatefulWidget {
|
||
const CameraScreen({super.key});
|
||
|
||
@override
|
||
State<CameraScreen> createState() => _CameraScreenState();
|
||
}
|
||
|
||
class _CameraScreenState extends State<CameraScreen> {
|
||
CameraController? _controller;
|
||
List<CameraDescription>? _cameras;
|
||
bool _isInitialized = false;
|
||
bool _isCapturing = false;
|
||
int _photoCount = 0;
|
||
Timer? _timer;
|
||
List<String> _savedPhotoPaths = [];
|
||
bool _isWebOrDesktop = false;
|
||
final YOLOService _yoloService = YOLOService();
|
||
|
||
@override
|
||
void initState() {
|
||
super.initState();
|
||
_initializeCamera();
|
||
}
|
||
|
||
Future<void> _initializeCamera() async {
|
||
try {
|
||
// Web 和桌面平台使用 image_picker
|
||
if (kIsWeb || (!Platform.isAndroid && !Platform.isIOS)) {
|
||
if (mounted) {
|
||
setState(() {
|
||
_isWebOrDesktop = true;
|
||
_isInitialized = true; // Web/桌面平台直接显示界面
|
||
});
|
||
}
|
||
return;
|
||
}
|
||
|
||
// 移动平台使用 camera 包
|
||
await _requestPermissions();
|
||
|
||
_cameras = await availableCameras();
|
||
if (_cameras == null || _cameras!.isEmpty) {
|
||
_showError('未找到可用的相机');
|
||
return;
|
||
}
|
||
|
||
_controller = CameraController(
|
||
_cameras![0],
|
||
ResolutionPreset.high,
|
||
enableAudio: false,
|
||
);
|
||
|
||
await _controller!.initialize();
|
||
|
||
if (mounted) {
|
||
setState(() {
|
||
_isInitialized = true;
|
||
});
|
||
}
|
||
} catch (e) {
|
||
_showError('初始化相机失败: $e');
|
||
}
|
||
}
|
||
|
||
Future<void> _requestPermissions() async {
|
||
if (kIsWeb || (!Platform.isAndroid && !Platform.isIOS)) {
|
||
return; // Web/桌面平台权限由浏览器/系统处理
|
||
}
|
||
|
||
final cameraStatus = await Permission.camera.request();
|
||
if (!cameraStatus.isGranted) {
|
||
_showError('需要相机权限才能使用此功能');
|
||
return;
|
||
}
|
||
|
||
if (Platform.isAndroid) {
|
||
final photosStatus = await Permission.photos.request();
|
||
if (photosStatus.isDenied || photosStatus.isPermanentlyDenied) {
|
||
final storageStatus = await Permission.storage.request();
|
||
if (!storageStatus.isGranted) {
|
||
_showError('需要存储权限才能保存图片');
|
||
return;
|
||
}
|
||
}
|
||
} else if (Platform.isIOS) {
|
||
final photosStatus = await Permission.photos.request();
|
||
if (!photosStatus.isGranted) {
|
||
_showError('需要照片权限才能保存图片');
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
|
||
Future<void> _startCapturing() async {
|
||
if (_isWebOrDesktop) {
|
||
_showError('Web/桌面平台不支持自动连续拍照，请使用移动设备');
|
||
return;
|
||
}
|
||
|
||
// 移动平台：自动拍照模式
|
||
if (!_isInitialized || _controller == null || !_controller!.value.isInitialized) {
|
||
_showError('相机未初始化');
|
||
return;
|
||
}
|
||
|
||
setState(() {
|
||
_isCapturing = true;
|
||
_photoCount = 0;
|
||
_savedPhotoPaths.clear();
|
||
});
|
||
|
||
// 立即拍摄第一张
|
||
await _takePhoto();
|
||
|
||
// 然后每2秒拍摄一张，直到点击停止
|
||
_timer = Timer.periodic(const Duration(seconds: 2), (timer) async {
|
||
if (_isCapturing) {
|
||
await _takePhoto();
|
||
} else {
|
||
timer.cancel();
|
||
}
|
||
});
|
||
}
|
||
|
||
Future<void> _stopCapturing() async {
|
||
setState(() {
|
||
_isCapturing = false;
|
||
});
|
||
_timer?.cancel();
|
||
|
||
if (_savedPhotoPaths.isEmpty) {
|
||
_showError('没有拍摄照片');
|
||
return;
|
||
}
|
||
|
||
// 显示分析中对话框
|
||
if (!mounted) return;
|
||
|
||
showDialog(
|
||
context: context,
|
||
barrierDismissible: false,
|
||
builder: (BuildContext context) {
|
||
return const AlertDialog(
|
||
content: Column(
|
||
mainAxisSize: MainAxisSize.min,
|
||
children: [
|
||
CircularProgressIndicator(),
|
||
SizedBox(height: 16),
|
||
Text('正在分析照片...'),
|
||
],
|
||
),
|
||
);
|
||
},
|
||
);
|
||
|
||
try {
|
||
// 使用YOLO分析照片
|
||
final analysisResults = await _yoloService.analyzeImages(_savedPhotoPaths);
|
||
|
||
if (!mounted) return;
|
||
|
||
Navigator.of(context).pop(); // 关闭分析中对话框
|
||
|
||
// 导航到结果页面
|
||
Navigator.of(context).pushReplacement(
|
||
MaterialPageRoute(
|
||
builder: (context) => ResultScreen(
|
||
photoPaths: _savedPhotoPaths,
|
||
analysisResults: analysisResults,
|
||
),
|
||
),
|
||
);
|
||
} catch (e) {
|
||
if (!mounted) return;
|
||
|
||
Navigator.of(context).pop(); // 关闭分析中对话框
|
||
_showError('分析照片失败: $e');
|
||
}
|
||
}
|
||
|
||
Future<void> _takePhoto() async {
|
||
if (_controller == null || !_controller!.value.isInitialized) {
|
||
return;
|
||
}
|
||
|
||
try {
|
||
final XFile photo = await _controller!.takePicture();
|
||
await _savePhoto(photo.path);
|
||
|
||
if (mounted) {
|
||
setState(() {
|
||
_photoCount++;
|
||
});
|
||
}
|
||
} catch (e) {
|
||
if (mounted) {
|
||
_showError('拍照失败: $e');
|
||
}
|
||
}
|
||
}
|
||
|
||
Future<void> _savePhoto(String photoPath) async {
|
||
try {
|
||
if (!kIsWeb && (Platform.isAndroid || Platform.isIOS)) {
|
||
// 移动平台：保存到相册
|
||
await Gal.putImage(photoPath);
|
||
} else {
|
||
// Web/桌面平台：保存到下载目录
|
||
final appDir = await getApplicationDocumentsDirectory();
|
||
final fileName = 'photo_${DateTime.now().millisecondsSinceEpoch}.jpg';
|
||
final savedPath = path.join(appDir.path, fileName);
|
||
await File(photoPath).copy(savedPath);
|
||
}
|
||
_savedPhotoPaths.add(photoPath);
|
||
} catch (e) {
|
||
// 保存失败不影响流程
|
||
_savedPhotoPaths.add(photoPath);
|
||
}
|
||
}
|
||
|
||
void _showError(String message) {
|
||
if (mounted) {
|
||
ScaffoldMessenger.of(context).showSnackBar(
|
||
SnackBar(
|
||
content: Text(message),
|
||
backgroundColor: Colors.red,
|
||
),
|
||
);
|
||
}
|
||
}
|
||
|
||
@override
|
||
void dispose() {
|
||
_timer?.cancel();
|
||
_controller?.dispose();
|
||
super.dispose();
|
||
}
|
||
|
||
@override
|
||
Widget build(BuildContext context) {
|
||
if (!_isInitialized) {
|
||
return Scaffold(
|
||
backgroundColor: Colors.black,
|
||
body: const Center(
|
||
child: CircularProgressIndicator(color: Colors.white),
|
||
),
|
||
);
|
||
}
|
||
|
||
// Web/桌面平台界面
|
||
if (_isWebOrDesktop) {
|
||
return Scaffold(
|
||
backgroundColor: Colors.black,
|
||
body: Center(
|
||
child: Padding(
|
||
padding: const EdgeInsets.all(24.0),
|
||
child: Column(
|
||
mainAxisAlignment: MainAxisAlignment.center,
|
||
children: [
|
||
const Icon(
|
||
Icons.camera_alt,
|
||
size: 100,
|
||
color: Colors.white,
|
||
),
|
||
const SizedBox(height: 32),
|
||
Text(
|
||
'$_photoCount 张',
|
||
style: const TextStyle(
|
||
color: Colors.white,
|
||
fontSize: 48,
|
||
fontWeight: FontWeight.bold,
|
||
),
|
||
),
|
||
const SizedBox(height: 16),
|
||
const Text(
|
||
'Web/桌面平台不支持自动连续拍照',
|
||
style: TextStyle(
|
||
color: Colors.white70,
|
||
fontSize: 18,
|
||
),
|
||
textAlign: TextAlign.center,
|
||
),
|
||
const SizedBox(height: 48),
|
||
IconButton(
|
||
icon: const Icon(Icons.arrow_back, color: Colors.white),
|
||
onPressed: () => Navigator.of(context).pop(),
|
||
tooltip: '返回',
|
||
),
|
||
],
|
||
),
|
||
),
|
||
),
|
||
);
|
||
}
|
||
|
||
// 移动平台：显示相机预览
|
||
if (_controller == null || !_controller!.value.isInitialized) {
|
||
return Scaffold(
|
||
backgroundColor: Colors.black,
|
||
body: const Center(
|
||
child: Text(
|
||
'相机初始化中...',
|
||
style: TextStyle(color: Colors.white),
|
||
),
|
||
),
|
||
);
|
||
}
|
||
|
||
return Scaffold(
|
||
backgroundColor: Colors.black,
|
||
body: Stack(
|
||
children: [
|
||
SizedBox.expand(child: CameraPreview(_controller!)),
|
||
|
||
SafeArea(
|
||
child: Container(
|
||
padding: const EdgeInsets.all(16),
|
||
decoration: BoxDecoration(
|
||
gradient: LinearGradient(
|
||
begin: Alignment.topCenter,
|
||
end: Alignment.bottomCenter,
|
||
colors: [
|
||
Colors.black.withOpacity(0.7),
|
||
Colors.transparent,
|
||
],
|
||
),
|
||
),
|
||
child: Row(
|
||
mainAxisAlignment: MainAxisAlignment.spaceBetween,
|
||
children: [
|
||
IconButton(
|
||
icon: const Icon(Icons.arrow_back, color: Colors.white),
|
||
onPressed: () {
|
||
if (_isCapturing) {
|
||
_stopCapturing();
|
||
}
|
||
Navigator.of(context).pop();
|
||
},
|
||
),
|
||
if (_isCapturing)
|
||
Container(
|
||
padding: const EdgeInsets.symmetric(
|
||
horizontal: 16,
|
||
vertical: 8,
|
||
),
|
||
decoration: BoxDecoration(
|
||
color: Colors.red.withOpacity(0.8),
|
||
borderRadius: BorderRadius.circular(20),
|
||
),
|
||
child: Row(
|
||
children: [
|
||
Container(
|
||
width: 12,
|
||
height: 12,
|
||
decoration: const BoxDecoration(
|
||
color: Colors.white,
|
||
shape: BoxShape.circle,
|
||
),
|
||
),
|
||
const SizedBox(width: 8),
|
||
Text(
|
||
'$_photoCount 张',
|
||
style: const TextStyle(
|
||
color: Colors.white,
|
||
fontSize: 16,
|
||
fontWeight: FontWeight.bold,
|
||
),
|
||
),
|
||
],
|
||
),
|
||
),
|
||
],
|
||
),
|
||
),
|
||
),
|
||
|
||
Align(
|
||
alignment: Alignment.bottomCenter,
|
||
child: Container(
|
||
padding: const EdgeInsets.all(32),
|
||
decoration: BoxDecoration(
|
||
gradient: LinearGradient(
|
||
begin: Alignment.bottomCenter,
|
||
end: Alignment.topCenter,
|
||
colors: [
|
||
Colors.black.withOpacity(0.7),
|
||
Colors.transparent,
|
||
],
|
||
),
|
||
),
|
||
child: _isCapturing
|
||
? Column(
|
||
mainAxisSize: MainAxisSize.min,
|
||
children: [
|
||
ElevatedButton.icon(
|
||
onPressed: _stopCapturing,
|
||
icon: const Icon(Icons.stop, size: 28),
|
||
label: const Text(
|
||
'停止拍照',
|
||
style: TextStyle(fontSize: 18),
|
||
),
|
||
style: ElevatedButton.styleFrom(
|
||
backgroundColor: Colors.red,
|
||
foregroundColor: Colors.white,
|
||
padding: const EdgeInsets.symmetric(
|
||
horizontal: 32,
|
||
vertical: 16,
|
||
),
|
||
shape: RoundedRectangleBorder(
|
||
borderRadius: BorderRadius.circular(30),
|
||
),
|
||
),
|
||
),
|
||
const SizedBox(height: 16),
|
||
Text(
|
||
'已拍摄 $_photoCount 张照片',
|
||
style: const TextStyle(
|
||
color: Colors.white,
|
||
fontSize: 16,
|
||
),
|
||
),
|
||
],
|
||
)
|
||
: ElevatedButton.icon(
|
||
onPressed: _startCapturing,
|
||
icon: const Icon(Icons.camera_alt, size: 28),
|
||
label: const Text(
|
||
'开始拍照',
|
||
style: TextStyle(fontSize: 18),
|
||
),
|
||
style: ElevatedButton.styleFrom(
|
||
backgroundColor: Colors.white,
|
||
foregroundColor: Colors.black,
|
||
padding: const EdgeInsets.symmetric(
|
||
horizontal: 32,
|
||
vertical: 16,
|
||
),
|
||
shape: RoundedRectangleBorder(
|
||
borderRadius: BorderRadius.circular(30),
|
||
),
|
||
),
|
||
),
|
||
),
|
||
),
|
||
],
|
||
),
|
||
);
|
||
}
|
||
}
|