Changeset 4:bcc93ecb8cb6

Timestamp:

01/29/12 14:00:30 (13 years ago)

Author:

István Váradi <ivaradi@…>

Branch:

default

Phase:

public

Message:

The aircraft model framework is basically implemented

Location:

src

Files:

• acft.py (added)
• const.py (added)
• fs.py (modified) (2 diffs)
• fsuipc.py (modified) (11 diffs)
• test.py (added)
• util.py (added)

src/fs.py

@@ -1 +1 @@
 # Module for generic flight-simulator interfaces
 
-# Flight simulator type: MS Flight Simulator 2004
-TYPE_FS2K4 = 1
+#-------------------------------------------------------------------------------
 
-# Flight simulator type: MS Flight Simulator X
-TYPE_FSX = 2
+import const
 
-# Flight simulator type: X-Plane 9
-TYPE_XPLANE9 = 3
+#-------------------------------------------------------------------------------
 
-# Flight simulator type: X-Plane 10
-TYPE_XPLANE10 = 4
-
-class ConnectionListener:
+class ConnectionListener(object):
     """Base class for listeners on connections to the flight simulator."""
     def connected(self, fsType, descriptor):
@@ -24 +18 @@
         print "fs.ConnectionListener.disconnected"
 
+#-------------------------------------------------------------------------------
+
+class SimulatorException(Exception):
+    """Exception thrown by the simulator interface for communication failure."""
+
+#-------------------------------------------------------------------------------
+
+def createSimulator(type, connectionListener):
+    """Create a simulator instance for the given simulator type with the given
+    connection listener.
+
+    The returned object should provide the following members:
+    FIXME: add info
+    """
+    assert type==const.TYPE_MSFS9, "Only MS Flight Simulator 2004 is supported"
+    import fsuipc
+    return fsuipc.Simulator(connectionListener)
+
+#-------------------------------------------------------------------------------
+
+class AircraftState(object):
+    """Base class for the aircraft state produced by the aircraft model based
+    on readings from the simulator.
+
+    The following data members should be provided at least:
+    - timestamp: the simulator time of the measurement in seconds since the
+    epoch (float)
+    - paused: a boolean indicating if the flight simulator is paused for
+    whatever reason (it could be a pause mode, or a menu, a dialog, or a
+    replay, etc.)
+    - trickMode: a boolean indicating if some "trick" mode (e.g. "SLEW" in
+    MSFS) is activated
+    - overspeed: a boolean indicating if the aircraft is in overspeed
+    - stalled: a boolean indicating if the aircraft is stalled
+    - onTheGround: a boolean indicating if the aircraft is on the ground
+    - grossWeight: the gross weight in kilograms (float)
+    - heading: the heading of the aircraft in degrees (float)
+    - pitch: the pitch of the aircraft in degrees. Positive means pitch down,
+    negative means pitch up (float)
+    - bank: the bank of the aircraft in degrees. Positive means bank left,
+    negative means bank right (float)
+    - ias: the indicated airspeed in knots (float)    
+    - vs: the vertical speed in feet/minutes (float)
+    - altitude: the altitude of the aircraft in feet (float)
+    - flapsSet: the selected degrees of the flaps (float)
+    - flaps: the actual degrees of the flaps (float)
+    - n1[]: the N1 values of the turbine engines (array of floats
+    of as many items as the number of engines, present only for aircraft with
+    turbines)
+    - rpm[]: the RPM values of the piston engines (array of floats
+    of as many items as the number of engines, present only for aircraft with
+    pistons)
+    - reverser[]: an array of booleans indicating if the thrust reversers are
+    activated on any of the engines. The number of items equals to the number
+    of engines.
+    - navLightsOn: a boolean indicating if the navigation lights are on
+    - antiCollisionLightsOn: a boolean indicating if the anti-collision lights are on
+    - strobeLightsOn: a boolean indicating if the strobe lights are on
+    - langingLightsOn: a boolean indicating if the landing lights are on
+    - pitotHeatOn: a boolean indicating if the pitot heat is on
+    - gearsDown: a boolean indicating if the gears are down
+    - spoilersArmed: a boolean indicating if the spoilers have been armed for
+    automatic deployment
+    - spoilersExtension: the percentage of how much the spoiler is extended
+    (float) 
+
+    FIXME: needed when taxiing only:
+    - zfw: the Zero Fuel Weight in klograms (float)
+
+    FIXME: needed for touchdown only:
+    - ambientWindDirection: the ambient wind direction around the aircraft (float)
+    - ambientWindSpeed: the ambient wind speed around the aircraft in knowns (float)
+    - tdRate: the touchdown rate calculated by FSUIPC (float)
+    """
+    

src/fsuipc.py

@@ -1 +1 @@
 # Module handling the connection to FSUIPC
+
+#------------------------------------------------------------------------------
+
+import fs
+import const
+import util
 
 import threading
 import os
-import fs
 import time
+import calendar
+import sys
 
 if os.name == "nt":
@@ -11 +18 @@
     import pyuipc_emu as pyuipc
 
+#------------------------------------------------------------------------------
+
 class Handler(threading.Thread):
     """The thread to handle the FSUIPC requests."""
+    @staticmethod
+    def _callSafe(fun):
+        """Call the given function and swallow any exceptions."""
+        try:
+            return fun()
+        except Exception, e:
+            print >> sys.stderr, str(e)
+            return None
+
     class Request(object):
         """A simple, one-shot request."""
@@ -23 +41 @@
             
         def process(self, time):
-            """Process the request."""
+            """Process the request."""            
             if self._forWrite:
                 pyuipc.write(self._data)
-                self._callback(self._extra)
+                Handler._callSafe(lambda: self._callback(True, self._extra))
             else:
                 values = pyuipc.read(self._data)
-                self._callback(values, self._extra)            
+                Handler._callSafe(lambda: self._callback(values, self._extra))
 
             return True
+
+        def fail(self):
+            """Handle the failure of this request."""
+            if self._forWrite:
+                Handler._callSafe(lambda: self._callback(False, self._extra))
+            else:
+                Handler._callSafe(lambda: self._callback(None, self._extra))
 
     class PeriodicRequest(object):
@@ -68 +93 @@
             values = pyuipc.read(self._preparedData)
 
-            self._callback(values, self._extra)
+            Handler._callSafe(lambda: self._callback(values, self._extra))
 
             while self._nextFire <= time:
@@ -74 +99 @@
             
             return True
+
+        def fail(self):
+            """Handle the failure of this request."""
+            pass
 
         def __cmp__(self, other):
@@ -103 +132 @@
 
         callback is a function that receives two pieces of data:
-        - the values retrieved
+        - the values retrieved or None on error
         - the extra parameter
 
@@ -120 +149 @@
         - the data to write
 
-        callback is a function that receives the extra data when writing was
-        succesful. It will called in the handler's thread!
+        callback is a function that receives two pieces of data:
+        - a boolean indicating if writing was successful
+        - the extra data
+        It will be called in the handler's thread!
         """
         with self._requestCondition:
             self._requests.append(Handler.Request(True, data, callback, extra))
             self._requestCondition.notify()
+
+    @staticmethod
+    def _readWriteCallback(data, extra):
+        """Callback for the read() and write() calls below."""
+        extra.append(data)
+        with extra[0] as condition:
+            condition.notify()
+
+    def read(self, data):
+        """Read the given data synchronously.
+
+        If a problem occurs, an exception is thrown."""
+        with threading.Condition() as condition:
+            extra = [condition]
+            self._requestRead(data, self._readWriteCallback, extra)
+            while len(extra)<2:
+                condition.wait()
+            if extra[1] is None:
+                raise fs.SimulatorException("reading failed")
+            else:
+                return extra[1]
+
+    def write(self, data):
+        """Write the given data synchronously.
+
+        If a problem occurs, an exception is thrown."""
+        with threading.Condition() as condition:
+            extra = [condition]
+            self._requestWrite(data, self._writeCallback, extra)
+            while len(extra)<2:
+                condition.wait()
+            if extra[1] is None:
+                raise fs.SimulatorException("writing failed")
 
     def requestPeriodicRead(self, period, data, callback, extra = None):
@@ -189 +253 @@
                     (pyuipc.fsuipc_version, pyuipc.lib_version, 
                      pyuipc.fs_version)
-                self._connectionListener.connected(fs.TYPE_FS2K4, description)
+                Handler._callSafe(lambda:     
+                                  self._connectionListener.connected(const.TYPE_MSFS9, 
+                                                                     description))
                 return True
             except Exception, e:
                 print "fsuipc.Handler._connect: connection failed: " + str(e)
+                time.sleep(0.1)
 
         return False
@@ -212 +279 @@
         """Disconnect from the flight simulator."""
         pyuipc.close()
-        self._connectionListener.disconnected()
+        Handler._callSafe(lambda: self._connectionListener.disconnected())
+
+    def _failRequests(self, request):
+        """Fail the outstanding, single-shot requuests."""
+        request.fail()
+        with self._requestCondition:
+            for request in self._requests:
+                try:
+                    self._requestCondition.release()
+                    request.fail()
+                finally:
+                    self._requestCondition.acquire()
+            self._requests = []        
 
     def _processRequest(self, request, time):
@@ -230 +309 @@
                 str(e) + ") reconnecting."
             self._disconnect()
+            self._failRequests(request)
             if not self._connect(): return None
             else: return True
@@ -254 +334 @@
 
         return self._connectionRequested
+
+#------------------------------------------------------------------------------
+
+class Simulator(object):
+    """The simulator class representing the interface to the flight simulator
+    via FSUIPC."""
+    def __init__(self, connectionListener):
+        """Construct the simulator."""
+        self._handler = Handler(connectionListener)
+        self._handler.start()
+        self._aircraft = None
+        self._aircraftName = None
+        self._aircraftModel = None
+        self._monitoringRequestID = None
+
+    def connect(self):
+        """Initiate a connection to the simulator."""
+        self._handler.connect()
+
+    def startMonitoring(self, aircraft):
+        """Start the periodic monitoring of the aircraft and pass the resulting
+        state to the given aircraft object periodically.
+        
+        The aircraft object passed must provide the following members:
+        - type: one of the AIRCRAFT_XXX constants from const.py
+        - modelChanged(aircraftName, modelName): called when the model handling
+        the aircraft has changed.
+        - handleState(aircraftState): handle the given state."""
+        assert self._aircraft is None
+
+        self._aircraft = aircraft
+        self._startMonitoring()
+
+    def stopMonitoring(self):
+        """Stop the periodic monitoring of the aircraft."""
+        self._aircraft = None
+
+    def disconnect(self):
+        """Disconnect from the simulator."""
+        self._handler.disconnect()
+
+    def _startMonitoring(self):
+        """The internal call to start monitoring."""
+        self._handler.requestRead([(0x3d00, -256)], self._monitoringStartCallback)
+
+    def _monitoringStartCallback(self, data, extra):
+        """Callback for the data read when the monitoring has started.
+        
+        The following data items are expected:
+        - the name of the aircraft
+        """
+        if self._aircraft is None:
+            return
+        elif data is None:
+            self._startMonitoring()
+        else:
+            self._setAircraftName(data[0])            
+
+    def _setAircraftName(self, name):
+        """Set the name of the aicraft and if it is different from the
+        previous, create a new model for it.
+        
+        If so, also notifty the aircraft about the change."""
+        if name==self._aircraftName:
+            return
+
+        self._aircraftName = name
+        if self._aircraftModel is None or \
+           not self._aircraftModel.doesHandle(name):            
+            self._setAircraftModel(AircraftModel.create(self._aircraft, name))
+        
+        self._aircraft.modelChanged(self._aircraftName, 
+                                    self._aircraftModel.name)
+
+    def _setAircraftModel(self, model):
+        """Set a new aircraft model.
+
+        It will be queried for the data to monitor and the monitoring request
+        will be replaced by a new one."""
+        self._aircraftModel = model
+        
+        if self._monitoringRequestID is not None:
+            self._handler.clearPeriodic(self._monitoringRequestID)
+            
+        self._monitoringRequestID = \
+            self._handler.requestPeriodicRead(1.0, 
+                                              model.getMonitoringData(),
+                                              self._handleMonitoringData)
+
+    def _handleMonitoringData(self, data, extra):
+        """Handle the monitoring data."""
+        if self._aircraft is None:
+            self._handler.clearPeriodic(self._monitoringRequestID)
+            return
+
+        self._setAircraftName(data[0])
+        aircraftState = self._aircraftModel.getAircraftState(self._aircraft, data)
+        self._aircraft.handleState(aircraftState)
+
+#------------------------------------------------------------------------------
+
+class AircraftModel(object):
+    """Base class for the aircraft models.
+
+    Aircraft models handle the data arriving from FSUIPC and turn it into an
+    object describing the aircraft's state."""
+    monitoringData = [("aircraftName", 0x3d00, -256),
+                      ("year", 0x0240, "H"),
+                      ("dayOfYear", 0x023e, "H"),
+                      ("zuluHour", 0x023b, "b"),
+                      ("zuluMinute", 0x023c, "b"),
+                      ("seconds", 0x023a, "b"),
+                      ("paused", 0x0264, "H"),
+                      ("frozen", 0x3364, "H"),
+                      ("slew", 0x05dc, "H"),
+                      ("overspeed", 0x036d, "b"),
+                      ("stalled", 0x036c, "b"),
+                      ("onTheGround", 0x0366, "H"),
+                      ("grossWeight", 0x30c0, "f"),
+                      ("heading", 0x0580, "d"),
+                      ("pitch", 0x0578, "d"),
+                      ("bank", 0x057c, "d"),
+                      ("ias", 0x02bc, "d"),
+                      ("vs", 0x02c8, "d"),
+                      ("altitude", 0x0570, "l"),
+                      ("flapsControl", 0x0bdc, "d"),
+                      ("flapsLeft", 0x0be0, "d"),
+                      ("flapsRight", 0x0be4, "d"),
+                      ("lights", 0x0d0c, "H"),
+                      ("pitot", 0x029c, "b"),
+                      ("noseGear", 0x0bec, "d"),
+                      ("spoilersArmed", 0x0bcc, "d"),
+                      ("spoilers", 0x0bd0, "d")]
+
+    @staticmethod
+    def create(aircraft, aircraftName):
+        """Create the model for the given aircraft name, and notify the
+        aircraft about it."""        
+        return AircraftModel([0, 10, 20, 30])
+
+    def __init__(self, flapsNotches):
+        """Construct the aircraft model.
+        
+        flapsNotches is a list of degrees of flaps that are available on the aircraft."""
+        self._flapsNotches = flapsNotches
+    
+    @property
+    def name(self):
+        """Get the name for this aircraft model."""
+        return "FSUIPC/Generic"
+    
+    def doesHandle(self, aircraftName):
+        """Determine if the model handles the given aircraft name.
+        
+        This default implementation returns True."""
+        return True
+
+    def addDataWithIndexMembers(self, dest, prefix, data):
+        """Add FSUIPC data to the given array and also corresponding index
+        member variables with the given prefix.
+
+        data is a list of triplets of the following items:
+        - the name of the data item. The index member variable will have a name
+        created by prepending the given prefix to this name.
+        - the FSUIPC offset
+        - the FSUIPC type
+        
+        The latter two items will be appended to dest."""
+        index = len(dest)
+        for (name, offset, type) in data:
+            setattr(self, prefix + name, index)
+            dest.append((offset, type))
+            index += 1
+            
+    def getMonitoringData(self):
+        """Get the data specification for monitoring.
+        
+        The first item should always be the aircraft name (0x3d00, -256)."""
+        data = []
+        self.addDataWithIndexMembers(data, "_monidx_",
+                                     AircraftModel.monitoringData)
+        return data
+    
+    def getAircraftState(self, aircraft, data):
+        """Get an aircraft state object for the given monitoring data."""
+        state = fs.AircraftState()
+        
+        timestamp = calendar.timegm(time.struct_time([data[self._monidx_year],
+                                                      1, 1, 0, 0, 0, -1, 1, 0]))
+        timestamp += data[self._monidx_dayOfYear] * 24 * 3600
+        timestamp += data[self._monidx_zuluHour] * 3600
+        timestamp += data[self._monidx_zuluMinute] * 60
+        timestamp += data[self._monidx_seconds]        
+        state.timestamp = timestamp
+        
+        state.paused = data[self._monidx_paused]!=0 or \
+            data[self._monidx_frozen]!=0
+        state.trickMode = data[self._monidx_slew]!=0
+
+        state.overspeed = data[self._monidx_overspeed]!=0
+        state.stalled = data[self._monidx_stalled]!=0
+        state.onTheGround = data[self._monidx_onTheGround]!=0
+
+        state.grossWeight = data[self._monidx_grossWeight] * util.LBSTOKG
+        
+        state.heading = data[self._monidx_heading]*360.0/65536.0/65536.0
+        if state.heading<0.0: state.heading += 360.0
+
+        state.pitch = data[self._monidx_pitch]*360.0/65536.0/65536.0
+        state.bank = data[self._monidx_bank]*360.0/65536.0/65536.0
+
+        state.ias = data[self._monidx_ias]/128.0
+        state.vs = data[self._monidx_vs]*60.0*3.28984/256.0
+
+        state.altitude = data[self._monidx_altitude]*3.28084/65536.0/65536.0
+        
+        numNotchesM1 = len(self._flapsNotches) - 1
+        flapsIncrement = 16383 / numNotchesM1
+        flapsControl = data[self._monidx_flapsControl]
+        flapsIndex = flapsControl / flapsIncrement
+        if flapsIndex < numNotchesM1:
+            if (flapsControl - (flapsIndex*flapsIncrement) >
+                (flapsIndex+1)*flapsIncrement - flapsControl):
+                flapsIndex += 1
+        state.flapsSet = self._flapsNotches[flapsIndex]
+        
+        flapsLeft = data[self._monidx_flapsLeft]
+        flapsIndex = flapsLeft / flapsIncrement
+        state.flaps = self._flapsNotches[flapsIndex]
+        if flapsIndex != numNotchesM1:
+            thisNotch = flapsIndex * flapsIncrement
+            nextNotch = thisNotch + flapsIncrement
+            
+            state.flaps += (self._flapsNotches[flapsIndex+1] - state.flaps) * \
+                (flapsLeft - thisNotch) / (nextNotch - thisNotch)
+        
+        lights = data[self._monidx_lights]
+        
+        state.navLightsOn = (lights%0x01) != 0
+        state.antiCollisionLightsOn = (lights%0x02) != 0
+        state.landingLightsOn = (lights%0x04) != 0
+        state.strobeLightsOn = (lights%0x10) != 0
+        
+        state.pitotHeatOn = data[self._monidx_pitot]!=0
+
+        state.gearsDown = data[self._monidx_noseGear]==16383
+
+        state.spoilersArmed = data[self._monidx_spoilersArmed]!=0
+        
+        spoilers = data[self._monidx_spoilers]
+        if spoilers<=4800:
+            state.spoilersExtension = 0.0
+        else:
+            state.spoilersExtension = (spoilers - 4800) * 100.0 / (16383 - 4800)
+        
+        return state