From f22551c6f717c012cf52f3562cdbdcade5fe2bfd Mon Sep 17 00:00:00 2001
From: Evan Hall <evan.hall@ligo.org>
Date: Fri, 18 Oct 2019 15:49:32 -0400
Subject: [PATCH] Update noises for CE2 budget

---
 gwinc/ifo/CE2/__init__.py | 63 ++++++++++++++++++++++++++++++++++-----
 1 file changed, 56 insertions(+), 7 deletions(-)

diff --git a/gwinc/ifo/CE2/__init__.py b/gwinc/ifo/CE2/__init__.py
index ca060b51..563e3e05 100644
--- a/gwinc/ifo/CE2/__init__.py
+++ b/gwinc/ifo/CE2/__init__.py
@@ -1,21 +1,70 @@
 from gwinc.ifo.noises import *
 
+class Newtonian(nb.Budget):
+    """Newtonian Gravity
 
-class CE2(nb.Budget):
+    """
 
-    name = 'Cosmic Explorer 2'
+    name = 'Newtonian'
+
+    style = dict(
+        label='Newtonian',
+        color='#15b01a',
+    )
+
+    noises = [
+        NewtonianRayleigh,
+        NewtonianBody,
+        NewtonianInfrasound,
+    ]
+
+class Coating(nb.Budget):
+    """Coating Thermal
+
+    """
+
+    name = 'Coating'
+
+    style = dict(
+        label='Coating Thermal',
+        color='#fe0002',
+    )
 
     noises = [
-        QuantumVacuum,
-        Seismic,
-        Newtonian,
-        AtmosphericInfrasound,
-        SuspensionThermal,
         CoatingBrownian,
         CoatingThermoOptic,
+    ]
+
+class Substrate(nb.Budget):
+    """Substrate Thermal
+
+    """
+
+    name = 'Substrate'
+
+    style = dict(
+        label='Substrate Thermal',
+        color='#fb7d07',
+        linestyle='--',
+    )
+
+    noises = [
         ITMThermoRefractive,
         ITMCarrierDensity,
         SubstrateBrownian,
         SubstrateThermoElastic,
+    ]
+
+class CE2(nb.Budget):
+
+    name = 'Cosmic Explorer 2'
+
+    noises = [
+        QuantumVacuum,
+        Seismic,
+        Newtonian,
+        SuspensionThermal,
+        Coating,
+        Substrate,
         ExcessGas,
     ]
-- 
GitLab