. */ declare(strict_types=1); namespace Fisharebest\Webtrees\Statistics\Google; use Fisharebest\Webtrees\I18N; use Fisharebest\Webtrees\Statistics\Repository\Interfaces\IndividualRepositoryInterface; use Fisharebest\Webtrees\Statistics\Repository\Interfaces\PlaceRepositoryInterface; use Fisharebest\Webtrees\Statistics\Service\CountryService; use Fisharebest\Webtrees\Tree; use Illuminate\Database\Capsule\Manager as DB; use Illuminate\Database\Query\Expression; use Illuminate\Database\Query\JoinClause; use function array_key_exists; use function preg_match_all; use function view; /** * A chart showing the distribution of different events on a map. */ class ChartDistribution { private Tree $tree; private CountryService $country_service; private IndividualRepositoryInterface $individual_repository; private PlaceRepositoryInterface $place_repository; /** * @var array */ private array $country_to_iso3166; /** * @param Tree $tree * @param CountryService $country_service */ public function __construct( Tree $tree, CountryService $country_service, IndividualRepositoryInterface $individual_repository, PlaceRepositoryInterface $place_repository ) { $this->tree = $tree; $this->country_service = $country_service; $this->individual_repository = $individual_repository; $this->place_repository = $place_repository; // Get the country names for each language $this->country_to_iso3166 = $this->getIso3166Countries(); } /** * Returns the country names for each language. * * @return array */ private function getIso3166Countries(): array { // Get the country names for each language $country_to_iso3166 = []; $current_language = I18N::languageTag(); foreach (I18N::activeLocales() as $locale) { I18N::init($locale->languageTag()); $countries = $this->country_service->getAllCountries(); foreach ($this->country_service->iso3166() as $three => $two) { $country_to_iso3166[$three] = $two; $country_to_iso3166[$countries[$three]] = $two; } } I18N::init($current_language); return $country_to_iso3166; } /** * Returns the data structure required by google geochart. * * @param array $places * * @return array */ private function createChartData(array $places): array { $data = [ [ I18N::translate('Country'), I18N::translate('Total'), ], ]; // webtrees uses 3-letter country codes and localised country names, but google uses 2 letter codes. foreach ($places as $country => $count) { $data[] = [ [ 'v' => $country, 'f' => $this->country_service->mapTwoLetterToName($country), ], $count ]; } return $data; } /** * Returns the google geochart data for birth fact. * * @return array */ private function getBirthChartData(): array { // Count how many people were born in each country $surn_countries = []; $b_countries = $this->place_repository->statsPlaces('INDI', 'BIRT', 0, true); foreach ($b_countries as $country => $count) { // Consolidate places (Germany, DEU => DE) if (array_key_exists($country, $this->country_to_iso3166)) { $country_code = $this->country_to_iso3166[$country]; if (array_key_exists($country_code, $surn_countries)) { $surn_countries[$country_code] += $count; } else { $surn_countries[$country_code] = $count; } } } return $this->createChartData($surn_countries); } /** * Returns the google geochart data for death fact. * * @return array */ private function getDeathChartData(): array { // Count how many people were death in each country $surn_countries = []; $d_countries = $this->place_repository->statsPlaces('INDI', 'DEAT', 0, true); foreach ($d_countries as $country => $count) { // Consolidate places (Germany, DEU => DE) if (array_key_exists($country, $this->country_to_iso3166)) { $country_code = $this->country_to_iso3166[$country]; if (array_key_exists($country_code, $surn_countries)) { $surn_countries[$country_code] += $count; } else { $surn_countries[$country_code] = $count; } } } return $this->createChartData($surn_countries); } /** * Returns the google geochart data for marriages. * * @return array */ private function getMarriageChartData(): array { // Count how many families got marriage in each country $surn_countries = []; $m_countries = $this->place_repository->statsPlaces('FAM'); // webtrees uses 3 letter country codes and localised country names, but google uses 2 letter codes. foreach ($m_countries as $place) { // Consolidate places (Germany, DEU => DE) if (array_key_exists($place->country, $this->country_to_iso3166)) { $country_code = $this->country_to_iso3166[$place->country]; if (array_key_exists($country_code, $surn_countries)) { $surn_countries[$country_code] += $place->tot; } else { $surn_countries[$country_code] = $place->tot; } } } return $this->createChartData($surn_countries); } /** * Returns the related database records. * * @param string $surname * * @return array */ private function queryRecords(string $surname): array { $query = DB::table('individuals') ->select(['i_gedcom']) ->join('name', static function (JoinClause $join): void { $join->on('n_id', '=', 'i_id') ->on('n_file', '=', 'i_file'); }) ->where('n_file', '=', $this->tree->id()) ->where(new Expression('n_surn /*! COLLATE ' . I18N::collation() . ' */'), '=', $surname); return $query->get()->all(); } /** * Returns the google geochart data for surnames. * * @param string $surname The surname used to create the chart * * @return array */ private function getSurnameChartData(string $surname): array { if ($surname === '') { $surname = $this->individual_repository->getCommonSurname(); } // Count how many people are events in each country $surn_countries = []; $records = $this->queryRecords($surname); foreach ($records as $row) { if (preg_match_all('/^2 PLAC (?:.*, *)*(.*)/m', $row->i_gedcom, $matches)) { // webtrees uses 3 letter country codes and localised country names, // but google uses 2 letter codes. foreach ($matches[1] as $country) { // Consolidate places (Germany, DEU => DE) if (array_key_exists($country, $this->country_to_iso3166)) { $country_code = $this->country_to_iso3166[$country]; if (array_key_exists($country_code, $surn_countries)) { $surn_countries[$country_code]++; } else { $surn_countries[$country_code] = 1; } } } } } return $this->createChartData($surn_countries); } /** * Returns the google geochart data for individuals. * * @return array */ private function getIndivdualChartData(): array { // Count how many people have events in each country $surn_countries = []; $a_countries = $this->place_repository->statsPlaces('INDI'); // webtrees uses 3 letter country codes and localised country names, but google uses 2 letter codes. foreach ($a_countries as $place) { // Consolidate places (Germany, DEU => DE) if (array_key_exists($place->country, $this->country_to_iso3166)) { $country_code = $this->country_to_iso3166[$place->country]; if (array_key_exists($country_code, $surn_countries)) { $surn_countries[$country_code] += $place->tot; } else { $surn_countries[$country_code] = $place->tot; } } } return $this->createChartData($surn_countries); } /** * Create a chart showing where events occurred. * * @param string $chart_shows The type of chart map to show * @param string $chart_type The type of chart to show * @param string $surname The surname for surname based distribution chart * * @return string */ public function chartDistribution( string $chart_shows = 'world', string $chart_type = '', string $surname = '' ): string { switch ($chart_type) { case 'surname_distribution_chart': $chart_title = I18N::translate('Surname distribution chart') . ': ' . $surname; $data = $this->getSurnameChartData($surname); break; case 'birth_distribution_chart': $chart_title = I18N::translate('Birth by country'); $data = $this->getBirthChartData(); break; case 'death_distribution_chart': $chart_title = I18N::translate('Death by country'); $data = $this->getDeathChartData(); break; case 'marriage_distribution_chart': $chart_title = I18N::translate('Marriage by country'); $data = $this->getMarriageChartData(); break; case 'indi_distribution_chart': default: $chart_title = I18N::translate('Individual distribution chart'); $data = $this->getIndivdualChartData(); break; } return view('statistics/other/charts/geo', [ 'chart_title' => $chart_title, 'chart_color2' => '84beff', 'chart_color3' => 'c3dfff', 'region' => $chart_shows, 'data' => $data, 'language' => I18N::languageTag(), ]); } }