Census definitions

1 files changed, 44 insertions, 0 deletions

diff --git a/app/Census/AbstractCensusColumn.php b/app/Census/AbstractCensusColumn.php
index 7186ca6ecb..68decf7722 100644
--- a/app/Census/AbstractCensusColumn.php
+++ b/app/Census/AbstractCensusColumn.php
@@ -16,6 +16,7 @@
 namespace Fisharebest\Webtrees\Census;
 
 use Fisharebest\Webtrees\Date;
+use Fisharebest\Webtrees\Individual;
 
 /**
  * Definitions for a census column
@@ -53,6 +54,19 @@ class AbstractCensusColumn {
 	}
 
 	/**
+	 * Extract the country (last part) of a place name.
+	 *
+	 * @param string $place - e.g. "London, England"
+	 *
+	 * @return string - e.g. "England"
+	 */
+	protected function country($place) {
+		$place = explode(', ', $place);
+
+		return end($place);
+	}
+
+	/**
 	 * When did this census occur
 	 *
 	 * @return Date
@@ -62,6 +76,36 @@ class AbstractCensusColumn {
 	}
 
 	/**
+	 * Find the father of an individual
+	 *
+	 * @return Individual|null $individual
+	 */
+	public function father(Individual $individual) {
+		$family = $individual->getPrimaryChildFamily();
+
+		if ($family) {
+			return $family->getHusband();
+		} else {
+			return null;
+		}
+	}
+
+	/**
+	 * Find the mother of an individual
+	 *
+	 * @return Individual|null $individual
+	 */
+	public function mother(Individual $individual) {
+		$family = $individual->getPrimaryChildFamily();
+
+		if ($family) {
+			return $family->getWife();
+		} else {
+			return null;
+		}
+	}
+
+	/**
 	 * Where did this census occur
 	 *
 	 * @return string