1 files changed, 20 insertions, 13 deletions
diff --git a/Project/DataReport.h b/Project/DataReport.h
index 61a7ccf..f262cb9 100644
--- a/Project/DataReport.h
+++ b/Project/DataReport.h
@@ -8,24 +8,25 @@ class ElectricCalculation;
 
 class DataReport : public DataReportBase
 {
-public:
-   enum GridSelection {
-       GRID_ALL = 0,
-       GRID_PF,
-       GRID_PFBUSES,
-       GRID_PFBRANCHES,
-       GRID_FAULT,
-       GRID_FAULTBUSES,
-       GRID_FAULTBRANCHES,
-       GRID_FAULTGENERATORS,
-   };
-   
+   public:
+    enum GridSelection {
+        GRID_ALL = 0,
+        GRID_PF,
+        GRID_PFBUSES,
+        GRID_PFBRANCHES,
+        GRID_FAULT,
+        GRID_FAULTBUSES,
+        GRID_FAULTBRANCHES,
+        GRID_FAULTGENERATORS,
+    };
+
     DataReport(wxWindow* parent, Workspace* workspace);
     virtual ~DataReport();
 
     virtual void SetHeaders();
     virtual void CreateGrids();
     virtual void FillValues(GridSelection gridToFill = GRID_ALL);
+    virtual void SetRowsColours(wxGrid* grid, int rowStart = 1);
 
    protected:
     virtual void OnFaulrGridChanged(wxGridEvent& event);
@@ -37,7 +38,7 @@ public:
     virtual void OnPFBusGridChanged(wxGridEvent& event);
     Workspace* m_workspace = NULL;
     bool m_changingValues = false;
-    
+
     // Headers choices
     wxArrayString m_voltageChoices;
     wxArrayString m_activePowerChoices;
@@ -46,5 +47,11 @@ public:
     wxArrayString m_indReactanceChoices;
     wxArrayString m_capSusceptanceChoices;
     wxArrayString m_currentChoices;
+
+    // Colors
+    wxColour m_headerColour;
+    wxColour m_offlineColour;
+    wxColour m_oddRowColour;
+    wxColour m_evenRowColour;
 };
 #endif  // DATAREPORT_H