| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498 |
- /* jshint esversion: 6 */
- var view = view || {};
- var base = base || require('./base');
- var zip = zip || require('./zip');
- var gzip = gzip || require('./gzip');
- var tar = tar || require('./tar');
- var json = json || require('./json');
- var protobuf = protobuf || require('./protobuf');
- var flatbuffers = flatbuffers || require('./flatbuffers');
- var python = python || require('./python');
- var sidebar = sidebar || require('./view-sidebar');
- var grapher = grapher || require('./view-grapher');
- view.View = class {
- constructor(host, id) {
- this._host = host;
- this._id = id ? ('-' + id) : '';
- this._host.initialize(this).then(() => {
- this._model = null;
- this._graphs = [];
- this._selection = [];
- this._sidebar = new sidebar.Sidebar(this._host, id);
- this._showAttributes = false;
- this._showInitializers = true;
- this._showNames = false;
- this._showHorizontal = false;
- this._searchText = '';
- this._modelFactoryService = new view.ModelFactoryService(this._host);
- this._getElementById('zoom-in-button').addEventListener('click', () => {
- this.zoomIn();
- });
- this._getElementById('zoom-out-button').addEventListener('click', () => {
- this.zoomOut();
- });
- this._getElementById('back-button').addEventListener('click', () => {
- this.popGraph();
- });
- this._getElementById('name-button').addEventListener('click', () => {
- this.showDocumentation(this.activeGraph);
- });
- this._getElementById('sidebar').addEventListener('mousewheel', (e) => {
- this._preventZoom(e);
- }, { passive: true });
- this._host.document.addEventListener('keydown', () => {
- this.clearSelection();
- });
- this._host.start();
- switch (this._host.environment('zoom')) {
- case 'scroll': {
- const userAgent = navigator.userAgent.toLowerCase();
- const safari = userAgent.indexOf('safari') !== -1 && userAgent.indexOf('chrome') === -1;
- const elements = [ 'graph', 'toolbar' ];
- for (const id of elements) {
- const element = this._getElementById(id);
- element.addEventListener('mousewheel', (e) => {
- this._mouseWheelHandler(e);
- });
- element.addEventListener('scroll', (e) => {
- this._scrollHandler(e);
- });
- element.addEventListener('wheel', (e) => {
- this._mouseWheelHandler(e);
- });
- if (safari) {
- element.addEventListener('gesturestart', (e) => {
- e.preventDefault();
- this._gestureZoom = this._zoom;
- }, false);
- element.addEventListener('gesturechange', (e) => {
- e.preventDefault();
- this._updateZoom(this._gestureZoom * e.scale, e);
- }, false);
- element.addEventListener('gestureend', (e) => {
- e.preventDefault();
- this._updateZoom(this._gestureZoom * e.scale, e);
- }, false);
- }
- else {
- element.addEventListener('touchstart', (e) => {
- if (e.touches.length === 2) {
- this._touchPoints = Array.from(e.touches);
- this._touchZoom = this._zoom;
- }
- }, { passive: true });
- element.addEventListener('touchmove', (e) => {
- if (Array.isArray(this._touchPoints) && this._touchPoints.length === 2 && e.touches.length === 2) {
- const distance = (points) => {
- const dx =(points[1].clientX - points[0].clientX);
- const dy =(points[1].clientY - points[0].clientY);
- return Math.sqrt(dx * dx + dy * dy);
- };
- const d1 = distance(Array.from(e.touches));
- const d2 = distance(this._touchPoints);
- if (d2 !== 0) {
- const points = this._touchPoints;
- const e = {
- pageX: (points[1].pageX + points[0].pageX) / 2,
- pageY: (points[1].pageY + points[0].pageY) / 2
- };
- const zoom = d2 === 0 ? d1 : d1 / d2;
- this._updateZoom(this._touchZoom * zoom, e);
- }
- }
- }, { passive: true });
- element.addEventListener('touchcancel', () => {
- delete this._touchPoints;
- delete this._touchZoom;
- }, { passive: true });
- element.addEventListener('touchend', () => {
- delete this._touchPoints;
- delete this._touchZoom;
- }, { passive: true });
- }
- }
- break;
- }
- case 'drag': {
- this._getElementById('toolbar').addEventListener('mousewheel', (e) => {
- this._preventZoom(e);
- }, { passive: true });
- break;
- }
- }
- }).catch((err) => {
- this.error(err, null, null);
- });
- }
- show(page) {
- if (!page) {
- page = (!this._model && !this.activeGraph) ? 'welcome' : 'default';
- }
- this._host.screen(page);
- if (this._sidebar) {
- this._sidebar.close();
- }
- this._host.document.body.setAttribute('class', page);
- }
- cut() {
- this._host.document.execCommand('cut');
- }
- copy() {
- this._host.document.execCommand('copy');
- }
- paste() {
- this._host.document.execCommand('paste');
- }
- selectAll() {
- this._host.document.execCommand('selectall');
- }
- find() {
- const graph = this.activeGraph;
- if (graph) {
- this.clearSelection();
- const graphElement = this._getElementById('canvas');
- const view = new sidebar.FindSidebar(this._host, graphElement, graph);
- view.on('search-text-changed', (sender, text) => {
- this._searchText = text;
- });
- view.on('select', (sender, selection) => {
- this._sidebar.close();
- this.select(selection);
- });
- this._sidebar.open(view.content, 'Find');
- view.focus(this._searchText);
- }
- }
- get model() {
- return this._model;
- }
- toggleAttributes() {
- this._showAttributes = !this._showAttributes;
- this._reload();
- }
- get showAttributes() {
- return this._showAttributes;
- }
- toggleInitializers() {
- this._showInitializers = !this._showInitializers;
- this._reload();
- }
- get showInitializers() {
- return this._showInitializers;
- }
- toggleNames() {
- this._showNames = !this._showNames;
- this._reload();
- }
- get showNames() {
- return this._showNames;
- }
- toggleDirection() {
- this._showHorizontal = !this._showHorizontal;
- this._reload();
- }
- get showHorizontal() {
- return this._showHorizontal;
- }
- _reload() {
- this.show('welcome spinner');
- if (this._model && this._graphs.length > 0) {
- this._updateGraph(this._model, this._graphs).catch((error) => {
- if (error) {
- this.error(error, 'Graph update failed.', 'welcome');
- }
- });
- }
- }
- _timeout(time) {
- return new Promise((resolve) => {
- setTimeout(() => { resolve(); }, time);
- });
- }
- _getElementById(id) {
- return this._host.document.getElementById(id + this._id);
- }
- zoomIn() {
- switch (this._host.environment('zoom')) {
- case 'scroll':
- this._updateZoom(this._zoom * 1.1);
- break;
- case 'drag':
- if (this._zoom) {
- this._zoom.scaleBy(1.2);
- }
- break;
- }
- }
- zoomOut() {
- switch (this._host.environment('zoom')) {
- case 'scroll':
- this._updateZoom(this._zoom * 0.9);
- break;
- case 'drag':
- if (this._zoom) {
- this._zoom.scaleBy(0.8);
- }
- break;
- }
- }
- resetZoom() {
- switch (this._host.environment('zoom')) {
- case 'scroll':
- this._updateZoom(1);
- break;
- case 'drag':
- if (this._zoom) {
- this._zoom.scaleTo(1);
- }
- break;
- }
- }
- _preventZoom(e) {
- if (e.shiftKey || e.ctrlKey) {
- e.preventDefault();
- }
- }
- _updateZoom(zoom, e) {
- const graphElement = this._getElementById('graph');
- const min = Math.min(Math.max(graphElement.clientHeight / this._height, 0.2), 1);
- zoom = Math.min(zoom, 1.4);
- zoom = Math.max(min, zoom);
- const scrollLeft = this._scrollLeft || graphElement.scrollLeft;
- const scrollTop = this._scrollTop || graphElement.scrollTop;
- const x = (e ? e.pageX : (graphElement.clientWidth / 2)) + scrollLeft;
- const y = (e ? e.pageY : (graphElement.clientHeight / 2)) + scrollTop;
- const canvasElement = this._getElementById('canvas');
- canvasElement.style.width = zoom * this._width;
- canvasElement.style.height = zoom * this._height;
- this._scrollLeft = ((x * zoom) / this._zoom) - (x - scrollLeft);
- this._scrollTop = ((y * zoom) / this._zoom) - (y - scrollTop);
- this._scrollLeft = Math.max(0, this._scrollLeft);
- this._scrollTop = Math.max(0, this._scrollTop);
- graphElement.scrollLeft = this._scrollLeft;
- graphElement.scrollTop = this._scrollTop;
- this._zoom = zoom;
- }
- _mouseWheelHandler(e) {
- if (e.shiftKey || e.ctrlKey) {
- this._updateZoom(this._zoom + (e.wheelDelta * 1.0 / 4000.0), e);
- e.preventDefault();
- }
- }
- _scrollHandler(e) {
- if (this._scrollLeft && e.target.scrollLeft !== Math.floor(this._scrollLeft)) {
- delete this._scrollLeft;
- }
- if (this._scrollTop && e.target.scrollTop !== Math.floor(this._scrollTop)) {
- delete this._scrollTop;
- }
- }
- select(selection) {
- this.clearSelection();
- if (selection && selection.length > 0) {
- const graphElement = this._getElementById('graph');
- switch (this._host.environment('zoom')) {
- case 'drag': {
- let x = 0;
- let y = 0;
- for (const element of selection) {
- element.classList.add('select');
- this._selection.push(element);
- const transform = element.transform.baseVal.consolidate();
- const box = element.getBBox();
- const ex = transform ? transform.matrix.e : box.x + (box.width / 2);
- const ey = transform ? transform.matrix.f : box.y + (box.height / 2);
- x += ex;
- y += ey;
- }
- x = x / selection.length;
- y = y / selection.length;
- const canvasElement = this._getElementById('canvas');
- const canvasRect = canvasElement.getBoundingClientRect();
- this._zoom.transform(view.Zoom.identity().translate((canvasRect.width / 2) - x, (canvasRect.height / 2) - y));
- break;
- }
- case 'scroll': {
- let x = 0;
- let y = 0;
- for (const element of selection) {
- element.classList.add('select');
- this._selection.push(element);
- const rect = element.getBoundingClientRect();
- x += rect.left + (rect.width / 2);
- y += rect.top + (rect.height / 2);
- }
- x = x / selection.length;
- y = y / selection.length;
- const rect = graphElement.getBoundingClientRect();
- const left = (graphElement.scrollLeft + x - rect.left) - (rect.width / 2);
- const top = (graphElement.scrollTop + y - rect.top) - (rect.height / 2);
- graphElement.scrollTo({ left: left, top: top, behavior: 'smooth' });
- break;
- }
- }
- }
- }
- clearSelection() {
- while (this._selection.length > 0) {
- const element = this._selection.pop();
- element.classList.remove('select');
- }
- }
- error(err, name, screen) {
- if (this._sidebar) {
- this._sidebar.close();
- }
- this._host.exception(err, false);
- const knowns = [
- { name: '', message: /^Invalid argument identifier/, url: 'https://github.com/lutzroeder/netron/issues/540' },
- { name: '', message: /^Cannot read property/, url: 'https://github.com/lutzroeder/netron/issues/647' },
- { name: '', message: /^Failed to render tensor/, url: 'https://github.com/lutzroeder/netron/issues/681' },
- { name: 'Error', message: /^EPERM: operation not permitted/, url: 'https://github.com/lutzroeder/netron/issues/551' },
- { name: 'Error', message: /^EACCES: permission denied/, url: 'https://github.com/lutzroeder/netron/issues/504' },
- { name: 'RangeError', message: /^Offset is outside the bounds of the DataView/, url: 'https://github.com/lutzroeder/netron/issues/563' },
- { name: 'RangeError', message: /^start offset of Int32Array/, url: 'https://github.com/lutzroeder/netron/issues/565' },
- { name: 'RangeError', message: /^Maximum call stack size exceeded/, url: 'https://github.com/lutzroeder/netron/issues/589' },
- { name: 'RangeError', message: /^Invalid string length/, url: 'https://github.com/lutzroeder/netron/issues/648' },
- { name: 'Error loading model.', message: /^Unsupported file content \(/, url: 'https://github.com/lutzroeder/netron/issues/550' },
- { name: 'Error loading model.', message: /^Unsupported Protocol Buffers content/, url: 'https://github.com/lutzroeder/netron/issues/593' },
- { name: 'Error loading model.', message: /^Unsupported Protocol Buffers text content/, url: 'https://github.com/lutzroeder/netron/issues/594' },
- { name: 'Error loading model.', message: /^Unsupported JSON content/, url: 'https://github.com/lutzroeder/netron/issues/595' },
- { name: 'Error loading Caffe model.', message: /^File format is not caffe\.NetParameter/, url: 'https://github.com/lutzroeder/netron/issues/563' },
- { name: 'Error loading Darknet model.', message: /^Invalid tensor shape/, url: 'https://github.com/lutzroeder/netron/issues/541' },
- { name: 'Error loading Keras model.', message: /^Unsupported data object header version/, url: 'https://github.com/lutzroeder/netron/issues/548' },
- { name: 'Error loading MNN model.', message: /^File format is not mnn\.Net/, url: 'https://github.com/lutzroeder/netron/issues/746' },
- { name: 'Error loading PyTorch model.', message: /^File does not contain root module or state dictionary/, url: 'https://github.com/lutzroeder/netron/issues/543' },
- { name: 'Error loading PyTorch model.', message: /^Module does not contain modules/, url: 'https://github.com/lutzroeder/netron/issues/544' },
- { name: 'Error loading PyTorch model.', message: /^Failed to resolve module/, url: 'https://github.com/lutzroeder/netron/issues/545' },
- { name: 'Error loading PyTorch model.', message: /^Unsupported function/, url: 'https://github.com/lutzroeder/netron/issues/546' },
- { name: 'Error loading PyTorch model.', message: /^Unsupported uninitialized argument/, url: 'https://github.com/lutzroeder/netron/issues/547' },
- { name: 'Error loading ONNX model.', message: /^File format is not onnx\.ModelProto/, url: 'https://github.com/lutzroeder/netron/issues/549' },
- { name: 'Error loading TensorFlow model.', message: /^File text format is not TensorFlow\.js graph-model/, url: 'https://github.com/lutzroeder/netron/issues/764' },
- { name: 'Error loading TensorFlow Lite model.', message: /^Offset is outside the bounds of the DataView/, url: 'https://github.com/lutzroeder/netron/issues/563' },
- { name: 'Error loading UFF model.', message: /^Unknown attribute/, url: 'https://github.com/lutzroeder/netron/issues/649' }
- ];
- const known = knowns.find((known) => (known.name.length === 0 || known.name === err.name) && err.message.match(known.message));
- const message = err.message + (known ? '\n\nPlease provide information about this issue at ' + known.url + '.' : '');
- name = name || err.name;
- this._host.error(name, message);
- this.show(screen !== undefined ? screen : 'welcome');
- if (known) {
- this._host.openURL(known.url);
- }
- }
- accept(file) {
- return this._modelFactoryService.accept(file);
- }
- open(context) {
- this._host.event('Model', 'Open', 'Size', context.stream ? context.stream.length : 0);
- this._sidebar.close();
- return this._timeout(2).then(() => {
- return this._modelFactoryService.open(context).then((model) => {
- const format = [];
- if (model.format) {
- format.push(model.format);
- }
- if (model.producer) {
- format.push('(' + model.producer + ')');
- }
- if (format.length > 0) {
- this._host.event('Model', 'Format', format.join(' '));
- }
- return this._timeout(20).then(() => {
- const graphs = Array.isArray(model.graphs) && model.graphs.length > 0 ? [ model.graphs[0] ] : [];
- return this._updateGraph(model, graphs);
- });
- });
- });
- }
- _updateActiveGraph(name) {
- this._sidebar.close();
- if (this._model) {
- const model = this._model;
- const graph = model.graphs.filter(graph => name === graph.name).shift();
- if (graph) {
- this.show('welcome spinner');
- this._timeout(200).then(() => {
- return this._updateGraph(model, [ graph ]).catch((error) => {
- if (error) {
- this.error(error, 'Graph update failed.', 'welcome');
- }
- });
- });
- }
- }
- }
- get activeGraph() {
- return Array.isArray(this._graphs) && this._graphs.length > 0 ? this._graphs[0] : null;
- }
- _updateGraph(model, graphs) {
- const update = () => {
- const nameButton = this._getElementById('name-button');
- const backButton = this._getElementById('back-button');
- if (this._graphs.length > 1) {
- // backButton.style.display = 'block';
- // nameButton.style.display = 'block';
- const graph = this.activeGraph;
- nameButton.innerHTML = graph ? graph.name : '';
- backButton.style.opacity = 1;
- nameButton.style.opacity = 1;
- }
- else {
- backButton.style.opacity = 0;
- nameButton.style.opacity = 0;
- }
- };
- const graph = Array.isArray(graphs) && graphs.length > 0 ? graphs[0] : null;
- return this._timeout(100).then(() => {
- if (graph && graph != this._graphs[0]) {
- const nodes = graph.nodes;
- if (nodes.length > 1400) {
- if (!this._host.confirm('Large model detected.', 'This graph contains a large number of nodes and might take a long time to render. Do you want to continue?')) {
- this._host.event('Graph', 'Render', 'Skip', nodes.length);
- this.show(null);
- return null;
- }
- }
- }
- return this.renderGraph(model, graph).then(() => {
- this._model = model;
- this._graphs = graphs;
- if (!graphs || graphs.length <= 1) {
- this.show('default');
- }
- update();
- return this._model;
- }).catch((error) => {
- return this.renderGraph(this._model, this.activeGraph).then(() => {
- if (!graphs || graphs.length <= 1) {
- this.show('default');
- }
- update();
- throw error;
- }).catch(() => {
- throw error;
- });
- });
- });
- }
- pushGraph(graph) {
- if (graph !== this.activeGraph) {
- return this._updateGraph(this._model, [ graph ].concat(this._graphs));
- }
- return Promise.resolve();
- }
- popGraph() {
- if (this._graphs.length > 1) {
- return this._updateGraph(this._model, this._graphs.slice(1));
- }
- }
- renderGraph(model, graph) {
- try {
- const graphElement = this._getElementById('graph');
- const canvasElement = this._getElementById('canvas');
- while (canvasElement.lastChild) {
- canvasElement.removeChild(canvasElement.lastChild);
- }
- if (!graph) {
- return Promise.resolve();
- }
- else {
- switch (this._host.environment('zoom')) {
- case 'scroll':
- this._zoom = 1;
- canvasElement.style.position = 'static';
- canvasElement.style.margin = 'auto';
- break;
- case 'drag':
- this._zoom = null;
- canvasElement.style.position = 'absolute';
- canvasElement.style.margin = '0';
- break;
- }
- const groups = graph.groups;
- const nodes = graph.nodes;
- this._host.event('Graph', 'Render', 'Size', nodes.length);
- const options = {};
- options.nodesep = 25;
- options.ranksep = 20;
- const rotate = graph.nodes.every((node) => node.inputs.filter((input) => input.arguments.every((argument) => !argument.initializer)).length === 0 && node.outputs.length === 0);
- const showHorizontal = rotate ? !this._showHorizontal : this._showHorizontal;
- if (showHorizontal) {
- options.rankdir = "LR";
- }
- if (nodes.length > 1500) {
- options.ranker = 'longest-path';
- }
- const viewGraph = new view.Graph(this, model, groups, options);
- const clusters = new Set();
- const clusterParentMap = new Map();
- if (groups) {
- for (const node of nodes) {
- if (node.group) {
- const path = node.group.split('/');
- while (path.length > 0) {
- const name = path.join('/');
- path.pop();
- clusterParentMap.set(name, path.join('/'));
- }
- }
- }
- }
- for (const node of nodes) {
- const viewNode = viewGraph.createNode(node);
- const inputs = node.inputs;
- for (const input of inputs) {
- for (const argument of input.arguments) {
- if (argument.name != '' && !argument.initializer) {
- viewGraph.createArgument(argument).to(viewNode);
- }
- }
- }
- let outputs = node.outputs;
- if (node.chain && node.chain.length > 0) {
- const chainOutputs = node.chain[node.chain.length - 1].outputs;
- if (chainOutputs.length > 0) {
- outputs = chainOutputs;
- }
- }
- for (const output of outputs) {
- for (const argument of output.arguments) {
- if (!argument) {
- throw new view.Error("Invalid null argument in '" + model.format + "'.");
- }
- if (argument.name != '') {
- viewGraph.createArgument(argument).from(viewNode);
- }
- }
- }
- if (node.controlDependencies && node.controlDependencies.length > 0) {
- for (const name of node.controlDependencies) {
- viewGraph.createArgument({ name: name, controlDependency: true }).to(viewNode);
- }
- }
- const createCluster = function(name) {
- if (!clusters.has(name)) {
- viewGraph.setNode({ name: name, rx: 5, ry: 5});
- clusters.add(name);
- const parent = clusterParentMap.get(name);
- if (parent) {
- createCluster(parent);
- viewGraph.setParent(name, parent);
- }
- }
- };
- if (groups) {
- let groupName = node.group;
- if (groupName && groupName.length > 0) {
- if (!clusterParentMap.has(groupName)) {
- const lastIndex = groupName.lastIndexOf('/');
- if (lastIndex != -1) {
- groupName = groupName.substring(0, lastIndex);
- if (!clusterParentMap.has(groupName)) {
- groupName = null;
- }
- }
- else {
- groupName = null;
- }
- }
- if (groupName) {
- createCluster(groupName);
- viewGraph.setParent(viewNode.name, groupName);
- }
- }
- }
- }
- for (const input of graph.inputs) {
- const viewInput = viewGraph.createInput(input);
- for (const argument of input.arguments) {
- viewGraph.createArgument(argument).from(viewInput);
- }
- }
- for (const output of graph.outputs) {
- const viewOutput = viewGraph.createOutput(output);
- for (const argument of output.arguments) {
- viewGraph.createArgument(argument).to(viewOutput);
- }
- }
- // Workaround for Safari background drag/zoom issue:
- // https://stackoverflow.com/questions/40887193/d3-js-zoom-is-not-working-with-mousewheel-in-safari
- const backgroundElement = this._host.document.createElementNS('http://www.w3.org/2000/svg', 'rect');
- backgroundElement.setAttribute('id', 'background');
- if (this._host.environment('zoom') === 'drag') {
- backgroundElement.setAttribute('width', '100%');
- backgroundElement.setAttribute('height', '100%');
- }
- backgroundElement.setAttribute('fill', 'none');
- backgroundElement.setAttribute('pointer-events', 'all');
- canvasElement.appendChild(backgroundElement);
- const originElement = this._host.document.createElementNS('http://www.w3.org/2000/svg', 'g');
- originElement.setAttribute('id', 'origin');
- canvasElement.appendChild(originElement);
- viewGraph.build(this._host.document, originElement);
- switch (this._host.environment('zoom')) {
- case 'drag': {
- this._zoom = new view.Zoom(canvasElement, 0.1, 1.4);
- this._zoom.on('zoom', (sender, data) => {
- originElement.setAttribute('transform', data.transform.toString());
- });
- this._zoom.transform(view.Zoom.identity());
- break;
- }
- case 'scroll': {
- this._zoom = 1;
- break;
- }
- }
- return this._timeout(20).then(() => {
- viewGraph.layout();
- const elements = Array.from(canvasElement.getElementsByClassName('graph-input') || []);
- if (elements.length === 0) {
- const nodeElements = Array.from(canvasElement.getElementsByClassName('graph-node') || []);
- if (nodeElements.length > 0) {
- elements.push(nodeElements[0]);
- }
- }
- switch (this._host.environment('zoom')) {
- case 'drag': {
- const svgSize = canvasElement.getBoundingClientRect();
- if (elements && elements.length > 0) {
- // Center view based on input elements
- const xs = [];
- const ys = [];
- for (let i = 0; i < elements.length; i++) {
- const transform = elements[i].transform.baseVal.consolidate();
- if (transform) {
- xs.push(transform.matrix.e);
- ys.push(transform.matrix.f);
- }
- }
- let x = xs[0];
- const y = ys[0];
- if (ys.every(y => y === ys[0])) {
- x = xs.reduce((a, b) => a + b, 0) / xs.length;
- }
- const sx = (svgSize.width / (this._showHorizontal ? 4 : 2)) - x;
- const sy = (svgSize.height / (this._showHorizontal ? 2 : 4)) - y;
- this._zoom.transform(view.Zoom.identity().translate(sx, sy));
- }
- else {
- this._zoom.transform(view.Zoom.identity().translate((svgSize.width - viewGraph.graph().width) / 2, (svgSize.height - viewGraph.graph().height) / 2));
- }
- break;
- }
- case 'scroll': {
- const size = canvasElement.getBBox();
- const margin = 100;
- const width = Math.ceil(margin + size.width + margin);
- const height = Math.ceil(margin + size.height + margin);
- originElement.setAttribute('transform', 'translate(' + margin.toString() + ', ' + margin.toString() + ') scale(1)');
- backgroundElement.setAttribute('width', width);
- backgroundElement.setAttribute('height', height);
- this._width = width;
- this._height = height;
- this._zoom = 1;
- delete this._scrollLeft;
- delete this._scrollRight;
- canvasElement.setAttribute('viewBox', '0 0 ' + width + ' ' + height);
- canvasElement.setAttribute('width', width);
- canvasElement.setAttribute('height', height);
- this._updateZoom(this._zoom);
- if (elements && elements.length > 0) {
- // Center view based on input elements
- const xs = [];
- const ys = [];
- for (let i = 0; i < elements.length; i++) {
- const element = elements[i];
- const rect = element.getBoundingClientRect();
- xs.push(rect.left + (rect.width / 2));
- ys.push(rect.top + (rect.height / 2));
- }
- let x = xs[0];
- const y = ys[0];
- if (ys.every(y => y === ys[0])) {
- x = xs.reduce((a, b) => a + b, 0) / xs.length;
- }
- // const canvasRect = graphElement.getBoundingClientRect();
- const graphRect = graphElement.getBoundingClientRect();
- // const sx = (canvasRect.width / (this._showHorizontal ? 4 : 2)) - x;
- // const sy = (canvasRect.height / (this._showHorizontal ? 2 : 4)) - y;
- const left = (graphElement.scrollLeft + x - graphRect.left) - (graphRect.width / 2);
- const top = (graphElement.scrollTop + y - graphRect.top) - (graphRect.height / 2);
- graphElement.scrollTo({ left: left, top: top, behavior: 'auto' });
- }
- else {
- const canvasRect = graphElement.getBoundingClientRect();
- const graphRect = graphElement.getBoundingClientRect();
- const left = (graphElement.scrollLeft + (canvasRect.width / 2) - graphRect.left) - (graphRect.width / 2);
- const top = (graphElement.scrollTop + (canvasRect.height / 2) - graphRect.top) - (graphRect.height / 2);
- graphElement.scrollTo({ left: left, top: top, behavior: 'auto' });
- }
- break;
- }
- }
- return;
- });
- }
- }
- catch (error) {
- return Promise.reject(error);
- }
- }
- applyStyleSheet(element, name) {
- let rules = [];
- for (const styleSheet of this._host.document.styleSheets) {
- if (styleSheet && styleSheet.href && styleSheet.href.endsWith('/' + name)) {
- rules = styleSheet.cssRules;
- break;
- }
- }
- const nodes = element.getElementsByTagName('*');
- for (const node of nodes) {
- for (const rule of rules) {
- if (node.matches(rule.selectorText)) {
- for (const item of rule.style) {
- node.style[item] = rule.style[item];
- }
- }
- }
- }
- }
- export(file) {
- const lastIndex = file.lastIndexOf('.');
- const extension = (lastIndex != -1) ? file.substring(lastIndex + 1) : '';
- if (this.activeGraph && (extension === 'png' || extension === 'svg')) {
- const graphElement = this._getElementById('canvas');
- const exportElement = graphElement.cloneNode(true);
- this.applyStyleSheet(exportElement, 'view-grapher.css');
- exportElement.setAttribute('id', 'export');
- exportElement.removeAttribute('width');
- exportElement.removeAttribute('height');
- exportElement.style.removeProperty('opacity');
- exportElement.style.removeProperty('display');
- const backgroundElement = exportElement.querySelector('#background');
- const originElement = exportElement.querySelector('#origin');
- originElement.setAttribute('transform', 'translate(0,0) scale(1)');
- backgroundElement.removeAttribute('width');
- backgroundElement.removeAttribute('height');
- const parentElement = graphElement.parentElement;
- parentElement.insertBefore(exportElement, graphElement);
- const size = exportElement.getBBox();
- parentElement.removeChild(exportElement);
- parentElement.removeChild(graphElement);
- parentElement.appendChild(graphElement);
- const delta = (Math.min(size.width, size.height) / 2.0) * 0.1;
- const width = Math.ceil(delta + size.width + delta);
- const height = Math.ceil(delta + size.height + delta);
- originElement.setAttribute('transform', 'translate(' + delta.toString() + ', ' + delta.toString() + ') scale(1)');
- exportElement.setAttribute('width', width);
- exportElement.setAttribute('height', height);
- backgroundElement.setAttribute('width', width);
- backgroundElement.setAttribute('height', height);
- backgroundElement.setAttribute('fill', '#fff');
- const data = new XMLSerializer().serializeToString(exportElement);
- if (extension === 'svg') {
- const blob = new Blob([ data ], { type: 'image/svg' });
- this._host.export(file, blob);
- }
- if (extension === 'png') {
- const imageElement = new Image();
- imageElement.onload = () => {
- const max = Math.max(width, height);
- const scale = Math.min(24000.0 / max, 2.0);
- const canvas = this._host.document.createElement('canvas');
- canvas.width = Math.ceil(width * scale);
- canvas.height = Math.ceil(height * scale);
- const context = canvas.getContext('2d');
- context.scale(scale, scale);
- context.drawImage(imageElement, 0, 0);
- canvas.toBlob((blob) => {
- if (blob) {
- this._host.export(file, blob);
- }
- else {
- const err = new Error();
- err.name = 'Error exporting image.';
- err.message = 'Image may be too large to render as PNG.';
- this._host.exception(err, false);
- this._host.error(err.name, err.message);
- }
- }, 'image/png');
- };
- imageElement.src = 'data:image/svg+xml;base64,' + this._host.window.btoa(unescape(encodeURIComponent(data)));
- }
- }
- }
- showModelProperties() {
- if (this._model) {
- const modelSidebar = new sidebar.ModelSidebar(this._host, this._model, this.activeGraph);
- modelSidebar.on('update-active-graph', (sender, name) => {
- this._updateActiveGraph(name);
- });
- this._sidebar.open(modelSidebar.render(), 'Model Properties');
- }
- }
- showNodeProperties(node, input) {
- if (node) {
- const nodeSidebar = new sidebar.NodeSidebar(this._host, node);
- nodeSidebar.on('show-documentation', (/* sender, e */) => {
- this.showDocumentation(node.type);
- });
- nodeSidebar.on('show-graph', (sender, graph) => {
- this.pushGraph(graph);
- });
- nodeSidebar.on('export-tensor', (sender, tensor) => {
- this._host.require('./numpy').then((numpy) => {
- const defaultPath = tensor.name ? tensor.name.split('/').join('_').split(':').join('_').split('.').join('_') : 'tensor';
- this._host.save('NumPy Array', 'npy', defaultPath, (file) => {
- try {
- const dataTypeMap = new Map([
- [ 'float16', 'f2' ], [ 'float32', 'f4' ], [ 'float64', 'f8' ],
- [ 'int8', 'i1' ], [ 'int16', 'i2'], [ 'int32', 'i4' ], [ 'int64', 'i8' ],
- [ 'uint8', 'u1' ], [ 'uint16', 'u2' ], [ 'uint32', 'u4' ], [ 'uint64', 'u8' ],
- [ 'qint8', 'i1' ], [ 'qint16', 'i2' ],
- [ 'quint8', 'u1' ], [ 'quint16', 'u2' ]
- ]);
- const array = new numpy.Array();
- array.shape = tensor.type.shape.dimensions;
- array.data = tensor.value;
- array.dataType = dataTypeMap.has(tensor.type.dataType) ? dataTypeMap.get(tensor.type.dataType) : tensor.type.dataType;
- const blob = new Blob([ array.toBuffer() ], { type: 'application/octet-stream' });
- this._host.export(file, blob);
- }
- catch (error) {
- this.error(error, 'Error saving NumPy tensor.', null);
- }
- });
- }).catch(() => {
- });
- });
- nodeSidebar.on('error', (sender, error) => {
- if (this._model) {
- error.message = error.message.replace(/\.$/, '') + " in format '" + this._model.format + "'.";
- }
- this.error(error, null, null);
- });
- if (input) {
- nodeSidebar.toggleInput(input.name);
- }
- this._sidebar.open(nodeSidebar.render(), 'Node Properties');
- }
- }
- showDocumentation(type) {
- if (type && (type.description || type.inputs || type.outputs || type.attributes)) {
- if (type.nodes) {
- this.pushGraph(type);
- }
- const documentationSidebar = new sidebar.DocumentationSidebar(this._host, type);
- documentationSidebar.on('navigate', (sender, e) => {
- this._host.openURL(e.link);
- });
- this._sidebar.push(documentationSidebar.render(), 'Documentation');
- }
- }
- };
- view.Graph = class extends grapher.Graph {
- constructor(view, model, compound, options) {
- super(compound, options);
- this.view = view;
- this.model = model;
- this._arguments = new Map();
- this._nodeKey = 0;
- }
- createNode(node) {
- const value = new view.Node(this, node);
- value.name = this._nodeKey++;
- this.setNode(value);
- return value;
- }
- createInput(input) {
- const value = new view.Input(this, input);
- value.name = this._nodeKey++;
- this.setNode(value);
- return value;
- }
- createOutput(output) {
- const value = new view.Output(this, output);
- value.name = this._nodeKey++;
- this.setNode(value);
- return value;
- }
- createArgument(argument) {
- const name = argument.name;
- if (!this._arguments.has(name)) {
- this._arguments.set(name, new view.Argument(this, argument));
- }
- return this._arguments.get(name);
- }
- createEdge(from, to) {
- const value = new view.Edge(from, to);
- return value;
- }
- build(document, originElement) {
- for (const argument of this._arguments.values()) {
- argument.build();
- }
- super.build(document, originElement);
- }
- };
- view.Node = class extends grapher.Node {
- constructor(context, value) {
- super();
- this.context = context;
- this.value = value;
- view.Node.counter = view.Node.counter || 0;
- this.id = 'node-' + (value.name ? 'name-' + value.name : 'id-' + (view.Node.counter++).toString());
- this._add(this.value);
- }
- get class() {
- return 'graph-node';
- }
- _add(node) {
- const header = this.header();
- const styles = [ 'node-item-type' ];
- const type = node.type;
- const category = type && type.category ? type.category : '';
- if (category) {
- styles.push('node-item-type-' + category.toLowerCase());
- }
- if (typeof type.name !== 'string' || !type.name.split) { // #416
- const format = this.context.model && this.context.model.format ? this.context.model.format : '?';
- throw new view.Error("Unknown node type '" + JSON.stringify(type.name) + "' in format '" + format + "'.");
- }
- const content = this.context.view.showNames && (node.name || node.location) ? (node.name || node.location) : type.name.split('.').pop();
- const tooltip = this.context.view.showNames && (node.name || node.location) ? type.name : (node.name || node.location);
- header.add(null, styles, content, tooltip, () => {
- this.context.view.showNodeProperties(node, null);
- });
- if (node.type.nodes) {
- header.add(null, styles, '\u238B', 'Show Function Definition', () => {
- this.context.view.pushGraph(node.type);
- });
- }
- if (node.nodes) {
- header.add(null, styles, '+', null, () => {
- // debugger;
- });
- }
- const initializers = [];
- let hiddenInitializers = false;
- if (this.context.view.showInitializers) {
- for (const input of node.inputs) {
- if (input.visible && input.arguments.length === 1 && input.arguments[0].initializer != null) {
- initializers.push(input);
- }
- if ((!input.visible || input.arguments.length > 1) &&
- input.arguments.some((argument) => argument.initializer != null)) {
- hiddenInitializers = true;
- }
- }
- }
- let sortedAttributes = [];
- const attributes = node.attributes || [];
- if (this.context.view.showAttributes) {
- sortedAttributes = attributes.filter((attribute) => attribute.visible).slice();
- }
- sortedAttributes.sort((a, b) => {
- const au = a.name.toUpperCase();
- const bu = b.name.toUpperCase();
- return (au < bu) ? -1 : (au > bu) ? 1 : 0;
- });
- if (initializers.length > 0 || hiddenInitializers || sortedAttributes.length > 0) {
- const block = this.list();
- block.handler = () => {
- this.context.view.showNodeProperties(node);
- };
- for (const initializer of initializers) {
- const argument = initializer.arguments[0];
- const type = argument.type;
- let shape = '';
- let separator = '';
- if (type && type.shape && type.shape.dimensions && Array.isArray(type.shape.dimensions)) {
- shape = '\u3008' + type.shape.dimensions.map((d) => d ? d : '?').join('\u00D7') + '\u3009';
- if (type.shape.dimensions.length === 0 && argument.initializer && !argument.initializer.state) {
- try {
- shape = argument.initializer.toString();
- if (shape && shape.length > 10) {
- shape = shape.substring(0, 10) + '\u2026';
- }
- separator = ' = ';
- }
- catch (err) {
- let type = '?';
- try {
- type = argument.initializer.type.toString();
- }
- catch (error) {
- // continue regardless of error
- }
- const format = this.context.view.model && this.context.view.model.format ? this.context.view.model.format : '?';
- throw new view.Error("Failed to render tensor of type '" + type + "' in format '" + format + "' (" + err.message + ").");
- }
- }
- }
- block.add(argument.name ? 'initializer-' + argument.name : '', initializer.name, shape, type ? type.toString() : '', separator);
- }
- if (hiddenInitializers) {
- block.add(null, '\u3008' + '\u2026' + '\u3009', '', null, '');
- }
- for (const attribute of sortedAttributes) {
- if (attribute.visible) {
- let value = sidebar.NodeSidebar.formatAttributeValue(attribute.value, attribute.type);
- if (value && value.length > 25) {
- value = value.substring(0, 25) + '\u2026';
- }
- block.add(null, attribute.name, value, attribute.type, ' = ');
- }
- }
- }
- if (Array.isArray(node.chain) && node.chain.length > 0) {
- for (const innerNode of node.chain) {
- this._add(innerNode);
- }
- }
- if (node.inner) {
- this._add(node.inner);
- }
- }
- };
- view.Input = class extends grapher.Node {
- constructor(context, value) {
- super();
- this.context = context;
- this.value = value;
- view.Input.counter = view.Input.counter || 0;
- const types = value.arguments.map((argument) => argument.type || '').join('\n');
- let name = value.name || '';
- if (name.length > 16) {
- name = name.split('/').pop();
- }
- const header = this.header();
- header.add(null, [ 'graph-item-input' ], name, types, () => this.context.view.showModelProperties());
- this.id = 'input-' + (name ? 'name-' + name : 'id-' + (view.Input.counter++).toString());
- }
- get class() {
- return 'graph-input';
- }
- };
- view.Output = class extends grapher.Node {
- constructor(context, value) {
- super();
- this.context = context;
- this.value = value;
- const types = value.arguments.map((argument) => argument.type || '').join('\n');
- let name = value.name || '';
- if (name.length > 16) {
- name = name.split('/').pop();
- }
- const header = this.header();
- header.add(null, [ 'graph-item-output' ], name, types, () => this.context.view.showModelProperties());
- }
- };
- view.Argument = class {
- constructor(context, argument) {
- this.context = context;
- this._argument = argument;
- }
- from(node) {
- this._from = node;
- }
- to(node) {
- this._to = this._to || [];
- this._to.push(node);
- }
- build() {
- this._edges = this._edges || [];
- if (this._from && this._to) {
- for (const to of this._to) {
- let text = '';
- const type = this._argument.type;
- if (type && type.shape && type.shape.dimensions && type.shape.dimensions.length > 0) {
- text = type.shape.dimensions.map((dimension) => dimension || '?').join('\u00D7');
- }
- if (this.context.view.showNames) {
- text = this._argument.name.split('\n').shift(); // custom argument id
- }
- const edge = this.context.createEdge(this._from, to);
- edge.v = this._from.name;
- edge.w = to.name;
- edge.label = text;
- edge.id = 'edge-' + this._argument.name;
- if (this._argument.controlDependency) {
- edge.class = 'edge-path-control-dependency';
- }
- this.context.setEdge(edge);
- this._edges.push(edge);
- }
- }
- }
- };
- view.Edge = class extends grapher.Edge {
- constructor(from, to) {
- super(from, to);
- }
- };
- view.ModelContext = class {
- constructor(context, formats) {
- this._context = context;
- this._tags = new Map();
- this._content = new Map();
- this._formats = formats || new Map();
- }
- get identifier() {
- return this._context.identifier;
- }
- get stream() {
- return this._context.stream;
- }
- request(file, encoding, base) {
- return this._context.request(file, encoding, base);
- }
- require(id) {
- return this._context.require(id);
- }
- exception(error, fatal) {
- this._context.exception(error, fatal);
- }
- entries(format) {
- return this._formats.get(format) || new Map();
- }
- open(type) {
- if (!this._content.has(type)) {
- this._content.set(type, undefined);
- const stream = this.stream;
- const position = stream.position;
- const skip =
- Array.from(this._tags).some((pair) => pair[0] !== 'flatbuffers' && pair[1].size > 0) ||
- Array.from(this._content.values()).some((obj) => obj !== undefined);
- if (!skip) {
- switch (type) {
- case 'json': {
- try {
- const reader = json.TextReader.open(stream);
- if (reader) {
- const obj = reader.read();
- this._content.set(type, obj);
- }
- }
- catch (err) {
- // continue regardless of error
- }
- break;
- }
- case 'pkl': {
- let unpickler = null;
- try {
- if (stream.length > 2) {
- const zlib = (stream) => {
- const buffer = stream.peek(2);
- if (buffer[0] === 0x78) {
- const check = (buffer[0] << 8) + buffer[1];
- if (check % 31 === 0) {
- const archive = zip.Archive.open(stream);
- return archive.entries.get('');
- }
- }
- return stream;
- };
- unpickler = python.Unpickler.open(zlib(stream));
- }
- }
- catch (err) {
- // continue regardless of error
- }
- if (unpickler) {
- const execution = new python.Execution(null, (error, fatal) => {
- const message = error && error.message ? error.message : error.toString();
- this.exception(new view.Error(message.replace(/\.$/, '') + " in '" + this.identifier + "'."), fatal);
- });
- const obj = unpickler.load((name, args) => execution.invoke(name, args));
- this._content.set(type, obj);
- }
- break;
- }
- }
- }
- if (stream.position !== position) {
- stream.seek(0);
- }
- }
- return this._content.get(type);
- }
- tags(type) {
- if (!this._tags.has(type)) {
- let tags = new Map();
- const stream = this.stream;
- const position = stream.position;
- if (stream) {
- const signatures = [
- [ 0x80, undefined, 0x8a, 0x0a, 0x6c, 0xfc, 0x9c, 0x46, 0xf9, 0x20, 0x6a, 0xa8, 0x50, 0x19 ], // PyTorch
- [ 0x50, 0x4b ], // Zip
- [ 0x1f, 0x8b ] // Gzip
- ];
- const skip =
- signatures.some((signature) => signature.length <= stream.length && stream.peek(signature.length).every((value, index) => signature[index] === undefined || signature[index] === value)) ||
- (Array.from(this._tags).some((pair) => pair[0] !== 'flatbuffers' && pair[1].size > 0) && type !== 'pb+') ||
- Array.from(this._content.values()).some((obj) => obj !== undefined);
- if (!skip) {
- try {
- switch (type) {
- case 'pbtxt': {
- const reader = protobuf.TextReader.open(stream);
- tags = reader ? reader.signature() : tags;
- break;
- }
- case 'pb': {
- const reader = protobuf.BinaryReader.open(stream);
- tags = reader.signature();
- break;
- }
- case 'pb+': {
- const reader = protobuf.BinaryReader.open(stream);
- tags = reader.decode();
- break;
- }
- case 'flatbuffers': {
- if (stream.length >= 8) {
- const buffer = stream.peek(Math.min(32, stream.length));
- const reader = flatbuffers.BinaryReader.open(buffer);
- const identifier = reader.identifier;
- if (identifier.length > 0) {
- tags.set('file_identifier', identifier);
- }
- }
- break;
- }
- }
- }
- catch (error) {
- tags.clear();
- }
- }
- }
- if (stream.position !== position) {
- stream.seek(position);
- }
- this._tags.set(type, tags);
- }
- return this._tags.get(type);
- }
- };
- view.ArchiveContext = class {
- constructor(host, entries, rootFolder, identifier, stream) {
- this._host = host;
- this._entries = new Map();
- if (entries) {
- for (const entry of entries) {
- if (entry[0].startsWith(rootFolder)) {
- const name = entry[0].substring(rootFolder.length);
- this._entries.set(name, entry[1]);
- }
- }
- }
- this._identifier = identifier.substring(rootFolder.length);
- this._stream = stream;
- }
- get identifier() {
- return this._identifier;
- }
- get stream() {
- return this._stream;
- }
- request(file, encoding, base) {
- if (base === undefined) {
- const stream = this._entries.get(file);
- if (!stream) {
- return Promise.reject(new Error('File not found.'));
- }
- if (encoding) {
- const decoder = new TextDecoder(encoding);
- const buffer = stream.peek();
- const value = decoder.decode(buffer);
- return Promise.resolve(value);
- }
- return Promise.resolve(stream);
- }
- return this._host.request(file, encoding, base);
- }
- require(id) {
- return this._host.require(id);
- }
- exception(error, fatal) {
- this._host.exception(error, fatal);
- }
- };
- view.ArchiveError = class extends Error {
- constructor(message) {
- super(message);
- this.name = 'Error loading archive.';
- }
- };
- view.ModelFactoryService = class {
- constructor(host) {
- this._host = host;
- this._extensions = [];
- this.register('./pytorch', [ '.pt', '.pth', '.pt1', '.pyt', '.pkl', '.pickle', '.h5', '.t7', '.model', '.dms', '.tar', '.ckpt', '.chkpt', '.tckpt', '.bin', '.pb', '.zip', '.nn', '.torchmodel' ]);
- this.register('./onnx', [ '.onnx', '.onn', '.pb', '.pbtxt', '.prototxt', '.model', '.pt', '.pth', '.pkl', '.ort', '.ort.onnx' ]);
- this.register('./mxnet', [ '.json', '.params' ]);
- this.register('./coreml', [ '.mlmodel', '.bin', 'manifest.json', 'metadata.json', 'featuredescriptions.json' ]);
- this.register('./caffe', [ '.caffemodel', '.pbtxt', '.prototxt', '.pt', '.txt' ]);
- this.register('./caffe2', [ '.pb', '.pbtxt', '.prototxt' ]);
- this.register('./torch', [ '.t7' ]);
- this.register('./tflite', [ '.tflite', '.lite', '.tfl', '.bin', '.pb', '.tmfile', '.h5', '.model', '.json', '.txt' ]);
- this.register('./tf', [ '.pb', '.meta', '.pbtxt', '.prototxt', '.pt', '.json', '.index', '.ckpt', '.graphdef', /.data-[0-9][0-9][0-9][0-9][0-9]-of-[0-9][0-9][0-9][0-9][0-9]$/, /^events.out.tfevents./ ]);
- this.register('./mediapipe', [ '.pbtxt' ]);
- this.register('./uff', [ '.uff', '.pb', '.pbtxt', '.uff.txt', '.trt', '.engine' ]);
- this.register('./tensorrt', [ '.trt', '.engine', '.model', '.txt', '.uff', '.pb', '.tmfile', '.onnx', '.pth' ]);
- this.register('./npz', [ '.npz', '.npy', '.pkl' ]);
- this.register('./lasagne', [ '.pkl', '.pickle', '.joblib', '.model', '.pkl.z', '.joblib.z' ]);
- this.register('./lightgbm', [ '.txt', '.pkl', '.model' ]);
- this.register('./sklearn', [ '.pkl', '.pickle', '.joblib', '.model', '.meta', '.pb', '.pt', '.h5', '.pkl.z', '.joblib.z' ]);
- this.register('./pickle', [ '.pkl', '.pickle', '.joblib', '.model', '.meta', '.pb', '.pt', '.h5', '.pkl.z', '.joblib.z' ]);
- this.register('./cntk', [ '.model', '.cntk', '.cmf', '.dnn' ]);
- this.register('./paddle', [ '.pdmodel', '.pdparams', '.pdiparams', '.paddle', '__model__', '.__model__', '.pbtxt', '.txt', '.tar', '.tar.gz' ]);
- this.register('./bigdl', [ '.model', '.bigdl' ]);
- this.register('./darknet', [ '.cfg', '.model', '.txt', '.weights' ]);
- this.register('./weka', [ '.model' ]);
- this.register('./rknn', [ '.rknn', '.onnx' ]);
- this.register('./dlc', [ '.dlc' ]);
- this.register('./keras', [ '.h5', '.hd5', '.hdf5', '.keras', '.json', '.cfg', '.model', '.pb', '.pth', '.weights', '.pkl', '.lite', '.tflite', '.ckpt' ]);
- this.register('./armnn', [ '.armnn', '.json' ]);
- this.register('./mnn', ['.mnn']);
- this.register('./ncnn', [ '.param', '.bin', '.cfg.ncnn', '.weights.ncnn' ]);
- this.register('./tnn', [ '.tnnproto', '.tnnmodel' ]);
- this.register('./tengine', ['.tmfile']);
- this.register('./mslite', [ '.ms']);
- this.register('./barracuda', [ '.nn' ]);
- this.register('./dnn', [ '.dnn' ]);
- this.register('./xmodel', [ '.xmodel' ]);
- this.register('./openvino', [ '.xml', '.bin' ]);
- this.register('./flux', [ '.bson' ]);
- this.register('./dl4j', [ '.zip' ]);
- this.register('./mlnet', [ '.zip' ]);
- this.register('./acuity', [ '.json' ]);
- this.register('./imgdnn', [ '.dnn', 'params', '.json' ]);
- }
- register(id, extensions) {
- for (const extension of extensions) {
- this._extensions.push({ extension: extension, id: id });
- }
- }
- open(context) {
- return this._openSignature(context).then((context) => {
- const containers = new Map();
- let stream = context.stream;
- const entries = context.entries;
- if (!stream && entries && entries.size > 0) {
- containers.set('', entries);
- }
- else {
- const identifier = context.identifier;
- try {
- const archive = gzip.Archive.open(stream);
- if (archive) {
- const entries = archive.entries;
- containers.set('gzip', entries);
- if (archive.entries.size === 1) {
- stream = archive.entries.values().next().value;
- }
- }
- }
- catch (error) {
- const message = error && error.message ? error.message : error.toString();
- throw new view.ArchiveError(message.replace(/\.$/, '') + " in '" + identifier + "'.");
- }
- try {
- const formats = new Map([ [ 'zip', zip ], [ 'tar', tar ] ]);
- for (const pair of formats) {
- const format = pair[0];
- const module = pair[1];
- const archive = module.Archive.open(stream);
- if (archive) {
- containers.set(format, archive.entries);
- containers.delete('gzip');
- break;
- }
- }
- }
- catch (error) {
- const message = error && error.message ? error.message : error.toString();
- throw new view.ArchiveError(message.replace(/\.$/, '') + " in '" + identifier + "'.");
- }
- }
- const modelContext = new view.ModelContext(context, containers);
- return this._openContext(modelContext).then((model) => {
- if (model) {
- return model;
- }
- if (containers.size > 0) {
- return this._openEntries(containers.values().next().value).then((context) => {
- if (context) {
- return this._openContext(context);
- }
- this._unsupported(modelContext);
- });
- }
- this._unsupported(modelContext);
- });
- });
- }
- _unsupported(context) {
- const identifier = context.identifier;
- const extension = identifier.split('.').pop().toLowerCase();
- const stream = context.stream;
- for (const module of [ zip, tar, gzip ]) {
- let archive = null;
- try {
- archive = module.Archive.open(stream);
- }
- catch (error) {
- // continue regardless of error
- }
- if (archive) {
- throw new view.Error("Archive contains no model files in '" + identifier + "'.", true);
- }
- }
- const skip = () => {
- const knownUnsupportedIdentifiers = new Set([
- 'natives_blob.bin',
- 'v8_context_snapshot.bin',
- 'snapshot_blob.bin',
- 'image_net_labels.json',
- 'package.json',
- 'models.json',
- 'LICENSE.meta',
- 'input_0.pb',
- 'output_0.pb'
- ]);
- return knownUnsupportedIdentifiers.has(context.identifier);
- };
- const json = () => {
- const obj = context.open('json');
- if (obj) {
- const formats = [
- { name: 'Netron metadata', tags: [ '[].name', '[].schema' ] },
- { name: 'Netron metadata', tags: [ '[].name', '[].attributes' ] },
- { name: 'Darkflow metadata', tags: [ 'net', 'type', 'model' ] },
- { name: 'keras-yolo2 configuration', tags: [ 'model', 'train', 'valid' ] },
- { name: 'Vulkan SwiftShader ICD manifest', tags: [ 'file_format_version', 'ICD' ] },
- { name: 'DeepLearningExamples configuration', tags: [ 'attention_probs_dropout_prob', 'hidden_act', 'hidden_dropout_prob', 'hidden_size', ] },
- { name: 'NuGet assets', tags: [ 'version', 'targets', 'packageFolders' ] },
- { name: 'NuGet data', tags: [ 'format', 'restore', 'projects' ] },
- { name: 'NPM package', tags: [ 'name', 'version', 'dependencies' ] },
- { name: 'NetworkX adjacency_data', tags: [ 'directed', 'graph', 'nodes' ] }
- ];
- const match = (obj, tag) => {
- if (tag.startsWith('[].')) {
- tag = tag.substring(3);
- return (Array.isArray(obj) && obj.some((item) => Object.prototype.hasOwnProperty.call(item, tag)));
- }
- return Object.prototype.hasOwnProperty.call(obj, tag);
- };
- for (const format of formats) {
- if (format.tags.every((tag) => match(obj, tag))) {
- throw new view.Error('Invalid file content. File contains ' + format.name + '.', true);
- }
- }
- const content = JSON.stringify(obj).substring(0, 100).replace(/\s/, '').substr(0, 48) + '...';
- throw new view.Error("Unsupported JSON content '" + (content.length > 64 ? content.substring(0, 100) + '...' : content) + "' for extension '." + extension + "' in '" + identifier + "'.", !skip());
- }
- };
- const pbtxt = () => {
- const formats = [
- { name: 'ImageNet LabelMap data', tags: [ 'entry', 'entry.target_class' ] },
- { name: 'StringIntLabelMapProto data', tags: [ 'item', 'item.id', 'item.name' ] },
- { name: 'caffe.LabelMap data', tags: [ 'item', 'item.name', 'item.label' ] },
- { name: 'Triton Inference Server configuration', tags: [ 'name', 'platform', 'input', 'output' ] },
- { name: 'TensorFlow OpList data', tags: [ 'op', 'op.name', 'op.input_arg' ] },
- { name: 'vitis.ai.proto.DpuModelParamList data', tags: [ 'model', 'model.name', 'model.kernel' ] },
- { name: 'object_detection.protos.DetectionModel data', tags: [ 'model', 'model.ssd' ] },
- { name: 'object_detection.protos.DetectionModel data', tags: [ 'model', 'model.faster_rcnn' ] },
- { name: 'tensorflow.CheckpointState data', tags: [ 'model_checkpoint_path', 'all_model_checkpoint_paths' ] }
- ];
- const tags = context.tags('pbtxt');
- if (tags.size > 0) {
- for (const format of formats) {
- if (format.tags.every((tag) => tags.has(tag))) {
- throw new view.Error('Invalid file content. File contains ' + format.name + '.', true);
- }
- }
- const entries = [];
- entries.push(...Array.from(tags).filter((pair) => pair[0].toString().indexOf('.') === -1));
- entries.push(...Array.from(tags).filter((pair) => pair[0].toString().indexOf('.') !== -1));
- const content = entries.map((pair) => pair[1] === true ? pair[0] : pair[0] + ':' + JSON.stringify(pair[1])).join(',');
- throw new view.Error("Unsupported Protocol Buffers text content '" + (content.length > 64 ? content.substring(0, 100) + '...' : content) + "' for extension '." + extension + "' in '" + identifier + "'.", !skip());
- }
- };
- const pb = () => {
- const tags = context.tags('pb+');
- if (tags.size > 0) {
- const formats = [
- { name: 'mediapipe.BoxDetectorIndex data', tags: [[1,[[1,[[1,[[1,5],[2,5],[3,5],[4,5],[6,0],[7,5],[8,5],[10,5],[11,0],[12,0]]],[2,5],[3,[]]]],[2,false],[3,false],[4,false],[5,false]]],[2,false],[3,false]] },
- { name: 'sentencepiece.ModelProto data', tags: [[1,[[1,2],[2,5],[3,0]]],[2,[[1,2],[2,2],[3,0],[4,0],[5,2],[6,0],[7,2],[10,5],[16,0],[40,0],[41,0],[42,0],[43,0]]],[3,[]],[4,[]],[5,[]]] },
- { name: 'third_party.tensorflow.python.keras.protobuf.SavedMetadata data', tags: [[1,[[1,[[1,0],[2,0]]],[2,0],[3,2],[4,2],[5,2]]]] }
- ];
- const match = (tags, schema) => {
- for (const pair of schema) {
- const key = pair[0];
- const inner = pair[1];
- if (!tags.has(key)) {
- continue;
- }
- else if (inner === false) {
- return false;
- }
- if (Array.isArray(inner)) {
- const value = tags.get(key);
- if (!(value instanceof Map) || !match(value, inner)) {
- return false;
- }
- }
- else if (inner !== tags.get(key)) {
- return false;
- }
- }
- return true;
- };
- const tags = context.tags('pb+');
- for (const format of formats) {
- if (match(tags, format.tags)) {
- throw new view.Error('Invalid file content. File contains ' + format.name + '.', true);
- }
- }
- const format = (tags) => {
- const text = Array.from(tags).map((pair) => {
- const key = pair[0];
- const value = pair[1];
- return key.toString() + ':' + (value instanceof Map ? '{' + format(value) + '}' : value.toString());
- });
- return text.join(',');
- };
- const content = format(tags);
- throw new view.Error("Unsupported Protocol Buffers content '" + (content.length > 64 ? content.substring(0, 100) + '...' : content) + "' for extension '." + extension + "' in '" + identifier + "'.", !skip());
- }
- };
- const flatbuffers = () => {
- const tags = context.tags('flatbuffers');
- if (tags.has('file_identifier')) {
- const file_identifier = tags.get('file_identifier');
- const formats = [
- { name: 'onnxruntime.experimental.fbs.InferenceSession data', identifier: 'ORTM' },
- { name: 'tflite.Model data', identifier: 'TFL3' }
- ];
- for (const format of formats) {
- if (file_identifier === format.identifier) {
- throw new view.Error('Invalid file content. File contains ' + format.name + '.', true);
- }
- }
- }
- };
- const unknown = () => {
- stream.seek(0);
- const buffer = stream.peek(Math.min(16, stream.length));
- const bytes = Array.from(buffer).map((c) => (c < 16 ? '0' : '') + c.toString(16)).join('');
- const content = stream.length > 268435456 ? '(' + bytes + ') [' + stream.length.toString() + ']': '(' + bytes + ')';
- throw new view.Error("Unsupported file content " + content + " for extension '." + extension + "' in '" + identifier + "'.", !skip());
- };
- json();
- pbtxt();
- pb();
- flatbuffers();
- unknown();
- }
- _openContext(context) {
- const modules = this._filter(context).filter((module) => module && module.length > 0);
- const errors = [];
- let match = false;
- const nextModule = () => {
- if (modules.length > 0) {
- const id = modules.shift();
- return this._host.require(id).then((module) => {
- const updateErrorContext = (error, context) => {
- const text = " in '" + context.identifier + "'.";
- if (error && !error.message.endsWith(text) && (error.context === undefined || error.context === true)) {
- error.message = error.message.replace(/\.$/, '') + text;
- }
- };
- if (!module.ModelFactory) {
- throw new view.Error("Failed to load module '" + id + "'.");
- }
- const modelFactory = new module.ModelFactory();
- try {
- if (!modelFactory.match(context)) {
- return nextModule();
- }
- }
- catch (error) {
- updateErrorContext(error, context);
- return Promise.reject(error);
- }
- match = true;
- return modelFactory.open(context).then((model) => {
- return model;
- }).catch((error) => {
- updateErrorContext(error, context);
- errors.push(error);
- return nextModule();
- });
- });
- }
- else {
- if (match) {
- if (errors.length === 1) {
- const error = errors[0];
- return Promise.reject(error);
- }
- return Promise.reject(new view.Error(errors.map((err) => err.message).join('\n')));
- }
- return Promise.resolve(null);
- }
- };
- return nextModule();
- }
- _openEntries(entries) {
- try {
- const rootFolder = (files) => {
- const map = files.map((file) => file.split('/').slice(0, -1));
- const at = index => list => list[index];
- const rotate = list => list.length === 0 ? [] : list[0].map((item, index) => list.map(at(index)));
- const equals = list => list.every((item) => item === list[0]);
- const folder = rotate(map).filter(equals).map(at(0)).join('/');
- return folder.length === 0 ? folder : folder + '/';
- };
- const filter = (queue) => {
- let matches = [];
- const nextEntry = () => {
- if (queue.length > 0) {
- const entry = queue.shift();
- const context = new view.ModelContext(new view.ArchiveContext(this._host, null, folder, entry.name, entry.stream));
- let modules = this._filter(context);
- const nextModule = () => {
- if (modules.length > 0) {
- const id = modules.shift();
- return this._host.require(id).then((module) => {
- if (!module.ModelFactory) {
- throw new view.ArchiveError("Failed to load module '" + id + "'.", null);
- }
- const factory = new module.ModelFactory();
- if (factory.match(context)) {
- matches.push(entry);
- modules = [];
- }
- return nextModule();
- });
- }
- else {
- return nextEntry();
- }
- };
- return nextModule();
- }
- else {
- if (matches.length === 0) {
- return Promise.resolve(null);
- }
- // MXNet
- if (matches.length === 2 &&
- matches.some((e) => e.name.toLowerCase().endsWith('.params')) &&
- matches.some((e) => e.name.toLowerCase().endsWith('-symbol.json'))) {
- matches = matches.filter((e) => e.name.toLowerCase().endsWith('.params'));
- }
- // TensorFlow.js
- if (matches.length > 0 &&
- matches.some((e) => e.name.toLowerCase().endsWith('.bin')) &&
- matches.some((e) => e.name.toLowerCase().endsWith('.json'))) {
- matches = matches.filter((e) => e.name.toLowerCase().endsWith('.json'));
- }
- // ncnn
- if (matches.length > 0 &&
- matches.some((e) => e.name.toLowerCase().endsWith('.bin')) &&
- matches.some((e) => e.name.toLowerCase().endsWith('.param'))) {
- matches = matches.filter((e) => e.name.toLowerCase().endsWith('.param'));
- }
- // ncnn
- if (matches.length > 0 &&
- matches.some((e) => e.name.toLowerCase().endsWith('.bin')) &&
- matches.some((e) => e.name.toLowerCase().endsWith('.param.bin'))) {
- matches = matches.filter((e) => e.name.toLowerCase().endsWith('.param.bin'));
- }
- // Paddle
- if (matches.length > 0 &&
- matches.some((e) => e.name.toLowerCase().endsWith('.pdmodel')) &&
- matches.some((e) => e.name.toLowerCase().endsWith('.pdiparams'))) {
- matches = matches.filter((e) => e.name.toLowerCase().endsWith('.pdmodel'));
- }
- // TensorFlow Bundle
- if (matches.length > 1 &&
- matches.some((e) => e.name.toLowerCase().endsWith('.data-00000-of-00001'))) {
- matches = matches.filter((e) => !e.name.toLowerCase().endsWith('.data-00000-of-00001'));
- }
- // TensorFlow SavedModel
- if (matches.length === 2 &&
- matches.some((e) => e.name.toLowerCase().split('/').pop() === 'keras_metadata.pb')) {
- matches = matches.filter((e) => e.name.toLowerCase().split('/').pop() !== 'keras_metadata.pb');
- }
- if (matches.length > 1) {
- return Promise.reject(new view.ArchiveError('Archive contains multiple model files.'));
- }
- const match = matches.shift();
- return Promise.resolve(new view.ModelContext(new view.ArchiveContext(this._host, entries, folder, match.name, match.stream)));
- }
- };
- return nextEntry();
- };
- const list = Array.from(entries).map((entry) => {
- return { name: entry[0], stream: entry[1] };
- });
- const files = list.filter((entry) => {
- if (entry.name.endsWith('/')) {
- return false;
- }
- if (entry.name.split('/').pop().startsWith('.')) {
- return false;
- }
- if (!entry.name.startsWith('./') && entry.name.startsWith('.')) {
- return false;
- }
- return true;
- });
- const folder = rootFolder(files.map((entry) => entry.name));
- const queue = files.slice(0).filter((entry) => entry.name.substring(folder.length).indexOf('/') < 0);
- return filter(queue).then((context) => {
- if (context) {
- return Promise.resolve(context);
- }
- const queue = files.slice(0).filter((entry) => entry.name.substring(folder.length).indexOf('/') >= 0);
- return filter(queue);
- });
- }
- catch (error) {
- return Promise.reject(new view.ArchiveError(error.message));
- }
- }
- accept(identifier) {
- const extension = identifier.indexOf('.') === -1 ? '' : identifier.split('.').pop().toLowerCase();
- identifier = identifier.toLowerCase().split('/').pop();
- for (const entry of this._extensions) {
- if ((typeof entry.extension === 'string' && identifier.endsWith(entry.extension)) ||
- (entry.extension instanceof RegExp && entry.extension.exec(identifier))) {
- this._host.event('File', 'Accept', extension, 1);
- return true;
- }
- }
- if (identifier.endsWith('.zip') ||
- identifier.endsWith('.tar') ||
- identifier.endsWith('.tar.gz') ||
- identifier.endsWith('.tgz') ||
- identifier.endsWith('.gz') ||
- identifier.endsWith('.mar') ||
- identifier.endsWith('.model') ||
- identifier.endsWith('.mlpackage')) {
- this._host.event('File', 'Accept', extension, 1);
- return true;
- }
- this._host.event('File', 'Reject', extension, 1);
- return false;
- }
- _filter(context) {
- const identifier = context.identifier.toLowerCase().split('/').pop();
- const list = this._extensions.filter((entry) =>
- (typeof entry.extension === 'string' && identifier.endsWith(entry.extension)) ||
- (entry.extension instanceof RegExp && entry.extension.exec(identifier)));
- return Array.from(new Set(list.map((entry) => entry.id)));
- }
- _openSignature(context) {
- const stream = context.stream;
- if (stream) {
- let empty = true;
- let position = 0;
- while (empty && position < stream.length) {
- const buffer = stream.read(Math.min(4096, stream.length - position));
- position += buffer.length;
- if (!buffer.every((value) => value === 0x00)) {
- empty = false;
- break;
- }
- }
- stream.seek(0);
- if (empty) {
- return Promise.reject(new view.Error('File has no content.', true));
- }
- /* eslint-disable no-control-regex */
- const entries = [
- { name: 'ELF executable', value: /^\x7FELF/ },
- { name: 'PNG image', value: /^\x89PNG/ },
- { name: 'Git LFS header', value: /^version https:\/\/git-lfs.github.com/ },
- { name: 'Git LFS header', value: /^\s*oid sha256:/ },
- { name: 'HTML markup', value: /^\s*<html>/ },
- { name: 'HTML markup', value: /^\s*<!doctype\s*html>/ },
- { name: 'HTML markup', value: /^\s*<!DOCTYPE\s*html>/ },
- { name: 'HTML markup', value: /^\s*<!DOCTYPE\s*HTML>/ },
- { name: 'HTML markup', value: /^\s*<!DOCTYPE\s*HTML\s+(PUBLIC|SYSTEM)?/ },
- { name: 'Unity metadata', value: /^fileFormatVersion:/ },
- { name: 'Python source code', value: /^\s*import[ ]+(os|sys|types|torch|argparse|onnx|numpy|tensorflow)(,|;|\s)/ },
- { name: 'Python source code', value: /^\s*import[ ]+([a-z])+[ ]+as[ ]+/ },
- { name: 'Python source code', value: /^\s*from[ ]+(torch)[ ]+import[ ]+/ },
- { name: 'TSD header', value: /^%TSD-Header-###%/ },
- { name: 'AppleDouble data', value: /^\x00\x05\x16\x07/ },
- { name: 'TensorFlow Hub module', value: /^\x08\x03$/, identifier: 'tfhub_module.pb' }
- ];
- /* eslint-enable no-control-regex */
- const buffer = stream.peek(Math.min(4096, stream.length));
- const text = String.fromCharCode.apply(null, buffer);
- for (const entry of entries) {
- if (text.match(entry.value) && (!entry.identifier || entry.identifier === context.identifier)) {
- return Promise.reject(new view.Error('Invalid file content. File contains ' + entry.name + '.', true));
- }
- }
- }
- return Promise.resolve(context);
- }
- };
- view.Error = class extends Error {
- constructor(message, telemetry) {
- super(message);
- this.name = 'Error loading model.';
- this.telemetry = telemetry;
- this.stack = undefined;
- }
- };
- view.Zoom = class {
- constructor(node, min, max) {
- this._scaleExtent = [ min, max ];
- this._translateExtent = [ [-Infinity, -Infinity], [Infinity, Infinity] ],
- this._touchStarting = false;
- this._touchFirst = false;
- this._touchEnding = false;
- this._touchDelay = 500;
- this._wheelDelay = 150;
- this._events = new Map([ [ 'start', [] ], [ 'zoom', [] ], [ 'end', [] ] ]);
- this._selection = new view.Zoom.Selection(node);
- this._selection.node.__zoom = view.Zoom.identity();
- this._selection.on('wheel.zoom', (event) => this._wheel(event), {passive: false});
- this._selection.on('mousedown.zoom', (event) => this._mouseDown(event));
- if (navigator.maxTouchPoints || node.ontouchstart) {
- this._selection.on('touchstart.zoom', (event) => this._touchStarted(event));
- this._selection.on('touchmove.zoom', (event) => this._touchMoved(event));
- this._selection.on('touchend.zoom', (event) => this._touchEnded(event));
- this._selection.on('touchcancel.zoom', (event) => this._touchEnded(event));
- node.style.setProperty('-webkit-tap-highlight-color', 'rgba(0,0,0,0)', '');
- }
- }
- static identity() {
- view.Zoom._identity = view.Zoom._identity || new view.Zoom.Transform(1, 0, 0);
- return view.Zoom._identity;
- }
- on(event, callback) {
- if (this._events.has(event)) {
- if (callback) {
- this._events.get(event).push(callback);
- }
- else {
- this._events.set([]);
- }
- }
- }
- raise(event, data) {
- if (this._events.has(event)) {
- const callbacks = this._events.get(event);
- for (const callback of callbacks) {
- callback(this, data);
- }
- }
- }
- transform(transform) {
- const node = this._selection.node;
- if (node) {
- this._gesture(node, arguments)
- .start()
- .zoom(null, typeof transform === 'function' ? transform() : transform)
- .end();
- }
- }
- scaleTo(k) {
- const node = this._selection.node;
- if (node) {
- this.transform(() => {
- const e = this.extent(node);
- const t0 = node.__zoom;
- const p0 = this._centroid(e);
- const p1 = t0.invert(p0);
- const k1 = typeof k === 'function' ? k() : k;
- const transform = this.translate(this.scale(t0, k1), p0, p1);
- return this._constrain(transform, e, this._translateExtent);
- });
- }
- }
- scaleBy(k) {
- const node = this._selection.node;
- if (node) {
- this.scaleTo(() => {
- const k0 = node.__zoom.k;
- const k1 = k;
- return k0 * k1;
- });
- }
- }
- scale(transform, k) {
- k = Math.max(this._scaleExtent[0], Math.min(this._scaleExtent[1], k));
- return k === transform.k ? transform : new view.Zoom.Transform(k, transform.x, transform.y);
- }
- translate(transform, p0, p1) {
- const x = p0[0] - p1[0] * transform.k, y = p0[1] - p1[1] * transform.k;
- return x === transform.x && y === transform.y ? transform : new view.Zoom.Transform(transform.k, x, y);
- }
- pointer(event, node) {
- while (event.sourceEvent) {
- event = event.sourceEvent;
- }
- if (node === undefined) {
- node = event.currentTarget;
- }
- if (node) {
- const svg = node.ownerSVGElement || node;
- if (svg.createSVGPoint) {
- let point = svg.createSVGPoint();
- point.x = event.clientX, point.y = event.clientY;
- point = point.matrixTransform(node.getScreenCTM().inverse());
- return [point.x, point.y];
- }
- if (node.getBoundingClientRect) {
- const rect = node.getBoundingClientRect();
- return [event.clientX - rect.left - node.clientLeft, event.clientY - rect.top - node.clientTop];
- }
- }
- return [event.pageX, event.pageY];
- }
- _filter(event) {
- return (!event.ctrlKey || event.type === 'wheel') && !event.button;
- }
- extent(node) {
- let e = node;
- if (e instanceof SVGElement) {
- e = e.ownerSVGElement || e;
- if (e.hasAttribute('viewBox')) {
- e = e.viewBox.baseVal;
- return [[e.x, e.y], [e.x + e.width, e.y + e.height]];
- }
- return [[0, 0], [e.width.baseVal.value, e.height.baseVal.value]];
- }
- return [[0, 0], [e.clientWidth, e.clientHeight]];
- }
- _wheelDelta(event) {
- return -event.deltaY * (event.deltaMode === 1 ? 0.05 : event.deltaMode ? 1 : 0.002) * (event.ctrlKey ? 10 : 1);
- }
- _constrain(transform, extent, translateExtent) {
- const dx0 = transform.invertX(extent[0][0]) - translateExtent[0][0];
- const dx1 = transform.invertX(extent[1][0]) - translateExtent[1][0];
- const dy0 = transform.invertY(extent[0][1]) - translateExtent[0][1];
- const dy1 = transform.invertY(extent[1][1]) - translateExtent[1][1];
- return transform.translate(
- dx1 > dx0 ? (dx0 + dx1) / 2 : Math.min(0, dx0) || Math.max(0, dx1),
- dy1 > dy0 ? (dy0 + dy1) / 2 : Math.min(0, dy0) || Math.max(0, dy1)
- );
- }
- _centroid(extent) {
- return [ (+extent[0][0] + +extent[1][0]) / 2, (+extent[0][1] + +extent[1][1]) / 2 ];
- }
- _gesture(node, clean) {
- return (!clean && node.__zooming) || new view.Zoom.Gesture(node, this);
- }
- _stopEvent(event) {
- event.preventDefault();
- event.stopImmediatePropagation();
- }
- _wheel(event) {
- const currentTarget = event.currentTarget;
- const wheelidled = (gesture) => {
- gesture.wheel = null;
- gesture.end();
- };
- if (this._filter(event)) {
- const gesture = this._gesture(currentTarget);
- const t = currentTarget.__zoom;
- const k = Math.max(this._scaleExtent[0], Math.min(this._scaleExtent[1], t.k * Math.pow(2, this._wheelDelta(event))));
- const p = this.pointer(event);
- if (gesture.wheel) {
- if (gesture.mouse[0][0] !== p[0] || gesture.mouse[0][1] !== p[1]) {
- gesture.mouse[1] = t.invert(gesture.mouse[0] = p);
- }
- clearTimeout(gesture.wheel);
- }
- else if (t.k === k) {
- return;
- }
- else {
- gesture.mouse = [p, t.invert(p)];
- gesture.start();
- }
- event.preventDefault();
- event.stopImmediatePropagation();
- gesture.wheel = setTimeout(() => wheelidled(gesture), this._wheelDelay);
- const transform = this.translate(this.scale(t, k), gesture.mouse[0], gesture.mouse[1]);
- gesture.zoom('mouse', this._constrain(transform, gesture.extent, this._translateExtent));
- }
- }
- _mouseDown(event) {
- const currentTarget = event.currentTarget;
- if (this._touchEnding || !this._filter(event)) return;
- const gesture = this._gesture(currentTarget, true);
- const selection = new view.Zoom.Selection(event.view)
- .on('mousemove.zoom', (event) => mousemoved(event), true)
- .on('mouseup.zoom', (event) => mouseupped(event), true);
- const p = this.pointer(event, currentTarget);
- const x0 = event.clientX;
- const y0 = event.clientY;
- const root = event.view.document.documentElement;
- selection.on('dragstart.drag', (event) => this._stopEvent(event), { capture: true, passive: false });
- if ('onselectstart' in root) {
- selection.on('selectstart.drag', (event) => this._stopEvent(event), { capture: true, passive: false });
- }
- else {
- root.__noselect = root.style.MozUserSelect;
- root.style.MozUserSelect = 'none';
- }
- event.stopImmediatePropagation();
- gesture.mouse = [ p, currentTarget.__zoom.invert(p) ];
- gesture.start();
- const mousemoved = (event) => {
- event.preventDefault();
- event.stopImmediatePropagation();
- if (!gesture.moved) {
- const dx = event.clientX - x0, dy = event.clientY - y0;
- gesture.moved = dx * dx + dy * dy > 0;
- }
- const transform = this.translate(gesture.node.__zoom, gesture.mouse[0] = this.pointer(event, currentTarget), gesture.mouse[1]);
- gesture.zoom('mouse', this._constrain(transform, gesture.extent, this._translateExtent));
- };
- const mouseupped = (event) => {
- selection.on('mousemove.zoom', null);
- selection.on('mouseup.zoom', null);
- const root = event.view.document.documentElement;
- selection.on('dragstart.drag', null);
- if (gesture.moved) {
- selection.on('click.drag', (event) => this._stopEvent(event), { capture: true, passive: false });
- setTimeout(function() { selection.on('click.drag', null); }, 0);
- }
- if ('onselectstart' in root) {
- selection.on('selectstart.drag', null);
- }
- else {
- root.style.MozUserSelect = root.__noselect;
- delete root.__noselect;
- }
- event.preventDefault();
- event.stopImmediatePropagation();
- gesture.end();
- };
- }
- _touchStarted(event) {
- const currentTarget = event.currentTarget;
- if (this._filter(event)) {
- const touches = event.touches;
- const gesture = this._gesture(currentTarget, event.changedTouches.length === touches.length);
- let started;
- let p;
- event.stopImmediatePropagation();
- for (let i = 0; i < touches.length; ++i) {
- const t = touches[i];
- p = this.pointer(t, currentTarget);
- p = [p, currentTarget.__zoom.invert(p), t.identifier];
- if (!gesture.touch0) {
- gesture.touch0 = p;
- started = true;
- gesture.taps = 1 + !!this._touchStarting;
- }
- else if (!gesture.touch1 && gesture.touch0[2] !== p[2]) {
- gesture.touch1 = p;
- gesture.taps = 0;
- }
- }
- if (this._touchStarting) {
- this._touchStarting = clearTimeout(this._touchStarting);
- }
- if (started) {
- if (gesture.taps < 2) {
- this._touchFirst = p[0];
- this._touchStarting = setTimeout(function() { this._touchStarting = null; }, this._touchDelay);
- }
- gesture.start();
- }
- }
- }
- _touchMoved(event) {
- const currentTarget = event.currentTarget;
- if (currentTarget.__zooming) {
- const gesture = this._gesture(currentTarget);
- const touches = event.changedTouches;
- let t, p, l;
- event.preventDefault();
- event.stopImmediatePropagation();
- for (let i = 0; i < touches.length; i++) {
- t = touches[i], p = this.pointer(t, currentTarget);
- if (gesture.touch0 && gesture.touch0[2] === t.identifier) {
- gesture.touch0[0] = p;
- }
- else if (gesture.touch1 && gesture.touch1[2] === t.identifier) {
- gesture.touch1[0] = p;
- }
- }
- t = gesture.node.__zoom;
- if (gesture.touch1) {
- const p0 = gesture.touch0[0];
- const l0 = gesture.touch0[1];
- const p1 = gesture.touch1[0];
- const l1 = gesture.touch1[1];
- let dp, dl;
- dp = (dp = p1[0] - p0[0]) * dp + (dp = p1[1] - p0[1]) * dp;
- dl = (dl = l1[0] - l0[0]) * dl + (dl = l1[1] - l0[1]) * dl;
- t = this.scale(t, Math.sqrt(dp / dl));
- p = [(p0[0] + p1[0]) / 2, (p0[1] + p1[1]) / 2];
- l = [(l0[0] + l1[0]) / 2, (l0[1] + l1[1]) / 2];
- }
- else if (gesture.touch0) {
- p = gesture.touch0[0], l = gesture.touch0[1];
- }
- else {
- return;
- }
- const transform = this.translate(t, p, l);
- gesture.zoom('touch', this._constrain(transform, gesture.extent, this._translateExtent));
- }
- }
- _touchEnded(event) {
- const currentTarget = event.currentTarget;
- if (currentTarget.__zooming) {
- const gesture = this._gesture(currentTarget);
- const touches = event.changedTouches;
- event.stopImmediatePropagation();
- if (this._touchEnding) {
- clearTimeout(this._touchEnding);
- }
- this._touchEnding = setTimeout(function() { this._touchEnding = null; }, this._touchDelay);
- for (let i = 0; i < touches.length; i++) {
- const touch = touches[i];
- if (gesture.touch0 && gesture.touch0[2] === touch.identifier) {
- delete gesture.touch0;
- }
- else if (gesture.touch1 && gesture.touch1[2] === touch.identifier) {
- delete gesture.touch1;
- }
- }
- if (gesture.touch1 && !gesture.touch0) {
- gesture.touch0 = gesture.touch1;
- delete gesture.touch1;
- }
- if (gesture.touch0) {
- gesture.touch0[1] = currentTarget.__zoom.invert(gesture.touch0[0]);
- }
- else {
- gesture.end();
- }
- }
- }
- };
- view.Zoom.Selection = class {
- constructor(node) {
- this._node = node;
- }
- get node() {
- return this._node;
- }
- each(callback) {
- if (this._node) {
- callback(this._node);
- }
- return this;
- }
- on(name, value, options) {
- const node = this._node;
- if (node) {
- const key = name.split('.');
- if (value) {
- node.__on = node.__on || [];
- const listener = (event) => value.call(node, event);
- let match = false;
- for (const handler of node.__on) {
- if (handler.type === key[0] && handler.name === key[1]) {
- node.removeEventListener(handler.type, handler.listener, handler.options);
- node.addEventListener(handler.type, handler.listener = listener, handler.options = options);
- handler.value = value;
- handler.options = options;
- match = true;
- break;
- }
- }
- if (!match) {
- node.addEventListener(key[0], listener, options);
- node.__on.push({ type: key[0], name: key[1], value: value, listener: listener, options: options });
- }
- }
- else if (node.__on) {
- node.__on = node.__on.filter((handler) => {
- if (handler.type === key[0] && handler.name === key[1]) {
- node.removeEventListener(handler.type, handler.listener, handler.options);
- return false;
- }
- return true;
- });
- if (node.__on.length === 0) {
- delete node.__on;
- }
- }
- }
- return this;
- }
- };
- view.Zoom.Transform = class {
- constructor(k, x, y) {
- this.k = k;
- this.x = x;
- this.y = y;
- }
- translate(x, y) {
- return x === 0 & y === 0 ? this : new view.Zoom.Transform(this.k, this.x + this.k * x, this.y + this.k * y);
- }
- invert(location) {
- return [(location[0] - this.x) / this.k, (location[1] - this.y) / this.k];
- }
- invertX(x) {
- return (x - this.x) / this.k;
- }
- invertY(y) {
- return (y - this.y) / this.k;
- }
- toString() {
- return 'translate(' + this.x + ',' + this.y + ') scale(' + this.k + ')';
- }
- };
- view.Zoom.Gesture = class {
- constructor(node, target) {
- this.node = node;
- this.active = 0;
- this.extent = target.extent(node);
- this.taps = 0;
- this.target = target;
- }
- start() {
- if (++this.active === 1) {
- this.node.__zooming = this;
- this.raise('start');
- }
- return this;
- }
- zoom(name, transform) {
- if (this.mouse && name !== 'mouse') {
- this.mouse[1] = transform.invert(this.mouse[0]);
- }
- if (this.touch0 && name !== 'touch') {
- this.touch0[1] = transform.invert(this.touch0[0]);
- }
- if (this.touch1 && name !== 'touch') {
- this.touch1[1] = transform.invert(this.touch1[0]);
- }
- this.node.__zoom = transform;
- this.raise('zoom');
- return this;
- }
- end() {
- if (--this.active === 0) {
- delete this.node.__zooming;
- this.raise('end');
- }
- return this;
- }
- raise(event) {
- this.target.raise(event, { transform: this.node.__zoom });
- }
- };
- if (typeof module !== 'undefined' && typeof module.exports === 'object') {
- module.exports.View = view.View;
- module.exports.ModelFactoryService = view.ModelFactoryService;
- }
|