Snap To Rail

lib/editor/components/pattern/EditSettings.js

@@ -8,8 +8,10 @@ import {updateEditSetting} from '../../actions/active'
 import {CLICK_OPTIONS, SNAP_TO_OPTIONS} from '../../util'
 import toSentenceCase from '../../../common/util/to-sentence-case'
 import type {EditSettingsState} from '../../../types/reducers'
+import type {GtfsRoute} from '../../../types'
 
 type Props = {
+  activeEntity: GtfsRoute,
   editSettings: EditSettingsState,
   patternSegment: number,
   updateEditSetting: typeof updateEditSetting
@@ -62,7 +64,7 @@ export default class EditSettings extends Component<Props, State> {
 
   render () {
     // Edit Settings passed in are present
-    const {editSettings, patternSegment, updateEditSetting} = this.props
+    const {editSettings, patternSegment, updateEditSetting, activeEntity} = this.props
     const {
       editGeometry,
       followStreets,
@@ -83,18 +85,33 @@ export default class EditSettings extends Component<Props, State> {
     ]
     if (!editGeometry) return null
     const noSegmentIsActive = !patternSegment && patternSegment !== 0
+
+    // If route_type is rail, add "snap to rail" option
+    const snapToOptions = Object.keys(SNAP_TO_OPTIONS).reduce((acc, key) => {
+      if (key === 'RAIL') {
+        const routeType = activeEntity && activeEntity.route_type
+        if ([0, 1, 2].includes(routeType)) {
+          acc[key] = SNAP_TO_OPTIONS[key]
+        }
+      } else {
+        acc[key] = SNAP_TO_OPTIONS[key]
+      }
+      return acc
+    }, {})
+
     return (
       <div>
         <ControlLabel><small>Snap to</small></ControlLabel>
         <FormControl
           componentClass='select'
-          value={currentSelection}
           name='followOption'
-          onChange={this._onFollowOptionChange}>
-          {Object.keys(SNAP_TO_OPTIONS).map(key => (
+          onChange={this._onFollowOptionChange}
+          value={currentSelection}
+        >
+          {Object.keys(snapToOptions).map(key => (
             <option
               key={key}
-              value={SNAP_TO_OPTIONS[key]}>
+              value={snapToOptions[key]}>
               {toSentenceCase(key)}
             </option>
           ))}

lib/editor/components/pattern/EditShapePanel.js

@@ -21,12 +21,13 @@ import {
   getPatternDistance,
   isValidStopControlPoint
 } from '../../util/map'
-import type {ControlPoint, LatLng, Pattern, GtfsStop} from '../../../types'
+import type {ControlPoint, GtfsRoute, LatLng, Pattern, GtfsStop} from '../../../types'
 import type {EditSettingsUndoState} from '../../../types/reducers'
 
 import EditSettings from './EditSettings'
 
 type Props = {
+  activeEntity: GtfsRoute,
   activePattern: Pattern,
   controlPoints: Array<ControlPoint>,
   editSettings: EditSettingsUndoState,
@@ -41,7 +42,7 @@ type Props = {
   undoActiveTripPatternEdits: typeof tripPatternActions.undoActiveTripPatternEdits,
   updateActiveGtfsEntity: typeof activeActions.updateActiveGtfsEntity,
   updateEditSetting: typeof activeActions.updateEditSetting,
-  updatePatternGeometry: typeof mapActions.updatePatternGeometry,
+  updatePatternGeometry: typeof mapActions.updatePatternGeometry
 }
 
 export default class EditShapePanel extends Component<Props> {
@@ -200,6 +201,7 @@ export default class EditShapePanel extends Component<Props> {
 
   render () {
     const {
+      activeEntity,
       activePattern,
       controlPoints, // FIXME use to describe which segment user is editing
       patternSegment,
@@ -360,6 +362,7 @@ export default class EditShapePanel extends Component<Props> {
               </Button>
             </ButtonToolbar>
             <EditSettings
+              activeEntity={activeEntity}
               editSettings={editSettings}
               patternSegment={patternSegment}
               updateEditSetting={updateEditSetting} />