160 lines
7.3 KiB
TypeScript
160 lines
7.3 KiB
TypeScript
"use client"
|
|
|
|
import { PieChart, Pie, Cell, ResponsiveContainer, Legend, Tooltip } from "recharts"
|
|
import { Loader2 } from "lucide-react"
|
|
|
|
interface DetectionDonutChartProps {
|
|
defectedSignboard: number
|
|
pothole: number
|
|
roadCrack: number
|
|
damagedRoadMarking: number
|
|
goodSignboard: number
|
|
isLoading?: boolean
|
|
}
|
|
|
|
const COLORS = {
|
|
defectedSignboard: "#3b82f6", // Blue
|
|
pothole: "#ef4444", // Red
|
|
roadCrack: "#f59e0b", // Amber/Orange
|
|
damagedRoadMarking: "#6366f1", // Indigo
|
|
goodSignboard: "#10b981" // Emerald
|
|
}
|
|
|
|
export function DetectionDonutChart({
|
|
defectedSignboard,
|
|
pothole,
|
|
roadCrack,
|
|
damagedRoadMarking,
|
|
goodSignboard,
|
|
isLoading = false
|
|
}: DetectionDonutChartProps) {
|
|
if (isLoading) {
|
|
return (
|
|
<div className="h-[250px] flex items-center justify-center">
|
|
<Loader2 className="h-8 w-8 text-primary/50" />
|
|
</div>
|
|
)
|
|
}
|
|
|
|
const total = defectedSignboard + pothole + roadCrack + damagedRoadMarking + goodSignboard
|
|
|
|
if (total === 0) {
|
|
return (
|
|
<div className="h-[250px] flex flex-col items-center justify-center text-muted-foreground">
|
|
<p className="text-sm">No detections found</p>
|
|
<p className="text-xs mt-1">Process videos to see data</p>
|
|
</div>
|
|
)
|
|
}
|
|
|
|
const data = [
|
|
{ name: "Defected Signboards", value: defectedSignboard, color: COLORS.defectedSignboard },
|
|
{ name: "Potholes", value: pothole, color: COLORS.pothole },
|
|
{ name: "Road Cracks", value: roadCrack, color: COLORS.roadCrack },
|
|
{ name: "Damaged Markings", value: damagedRoadMarking, color: COLORS.damagedRoadMarking },
|
|
{ name: "Good Signboards", value: goodSignboard, color: COLORS.goodSignboard }
|
|
].filter(item => item.value > 0)
|
|
|
|
return (
|
|
<div className="h-[220px] relative">
|
|
<ResponsiveContainer width="100%" height="100%">
|
|
<PieChart>
|
|
<defs>
|
|
<linearGradient id="gradPothole" x1="0" y1="0" x2="0" y2="1">
|
|
<stop offset="0%" stopColor="#ff8a8a" />
|
|
<stop offset="100%" stopColor="#ef4444" />
|
|
</linearGradient>
|
|
<linearGradient id="gradDefectedSign" x1="0" y1="0" x2="0" y2="1">
|
|
<stop offset="0%" stopColor="#60a5fa" />
|
|
<stop offset="100%" stopColor="#3b82f6" />
|
|
</linearGradient>
|
|
<linearGradient id="gradCrack" x1="0" y1="0" x2="0" y2="1">
|
|
<stop offset="0%" stopColor="#fbbf24" />
|
|
<stop offset="100%" stopColor="#f59e0b" />
|
|
</linearGradient>
|
|
<linearGradient id="gradMarking" x1="0" y1="0" x2="0" y2="1">
|
|
<stop offset="0%" stopColor="#818cf8" />
|
|
<stop offset="100%" stopColor="#6366f1" />
|
|
</linearGradient>
|
|
<linearGradient id="gradGoodSign" x1="0" y1="0" x2="0" y2="1">
|
|
<stop offset="0%" stopColor="#34d399" />
|
|
<stop offset="100%" stopColor="#10b981" />
|
|
</linearGradient>
|
|
</defs>
|
|
<Pie
|
|
data={data}
|
|
cx="50%"
|
|
cy="45%"
|
|
innerRadius={50}
|
|
outerRadius={70}
|
|
paddingAngle={5}
|
|
dataKey="value"
|
|
strokeWidth={0}
|
|
isAnimationActive={false}
|
|
>
|
|
{data.map((entry, index) => {
|
|
const gradId = entry.name === "Potholes" ? "gradPothole" :
|
|
entry.name === "Defected Signboards" ? "gradDefectedSign" :
|
|
entry.name === "Road Cracks" ? "gradCrack" :
|
|
entry.name === "Damaged Markings" ? "gradMarking" : "gradGoodSign"
|
|
return (
|
|
<Cell
|
|
key={`cell-${index}`}
|
|
fill={`url(#${gradId})`}
|
|
fillOpacity={1}
|
|
className="hover:fill-opacity-80"
|
|
/>
|
|
)
|
|
})}
|
|
</Pie>
|
|
<Tooltip
|
|
content={({ active, payload }) => {
|
|
if (active && payload && payload.length) {
|
|
const data = payload[0]
|
|
return (
|
|
<div className="bg-background/95 backdrop-blur-sm border border-border rounded-lg px-3 py-2 shadow-lg">
|
|
<p className="font-medium text-sm">{data.name}</p>
|
|
<p className="text-sm text-muted-foreground">
|
|
Count: <span className="font-semibold text-foreground">{data.value}</span>
|
|
</p>
|
|
<p className="text-xs text-muted-foreground">
|
|
{((Number(data.value) / total) * 100).toFixed(1)}% of total
|
|
</p>
|
|
</div>
|
|
)
|
|
}
|
|
return null
|
|
}}
|
|
/>
|
|
<Legend
|
|
verticalAlign="bottom"
|
|
height={40}
|
|
content={({ payload }) => (
|
|
<div className="flex flex-wrap items-center justify-center gap-x-4 gap-y-1 mt-2">
|
|
{payload?.map((entry, index) => (
|
|
<div key={`legend-${index}`} className="flex items-center gap-1.5">
|
|
<div
|
|
className="w-2.5 h-2.5 rounded-full"
|
|
style={{ backgroundColor: entry.color }}
|
|
/>
|
|
<span className="text-[9px] text-muted-foreground whitespace-nowrap">
|
|
{entry.value}: {data[index].value}
|
|
</span>
|
|
</div>
|
|
))}
|
|
</div>
|
|
)}
|
|
/>
|
|
</PieChart>
|
|
</ResponsiveContainer>
|
|
{/* Center label */}
|
|
<div className="absolute inset-0 flex items-center justify-center pointer-events-none" style={{ marginTop: '-55px' }}>
|
|
<div className="text-center">
|
|
<p className="text-xl font-extrabold text-[#2563eb]">{total}</p>
|
|
<p className="text-[9px] uppercase tracking-wider text-muted-foreground">Total</p>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
)
|
|
}
|