2b:T1575,viewof chart = { let i = 0; let root = d3.hierarchy(tracking_info[0]).eachBefore(d => d.index = i++); const nodes = root.descendants(); const svg = d3.create("svg") .attr("viewBox", [-nodeSize / 2, -nodeSize * 3 / 2, width, (nodes.length + 1) * nodeSize]) .attr("font-family", "sans-serif") .attr("font-size", 10) .style("overflow", "visible"); let returnObject = svg.node(); returnObject.value = ""; svg.on ("click", (d)=>{returnObject.value="";returnObject.dispatchEvent(new CustomEvent('input'))}); const link = svg.append("g") .attr("fill", "none") .attr("stroke", "#999") .selectAll("path") .data(root.links()) .join("path") .attr("stroke-width", (d)=>2*tracking_info[d.target.data.id].CPI) .attr("d", d => ` M${d.source.depth * nodeSize},${d.source.index * nodeSize} V${d.target.index * nodeSize} h${nodeSize} `); link.append ("title").text ((d)=>tracking_info[d.target.data.id].CPI); /*const link_labels = svg.append ("g").selectAll("text") .data(root.links()) .join("text") .attr ("x", (d)=>d.target.depth * nodeSize) .attr ("dx", "-0.2em") .attr ("y", (d)=>d.target.index * nodeSize) .attr("font-size", 7) .attr ("dy", "1em") .attr("text-anchor", "end") .attr("font-weight", "lighter") .text ((d)=>tracking_info[d.target.data.id].CPI);*/ let size_scaler = d3.scaleLog() .domain([1, N]) .rangeRound([1,6]) const node = svg.append("g") .selectAll("g") .data(nodes) .join("g") .attr("transform", d => `translate(0,${d.index * nodeSize})`); _.each ([[100,510,"timeSeries","Mutation frequency"],[100,700,"timeSeriesSig","Signature frequency"]], (settings)=> { let tc_width = 100; let tc_offset = 400; let tc_data = "timeSeries"; let tc_title = "Frequency over time"; [tc_width,tc_offset,tc_data,tc_title] = settings; const time_scale = d3.scaleTime() .domain(date_range) .range([tc_offset, tc_offset+tc_width]); const freq_scale = d3.scaleLinear() .domain ([0,1]) .range ([nodeSize/3,-nodeSize/3]); const time_area = d3.area() .curve(d3.curveBasis) .x(d => time_scale(d.date)) .y0(freq_scale(0)) .y1(d => freq_scale(d.count)); let firstSeenTime = []; node.each ((d,i)=> { if ( tracking_mutations_table[d.data.id][tc_data].length) { firstSeenTime.push ({'y' :d.index * nodeSize + nodeSize/3 , 'x' : tracking_mutations_table[d.data.id][tc_data][0].date})}; }); /* trace the first time a mutation was seen */ firstSeenTime = _.sortBy (firstSeenTime, (d)=>d.y); const first_seen = d3.line() .curve(d3.curveLinear) .x(d => time_scale(d.x)) .y(d => d.y); svg.append ("path").datum (firstSeenTime).attr("stroke", "grey") .attr("stroke-width", 0.5) .attr("fill", "none") .attr("d", first_seen); const axis_g = svg.append ("g").attr("transform", `translate(0,${-2*nodeSize})`) .call(d3.axisTop(time_scale) .ticks(d3.timeMonth.every(2)) .tickFormat(d3.timeFormat("%b"))) .call(g => g.select(".domain") .remove()); //.tickFormat(d => d <= d3.timeYear(d) ? d.getFullYear() : null)) svg.append ("text").attr ("x", tc_offset) .attr ("y", -nodeSize) .attr("dy", "0.32em") .attr("text-anchor", "start") .attr("font-weight", "bold") .text(tc_title); node.append("path") .datum((d)=>tracking_mutations_table[d.data.id][tc_data]) .attr("stroke", "black") .attr("fill", "#999") .attr("d", time_area); }); node.append("circle") .attr("cx", d => d.depth * nodeSize) .attr("r", d=>size_scaler(1+tracking_mutations_table[d.data.id]["mapped"][0])) .attr("fill", d => d.children ? null : "#999") .on ("click", (e,d)=>{returnObject.value=sequences_with_tracking_info.mutation_paths[d.data.name].join (",");returnObject.dispatchEvent(new CustomEvent('input'));e.stopPropagation()}) .append("title").text(d=>tracking_mutations_table[d.data.id]["mapped"][0]); node.append("text") .attr("dy", "0.32em") .attr("x", d => d.depth * nodeSize + 8) .on ("click", (e,d)=>{returnObject.value=d.data.name;returnObject.dispatchEvent(new CustomEvent('input'))}) .text(d => d.data.name); for (const {label, displayed, format, x, textOnly, clicker} of columns) { svg.append("text") .attr("dy", "0.32em") .attr("y", -nodeSize) .attr("x", x) .attr("text-anchor", "end") .attr("font-weight", "bold") .text(displayed); if (textOnly) { let nodes = node.append("text") .attr("dy", "0.32em") .attr("x", x) .attr("text-anchor", "end") .data(root.descendants().map ((d)=>{return {'value' :label (d.data)};})) .text(d => format(d.value, d)); if (clicker) { nodes.on ("click", clicker); } } else { node.each (function (d) { format (d, d3.select (this),nodeSize) }); } } return returnObject; }2c:T811,animatedChart = { const container = d3.create("div"); const svg = container .append('svg') .attr('width', mapWidth) .attr('height', mapHeight); const tooltip = container .append("div") .attr("class", "svg-tooltip") .style("position", "absolute") .style("bottom", "10px") .style("right", "10px") .style("background-color", "#e6e6e6") .style("padding", "10px") .style("border", "1px solid") .style("visibility", "hidden"); const path = d3.geoPath(mapProjection); /*svg .append('path') .attr('fill', 'none') .attr('stroke', '#ccc') .attr('d', path(graticule));*/ let pf = d3.format (".2p"); let colorFunction = function(d) { if (d.id in forMapPlotID) { if (mapDate) { if (mapDate in forMapPlotID[d.id]["fraction_by_week"]) return colorScale(forMapPlotID[d.id]["fraction_by_week"][mapDate]); return "#ccc"; } return colorScale (forMapPlotID[d.id].fraction); } return "#ccc"; } let makeLabel = function(d) { if (d.id in forMapPlotID) { if (mapDate) { if (mapDate in forMapPlotID[d.id]["fraction_by_week"]) return forMapPlotID[d.id].name + " :" + pf(forMapPlotID[d.id]["fraction_by_week"][mapDate]) + ", " + forMapPlotID[d.id]["byweek"][mapDate] + " sequences"; return "#ccc"; } return forMapPlotID[d.id].name + " :" + pf(forMapPlotID[d.id].fraction) + ", " + forMapPlotID[d.id].total + " sequences"; } return "N/A"; } let c = svg .append('g') .selectAll('path') .data(countryData.features) .join('path') .attr('fill', f => colorFunction(f)) .attr('stroke', '#ccc') .attr('stroke-width', 0.5) .attr('d', path) .append('title').text (d=>makeLabel (d)); svg .append('path') .attr('fill', 'none') .attr('stroke', 'black') .attr('pointer-events', 'none') .attr('d', path(countryData)); return container.node(); }2d:T78b,genomicDist = vegalite ({ "$schema": "https://vega.github.io/schema/vega-lite/v4.json", "config" : { "bar": {continuousBandSize : 3} }, "data": { "values": other_variants, }, "vconcat" : [ { "width": 800, "height" : 100, "selection": { "brush": {"type": "interval", "encodings": ["x"]} }, "mark": {"type": "bar", "tooltip": {"content": "data"}, "filled": true, "clip" : true}, "encoding": { "y": { "field": "fraction", "type": "quantitative", "axis" : {"grid" : false, "title" : "Mutation frequency", "orient" : "left"}, "scale" : {"type" : "sqrt", domain: [0,1]} }, "x" : { "field": "coord", "type": "quantitative", "axis" : {"grid" : false, "title" : "Genomic Position"}, "scale" : {domain : [260,29880]} }, "color" : { "field" : "kind", "type" : "nominal", "title" : "SNP kind" } } }, { "width": 800, "height" : 100, "selection": { "brush": {"type": "interval", "encodings": ["x"]} }, "mark": {"type": "bar", "tooltip": {"content": "data"}, "filled": true, "clip" : true}, "encoding": { "y": { "field": "fraction", "type": "quantitative", "axis" : {"grid" : false, "title" : "Mutation frequency", "orient" : "left"}, "scale" : {"type" : "sqrt", domain: [0,1]} }, "x" : { "field": "coord", "type": "quantitative", "axis" : {"grid" : false, "title" : "Genomic Position"}, "scale" : {"domain": {"selection": "brush"}} } } //"shape": {"field": "species", "type": "nominal"} }] //"resolve": {"scale": {"y": "independent"}} })2e:Ta5d,trends = { const marginBottom = 20, marginLeft = 40, marginTop = 10, marginRight = 50; const x = d3.scaleTime() .domain(date_range) .range([marginLeft, trendWidth-marginRight]); const sampleCount = _.map (sequencesByWeekWithAttr, (d,k)=>{return {'date' : deltaweekToDate(k), 'count' : d}}); const y = d3.scaleLinear([0, 1], [trendHeight-marginBottom, marginTop]); const y2 = d3.scaleLinear([0, d3.max (sampleCount, (d)=>d.count)], [trendHeight-marginBottom, marginTop]); const z = d3.interpolateTurbo; const area = d3.area().curve(d3.curveBasis) .x((d, i) => { //console.log (d.data.date); return x(d.data.date); }) .y0(d => y(d[0])) .y1(d => y(d[1])); const sampleCurve = d3.line().curve(d3.curveBasis) .x((d, i) => { //console.log (d); return x(d.date); }) .y(d => y2(d.count)); const svg = d3.create("svg") .attr("viewBox", [0, 0, trendWidth, trendHeight]); const path = svg.selectAll("path") .data(binStackstack (binnedDataForStacking)) .join("path") .attr("d", area) .attr ("stroke", "grey") .attr ("stroke-width", "0.5px") .attr("fill", (d,i) => { return binColorsWithPredef(binMutationClasses[i], i); }); const samplesPath = svg.append("path") .datum(sampleCount) .attr("d", sampleCurve) .attr("fill", "none") .attr ("stroke-width", "3px") .attr ("stroke", "#444444"); const axis_g = svg.append ("g").attr("transform", `translate(0,${trendHeight-marginBottom})`) .call(d3.axisBottom(x) .ticks(d3.timeMonth.every(1))) .call(g => g.select(".domain") .remove()); const axis_yg = svg.append ("g").attr("transform", `translate(${marginLeft},0)`) .call(d3.axisLeft(y).tickFormat (report_fraction)) .call(g => g.select(".domain") .remove()); const countFormat = d3.format(".2s"); const axis_y2g = svg.append ("g").attr("transform", `translate(${trendWidth-marginRight},0)`) .call(d3.axisRight(y2).tickFormat (countFormat)) .call(g => g.select(".domain") .remove()); svg.append ("text").attr("transform", "translate(" + (trendWidth-marginRight) + "," + (trendHeight-marginBottom)/2 + ") rotate(90)").text ("Sample Count").attr ("dy", "-2em").attr ("text-anchor", "middle").attr ("font-family","Helvetica").attr ("font-size", "14"); return svg.node(); /*while (true) { yield svg.node(); await path .data(randomize) .transition() .delay(1000) .duration(1500) .attr("d", area) .end(); }*/ }2f:T4de,columns = [ { label: function (d) {return "" + d["pos"] + d["variant"]}, displayed: "Variant", format: (d)=>d, x: 315, textOnly: true, clicker: (e,d)=>{ window.open( "https://nextstrain.org/ncov/global?c=gt-nuc_" + parseInt (d.value), "_blank") } }, { label: function (d) {return tracking_mutations_table[d.id]["count"];}, displayed: "Freq.", format: report_fraction, x: 355, textOnly: true }, { label: function (d) {return tracking_mutations_table[d.id]["mapped"][0];}, displayed: "Mapped", format: report_int, x: 400, textOnly: true }, // { label: function (d) {return tracking_mutations_table[d.id]["firstSeenMut"];}, displayed: "First mutation", /*format: function (d,obj,size) { //console.log (obj); //let flag = obj.append ()); }, */ format : (d)=>d, x: 505, textOnly: true }, { label: function (d) {return tracking_mutations_table[d.id]["firstSeen"];}, displayed: "First signature", /*format: function (d,obj,size) { //console.log (obj); //let flag = obj.append ()); }, */ format : (d)=>d, x: 695, textOnly: true } ]30:T7d7,sequencesByCountryAndWeek = (seqSet, sequenceKey) => { let sbcw = {"allweeks" : {}}; /* "country" -> {"total" : count "byweek" : { "week": X, "count" : Y } "allweeks" -> [array of weeks since start for which we have data] ... */ sequenceKey = sequenceKey || ((d,id)=>resolved_name (d[1])); _.each (seqSet || _.keys(linkage_data.sequences), (id)=> { let seqDate = linkage_data.records[id][1]; let seqLocation = linkage_data.records[id][2]; if (seqLocation) { if (seqLocation[1]) { seqLocation = sequenceKey(seqLocation,id); } else { seqLocation = null; } } else { seqLocation = null; } if (!(seqLocation in sbcw)) { sbcw [seqLocation] = {"total" : 0, byweek : {}}; } sbcw [seqLocation].total ++; if (seqDate) { seqDate = deltaWeeks (seqDate); sbcw["allweeks"][seqDate] = 0; if (!(seqDate in sbcw[seqLocation]["byweek"])) { sbcw [seqLocation]["byweek"][seqDate] = 0; } sbcw [seqLocation]["byweek"][seqDate]++; } }); sbcw["allweeks"] = _.sortBy (_.map(_.keys(sbcw["allweeks"]), (k)=>+k)); _.each (sbcw, (data, name) => { let country = _.find (country_equiv, (c)=>(c.name.common == name || c.name.official == name || c.cca3 == name || c.altSpellings.find ((e)=>e==name)!=undefined)); if (country) { data ['id'] = country.ccn3; data ['name'] = name; if (doCumulativeCounts) { let sum = 0; _.each (sbcw["allweeks"], (wk,i)=> { if (wk in data['byweek']) { sum += data['byweek'][wk]; } data['byweek'][wk] = sum; }); } } }); return sbcw; }31:T5c3,sequences_with_tracking_info = { let tagged_sequences = {}; // check if "reference" genome has some mutations in it already _.each (tracking_info, (data)=> { if (linkage_data.reference_base[data.pos-1] == data.variant) { data.reference_value = true; } }); _.each (linkage_data.sequences, (data, id)=>{ if (!sequenceFilter (id, data)) return; let tags = {}; _.each (tracking_info, (d) => { if (d.reference_value) { if (_.every (["A","C","G","T"], (n)=> (("" + d.pos + n) in data) == false)) { tags[d.name] = 1; } } else { let sig = "" + d.pos + d.variant; if (sig in data) { tags[d.name] = 1; } } }); tagged_sequences [id] = tags; }); let md = {}; _.each (tracking_info, (d)=> { md[d.name] = d; }); let mutation_data = []; let mutation_dict = {}; _.each (tracking_info, (d) => { let path = new Set ([d.name]); let parent = d.parent_name; let k = 0; while (parent && k < 20) { parent = md[parent]; path.add (parent.name); parent = parent.parent_name; k++; } mutation_dict [d.name] = [...path]; mutation_data.push (path); }); return { 'sequences' : tagged_sequences, 'mutations' : mutation_data, 'mutation_paths': mutation_dict }; }32:T448,earliest_seq_from_set = (s)=>{ let date = null, date1 = null; let locations = [], locations1 = []; let set = _.sortBy (_.filter (s, (d)=>linkage_data.records[d][1]) , (m)=>linkage_data.records[m][1]); if (set.length) { date = linkage_data.records[set[0]][1]; _.some (set, (s)=> { if ((linkage_data.records[s][1] ? linkage_data.records[s][1].getTime() : null) == date.getTime()) { locations.push (linkage_data.records[s][2]); return false; } return true; }); date1 = linkage_data.records[set[set.length-1]][1]; locations1 = [linkage_data.records[set[set.length-1]][2]]; for (let k = set.length-2; k >= 0; k--) { if ((linkage_data.records[set[k]][1] ? linkage_data.records[set[k]][1].getTime() : null) == date1.getTime()) { locations1.push (linkage_data.records[set[k]][2]); } else { break; } } } return [date, merge_location_info(locations), date1, merge_location_info(locations1), locations]; }33:Td68,tracking_mutations_table = { let by_mutation = [], by_position = {}, earliest_by_position = {}; /*for (const m of sequences_with_tracking_info.mutations[5].values()) { by_mutation.push (m); } return by_mutation;*/ let SQ = _.size (sequences_with_tracking_info.sequences); let sk = _.map (sequences_with_tracking_info.sequences, (s,k)=>[k,s]); let map_loc = (l)=> { if (!l) return ""; if (l[1].length) return l[1]; return l[0]; }; _.each (sequences_with_tracking_info.sequences, (d,seq_name)=>{ _.each (d, (ignore, m)=> { if (! (m in by_position)) by_position[m] = []; by_position[m].push (seq_name); } )}); _.each (sequences_with_tracking_info.mutations, (d, index)=> { if (! (tracking_info[index].name in by_position)) { by_position[tracking_info[index].name] = []; } let matched_seqs = _.map(_.filter (sk, (s)=> { for (const m of d.values()) { if (! (m in s[1])) { return false; } } return d.size == _.size (s[1]); }), (s)=>s[0]); //[date, location, date1,location1] = earliest_seq_from_set (matched_seqs); let locations = []; //[d1,l1,d2,l2,l3] = earliest_seq_from_set (by_position[tracking_info[index].name]); let time_series = _.map ([matched_seqs, by_position[tracking_info[index].name]], (ms,i)=> { let timeSeries = {}; let min_date = new Date(); locations.push ([]); min_date = min_date.getTime(); locations[i] = []; _.each(_.filter (ms, (d)=>(linkage_data.records[d]) && linkage_data.records[d][1]), (d)=> { //console.log (linkage_data.records[d][1]); let dtw = deltaWeeks(linkage_data.records[d][1]); let ttm = linkage_data.records[d][1].getTime(); if (ttm < min_date) { min_date = ttm; locations[i] = []; } if (ttm == min_date) { locations[i].push (linkage_data.records[d][2]); } if (timeSeries[dtw]) { timeSeries[dtw] ++; } else { timeSeries[dtw] = 1; } }); timeSeries = _.map (timeSeries, (d,i)=>{ return { 'date': deltaweekToDate (i), 'count' : d/sequencesByWeek[i]} }); return timeSeries; }); by_mutation.push ({ name: tracking_info[index].name, variant: ["" + tracking_info[index].pos + tracking_info[index].variant, tracking_info[index].reference_value], count: by_position[tracking_info[index].name].length/SQ, //sampled_on: [date, date1], //location: [map_loc(location),map_loc(location1)], mapped: [matched_seqs.length, [...d].join (",")], firstSeen: map_loc(merge_location_info(locations[0])), firstSeenMut: map_loc(merge_location_info(locations[1])), //last_seen_mut_location: map_loc(l2), timeSeries: time_series[1], timeSeriesSig: time_series[0]}); }); return by_mutation; }34:[] 2a:{"title":"Tracking signature mutations in SARS-CoV-2 relative to the inferred progenitor strain","cells":[{"id":0,"value":"intro = md`# Tracking signature mutations in SARS-CoV-2 relative to the inferred progenitor strain\n\n> _See [the preprint](https://www.biorxiv.org/content/10.1101/2020.09.24.311845v2) describing the approach_\n\nThis analysis is based on **${N}** whole genome sequences from GISAID, which have been sampled between ${report_date_f(date_range[0])} and ${report_date_f(date_range[1])}.\n`","pinned":false,"hidden":false,"mode":"ojs"},{"id":977,"value":"figure1Desc = md`**Figure 1** Signature mutations away from the inferred progenitor SARS-CoV-2 strain, in estimated order of appearance. _Mapped_ refers to the count of the genomes that include all the mutations in the **signature** (path leading to the listed mutation from the root of the tree), and no other signature mutations; click on the **node** to view geographic distribution of the signature; click on the **label** to view the geographic distribution of just this one mutation. Size of the nodes ~ to the number of mapped genomes, thickness of branch ~ co-occurence proportion (mouse over to view). Click on **variant** (e.g. 2416C) to view this mutation on [NextStrain](http://nextstrain.org).`","pinned":false,"hidden":false,"mode":"ojs"},{"id":1794,"value":"viewof as = autoSelect({\n options: _.uniq (_.flatten(_.map (linkage_data.sequences, (d,k)=>lookupLocation (k)))),\n title: `Restrict to specific country or region or locale (empty string to use all)`,\n placeholder: \"Select a location \"\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":1837,"value":"figure1View = md`Viewing data on ${_.size (sequences_with_tracking_info.sequences)} sequences with time and location information)`","pinned":false,"hidden":false,"mode":"ojs"},{"id":1465,"value":"$2b","pinned":false,"hidden":false,"mode":"ojs"},{"id":732,"value":"figure2Desc = md`**Figure 2** Geographic distribution of sequences for ${chart.length ? chart : \"any\"} signature`","pinned":false,"hidden":false,"mode":"ojs"},{"id":2261,"value":"viewof mapDate = Scrubber(forMapPlot.allweeks, {\n delay: 500,\n loopDelay: 100000,\n autoplay: false,\n format: sliderDate\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":2300,"value":"geospatialLegend = legend({\n color: colorScale,\n title: \"Cumulative fraction of sequences with the \" + chart + \" signature\",\n tickFormat: \"%\"\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":2175,"value":"$2c","pinned":false,"hidden":false,"mode":"ojs"},{"id":2335,"value":"viewof doCumulativeCounts = checkbox({\n options: [{ value: \"toggle\", label: \"Cumulative\" }],\n value: \"toggle\"\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":752,"value":"figure3Desc = md`**Figure 3** Distribution of other SNPs across the genome. _Click and drag in the top plot to zoom in_.`","pinned":false,"hidden":false,"mode":"ojs"},{"id":739,"value":"$2d","pinned":false,"hidden":false,"mode":"ojs"},{"id":2346,"value":"figure4Desc = md`**Figure 4** Temporal distribution of signature mutation groups that reach a minimum frequency of ${report_fraction(minFreqBin)} during at least one week ${as.length? \"restricted to \" + as : \"\"}. The thick line shows the number of sequences during each week.`","pinned":false,"hidden":false,"mode":"ojs"},{"id":2519,"value":"trendLegend = swatches({\n color: d3.scaleOrdinal(binMutationClasses, _.map (binMutationClasses, (d,i)=>binColorsWithPredef(d,i))),\n title: \"Proportion of sequences with a specific mutation signature by week\",\n //tickFormat: \"%\"\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":2427,"value":"$2e","pinned":false,"hidden":false,"mode":"ojs"},{"id":2170,"value":"md`----- \n### Data and libraries from this point on`","pinned":false,"hidden":false,"mode":"ojs"},{"id":1254,"value":"trackingSignature = {\n let dict = {};\n _.each (chart.split (\",\"), (m)=>dict[m]=true);\n return _.map (_.filter (tracking_info, (d)=>d.name in dict), (d)=>toSignature(d)).join (\", \");\n \n};","pinned":false,"hidden":false,"mode":"ojs"},{"id":1514,"value":"$2f","pinned":false,"hidden":false,"mode":"ojs"},{"id":29,"value":"md`---- `","pinned":false,"hidden":false,"mode":"ojs"},{"id":2783,"value":"raw_data = await d3.json (\"https://raw.githubusercontent.com/veg/SARS-CoV-2/segmented/data/fasta/20201012/variants.json\");","pinned":false,"hidden":false,"mode":"ojs"},{"id":20,"value":"linkage_data = {\n \n let as_dicts = {};\n let date_parser = d3.timeParse (\"%Y%m%d\");\n let today = new Date();\n _.each (raw_data.records, (d, k)=> {\n let vars = {};\n for (let i = 0; i < d[3].length; i+=2) {\n vars[\"\" + (d[3][i]+1) + d[3][i+1]] = 1; \n }\n //d[0] = +k;\n if (d[1] == \"20200101\") {\n if (d[2] && d[2][1] != \"China\" || !d[2]) {\n d[1] = null;\n }\n } \n if (d[1]) {\n d[1] = date_parser(d[1]);\n if (d[1] > today) {\n d[1] = null;\n }\n }\n as_dicts[k] = vars;\n });\n raw_data.sequences = as_dicts;\n return raw_data;\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":1760,"value":"mutation2track = \"\"","pinned":false,"hidden":false,"mode":"ojs"},{"id":499,"value":"countries = _.flatten ([[['World', -1,'']],_.sortBy (_.filter (_.map (_.countBy (linkage_data.records, (d)=>d[2] ? d[2][1] : \"\"), (d,k)=>[k,d,\"country\"]), (d)=>d[1]>=100), (d)=>-d[1]),_.sortBy (_.filter (_.map (_.countBy (linkage_data.records, (d)=>d[2] ? d[2][2] : \"\"), (d,k)=>[k,d,\"state\"]), (d)=>d[1]>=100 && d[0].length), (d)=>-d[1])],true)","pinned":false,"hidden":false,"mode":"ojs"},{"id":1803,"value":"lookupLocation = (id, level) => {\n if ((id in linkage_data.records) && linkage_data.records[id][2]) {\n if (level) {\n \n }\n return _.filter (linkage_data.records[id][2], (d)=>d.length);\n }\n return null;\n};","pinned":false,"hidden":false,"mode":"ojs"},{"id":1677,"value":"deltaWeeks = (d)=>Math.floor(((d)-date_range[0])/(3600*24*1000*7))","pinned":false,"hidden":false,"mode":"ojs"},{"id":1683,"value":"deltaweekToDate = (d)=> {\n return new Date (d*24*3600*7*1000+date_range[0].getTime());\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":1664,"value":"sequencesByWeek = _.countBy (_.filter (linkage_data.records, (k)=>k[1] && sequenceFilter (+(k[0])-1)),(d)=>deltaWeeks(d[1]))","pinned":false,"hidden":false,"mode":"ojs"},{"id":2480,"value":"sequencesByWeekWithAttr = _.countBy (_.filter (linkage_data.records, (k)=>k[1] && sequenceFilter (+(k[0])-1)),(d)=>deltaWeeks(d[1]))","pinned":false,"hidden":false,"mode":"ojs"},{"id":2087,"value":"extractSigMutations = ()=> {\n return _.map (_.map (trackingSignature.split (\",\"), (d)=>d.replace (\" \",\"\")), (s)=> {\n \n if (isNaN(parseInt(s[0]))) {\n return [s.substr (1,s.length-1), true]; \n }\n return [s, false];\n });\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":632,"value":"matched_sequences = {\n if (trackingSignature.length == 0) {\n return _.map (linkage_data.sequences, (d,k)=>k);\n }\n \n let sigs = extractSigMutations();\n \n return _.map (_.filter (_.map (linkage_data.sequences, (sd,sk)=>[sk,sd]), (s)=>_.every (sigs, (sig)=>{\n if (sig[1] == false) {\n return sig[0] in s[1];\n }\n return _.every ([\"A\",\"C\",\"G\",\"T\"], (n)=>((sig[0]+n) in s[1]) == false);\n })), (d)=>d[0]);\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":2836,"value":"matched_sequences_with_attr = _.filter (matched_sequences, (k)=>linkage_data.records[k][1] && _.some (linkage_data.records[k][2], (l)=>l.length))","pinned":false,"hidden":false,"mode":"ojs"},{"id":756,"value":"other_variants = {\n let variant_list = {};\n let sigs = {};\n _.each (extractSigMutations(), (sgm)=> {\n if (sgm[1]) {\n sigs[sgm[0]] = true;\n } else {\n sigs[sgm[0]] = false; \n }\n });\n let M = matched_sequences.length;\n _.each (matched_sequences, (id) => {\n if (id in linkage_data.sequences) {\n _.each (linkage_data.sequences[id], (m,k)=> {\n if (!(k in variant_list)) {\n variant_list[k] = 0;\n }\n variant_list[k]++;\n });\n }\n });\n return _.map (variant_list, (m,k)=> {\n return {'coord' : parseInt (k), 'fraction' : m/M, 'count' : m, 'mutation' : k, 'kind' : k in sigs ? 'Selected' : 'Other'};\n });\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":392,"value":"plot_divergence = _.map (matched_sequences, (k)=>{\n return {\n 'date' : linkage_data.records[k][1],\n 'location' : linkage_data.records[k][2] ? linkage_data.records[k][2][2] : \"\"\n }\n});","pinned":false,"hidden":false,"mode":"ojs"},{"id":2107,"value":"resolved_name = _.memoize ((name)=> {\n if (name == \"England\") {\n name = \"United Kingdom\"; \n }\n let c = _.find (country_equiv, (c)=>(c.name.common == name || c.name.official == name || c.cca3 == name || c.altSpellings.find ((e)=>e==name)!=undefined));\n if (c) {\n return c.name.common; \n }\n return \"N/A\";\n });","pinned":false,"hidden":false,"mode":"ojs"},{"id":2103,"value":"$30","pinned":false,"hidden":false,"mode":"ojs"},{"id":2117,"value":"unfilteredByCountryAndWeek = sequencesByCountryAndWeek()","pinned":false,"hidden":false,"mode":"ojs"},{"id":2123,"value":"filteredByCountryAndWeek = sequencesByCountryAndWeek(_.filter (matched_sequences, sequenceFilter))","pinned":false,"hidden":false,"mode":"ojs"},{"id":2128,"value":"forMapPlot = {\n _.each (filteredByCountryAndWeek, (data,country) => {\n data.fraction = data.total / unfilteredByCountryAndWeek[country].total;\n data.fraction_by_week = {};\n _.each (data.byweek, (dt,wk)=> {\n if (unfilteredByCountryAndWeek[country][\"byweek\"][wk]) {\n data.fraction_by_week[wk] = dt / unfilteredByCountryAndWeek[country][\"byweek\"][wk];\n } else {\n data.fraction_by_week[wk] = null;\n }\n });\n });\n return filteredByCountryAndWeek;\n} ","pinned":false,"hidden":false,"mode":"ojs"},{"id":2208,"value":"forMapPlotID = {\n let fmpi = {};\n _.each (forMapPlot, (d)=> {\n fmpi[d.id] = d;\n });\n return fmpi;\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":707,"value":"country_equiv = d3.json (\"https://raw.githubusercontent.com/mledoze/countries/master/dist/countries.json\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":1981,"value":"countryToCode = {\n let c2c = {};\n _.each (country_equiv, (country_info)=> {\n c2c[country_info.name.common] = country_info.cca2.toLowerCase();\n });\n return c2c;\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":711,"value":"country_codes = d3.json(\"https://raw.githubusercontent.com/lukes/ISO-3166-Countries-with-Regional-Codes/master/slim-3/slim-3.json\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":714,"value":"world = d3.json(\"https://unpkg.com/world-atlas@1/world/110m.json\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":2164,"value":"countryData = topojson.feature(world, world.objects.countries)","pinned":false,"hidden":false,"mode":"ojs"},{"id":2167,"value":"graticule = d3.geoGraticule10()","pinned":false,"hidden":false,"mode":"ojs"},{"id":52,"value":"N = _.size (linkage_data.sequences)","pinned":false,"hidden":false,"mode":"ojs"},{"id":9,"value":"d3 = require (\"d3\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":11,"value":"vegalite = require (\"vega-embed\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":16,"value":"_=require(\"lodash\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":256,"value":"import {select, slider, checkbox, text, autoSelect} from \"@jashkenas/inputs\"","pinned":false,"hidden":false,"mode":"ojs"},{"id":571,"value":"genes = {\n let genes = [\n [266,'ORF1a',13483],\n [13471, 'ORF1b',21555],\n [21563,'S',25385],\n [25393,'ORF3a',26220],\n [26523,'M',27191],\n [27202,'ORF6',27387],\n [27394,'ORF7a',27759],\n [27894,'ORF8',28259],\n [28274,'N',29533],\n [266, \"leader\", 805],\n [806, \"nsp2\", 2719],\n [2720, \"nsp3\", 8554],\n [8555, \"nsp4\", 10054],\n [10055, \"3C\", 10972],\n [10973, \"nsp6\",11842],\n [11843, \"nsp7\", 12091],\n [12092, \"nsp8\", 12685],\n [12686, \"nsp9\", 13024],\n [13025, \"nsp10\",13442],\n [16237, \"helicase\",18039],\n [18040, \"exonuclease\", 19620],\n [19621, \"endornase\", 20658],\n [20659, \"methyltransferase\", 21552],\n [26244, \"E\",26471],\n [29557, \"ORF10\",29673],\n [13441,\"RdRp\",16235]];\n \n return _.flatten ([[[0,\"genome\",30000]],_.sortBy (genes, (d)=>d[1])], true);\n }","pinned":false,"hidden":false,"mode":"ojs"},{"id":1478,"value":"nodeSize=18","pinned":false,"hidden":false,"mode":"ojs"},{"id":786,"value":"tracking_info = {\n let raw_list = await d3.json (\"https://dl.dropbox.com/s/t2aglsv6j895wuu/SC2-tracking-mutations.json\");\n let name_to_id = {};\n _.each (raw_list, (d, i) => {\n d[\"id\"] = i;\n name_to_id [d.name] = i;\n if (d.parent) {\n let parent_object = raw_list[name_to_id[d.parent]];\n if (!(\"children\" in parent_object)) {\n parent_object[\"children\"] = []; \n }\n parent_object[\"children\"].push (d);\n d.parent_name = d.parent;\n d.parent = parent_object;\n }\n });\n return raw_list;\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":1775,"value":"sequenceFilter = (id,data)=>{\n if (id in linkage_data.records) {\n if (linkage_data.records[id][2]) {\n if (as.length) {\n return _.some (linkage_data.records[id][2], (d)=>d == as);\n } \n return true;\n }\n } else {\n console.log (\"Missing => \" + id); \n }\n return false;\n};","pinned":false,"hidden":false,"mode":"ojs"},{"id":792,"value":"$31","pinned":false,"hidden":false,"mode":"ojs"},{"id":2368,"value":"binnedByMutationClassFiltered = sequencesByCountryAndWeek (_.filter (matched_sequences_with_attr, sequenceFilter), (d,id)=>sequencesByMutationClass [id])","pinned":false,"hidden":false,"mode":"ojs"},{"id":2384,"value":"binnedDataForStacking = {\n let result = [];\n let maxbyclass = {};\n _.each (binnedByMutationClassFiltered.allweeks, (wk)=> {\n let weekrecord = {'date' : deltaweekToDate(wk)};\n \n _.each (binnedByMutationClassFiltered, (data, mut) => {\n if (\"byweek\" in data) {\n weekrecord[mut] = (wk in data.byweek) ? data.byweek[wk] : 0;\n weekrecord[mut] = weekrecord[mut] / sequencesByWeekWithAttr[wk];\n if (! maxbyclass[mut] || weekrecord[mut] > maxbyclass[mut]){\n maxbyclass[mut] = weekrecord[mut];\n }\n }\n });\n result.push (weekrecord);\n });\n \n let ignore_keys = _.map (_.filter (_.map (maxbyclass, (d,k)=>[k,d]), (d)=>d[1] <= minFreqBin), (d)=>d[0]);\n return _.map (result, (d)=>_.omit(d, ignore_keys)) ;\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":2896,"value":"binMutationClasses = _.filter (_.keys (binnedDataForStacking[0]), (d)=>d!=\"date\")","pinned":true,"hidden":false,"mode":"ojs"},{"id":2863,"value":"binColorsWithPredef = (value, idx)=> {\n const lookup = {\n 'μ' : \"rgb(153, 0, 0)\",\n 'μ+ν' : \"rgb(255, 204, 255)\",\n 'μ+ɑ' : \"rgb(112, 173, 71)\",\n 'μ+ɑ+β' : \"rgb(91, 155, 213)\",\n 'μ+ɑ+β+ε' : \"rgb(255, 192, 0)\",\n 'μ+ɑ+β+ɣ' : \"rgb(0, 0, 0)\",\n 'μ+ɑ+β+ɣ+δ' : \"rgb(165, 165, 165)\"\n };\n return lookup[value] || binColorScale (idx);\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":2495,"value":"binColorScale = d3.scaleSequential((d)=>d3.interpolateTurbo((1+d)/binMutationClasses.length))","pinned":false,"hidden":false,"mode":"ojs"},{"id":2408,"value":"binStackstack = d3.stack()\n .keys(binMutationClasses)\n //.offset(0)\n .order(d3.stackOrderNone)","pinned":false,"hidden":false,"mode":"ojs"},{"id":2401,"value":"minFreqBin = 0.005","pinned":false,"hidden":false,"mode":"ojs"},{"id":2353,"value":"sequencesByMutationClass = {\n let tm_dict = {};\n _.each (tracking_info, (d)=>tm_dict[d.name] = d);\n let result = {};\n \n _.each (sequences_with_tracking_info.sequences, (data, id)=> {\n //if (sequenceFilter (id)) {\n let groups = {};\n _.each (data, (ignore, mutation)=> {\n groups[tm_dict[mutation].group] = 1;\n });\n\n result [id] = _.map (groups, (d,k)=>k).join (\"+\");\n //}\n });\n \n return result;\n \n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":1032,"value":"merge_location_info = (set)=> {\n let result = _.clone (set[0]);\n const levels = [3,2,1,0];\n for (let i = 1; i < set.length; i++) {\n for (let j = 0; j < levels.length; j++) {\n let lvl = levels[j];\n if (result[lvl].length == 0 && levels.length > 1) {\n levels.splice (0,1); \n j--;\n continue;\n } \n \n if (result[lvl] != set[i][lvl]) {\n if (j == levels.length - 1\n ) {\n result[lvl] = \"Multiple Regions\"; \n } else {\n result[lvl] = \"\"; \n }\n }\n }\n }\n return result;\n};","pinned":false,"hidden":false,"mode":"ojs"},{"id":926,"value":"$32","pinned":false,"hidden":false,"mode":"ojs"},{"id":935,"value":"report_fraction = d3.format (\".2p\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":969,"value":"report_date_f = d3.timeFormat (\"%m/%d/%Y\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":1010,"value":"report_date = (f)=> {\n if (f) return report_date_f(f);\n return \"\";\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":1553,"value":"report_int = d3.format (\",\");","pinned":false,"hidden":false,"mode":"ojs"},{"id":824,"value":"$33","pinned":false,"hidden":false,"mode":"ojs"},{"id":1168,"value":"date_range = d3.extent (_.map(linkage_data.records, (d)=>d[1]))","pinned":false,"hidden":false,"mode":"ojs"},{"id":1312,"value":"toSignature = (m)=>{\n if (m.reference_value) {\n return linkage_data.reference_base[m.pos-1]+m.pos;\n } \n return \"\"+m.pos+m.variant;\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":2188,"value":"mapHeight = 550","pinned":false,"hidden":false,"mode":"ojs"},{"id":2192,"value":"mapWidth = 960","pinned":false,"hidden":false,"mode":"ojs"},{"id":2428,"value":"trendWidth = 960","pinned":false,"hidden":false,"mode":"ojs"},{"id":2432,"value":"trendHeight = 500","pinned":false,"hidden":false,"mode":"ojs"},{"id":2157,"value":"topojson = require(\"topojson-client@3\")","pinned":false,"hidden":false,"mode":"ojs"},{"id":2179,"value":"mapProjection = d3.geoMercator().translate ([mapWidth/2,mapHeight*0.65]);//.fitSize([mapWidth, mapHeight], { type: \"Sphere\" })","pinned":false,"hidden":false,"mode":"ojs"},{"id":2258,"value":"import { Scrubber } from \"@mbostock/scrubber\"","pinned":false,"hidden":false,"mode":"ojs"},{"id":2263,"value":"sliderDate = wk =>\n report_date_f (deltaweekToDate(wk))","pinned":false,"hidden":false,"mode":"ojs"},{"id":2305,"value":"colorScheme = d3.interpolateOrRd\n","pinned":false,"hidden":false,"mode":"ojs"},{"id":2310,"value":"colorScale = d3.scaleSequential([0,1],colorScheme);","pinned":false,"hidden":false,"mode":"ojs"},{"id":2298,"value":"import { legend, swatches } from \"@d3/color-legend\"","pinned":false,"hidden":false,"mode":"ojs"},{"id":1139,"value":"html``","pinned":false,"hidden":false,"mode":"ojs"}],"files":"$W34","stdlib":"1","titling":"auto"}