workbench/tools/SysExplorer/Modules/ATA/ata_enum.c

   1 /*
   2     Copyright (C) 2015-2019, The AROS Development Team.
   3     $Id$
   4 */
   5
   6 #include <aros/debug.h>
   7
   8 #define __STORAGE_NOLIBBASE__
   9
  10 #include <proto/sysexp.h>
  11 #include <proto/storage.h>
  12
  13 #include <proto/alib.h>
  14 #include <proto/dos.h>
  15 #include <proto/exec.h>
  16 #include <proto/muimaster.h>
  17 #include <proto/oop.h>
  18 #include <proto/utility.h>
  19 #include <proto/intuition.h>
  20
  21 #include <zune/customclasses.h>
  22 #include <mui/NListtree_mcc.h>
  23 #include <mui/NListview_mcc.h>
  24 #include <utility/tagitem.h>
  25 #include <utility/hooks.h>
  26
  27 #include <hidd/storage.h>
  28 #include <hidd/hidd.h>
  29 #include <hidd/ata.h>
  30
  31 #include "locale.h"
  32 #include "storage.h"
  33
  34 #include "ata_classes.h"
  35 #include "ata_intern.h"
  36
  37 extern int ATABusWindow_Initialize(struct SysexpBase *SysexpBase);
  38 extern void ATABusWindow_Deinitialize(struct SysexpBase *SysexpBase);
  39 extern int ATAUnitWindow_Initialize(struct SysexpBase *SysexpBase);
  40 extern void ATAUnitWindow_Deinitialize(struct SysexpBase *SysexpBase);
  41
  42 #if (1) // TODO : Move into libbase
  43 extern OOP_AttrBase HiddStorageUnitAB;
  44 extern OOP_AttrBase HiddATABusAB;
  45 extern OOP_AttrBase HiddATAUnitAB;
  46
  47 extern OOP_MethodID HiddStorageControllerBase;
  48 #endif
  49
  50 AROS_UFH3S(BOOL, ataunitenumFunc,
  51     AROS_UFHA(struct Hook *, h,  A0),
  52     AROS_UFHA(OOP_Object*, unit, A2),
  53     AROS_UFHA(void *, parent, A1))
  54 {
  55     AROS_USERFUNC_INIT
  56
  57     struct SysexpEnum_data *edata = (struct SysexpEnum_data *)h->h_Data;
  58     struct InsertObjectMsg msg =
  59     {
  60         .obj      = unit,
  61     };
  62     CONST_STRPTR name;
  63     struct SysexpBase *SysexpBase;
  64     struct SysexpATABase *AtaBase;
  65
  66     D(bug("[ata.sysexp] %s()\n", __func__));
  67
  68     if (!edata)
  69         return TRUE;
  70
  71     SysexpBase = edata->ed_sysexpbase;
  72
  73     D(bug("[ata.sysexp] ataunit @ %p\n", unit));
  74     if (!unit)
  75         return TRUE;
  76
  77     AtaBase = GetBase("ATA.Module");
  78     msg.winClass = AtaBase->seab_ATAUnitWindowCLASS;
  79     SysexpBase->GlobalCount++;
  80
  81     OOP_GetAttr(unit, aHidd_StorageUnit_Model, (IPTR *)&name);
  82     DoMethod(SysexpBase->sesb_Tree, MUIM_NListtree_Insert, name, &msg,
  83              parent, MUIV_NListtree_Insert_PrevNode_Tail, 0);
  84
  85     return FALSE; /* Continue enumeration */
  86
  87     AROS_USERFUNC_EXIT
  88 }
  89
  90 struct SysexpStorageBase *StorageBase;
  91 static struct SysexpEnum_data privatehookdata =
  92 {
  93     NULL,
  94     NULL
  95 };
  96
  97 static void ataBusEnum(OOP_Object *obj, struct MUI_NListtree_TreeNode *parent)
  98 {
  99     D(bug("[ata.sysexp] atabus: \n"));
 100
 101     EnumBusUnits(obj, parent, ataunitenumFunc, &privatehookdata);
 102 }
 103
 104 void ATAStartup(struct SysexpBase *SysexpBase)
 105 {
 106     struct SysexpATABase *AtaBase;
 107     D(bug("[ata.sysexp] %s(%p)\n", __func__, SysexpBase));
 108
 109     StorageBase = GetBase("Storage.Module");
 110     D(bug("[ata.sysexp] %s: StorageBase @ %p\n", __func__, StorageBase));
 111     if (!StorageBase)
 112     {
 113             __showerror
 114         ( (char *)
 115             "ATA enumerator requires `storage.sysexp'.", NULL
 116         );
 117     }
 118     AtaBase = GetBase("ATA.Module");
 119     D(bug("[ata.sysexp] %s: AtaBase @ %p\n", __func__, AtaBase));
 120
 121     privatehookdata.ed_sysexpbase = SysexpBase;
 122
 123     AtaBase->seab_DevicePageCLASS = GetBase("DevicePage.Class");
 124     AtaBase->seab_GenericWindowCLASS = GetBase("GenericWindow.Class");
 125
 126     ATABusWindow_Initialize(SysexpBase);
 127     ATAUnitWindow_Initialize(SysexpBase);
 128
 129     RegisterStorageControllerHandler(CLID_Hidd_ATA, 90, AtaBase->seab_GenericWindowCLASS, NULL, NULL);
 130     RegisterStorageBusHandler(CLID_Hidd_ATABus, 90, AtaBase->seab_ATABusWindowCLASS, ataBusEnum, NULL);
 131 }
 132
 133 void ATAShutdown(struct SysexpBase *SysexpBase)
 134 {
 135     D(bug("[ata.sysexp] %s(%p)\n", __func__, SysexpBase));
 136
 137     ATAUnitWindow_Deinitialize(SysexpBase);
 138     ATABusWindow_Deinitialize(SysexpBase);
 139 }