X-Git-Url: http://plrg.eecs.uci.edu/git/?p=smartthings-infrastructure.git;a=blobdiff_plain;f=Extractor%2FExtractorScript.py;h=3a87c2389d32699e6d6eba606e052981bf8211bd;hp=bd7a320e6726a38726a7d6d864a66dc0b6c281b1;hb=752df0b3baa7349d6a0f5bdefb7d644a552bea2e;hpb=327f179b820ab60fedb776897e490b52330b48c9 diff --git a/Extractor/ExtractorScript.py b/Extractor/ExtractorScript.py index bd7a320..3a87c23 100644 --- a/Extractor/ExtractorScript.py +++ b/Extractor/ExtractorScript.py @@ -15,10 +15,6 @@ app2Subscribe = False # Mapping for event type and number of events # e.g., switch has 2 events: on and off eventTypeCounterMap = {'lock': 2, - 'unlock' : 1, - 'contact.open' : 1, - 'contact.closed' : 1, - 'tamper.tampered' : 1, 'nfcTouch' : 1, 'app' : 1, 'button' : 2, @@ -26,26 +22,15 @@ eventTypeCounterMap = {'lock': 2, 'water.dry' : 1, 'water.wet' : 1, 'presence' : 2, - 'presence.present' : 1, 'doorState' : 2, 'motion' : 2, - 'motion.active' : 1, - 'motion.inactive' : 1, 'smoke' : 3, - 'smoke.clear' : 1, - 'smoke.detected' : 1, - 'smoke.tested' : 1, 'carbonMonoxide' : 3, - 'carbonMonoxide.clear' : 1, - 'carbonMonoxide.detected' : 1, - 'carbonMonoxide.tested' : 1, 'battery' : 1, 'thermostatMode' : 5, 'thermostatFanMode' : 5, 'thermostatOperatingState' : 5, 'switch' : 2, - 'switch.on' : 1, - 'switch.off' : 1, 'location' : 3, 'mode' : 3, 'sunset' : 1, @@ -53,17 +38,15 @@ eventTypeCounterMap = {'lock': 2, 'sunrise' : 1, 'sunriseTime' : 1, 'acceleration' : 2, - 'acceleration.active' : 1, - 'acceleration.inactive' : 1, 'sleeping' : 2, - 'goal' : 1, - 'steps' : 1, + 'goal' : 2, + 'steps' : 2, 'color' : 1, 'colorTemperature' : 1, 'hue' : 1, 'saturation' : 1, 'energy' : 1, - 'power' : 1, + 'power' : 2, 'illuminance' : 1, 'humidity' : 1, 'alarm' : 4, @@ -78,6 +61,65 @@ eventTypeCounterMap = {'lock': 2, 'coolingSetpoint' : 1, 'thermostatSetpoint' : 1} +# Mapping for specific event types +eventTypesMap = {'alarm.both': 'alarm', + 'alarm.siren' : 'alarm', + 'alarm.strobe' : 'alarm', + 'alarm.off' : 'alarm', + 'contact.open' : 'contact', + 'contact.closed' : 'contact', + 'tamper.tampered' : 'contact', + 'doorState.open' : 'doorState', + 'doorState.closed' : 'doorState', + 'mode' : 'location', + 'mode.away' : 'location', + 'mode.home' : 'location', + 'mode.night' : 'location', + 'lock.locked' : 'lock', + 'lock.unlocked' : 'lock', + 'motion.active' : 'motion', + 'motion.inactive' : 'motion', + 'status.playing' : 'status', + 'status.stopped' : 'status', + 'status.paused' : 'status', + 'mute.muted' : 'mute', + 'mute.unmuted' : 'mute', + 'presence.present' : 'presence', + 'presence.not present' : 'presence', + 'presence.not' : 'presence', + 'smoke.detected' : 'smoke', + 'smoke.clear' : 'smoke', + 'smoke.tested' : 'smoke', + 'carbonMonoxide.detected' : 'carbonMonoxide', + 'carbonMonoxide.clear' : 'carbonMonoxide', + 'carbonMonoxide.tested' : 'carbonMonoxide', + 'thermostatMode.cool' : 'thermostatMode', + 'thermostatMode.auto' : 'thermostatMode', + 'thermostatMode.emergencyHeat' : 'thermostatMode', + 'thermostatMode.heat' : 'thermostatMode', + 'thermostatMode.off' : 'thermostatMode', + 'thermostatFanMode.auto' : 'thermostatFanMode', + 'thermostatFanMode.fanCirculate' : 'thermostatFanMode', + 'thermostatFanMode.circulate' : 'thermostatFanMode', + 'thermostatFanMode.fanOn' : 'thermostatFanMode', + 'thermostatFanMode.on' : 'thermostatFanMode', + 'fanMode' : 'thermostatFanMode', + 'fanMode.auto' : 'thermostatFanMode', + 'fanMode.fanCirculate' : 'thermostatFanMode', + 'fanMode.circulate' : 'thermostatFanMode', + 'fanMode.fanOn' : 'thermostatFanMode', + 'fanMode.on' : 'thermostatFanMode', + 'switch.on' : 'switch', + 'switch.off' : 'switch', + 'button.pushed' : 'button', + 'button.held' : 'button', + 'acceleration.active' : 'acceleration', + 'acceleration.inactive' : 'acceleration', + 'sleeping.sleeping' : 'sleeping', + 'sleeping.not sleeping' : 'sleeping', + 'water.dry' : 'water', + 'water.wet' : 'water'} + def GetToken(f): global readyToReturn global ToReturn @@ -111,6 +153,7 @@ def GetToken(f): def ExtractFunctions(F, appName): global eventList global eventAppList + global eventTypesMap global app1Subscribe global app2Subscribe Temp = GetToken(F) @@ -169,6 +212,13 @@ def ExtractFunctions(F, appName): #print "DEBUG: %s - %s" % (variable, Temp) #print capabilityMap #print "DEBUG: location variable: %s" % Temp + + #This is a bogus event, just skip it... + if Temp == "unlock": + continue + #Translate and reduce through mapping + if Temp in eventTypesMap.keys(): + Temp = eventTypesMap[Temp] if Temp == "app" or Temp == "location" or Temp == "mode" or Temp == "sunset" or Temp == "sunrise" or Temp == "sunsetTime" or Temp == "sunriseTime": if Temp not in eventList: eventList.append(Temp) @@ -256,6 +306,7 @@ def CountEvents(): global eventList numOfActualEvents = 0 #print "DEBUG: eventlist: %d" % len(eventList) + #print "DEBUG: eventlist: %s" % eventList for event in eventList: #print "DEBUG: Event: %s %d" % (event, eventTypeCounterMap[event]) numOfActualEvents = numOfActualEvents + eventTypeCounterMap[event] @@ -274,11 +325,15 @@ def ExtractEvents(extractedEvents): extractedEvents.write("\tswitch(eventNumber) {\n") counter = 0 indexApp1 = 0 - indexApp2 = eventAppList.index("App2") + if "App2" in eventAppList: + indexApp2 = eventAppList.index("App2") + else: + indexApp2 = len(eventList) indexApp2Start = indexApp2 #print "DEBUG: App1: %d" % indexApp1 #print "DEBUG: App2: %d" % indexApp2 #print "DEBUG: eventList: %d" % len(eventList) + #print eventList isApp1 = True while counter < numOfActualEvents: # Interleave events from App1 and App2 @@ -307,32 +362,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "unlock": - #Don't do anything for this case - counter = counter - 1 - elif eventList[i] == "contact.open": - event = open("eventSimulator/contactOpenEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "contact.closed": - event = open("eventSimulator/contactClosedEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "tamper.tampered": - #Write two events subsequently - event = open("eventSimulator/tamperTamperedOpenEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - extractedEvents.write("\n\t\t\tbreak\n") - counter = counter + 1 - extractedEvents.write("\t\tcase %d:\n" % counter) - event = open("eventSimulator/tamperTamperedClosedEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "nfcTouch": event = open("eventSimulator/nfcTouchEvent.groovy", "r") for line in event: @@ -384,16 +413,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "water.dry": - event = open("eventSimulator/waterDefaultDryEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "water.wet": - event = open("eventSimulator/waterDefaultWetEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "presence": #Check which capability variable = eventVarMap[eventList[i]] @@ -422,11 +441,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "presence.present": #Case for Touched event - event = open("eventSimulator/presencePresencePresentEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "doorState": #Write two events subsequently event = open("eventSimulator/doorOpenEvent.groovy", "r") @@ -453,16 +467,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "motion.active": - event = open("eventSimulator/motionDefaultActiveEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "motion.inactive": - event = open("eventSimulator/motionDefaultInactiveEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "smoke": #Write three events subsequently event = open("eventSimulator/smokeClearEvent.groovy", "r") @@ -483,21 +487,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "smoke.clear": - event = open("eventSimulator/smokeDefaultClearEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "smoke.detected": - event = open("eventSimulator/smokeDefaultDetectedEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "smoke.tested": - event = open("eventSimulator/smokeDefaultTestedEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "carbonMonoxide": #Check which capability variable = eventVarMap[eventList[i]] @@ -536,57 +525,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "carbonMonoxide.clear": - #Check which capability - variable = eventVarMap[eventList[i]] - if eventList[i] not in eventVarCounterMap.keys(): - eventVarCounterMap[eventList[i]] = 1 - eventVarCount = 0 - else: - eventVarCount = eventVarCounterMap[eventList[i]] - eventVarCounterMap[eventList[i]] = eventVarCount + 1 - capability = capabilityMap[variable[eventVarCount]] - if capability == "capability.smokeDetector": - event = open("eventSimulator/smokeCarbonMonoxideDefaultClearEvent.groovy", "r") - elif capability == "capability.carbonMonoxideDetector": - event = open("eventSimulator/carbonMonoxideDefaultClearEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "carbonMonoxide.detected": - #Check which capability - variable = eventVarMap[eventList[i]] - if eventList[i] not in eventVarCounterMap.keys(): - eventVarCounterMap[eventList[i]] = 1 - eventVarCount = 0 - else: - eventVarCount = eventVarCounterMap[eventList[i]] - eventVarCounterMap[eventList[i]] = eventVarCount + 1 - capability = capabilityMap[variable[eventVarCount]] - if capability == "capability.smokeDetector": - event = open("eventSimulator/smokeCarbonMonoxideDefaultDetectedEvent.groovy", "r") - elif capability == "capability.carbonMonoxideDetector": - event = open("eventSimulator/carbonMonoxideDefaultDetectedEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "carbonMonoxide.tested": - #Check which capability - variable = eventVarMap[eventList[i]] - if eventList[i] not in eventVarCounterMap.keys(): - eventVarCounterMap[eventList[i]] = 1 - eventVarCount = 0 - else: - eventVarCount = eventVarCounterMap[eventList[i]] - eventVarCounterMap[eventList[i]] = eventVarCount + 1 - capability = capabilityMap[variable[eventVarCount]] - if capability == "capability.smokeDetector": - event = open("eventSimulator/smokeCarbonMonoxideDefaultTestedEvent.groovy", "r") - elif capability == "capability.carbonMonoxideDetector": - event = open("eventSimulator/carbonMonoxideDefaultTestedEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "battery": #Check which capability variable = eventVarMap[eventList[i]] @@ -746,54 +684,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "switch.on": - #Check which capability - variable = eventVarMap[eventList[i]] - if eventList[i] not in eventVarCounterMap.keys(): - eventVarCounterMap[eventList[i]] = 1 - eventVarCount = 0 - else: - eventVarCount = eventVarCounterMap[eventList[i]] - eventVarCounterMap[eventList[i]] = eventVarCount + 1 - capability = capabilityMap[variable[eventVarCount]] - #Write two events subsequently - if capability == "capability.switch": - event = open("eventSimulator/switchOnDefaultEvent.groovy", "r") - elif capability == "capability.switchLevel": - event = open("eventSimulator/switchLevelOnDefaultEvent.groovy", "r") - elif capability == "capability.relaySwitch": - event = open("eventSimulator/relaySwitchOnDefaultEvent.groovy", "r") - elif capability == "capability.colorControl": - event = open("eventSimulator/colorControlSwitchOnDefaultEvent.groovy", "r") - elif capability == "capability.colorTemperature": - event = open("eventSimulator/colorTemperatureSwitchOnDefaultEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - elif eventList[i] == "switch.off": - #Check which capability - variable = eventVarMap[eventList[i]] - if eventList[i] not in eventVarCounterMap.keys(): - eventVarCounterMap[eventList[i]] = 1 - eventVarCount = 0 - else: - eventVarCount = eventVarCounterMap[eventList[i]] - eventVarCounterMap[eventList[i]] = eventVarCount + 1 - capability = capabilityMap[variable[eventVarCount]] - #Write two events subsequently - if capability == "capability.switch": - event = open("eventSimulator/switchOffDefaultEvent.groovy", "r") - elif capability == "capability.switchLevel": - event = open("eventSimulator/switchLevelOffDefaultEvent.groovy", "r") - elif capability == "capability.relaySwitch": - event = open("eventSimulator/relaySwitchOffDefaultEvent.groovy", "r") - elif capability == "capability.colorControl": - event = open("eventSimulator/colorControlSwitchOffDefaultEvent.groovy", "r") - elif capability == "capability.colorTemperature": - event = open("eventSimulator/colorTemperatureSwitchOffDefaultEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "location": #Case for Location event #Write three events subsequently event = open("eventSimulator/locationHomeEvent.groovy", "r") @@ -814,26 +704,6 @@ def ExtractEvents(extractedEvents): for line in event: extractedEvents.write(line) event.close() - elif eventList[i] == "mode": - #Write three events subsequently - event = open("eventSimulator/modeHomeEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - extractedEvents.write("\n\t\t\tbreak\n") - counter = counter + 1 - extractedEvents.write("\t\tcase %d:\n" % counter) - event = open("eventSimulator/modeAwayEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() - extractedEvents.write("\n\t\t\tbreak\n") - counter = counter + 1 - extractedEvents.write("\t\tcase %d:\n" % counter) - event = open("eventSimulator/modeNightEvent.groovy", "r") - for line in event: - extractedEvents.write(line) - event.close() elif eventList[i] == "sunrise": event = open("eventSimulator/locationSunriseEvent.groovy", "r") for line in event: @@ -891,12 +761,28 @@ def ExtractEvents(extractedEvents): extractedEvents.write(line) event.close() elif eventList[i] == "goal": - event = open("eventSimulator/stepGoalEvent.groovy", "r") + #Write two events subsequently + event = open("eventSimulator/stepGoalHighEvent.groovy", "r") + for line in event: + extractedEvents.write(line) + event.close() + extractedEvents.write("\n\t\t\tbreak\n") + counter = counter + 1 + extractedEvents.write("\t\tcase %d:\n" % counter) + event = open("eventSimulator/stepGoalLowEvent.groovy", "r") for line in event: extractedEvents.write(line) event.close() elif eventList[i] == "steps": - event = open("eventSimulator/stepStepsEvent.groovy", "r") + #Write two events subsequently + event = open("eventSimulator/stepStepsHighEvent.groovy", "r") + for line in event: + extractedEvents.write(line) + event.close() + extractedEvents.write("\n\t\t\tbreak\n") + counter = counter + 1 + extractedEvents.write("\t\tcase %d:\n" % counter) + event = open("eventSimulator/stepStepsLowEvent.groovy", "r") for line in event: extractedEvents.write(line) event.close() @@ -926,7 +812,15 @@ def ExtractEvents(extractedEvents): extractedEvents.write(line) event.close() elif eventList[i] == "power": - event = open("eventSimulator/powerMeterEvent.groovy", "r") + #Write two events subsequently + event = open("eventSimulator/powerMeterHighPowerEvent.groovy", "r") + for line in event: + extractedEvents.write(line) + event.close() + extractedEvents.write("\n\t\t\tbreak\n") + counter = counter + 1 + extractedEvents.write("\t\tcase %d:\n" % counter) + event = open("eventSimulator/powerMeterLowPowerEvent.groovy", "r") for line in event: extractedEvents.write(line) event.close()