1 files changed, 7 insertions, 9 deletions
diff --git a/Affinity.h b/Affinity.h
index 5e7bfe2..c218906 100644
--- a/Affinity.h
+++ b/Affinity.h
@@ -3,20 +3,18 @@
 /*
 htop - Affinity.h
 (C) 2004-2011 Hisham H. Muhammad
-(C) 2020 Red Hat, Inc.  All Rights Reserved.
+(C) 2020,2023 Red Hat, Inc.  All Rights Reserved.
 Released under the GNU GPLv2+, see the COPYING file
 in the source distribution for its full text.
 */
 
-#include "config.h" // IWYU pragma: keep
-
-#include "ProcessList.h"
+#include "Machine.h"
 
 #if defined(HAVE_LIBHWLOC) || defined(HAVE_AFFINITY)
 #include <stdbool.h>
 
 #include "Object.h"
-#include "Process.h"
+#include "Row.h"
 #endif
 
 
@@ -26,13 +24,13 @@ in the source distribution for its full text.
 
 
 typedef struct Affinity_ {
-   ProcessList* pl;
+   Machine* host;
    unsigned int size;
    unsigned int used;
    unsigned int* cpus;
 } Affinity;
 
-Affinity* Affinity_new(ProcessList* pl);
+Affinity* Affinity_new(Machine* host);
 
 void Affinity_delete(Affinity* this);
 
@@ -40,9 +38,9 @@ void Affinity_add(Affinity* this, unsigned int id);
 
 #if defined(HAVE_LIBHWLOC) || defined(HAVE_AFFINITY)
 
-Affinity* Affinity_get(const Process* proc, ProcessList* pl);
+Affinity* Affinity_rowGet(const Row* row, Machine* host);
 
-bool Affinity_set(Process* proc, Arg arg);
+bool Affinity_rowSet(Row* row, Arg arg);
 
 #endif /* HAVE_LIBHWLOC || HAVE_AFFINITY */