Add beam current to wall display

Author: Tom-Willemsen

app/components/InstrumentsDisplay.tsx

@@ -8,10 +8,12 @@ import {
 } from "@/app/types";
 import useWebSocket from "react-use-websocket";
 import { instListPV, instListSubscription, socketURL } from "@/app/commonVars";
+import { IfcPVWSRequest, PVWSRequestType } from "@/app/types";
 import { instListFromBytes } from "@/app/components/dehex_and_decompress";
 import {
   updateInstrumentRunstate,
   updateInstrumentRunstatePV,
+  updateTargetStationBeamCurrent,
 } from "@/app/wall/utils";
 import TargetStation from "@/app/components/TargetStation";
 import ScienceGroup from "@/app/components/ScienceGroup";
@@ -52,6 +54,7 @@ export default function InstrumentsDisplay({
   const [data, setData] = useState<Array<targetStation>>([
     {
       targetStation: "Target Station 1",
+      beamCurrentPv: "AC:TS1:BEAM:CURR",
       instruments: [
         { name: "ALF" },
         { name: "ARGUS" },
@@ -86,6 +89,7 @@ export default function InstrumentsDisplay({
     },
     {
       targetStation: "Target Station 2",
+      beamCurrentPv: "AC:TS2:BEAM:CURR",
       instruments: [
         { name: "CHIPIR" },
         { name: "IMAT" },
@@ -146,10 +150,21 @@ export default function InstrumentsDisplay({
     shouldReconnect: (closeEvent) => true,
   });
 
+  let targetStationCurrentPvs = []
+  for (const ts of data) {
+    if (ts.beamCurrentPv != undefined) {
+      targetStationCurrentPvs.push(ts.beamCurrentPv)
+    }
+  }
+
   useEffect(() => {
     // On page load, subscribe to the instrument list as it's required to get each instrument's PV prefix.
     sendJsonMessage(instListSubscription);
-  }, [sendJsonMessage]);
+    sendJsonMessage({
+      type: PVWSRequestType.subscribe,
+      pvs: targetStationCurrentPvs,
+    })
+  }, [sendJsonMessage, targetStationCurrentPvs]);
 
   useEffect(() => {
     // This is a PV update, it could be either the instlist or an instrument's runstate that has changed
@@ -161,25 +176,30 @@ export default function InstrumentsDisplay({
     const updatedPVName: string = updatedPV.pv;
     const updatedPVbytes: string | null | undefined = updatedPV.b64byt;
     let updatedPVvalue: string | null | undefined = updatedPV.text;
+    let updatedPVnum: number = updatedPV.value;
 
     if (updatedPVName == instListPV && updatedPVbytes != null) {
       const instListDict = instListFromBytes(updatedPVbytes);
       for (const instrument of instListDict) {
         setData((prev) => {
           return updateInstrumentRunstatePV(
-            prev,
-            instrument,
-            runstatePV,
-            sendJsonMessage,
+              prev,
+              instrument,
+              runstatePV,
+              sendJsonMessage,
           );
         });
       }
+    } else if (targetStationCurrentPvs.includes(updatedPVName)) {
+      setData((prev) => {
+        return updateTargetStationBeamCurrent(prev, updatedPVName, updatedPVnum);
+      });
     } else if (updatedPVvalue) {
       setData((prev) => {
         return updateInstrumentRunstate(prev, updatedPVName, updatedPVvalue);
       });
     }
-  }, [lastJsonMessage, sendJsonMessage]);
+  }, [lastJsonMessage, sendJsonMessage, targetStationCurrentPvs]);
 
   return (
     <div>
@@ -198,6 +218,7 @@ export default function InstrumentsDisplay({
               key={targetStation.targetStation}
               name={targetStation.targetStation}
               instruments={targetStation.instruments}
+              beamCurrent={targetStation.beamCurrent}
             />
           );
         })}

app/components/JenkinsJobs.tsx

@@ -41,7 +41,6 @@ export default function JenkinsJobs() {
       const jobs: Array<IfcWallDisplayJob> = resData["jobs"].filter(
         (job) => job["color"] != "disabled",
       );
-      console.log(jobs);
       setData(jobs);
     }
     fetchPosts();

app/components/TargetStation.test.tsx

@@ -5,6 +5,7 @@ it("renders targetstation unchanged", () => {
   const { container } = render(
     <TargetStation
       name={"TS5"}
+      beamCurrent={1234.5678}
       instruments={[
         { name: "Instrument", runstate: "RESUMING" },
         {

app/components/TargetStation.tsx

@@ -5,14 +5,16 @@ import { IfcInstrumentStatus } from "@/app/types";
 export default function TargetStation({
   name,
   instruments,
+  beamCurrent,
 }: {
   name: string;
   instruments: Array<IfcInstrumentStatus>;
+  beamCurrent: number;
 }) {
   return (
     <div className="flex flex-col justify-center items-start w-full">
       <h1 className="w-full text-left text-gray-600 dark:text-gray-200 font-semibold text-md mt-2 py-2 ">
-        {name}
+        {name} {beamCurrent !== undefined ? ("- " + beamCurrent.toFixed(2) + " μA") : ""}
       </h1>
       <div className="flex flex-wrap gap-1">
         {instruments.map((instrument) => (

app/components/__snapshots__/TargetStation.test.tsx.snap

@@ -9,6 +9,8 @@ exports[`renders targetstation unchanged 1`] = `
       class="w-full text-left text-gray-600 dark:text-gray-200 font-semibold text-md mt-2 py-2 "
     >
       TS5
+       
+      - 1234.57 μA
     </h1>
     <div
       class="flex flex-wrap gap-1"

app/wall/utils.ts

@@ -26,6 +26,28 @@ export function updateInstrumentRunstate(
   return newData;
 }
 
+/**
+ * Copy the original array, update the given target stations' beam current value, then return the copied array.
+ * @param prev the previous array of target stations, containing instruments.
+ * @param updatedPVName the runstate PV address
+ * @param updatedPVvalue the runstate
+ */
+export function updateTargetStationBeamCurrent(
+  prev: Array<targetStation>,
+  updatedPVName: string,
+  updatedPVvalue: string,
+) {
+  const newData: Array<targetStation> = [...prev];
+
+  let foundTs = newData.find(
+    (ts) => ts.beamCurrentPv === updatedPVName,
+  );
+  if (foundTs) {
+    foundTs.beamCurrent = updatedPVvalue;
+  }
+  return newData;
+}
+
 /**
  * Copy an original array then update an instrument's runstate PV, then subscribe to it. return the copied array.
  * @param prev the original array of target stations containing instrument runstate information