package Ġcom . d ail yst udio . t flite . example . image . style transfer Ċ Ċ import Ġandroid . os . Bundle Ċ import Ġandroid . view . View Ċ import Ġandroidx . camera . core . Camera Selector Ċ import Ġandroidx . fragment . app . Fragment Ċ import Ġandroidx . lifecycle . ViewModel Provider Ċ import Ġandroidx . lifecycle . lifecycle Scope Ċ import Ġcom . d ail yst udio . dev br icks x . development . Logger Ċ import Ġcom . d ail yst udio . dev br icks x . settings . Abs Settings Dialog Fragment Ċ import Ġcom . d ail yst udio . t flite . example . common . Abs Example Activity Ċ import Ġcom . d ail yst udio . t flite . example . common . image . Advance Inference Info Ċ import Ġcom . d ail yst udio . t flite . example . common . ui . Inference Settings Fragment Ċ import Ġcom . d ail yst udio . t flite . example . image . style transfer . fragment . Pick Style Dialog Fragment Ċ import Ġcom . d ail yst udio . t flite . example . image . style transfer . fragment . Style Transfer Camera Fragment Ċ import Ġcom . d ail yst udio . t flite . example . image . style transfer . model . Style Image ViewModel Ċ import Ġcom . d ail yst udio . t flite . example . image . style transfer . ui . Styled Overlay Ċ import Ġcom . google . android . material . floating action button . Floating ActionButton Ċ import Ġkotlinx . coroutines . Dispatch ers Ċ import Ġkotlinx . coroutines . launch Ċ import Ġorg . tensorflow . lite . examples . style transfer . Style Transfer Result ĊĊ Ċ class ĠExample Activity : ĠAbs Example Activity < Advance Inference Info , ĠStyle Transfer Result >() Ġ{ ĊĊĠĠĠ Ġcompanion Ġobject Ġ{ ĊĠĠĠĠĠĠĠ Ġconst Ġval ĠSTY LES _ DIRECTORY Ġ= Ġ" thumb nails " ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġprivate Ġvar Ġstyled Overlay : ĠStyled Overlay ? Ġ= Ġnull ĊĠĠĠ Ġprivate Ġvar Ġfab Styles : ĠFloating ActionButton ? Ġ= Ġnull ĊĠĠĠ Ġprivate Ġvar Ġfab Mode : ĠFloating ActionButton ? Ġ= Ġnull ĊĠĠĠ Ġprivate Ġvar Ġfab Camera Selector : ĠFloating ActionButton ? Ġ= Ġnull ĊĊĠĠĠ Ġprivate Ġvar Ġfragment Camera : ĠStyle Transfer Camera Fragment ? Ġ= Ġnull ĊĊĠĠĠ Ġoverride Ġfun ĠonCreate ( savedInstanceState : ĠBundle ?) Ġ{ ĊĠĠĠĠĠĠĠ Ġsuper . onCreate ( savedInstanceState ) ĊĊĠĠĠĠĠĠĠ Ġlifecycle Scope . launch ( Dispatch ers . IO ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġload Style Images () ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġsetup Views () Ġ{ ĊĠĠĠĠĠĠĠ Ġsuper . setup Views () ĊĊĠĠĠĠĠĠĠ Ġstyled Overlay Ġ= ĠfindViewById ( R . id . styled _ overlay ) ĊĊĠĠĠĠĠĠĠ Ġfab Styles Ġ= ĠfindViewById ( R . id . fab _ styles ) ĊĠĠĠĠĠĠĠ Ġfab Styles ?. setOnClickListener Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġval Ġdialog Ġ= ĠPick Style Dialog Fragment () ĊĊĠĠĠĠĠĠĠĠĠĠĠ Ġdialog . show ( support FragmentManager , Ġ" pick - styles ") ĊĠĠĠĠĠĠĠ Ġ} ĊĊĠĠĠĠĠĠĠ Ġfab Mode Ġ= ĠfindViewById ( R . id . fab _ mode ) ĊĠĠĠĠĠĠĠ Ġfab Mode ?. setOnClickListener Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġstyled Overlay ?. let Ġ{ Ġoverlay Ġ-> ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġval Ġpreview Mode Ġ= Ġoverlay . isIn Preview Mode () ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġoverlay . set Preview Mode (! preview Mode ) ĊĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfab Mode ?. setImage Resource ( if Ġ( preview Mode ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ ĠR . drawable . ic _ preview _ mode ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} Ġelse Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ ĠR . drawable . ic _ fullscreen _ mode ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ}) ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠ Ġ} ĊĊĠĠĠĠĠĠĠ Ġfab Camera Selector Ġ= ĠfindViewById ( R . id . fab _ camera _ selector ) ĊĠĠĠĠĠĠĠ Ġfab Camera Selector ?. setOnClickListener Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġfragment Camera ?. let Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġval Ġlens Facing Ġ= Ġit . getCurrent Lens Facing () ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ( Camera Selector . LEN S _ F ACING _ FRONT Ġ== Ġlens Facing ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġit . change Lens Facing ( Camera Selector . LEN S _ F ACING _ BACK ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfab Camera Selector ?. setImage Resource ( R . drawable . ic _ front _ camera ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} Ġelse Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġit . change Lens Facing ( Camera Selector . LEN S _ F ACING _ FRONT ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfab Camera Selector ?. setImage Resource ( R . drawable . ic _ back _ camera ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĠĠĠ Ġoverride Ġfun Ġget Layout Res Id (): ĠInt Ġ{ ĊĠĠĠĠĠĠĠ Ġreturn ĠR . layout . activity _ example _ style _ transfer ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġcreate Base Fragment (): ĠFragment Ġ{ ĊĠĠĠĠĠĠĠ Ġval Ġfragment Ġ= ĠStyle Transfer Camera Fragment () ĊĊĠĠĠĠĠĠĠ Ġfragment Camera Ġ= Ġfragment ĊĊĠĠĠĠĠĠĠ Ġreturn Ġfragment ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġcreate Results View (): ĠView ? Ġ{ ĊĠĠĠĠĠĠĠ Ġreturn Ġnull ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġon Results Updated ( results : ĠStyle Transfer Result ) Ġ{ ĊĠĠĠĠĠĠĠ Ġstyled Overlay ?. set Styled Overlay ( results . styled Bitmap ) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġprivate Ġfun Ġload Style Images () Ġ{ ĊĠĠĠĠĠĠĠ Ġval Ġstyles : ĠArray < String > Ġ= Ġtry Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġassets . list ( STY LES _ DIRECTORY ) ĊĠĠĠĠĠĠĠ Ġ} Ġcatch Ġ( e : ĠException ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ ĠLogger . error (" failed Ġto Ġlist Ġstyles Ġin Ġ[$ STY LES _ DIRECTORY ]: Ġ$ e ") ĊĊĠĠĠĠĠĠĠĠĠĠĠ Ġnull ĊĠĠĠĠĠĠĠ Ġ} Ġ?: Ġreturn ĊĊĠĠĠĠĠĠĠ Ġval ĠviewModel Ġ= ĠViewModel Provider ( this ). get ( Style Image ViewModel :: class . java ) ĊĊĠĠĠĠĠĠĠ Ġfor Ġ(( i , Ġstyle ) Ġin Ġstyles . with Index ()) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġval Ġstyle Image Ġ= ĠStyle Image ( i , ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġstyle , ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ"${ STY LES _ DIRECTORY }/${ style }") ĊĊĠĠĠĠĠĠĠĠĠĠĠ ĠviewModel . insert Style Image ( style Image ) ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġget Example Name (): ĠCharSequence ? Ġ{ ĊĠĠĠĠĠĠĠ Ġreturn ĠgetString ( R . string . app _ name ) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġget Example Icon Resource (): ĠInt Ġ{ ĊĠĠĠĠĠĠĠ Ġreturn ĠR . drawable . about _ icon ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġoverride Ġfun Ġget Example Desc (): ĠCharSequence ? Ġ{ ĊĠĠĠĠĠĠĠ Ġreturn ĠgetString ( R . string . app _ desc ) ĊĠĠĠ Ġ} Ċ Ċ }