From ace7ce7b21fd777fd5392c3c11e510390c7f2039 Mon Sep 17 00:00:00 2001
From: Vahid Ahmadi <va.vahidahmadi@gmail.com>
Date: Mon, 12 May 2025 12:01:48 +0100
Subject: [PATCH 1/2] Add PRCITSHP variable to census_cps.py

---
 policyengine_us_data/datasets/cps/census_cps.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/policyengine_us_data/datasets/cps/census_cps.py b/policyengine_us_data/datasets/cps/census_cps.py
index 287f54f5..eb13916b 100644
--- a/policyengine_us_data/datasets/cps/census_cps.py
+++ b/policyengine_us_data/datasets/cps/census_cps.py
@@ -303,4 +303,5 @@ class CensusCPS_2018(CensusCPS):
     "POTC_VAL",
     "PMED_VAL",
     "PEMCPREM",
+    "PRCITSHP",
 ]

From 1496832422086f65d59b1aa2146bc202c683f657 Mon Sep 17 00:00:00 2001
From: Vahid Ahmadi <va.vahidahmadi@gmail.com>
Date: Mon, 12 May 2025 12:43:31 +0100
Subject: [PATCH 2/2] Add add_ssn_card_type function to cps.py

---
 policyengine_us_data/datasets/cps/cps.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/policyengine_us_data/datasets/cps/cps.py b/policyengine_us_data/datasets/cps/cps.py
index 26579b82..ac5ec01f 100644
--- a/policyengine_us_data/datasets/cps/cps.py
+++ b/policyengine_us_data/datasets/cps/cps.py
@@ -648,6 +648,25 @@ def add_previous_year_income(self, cps: h5py.File) -> None:
     ].values
 
 
+def add_ssn_card_type(cps: h5py.File, person: pd.DataFrame) -> None:
+    """
+    Deterministically assign SSA card type based on PRCITSHP (citizenship status).
+
+    Mapping:
+    - PRCITSHP 1–4: Citizens or naturalized → code 1 (unrestricted SSN)
+    - PRCITSHP 5: Non-citizens → code 0 (default, ITIN or restricted)
+    """
+    # Default to 0 (no SSN / ITIN only)
+    ssn_card_type = np.full(len(person), 0)
+
+    # Assign code 1 for citizens (native/naturalized)
+    citizen_mask = person.PRCITSHP.isin([1, 2, 3, 4])
+    ssn_card_type[citizen_mask] = 1
+
+    # Save to CPS
+    cps["ssn_card_type"] = ssn_card_type
+
+
 class CPS_2019(CPS):
     name = "cps_2019"
     label = "CPS 2019"