#!/bin/bash
#$Id: cpuinfo,v 1.2 2022/03/16 08:31:51 RogerSeguin Exp $
########################################################################
# Copyright (c) 2022 Roger Seguin <rgrsgn@gmail.com>
#				https://frogzie.duckdns.org

# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:

# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.

# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
########################################################################

# Description:
#	Display processor's thread/core mapping

[ $# -eq 0 ] || {
	prg="$(basename $0)"
cat << EndUsage:
# $prg - Display processor's thread/core mapping
	Usage:	$prg	(no parameters)
EndUsage:
	exit 1
}

declare cps=$(lscpu | grep 'Core(s) per socket:' | awk '{print $4;}')
declare tpc=$(lscpu | grep 'Thread(s) per core:' | awk '{print $4;}')
declare -a map=($(grep 'core id' /proc/cpuinfo | awk '{print $4;}'))

printf "$cps cores\n"
printf "$tpc threads per core\n"
echo "Thread/core mapping: ${map[@]}"

declare nvp=$((cps * tpc))	# Number of virtual processors
for ((c = 0; c < cps; c++)); do
	for ((t = 0; t < nvp; t++)); do
		[ ${map[t]} -eq $c ] && eval core$c+=\($t\)
	done
done

for ((c = 0; c < cps; c++)); do
	printf "\tcore$c: threads $(eval echo \${core$c[@]})\n"
done

exit 0