Refactoring + who icd 10 support

.github/workflows/compile_go.yml

@@ -0,0 +1,28 @@
+# This workflow will build a golang project
+# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-go
+
+name: Go
+
+on:
+  push:
+    branches: [ "master" ]
+  pull_request:
+
+jobs:
+
+  build:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Set up Go
+      uses: actions/setup-go@v4
+      with:
+        go-version: '1.23'
+
+    - name: Build
+      run: go build -v -o ptra cmd/main.go
+
+    - name: Test
+      run: cd ptra_test && go test
+

LICENSE.txt

@@ -1,5 +1,5 @@
 PTRA: Patient Trajectory Analysis Library.
-Copyright (c) 2022 imec vzw.
+Copyright (c) 2022-2026 imec vzw.
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU Affero General Public License as

app/filters.go

@@ -269,3 +269,36 @@ func BladderCancerTrajectoryFilter(exp *trajectory.Experiment) trajectory.Trajec
 		return false
 	}
 }
+
+func codeMember(code string, categories []string) bool {
+	for _, cat := range categories {
+		if code == cat {
+			return true
+		}
+	}
+	return false
+}
+
+func Icd10TrajectoryFilter(exp *trajectory.Experiment, codes []string) trajectory.TrajectoryFilter {
+	//Determine all diagnoses that match the given codes, either exactly or either the top-level category
+	codeRelatedMap := map[int]bool{}
+	for did, _ := range exp.NameMap {
+		icdCode := exp.IdMap[did]
+		if len(icdCode) >= 3 {
+			subCode := icdCode[0:3]
+			if codeMember(subCode, codes) {
+				codeRelatedMap[did] = true
+			} else {
+				codeRelatedMap[did] = false
+			}
+		}
+	}
+	return func(t *trajectory.Trajectory) bool {
+		for _, did := range t.Diagnoses {
+			if codeRelatedMap[did] {
+				return true
+			}
+		}
+		return false
+	}
+}