580 lines
19 KiB
Dart
580 lines
19 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';
|
|
import '../utils/target_colors.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';
|
|
late Color _backgroundColor;
|
|
|
|
// Brightness and contrast controls
|
|
double _brightness = 0.0; // -1.0 to 1.0
|
|
double _contrast = 1.0; // 0.0 to 2.0
|
|
|
|
@override
|
|
void initState() {
|
|
super.initState();
|
|
// Initialize with default target background color
|
|
_backgroundColor = TargetColors.getBackgroundColor('assets/target.png');
|
|
_loadBulletCount();
|
|
_initCamera();
|
|
_loadBulletColor(); // 👈
|
|
_loadLanguage();
|
|
_loadSelectedTargetColor();
|
|
_loadBrightnessContrast();
|
|
YOLODetector.create().then((detector) => _yoloDetector = detector);
|
|
}
|
|
|
|
Future<void> _loadBrightnessContrast() async {
|
|
final prefs = await SharedPreferences.getInstance();
|
|
setState(() {
|
|
_brightness = prefs.getDouble('brightness') ?? 0.0;
|
|
_contrast = prefs.getDouble('contrast') ?? 1.0;
|
|
});
|
|
}
|
|
|
|
Future<void> _saveBrightnessContrast() async {
|
|
final prefs = await SharedPreferences.getInstance();
|
|
await prefs.setDouble('brightness', _brightness);
|
|
await prefs.setDouble('contrast', _contrast);
|
|
}
|
|
|
|
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> _loadSelectedTargetColor() async {
|
|
final prefs = await SharedPreferences.getInstance();
|
|
final selectedTarget = prefs.getString('selectedTarget') ?? 'assets/target.png';
|
|
setState(() {
|
|
_backgroundColor = TargetColors.getBackgroundColor(selectedTarget);
|
|
});
|
|
}
|
|
|
|
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,
|
|
),
|
|
),
|
|
);
|
|
}),
|
|
),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
);
|
|
},
|
|
);
|
|
}
|
|
|
|
|
|
void _showHelpDialog() {
|
|
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),
|
|
child: Column(
|
|
mainAxisSize: MainAxisSize.min,
|
|
crossAxisAlignment: CrossAxisAlignment.start,
|
|
children: [
|
|
Text(
|
|
_t('Help'),
|
|
style: const TextStyle(color: Colors.white, fontSize: 24, fontWeight: FontWeight.bold),
|
|
),
|
|
const SizedBox(height: 20),
|
|
_buildHelpItem(Icons.camera_alt, _t('Tap camera button to capture and detect bullets')),
|
|
const SizedBox(height: 12),
|
|
_buildHelpItem(Icons.filter_9_plus, _t('Bottom left number: Set expected bullet count (1-5)')),
|
|
const SizedBox(height: 12),
|
|
_buildHelpItem(Icons.brightness_6, _t('Sun icon: Adjust brightness and contrast for better visibility')),
|
|
const SizedBox(height: 12),
|
|
_buildHelpItem(Icons.crop_square, _t('Align target inside the square frame')),
|
|
const SizedBox(height: 20),
|
|
Center(
|
|
child: ElevatedButton(
|
|
onPressed: () => Navigator.of(context).pop(),
|
|
style: ElevatedButton.styleFrom(
|
|
backgroundColor: _bulletColor,
|
|
foregroundColor: Colors.white,
|
|
),
|
|
child: Text(_t('Got it!')),
|
|
),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
);
|
|
},
|
|
);
|
|
}
|
|
|
|
Widget _buildHelpItem(IconData icon, String text) {
|
|
return Row(
|
|
crossAxisAlignment: CrossAxisAlignment.start,
|
|
children: [
|
|
Icon(icon, color: _bulletColor, size: 20),
|
|
const SizedBox(width: 12),
|
|
Expanded(
|
|
child: Text(
|
|
text,
|
|
style: const TextStyle(color: Colors.white, fontSize: 14),
|
|
),
|
|
),
|
|
],
|
|
);
|
|
}
|
|
|
|
void _showBrightnessContrastDialog() {
|
|
showDialog(
|
|
context: context,
|
|
builder: (BuildContext context) {
|
|
return StatefulBuilder(
|
|
builder: (context, setDialogState) {
|
|
return Dialog(
|
|
backgroundColor: Colors.grey[900],
|
|
shape: RoundedRectangleBorder(borderRadius: BorderRadius.circular(16)),
|
|
child: Padding(
|
|
padding: const EdgeInsets.all(24.0),
|
|
child: Column(
|
|
mainAxisSize: MainAxisSize.min,
|
|
children: [
|
|
Text(
|
|
_t('Brightness & Contrast'),
|
|
style: const TextStyle(color: Colors.white, fontSize: 20, fontWeight: FontWeight.bold),
|
|
),
|
|
const SizedBox(height: 24),
|
|
// Brightness slider
|
|
Column(
|
|
crossAxisAlignment: CrossAxisAlignment.start,
|
|
children: [
|
|
Text(
|
|
'${_t('Brightness')}: ${(_brightness * 100).toInt()}',
|
|
style: const TextStyle(color: Colors.white, fontSize: 16),
|
|
),
|
|
Slider(
|
|
value: _brightness,
|
|
min: -1.0,
|
|
max: 1.0,
|
|
divisions: 40,
|
|
activeColor: _bulletColor,
|
|
onChanged: (value) {
|
|
setState(() {
|
|
_brightness = value;
|
|
});
|
|
setDialogState(() {});
|
|
_saveBrightnessContrast();
|
|
},
|
|
),
|
|
],
|
|
),
|
|
const SizedBox(height: 16),
|
|
// Contrast slider
|
|
Column(
|
|
crossAxisAlignment: CrossAxisAlignment.start,
|
|
children: [
|
|
Text(
|
|
'${_t('Contrast')}: ${(_contrast * 100).toInt()}%',
|
|
style: const TextStyle(color: Colors.white, fontSize: 16),
|
|
),
|
|
Slider(
|
|
value: _contrast,
|
|
min: 0.5,
|
|
max: 2.0,
|
|
divisions: 30,
|
|
activeColor: _bulletColor,
|
|
onChanged: (value) {
|
|
setState(() {
|
|
_contrast = value;
|
|
});
|
|
setDialogState(() {});
|
|
_saveBrightnessContrast();
|
|
},
|
|
),
|
|
],
|
|
),
|
|
const SizedBox(height: 16),
|
|
Row(
|
|
mainAxisAlignment: MainAxisAlignment.spaceEvenly,
|
|
children: [
|
|
ElevatedButton(
|
|
onPressed: () {
|
|
setState(() {
|
|
_brightness = 0.0;
|
|
_contrast = 1.0;
|
|
});
|
|
setDialogState(() {});
|
|
_saveBrightnessContrast();
|
|
},
|
|
style: ElevatedButton.styleFrom(
|
|
backgroundColor: Colors.grey[800],
|
|
foregroundColor: Colors.white,
|
|
),
|
|
child: Text(_t('Reset')),
|
|
),
|
|
ElevatedButton(
|
|
onPressed: () {
|
|
Navigator.of(context).pop();
|
|
},
|
|
style: ElevatedButton.styleFrom(
|
|
backgroundColor: _bulletColor,
|
|
foregroundColor: Colors.white,
|
|
),
|
|
child: Text(_t('Done')),
|
|
),
|
|
],
|
|
),
|
|
],
|
|
),
|
|
),
|
|
);
|
|
},
|
|
);
|
|
},
|
|
);
|
|
}
|
|
|
|
// 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(
|
|
backgroundColor: _backgroundColor,
|
|
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: [
|
|
IconButton(
|
|
icon: const Icon(Icons.help_outline, color: Colors.white),
|
|
onPressed: _showHelpDialog,
|
|
),
|
|
const SizedBox(width: 16),
|
|
],
|
|
),
|
|
bottomNavigationBar: BottomAppBar(
|
|
color: Colors.grey[900],
|
|
shape: const CircularNotchedRectangle(),
|
|
notchMargin: 8.0,
|
|
child: SizedBox(
|
|
height: 60,
|
|
child: Row(
|
|
children: [
|
|
// Left - Bullet count (just number)
|
|
Expanded(
|
|
child: Align(
|
|
alignment: Alignment.centerLeft,
|
|
child: Padding(
|
|
padding: const EdgeInsets.only(left: 16.0),
|
|
child: InkWell(
|
|
onTap: _showBulletDialog,
|
|
child: Text(
|
|
'$_numberOfBullets',
|
|
style: const TextStyle(
|
|
fontSize: 24,
|
|
fontWeight: FontWeight.bold,
|
|
color: Colors.white,
|
|
),
|
|
),
|
|
),
|
|
),
|
|
),
|
|
),
|
|
// Center space for FAB
|
|
const SizedBox(width: 80),
|
|
// Right section - Brightness
|
|
Expanded(
|
|
child: Row(
|
|
mainAxisAlignment: MainAxisAlignment.end,
|
|
children: [
|
|
IconButton(
|
|
icon: const Icon(Icons.brightness_6, color: Colors.white, size: 24),
|
|
onPressed: _showBrightnessContrastDialog,
|
|
),
|
|
],
|
|
),
|
|
),
|
|
],
|
|
),
|
|
),
|
|
),
|
|
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.withValues(alpha: 0.9),
|
|
width: 3,
|
|
),
|
|
),
|
|
),
|
|
),
|
|
],
|
|
);
|
|
} else {
|
|
return const Center(child: CircularProgressIndicator());
|
|
}
|
|
},
|
|
),
|
|
);
|
|
}
|
|
}
|