346 lines
10 KiB
Dart
346 lines
10 KiB
Dart
import 'dart:io';
|
|
import 'package:flutter/material.dart';
|
|
import 'package:camera/camera.dart';
|
|
import 'package:image/image.dart' as img;
|
|
import 'package:shared_preferences/shared_preferences.dart';
|
|
import 'yolo_detector.dart';
|
|
import 'preview_screen.dart';
|
|
import '../utils/translations.dart';
|
|
|
|
class CameraScreen extends StatefulWidget {
|
|
const CameraScreen({super.key});
|
|
|
|
@override
|
|
State<CameraScreen> createState() => _CameraScreenState();
|
|
}
|
|
|
|
class _CameraScreenState extends State<CameraScreen> {
|
|
CameraController? _controller;
|
|
Future<void>? _initializeControllerFuture;
|
|
late YOLODetector _yoloDetector;
|
|
Color _bulletColor = Colors.blue; // default fallback
|
|
|
|
// The variable to hold the number of bullets entered by the user (default set to 5)
|
|
int _numberOfBullets = 5;
|
|
String _selectedLanguage = 'English';
|
|
|
|
@override
|
|
void initState() {
|
|
super.initState();
|
|
_loadBulletCount();
|
|
_initCamera();
|
|
_loadBulletColor(); // 👈
|
|
_loadLanguage();
|
|
YOLODetector.create().then((detector) => _yoloDetector = detector);
|
|
}
|
|
|
|
String _t(String key) {
|
|
return AppTranslations.translate(key, _selectedLanguage);
|
|
}
|
|
|
|
Future<void> _loadLanguage() async {
|
|
final prefs = await SharedPreferences.getInstance();
|
|
setState(() {
|
|
_selectedLanguage = prefs.getString('language') ?? 'English';
|
|
});
|
|
}
|
|
|
|
Future<void> _loadBulletColor() async {
|
|
final prefs = await SharedPreferences.getInstance();
|
|
final selected = prefs.getString('bulletColor') ?? 'Blue';
|
|
setState(() {
|
|
_bulletColor = _getColorFromName(selected);
|
|
});
|
|
}
|
|
|
|
Color _getColorFromName(String colorName) {
|
|
switch (colorName.toLowerCase()) {
|
|
case 'blue':
|
|
return Colors.blue;
|
|
case 'green':
|
|
return Colors.green;
|
|
case 'yellow':
|
|
return Colors.yellow;
|
|
case 'purple':
|
|
return Colors.purple;
|
|
case 'cyan':
|
|
return Colors.cyan;
|
|
case 'teal':
|
|
return Colors.teal;
|
|
case 'indigo':
|
|
return Colors.indigo;
|
|
case 'pink':
|
|
return Colors.pink;
|
|
case 'lime':
|
|
return Colors.lime;
|
|
case 'mint':
|
|
return Colors.greenAccent;
|
|
case 'orange':
|
|
return Colors.orange;
|
|
case 'violet':
|
|
return Colors.deepPurpleAccent;
|
|
case 'amber':
|
|
return Colors.amber;
|
|
case 'brown':
|
|
return Colors.brown;
|
|
case 'grey':
|
|
return Colors.grey;
|
|
default:
|
|
return Colors.blue; // fallback
|
|
}
|
|
}
|
|
|
|
|
|
|
|
Future<void> _loadBulletCount() async {
|
|
final prefs = await SharedPreferences.getInstance();
|
|
setState(() {
|
|
_numberOfBullets = prefs.getInt('bulletCount') ?? 5; // Load stored bullet count, default to 5
|
|
});
|
|
}
|
|
|
|
Future<void> _saveBulletCount(int count) async {
|
|
final prefs = await SharedPreferences.getInstance();
|
|
await prefs.setInt('bulletCount', count); // Save the bullet count
|
|
}
|
|
|
|
Future<void> _initCamera() async {
|
|
final cameras = await availableCameras();
|
|
final camera = cameras.first;
|
|
|
|
_controller = CameraController(
|
|
camera,
|
|
ResolutionPreset.high,
|
|
enableAudio: false,
|
|
);
|
|
|
|
_initializeControllerFuture = _controller!.initialize();
|
|
if (mounted) setState(() {});
|
|
}
|
|
|
|
@override
|
|
void dispose() {
|
|
_controller?.dispose();
|
|
_yoloDetector.close();
|
|
super.dispose();
|
|
}
|
|
|
|
Future<void> _takePicture() async {
|
|
try {
|
|
await _initializeControllerFuture;
|
|
final file = await _controller!.takePicture();
|
|
final imageFile = File(file.path);
|
|
final imageBytes = await imageFile.readAsBytes();
|
|
final decodedImage = img.decodeImage(imageBytes)!;
|
|
|
|
// Crop center square from full image
|
|
final width = decodedImage.width;
|
|
final height = decodedImage.height;
|
|
final size = width < height ? width : height;
|
|
final offsetX = (width - size) ~/ 2;
|
|
final offsetY = (height - size) ~/ 2;
|
|
|
|
final cropped = img.copyCrop(
|
|
decodedImage,
|
|
x: offsetX,
|
|
y: offsetY,
|
|
width: size,
|
|
height: size,
|
|
);
|
|
|
|
final croppedFile = File('${file.path}_square.jpg')
|
|
..writeAsBytesSync(img.encodeJpg(cropped));
|
|
|
|
final detections = await _yoloDetector.detect(croppedFile);
|
|
|
|
if (!mounted) return;
|
|
|
|
Navigator.push(
|
|
context,
|
|
MaterialPageRoute(
|
|
builder: (context) => PreviewScreen(
|
|
imagePath: croppedFile.path,
|
|
detections: detections,
|
|
bulletCount: _numberOfBullets, // Pass the bullet count
|
|
),
|
|
),
|
|
);
|
|
} catch (e) {
|
|
print('Error taking picture: $e');
|
|
}
|
|
}
|
|
|
|
// Method to show the dialog for selecting the number of bullets
|
|
void _showBulletDialog() {
|
|
showDialog(
|
|
context: context,
|
|
builder: (BuildContext context) {
|
|
return Dialog(
|
|
backgroundColor: Colors.grey[900],
|
|
shape: RoundedRectangleBorder(borderRadius: BorderRadius.circular(16)),
|
|
child: Padding(
|
|
padding: const EdgeInsets.all(24.0), // Increased padding for better spacing
|
|
child: Column(
|
|
mainAxisSize: MainAxisSize.min,
|
|
crossAxisAlignment: CrossAxisAlignment.center,
|
|
children: [
|
|
Padding(
|
|
padding: const EdgeInsets.only(bottom: 16.0),
|
|
child: Text(
|
|
_t('Select the number of bullets'),
|
|
textAlign: TextAlign.center,
|
|
style: const TextStyle(
|
|
color: Colors.white,
|
|
fontSize: 20,
|
|
fontWeight: FontWeight.bold,
|
|
),
|
|
),
|
|
),
|
|
Center(
|
|
child: Wrap(
|
|
alignment: WrapAlignment.center,
|
|
spacing: 12,
|
|
runSpacing: 12,
|
|
children: List.generate(5, (index) {
|
|
final bulletNum = index + 1;
|
|
final isSelected = _numberOfBullets == bulletNum;
|
|
return ElevatedButton(
|
|
onPressed: () {
|
|
_selectBulletCount(bulletNum);
|
|
Navigator.of(context).pop();
|
|
},
|
|
style: ElevatedButton.styleFrom(
|
|
backgroundColor:
|
|
isSelected ? _bulletColor : Colors.grey[800],
|
|
foregroundColor: Colors.white,
|
|
padding:
|
|
const EdgeInsets.symmetric(horizontal: 20, vertical: 12),
|
|
shape: RoundedRectangleBorder(
|
|
borderRadius: BorderRadius.circular(12),
|
|
),
|
|
),
|
|
child: Text(
|
|
'$bulletNum',
|
|
style: TextStyle(
|
|
fontSize: 18,
|
|
fontWeight:
|
|
isSelected ? FontWeight.bold : FontWeight.normal,
|
|
),
|
|
),
|
|
);
|
|
}),
|
|
),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
);
|
|
},
|
|
);
|
|
}
|
|
|
|
|
|
// Update the bullet count and save it
|
|
void _selectBulletCount(int number) {
|
|
setState(() {
|
|
_numberOfBullets = number; // Update the selected number
|
|
});
|
|
_saveBulletCount(number); // Save the selected number
|
|
}
|
|
|
|
@override
|
|
Widget build(BuildContext context) {
|
|
if (_controller == null || _initializeControllerFuture == null) {
|
|
return const Scaffold(
|
|
body: Center(child: CircularProgressIndicator()),
|
|
);
|
|
}
|
|
|
|
return Scaffold(
|
|
extendBody: true,
|
|
appBar: AppBar(
|
|
backgroundColor: Colors.grey[900],
|
|
title: Text(
|
|
_t("Align the target inside the square"),
|
|
style: const TextStyle(
|
|
fontSize: 16,
|
|
fontWeight: FontWeight.bold,
|
|
color: Colors.white,
|
|
),
|
|
),
|
|
leading: IconButton(
|
|
icon: const Icon(Icons.arrow_back, color: Colors.white),
|
|
onPressed: () => Navigator.pop(context),
|
|
),
|
|
centerTitle: true,
|
|
actions: [
|
|
// Display the selected bullet count
|
|
IconButton(
|
|
icon: Text(
|
|
'$_numberOfBullets',
|
|
style: const TextStyle(
|
|
fontSize: 20,
|
|
fontWeight: FontWeight.bold,
|
|
color: Colors.white,
|
|
),
|
|
),
|
|
onPressed: _showBulletDialog, // Show dialog on tap
|
|
),
|
|
const SizedBox(width: 16), // Space after the number
|
|
],
|
|
),
|
|
bottomNavigationBar: BottomAppBar(
|
|
color: Colors.grey[900],
|
|
shape: const CircularNotchedRectangle(),
|
|
notchMargin: 8.0,
|
|
child: SizedBox(height: 60),
|
|
),
|
|
floatingActionButtonLocation: FloatingActionButtonLocation.centerDocked,
|
|
floatingActionButton: FloatingActionButton(
|
|
backgroundColor: Colors.white,
|
|
onPressed: _takePicture,
|
|
child: const Icon(Icons.camera_alt, color: Colors.black),
|
|
),
|
|
body: FutureBuilder(
|
|
future: _initializeControllerFuture,
|
|
builder: (context, snapshot) {
|
|
if (snapshot.connectionState == ConnectionState.done) {
|
|
final screenSize = MediaQuery.of(context).size;
|
|
return Stack(
|
|
children: [
|
|
Positioned.fill(
|
|
child: Container(
|
|
color: Colors.black, // covers any empty space
|
|
child: FittedBox(
|
|
fit: BoxFit.cover,
|
|
child: SizedBox(
|
|
width: _controller!.value.previewSize!.height,
|
|
height: _controller!.value.previewSize!.width,
|
|
child: CameraPreview(_controller!),
|
|
),
|
|
),
|
|
),
|
|
),
|
|
Center(
|
|
child: Container(
|
|
width: screenSize.width * 0.9,
|
|
height: screenSize.width * 0.9,
|
|
decoration: BoxDecoration(
|
|
border: Border.all(
|
|
color: _bulletColor.withOpacity(0.9),
|
|
width: 3,
|
|
),
|
|
),
|
|
),
|
|
),
|
|
],
|
|
);
|
|
} else {
|
|
return const Center(child: CircularProgressIndicator());
|
|
}
|
|
},
|
|
),
|
|
);
|
|
}
|
|
}
|