123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089 |
- <template src="./viewMask.html">
- </template>
- <script>
- const util = require('@/utils/util.js').default;
- import router from "@/router";
- import { MessageBox } from 'mint-ui';
- import spaceTypes from '@/static/spaceTypesIE.js';
- // const config = require('@/services/url$config.js');
- import touchHandle from '@/mixins/touchHandle.js';
- import {
- setStorage,
- getStorage,
- } from '@/utils/localStorage';
- // import requestConfig from '@/static/lib/requestConfig';
- // const app = getApp(); //获取应用实例
- export default {
- mixins: [touchHandle],
- data: function() {
- return {
- currentIndex:0, //当前看到的图片序号
- showAIImage: false,//是否显示当前AI结果集合 默认不显示,因为没有
- aiImagesList:[
- ],//AI生成的图片列表
-
- styleList: [], //风格列表数据
- curStyleIndex: 0, //当前选中的风格序号
- aiFlag:false,
- aiImage: "",
- inputBase64Url: "",
- aiPicId: "", // 图生图任务ID
- randomTimer:null,
- timer: null,
- outTimer: null,
- imageWidth: 750,
- imageHeight: 448,
- showAIFlag:false,//继续生成状态切换标志
- checked:false,
- shottingImg:'',
- count:0,
- random:1,
- // disableAble:false,
-
- myloading:false,
- loadingMsg:'',
- screenWidth:0,
- screenHeight:0,
- modelType:2, //当前模型类型,1指创意设计 2指精准设计 默认2
- themeIndex:0,
- themeList:[
- {
- id:1,
- text:'默认',
- prot:'',
- },
- {
- id:2,
- leftColor:'#F4EBDE',
- rightColor:'#C1C9B4',
- text:'莫兰素雅',
- prot:'Soft grays, warm earth tones, subtle blues, soft pinks,',
- },
- {
- id:3,
- leftColor:'#1F1F1F',
- rightColor:'#FFFFFF',
- text:'经典黑白',
- prot:'Classic black and pure white,',
- },
- {
- id:4,
- leftColor:'#6FA5BE',
- rightColor:'#E2CCC5',
- text:'海岸珊瑚',
- prot:'Deep ocean blue paired with vibrant coral red,',
- },
- {
- id:5,
- leftColor:'#849059',
- rightColor:'#E6CEB4',
- text:'森林璀璨',
- prot:'Deep forest green paired with luxurious gold,'
- },
- {
- id:6,
- leftColor:'#A1BACB',
- rightColor:'#E6DFD8',
- text:'梦幻静谧',
- prot:'Dreamy powder blue paired with neutral gray,'
- },
- {
- id:7,
- leftColor:'#303B2B',
- rightColor:'#D8C8A6',
- text:'自然和谐',
- prot:'Natural olive green paired with warm beige,'
- },
- {
- id:8,
- leftColor:'#652C30',
- rightColor:'#D5C3B0',
- text:'醇酒乳香',
- prot:'Rich burgundy paired with soft cream,'
- },
- {
- id:9,
- leftColor:'#E3C7C5',
- rightColor:'#EDE4DB',
- text:'雅致浪漫',
- prot:'Understated taupe paired with romantic pink,'
- },
- {
- id:10,
- leftColor:'#2C3C6E',
- rightColor:'#F0DBBD',
- text:'夜空璀璨',
- prot:'Elegant navy blue paired with glamorous gold,'
- },
- {
- id:11,
- leftColor:'#ACC9A1',
- rightColor:'#E6DCC6',
- text:'复古翠绿',
- prot:'Vintage tan paired with vibrant green,'
- },
- {
- id:12,
- leftColor:'#AAA8A7',
- rightColor:'#D8A176',
- text:'暖阳灰影',
- prot:'Warm orange paired with cool gray,'
- },
- {
- id:13,
- leftColor:'#413D41',
- rightColor:'#D1B687',
- text:'耀眼黑金',
- prot:'black and gold,'
- },
- ],
- timeOut:20000, //超时时间
- subType:0,
- currentImg:null,
- reqList:[
- {
- model:'batouresearch/sdxl-controlnet-lora',
- level:'5',
- aiPicId:'',
- resultImg:'',
- }
- ],
- reqListReset:[
- {
- model:'deployments/elabgroup/elab-sdxl-controlnet-lora',
- level:'3',
- aiPicId:'',
- resultImg:'',
- },{
- model:'batouresearch/sdxl-controlnet-lora',
- level:'5',
- aiPicId:'',
- resultImg:'',
- }
- ],
- // dialogVisible: false,
- // dialogAIVisible: false,
- datalist:[],
- connectUsImg:'',
- curSpaceId:'',
- }
- },
- props:{
- spaceObj: {
- type: Object,
- default: null,
- },
- spaceList: {
- type: [Array ,Object],
- default: [],
- },
- layoutList:{
- type: [Array ,Object],
- default: [],
- }
- },
- watch: {
- spaceObj(newVal, oldVal) {
- if (newVal == null) {
- return;
- }
- console.log("当前空间数据view-mark-watch:", newVal,newVal.layoutSpaceName,newVal.layoutSpaceType);
- this.getAiBeautyFamily();
- // this.curSpaceArea = parseFloat(
- // (newVal.spaceWidth * newVal.spaceHeight) / 10000
- // ).toFixed(2);
- // this.getOverallArrangementDetailsList();
- },
- spaceList(newVal, oldVal) {
- if (newVal == null) {
- return;
- }
- console.log("当前空间数据spaceList-watch:", newVal);
- if(newVal && newVal.length>0){
- this.datalist = [];
- newVal.forEach(it=>{
- if(it.spaceName && !it.isSizeLock){
- this.datalist.push(it);
- }
- })
- }
-
- // this.curSpaceArea = parseFloat(
- // (newVal.spaceWidth * newVal.spaceHeight) / 10000
- // ).toFixed(2);
- // this.getOverallArrangementDetailsList();
- },
- layoutList(newVal, oldVal) {
- if (newVal == null) {
- return;
- }
- this.updataLable();
- },
- },
- filters:{
- // spaceTypeFilter(type){
- // let name = spaceTypes[type - 1];
- // return name || '-'
- // },
- },
- mounted() {//组件挂载时事件
- // console.warn("***mounted-nav***",this.seedItem)
- // if(this.seedItem){
- // this.title = this.seedItem.seedText;
- // }
- // wx.checkJsApi({
- // jsApiList: ['previewImage'], // 需要检测的JS接口列表,所有JS接口列表见附录2,
- // success: function(res) {
- // // 以键值对的形式返回,可用的api值true,不可用为false
- // // 如:{"checkResult":{"chooseImage":true},"errMsg":"checkJsApi:ok"}
- // console.warn("checkJsApi",res)
- // // alert('su'+ JSON.stringify(res))
- // },
- // fail:(err)=>{
- // console.warn("checkJsApi-err",err)
- // // alert('err'+ JSON.stringify(err) )
- // }
- // })
- this.screenWidth = window.screen.width;
- this.screenHeight = window.screen.height;
- if(window.innerWidth && window.screen.width){
- this.screenWidth = Math.min(window.innerWidth,window.screen.width)
- }
- if(window.innerHeight && window.screen.height){
- this.screenHeight = Math.min(window.innerHeight,window.screen.height)
- }
- let firstData = JSON.parse(getStorage('firstData'));
- this.connectUsImg = firstData[0].selectOptionImageUrl;
- },
- beforeDestroy:function(){
- console.warn("***beforeDestroy***");//更新到页面上的数据
- this.stopInterval();
- },
- computed: {
- aiData() {
- return this.$store.state.aiData;
- },
- curHouseObj() {
- return this.$store.state.curHouseObj;
- },
- },
- methods:{
- //更改空间显示名称 X空间 类型 根据布局所属类型来显示
- updataLable(){
- let list = this.layoutList;
- this.datalist.forEach(lable=>{
- if(lable.spaceType==15){
- let layoutId = lable.layoutId;
- let layout = list.find(it=>it.id == layoutId);
- if(layout){
- let name = spaceTypes[layout.type - 1];
- lable.text = name;
- }
- }else{
- let name = spaceTypes[lable.spaceType - 1];
- lable.text = name;
- }
- })
- this.$forceUpdate()
- console.warn("***updataLable***",this.datalist)
- },
- swiperChangeImg(e){
- this.currentIndex = e.detail?e.detail.current:e;
- this.checked = this.aiImagesList[this.currentIndex].checked;
- console.warn("***swiperChangeImg***",this.checked);//更新到页面上的数据
- },
- //切换空间
- changeSpace(item){
- this.$parent.changeSpace(item.spaceId);
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080417', //点击ID
- clkName: 'changeangle_clk', //点击前往的页面名称
- clkParams: {
- locusName: "空间切换",
- locusValue:item.text,
- spaceId:item.spaceId,
- type:item.text,
- }
- };
- util.trackRequest(param);
- },
- //视角切换
- switchActor(){
- // this.$parent.clearHandle()
- this.$emit('switchActor');
- this.showAIImage = false;//隐藏AI结果集合-执行切换视角
- let index = this.$parent.currentActor.userIndex;//当前视角的序号
- let nextIndex = (index + 1) % this.$parent.actors.length;
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080417', //点击ID
- clkName: 'changeangle_clk', //点击前往的页面名称
- clkParams: {
- locusName: "视角切换",
- type:this.$parent.actors[nextIndex].actorEum
- }
- };
- util.trackRequest(param);
- },
- showOrHideWebGl(){
- // this.$parent.clearHandle()
- this.showAIImage = !this.showAIImage;
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080415', //点击ID
- clkName: 'contrast_clk', //点击前往的页面名称
- clkParams: {
- locusName: "对比",
- }
- };
- util.trackRequest(param);
- },
- //空间或者风格切换
- selectAction(selItem,type,index) {
- console.log('点击动作111:', selItem,type,index)
- var self = this;
- if(this.themeIndex == index){
- return false
- }
- this.themeIndex = index;
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23112701', //点击ID
- clkName: 'AIstyle_theme_clk', //点击前往的页面名称
- clkParams: {
- locusName: "AI 主题",
- type:this.themeList[this.themeIndex].text
- }
- };
- util.trackRequest(param);
- },
- //切换模式
- changeModel(){
- this.modelType = (this.modelType + 1) >2 ? 1 : 2;
- },
- //保存到相册
- save() {
- //表示canvas正在绘制,不能进行保存
- if (!this.aiImagesList || this.aiImagesList.length==0) {
- return false;
- }
- var _resultImg = this.aiImagesList[this.currentIndex].image;
- //正在选择照片,不能生效
- if (!_resultImg || _resultImg.length==0) {
- this.showToast("请选中图片后再试!");
- return false;
- }
- var para = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_56', //点击ID,固定
- clkName: "share-savepic", //点击名称
- expand: {
- "resultImg": (_resultImg || ""),
- },
- }
- util.trackRequest(para);
- if(navigator.userAgent.toLocaleLowerCase().includes('micromessenger')){
- this.$message("请长按图片保存!");
- }else{
- this.saveImageHandle(_resultImg);
- }
- },
- downloadIamge(imgsrc) { //下载图片地址和图片名
- var fileName = "4DImage" + util.formatDate(new Date(), "yyyyMMddhhmmss") + '.jpg';
- const image = new Image();
- // 解决跨域 Canvas 污染问题
- image.setAttribute("crossOrigin", "anonymous");
- image.onload = ()=> {
- // let canvas = document.createElement("canvas");
- // canvas.width = image.width;
- // canvas.height = image.height;
- // const context = canvas.getContext("2d");
- // context.drawImage(image, 0, 0, image.width, image.height);
- // const url = canvas.toDataURL("image/jpg"); //得到图片的base64编码数据
- const a = document.createElement("a"); // 生成一个a元素
- const event = new MouseEvent("click"); // 创建一个单击事件
- a.download = fileName || "photo"; // 设置图片名称
- a.href = imgsrc; // 将生成的URL设置为a.href属性
- a.dispatchEvent(event); // 触发a的单击事件
- this.showToast("保存成功!");
- };
- image.src = imgsrc;
- },
- saveImageHandle(_resultImg){
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080414', //点击ID
- clkName: 'download_clk', //点击前往的页面名称
- clkParams: {
- locusName: "下载",
- img:_resultImg
- }
- };
- util.trackRequest(param);
- this.downloadIamge(_resultImg);
- },
-
- //点赞喜欢
- changeAIImg() {
- // let lastPage = getCurrentPages()[getCurrentPages().length - 2] ? getCurrentPages()[getCurrentPages().length - 2].$vm : null;
- this.checked = !this.checked;//变更选项
- this.aiImagesList[this.currentIndex].checked = this.checked;
- console.warn("***changeAIImg***",this.checked,this.aiData)
- // if(this.aiData){//给上一个页面回传生成的数据
- // let space = this.aiData.find(it=>{
- // return it.spaceId == this.spaceObj.spaceId
- // })
- // if(space){
- // space.aiImagesList[this.currentIndex].checked = this.checked;
- // }
- // }
- if(this.checked){
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080418', //点击ID
- clkName: 'chooseprogramme_clk', //点击前往的页面名称
- clkParams: {
- locusName: "选定风格/视角",
- style:this.styleList[this.curStyleIndex].styleName,
- img:this.aiImagesList[this.currentIndex].image,
- }
- };
- util.trackRequest(param);
- }
- },
- rightScroll(){//右滑
- if(this.currentIndex <= (this.aiImagesList.length - 1) && this.currentIndex > 0){
- // this.currentIndex --;
- this.$refs.carousel.prev();
- }
- },
- leftScroll(){//继续生成
- if(this.currentIndex != this.aiImagesList.length - 1){
- if(this.currentIndex<this.aiImagesList.length - 1){
- // this.currentIndex ++;
- this.$refs.carousel.next();
- }
- return false;
- }
- console.log("***leftScroll***",this.currentIndex,this.aiImagesList.length)
- this.aiSubmit(2);//继续生成下一张
- },
- //预览图片-使用微信自带的预览功能,即H5中不可用
- previewAction() {
- if(wx && typeof(wx.previewImage)=='function'){
- let imglist = this.aiImagesList.map(it=>it.image);
- wx.previewImage({
- current: this.currentIndex,
- urls: imglist,
- success: (res) => {
- console.log('图片预览:', res);
- },
- fail:(err)=>{
- console.warn('图片预览-err:', res);
- }
- })
- }
- },
- mynavigateFuc(e) {
- if (e) {
- // this.$parent.clearHandle();
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080408', //点击ID
- clkName: 'WeCom_clk', //点击前往的页面名称
- clkParams: {
- locusName: "联系定制",
- }
- };
- util.trackRequest(param);
- if(window.__wxjs_environment === 'miniprogram'){
- wx.miniProgram.navigateTo({url: '/pages/transfer/transfer?event=openCustomerServiceChat&houseId='+this.$store.state.houseId})
- }else{
- this.$message.warning("敬请期待");
- }
- }
- },
- catchTapEvent:function(){
- return false;
- },
- //获取AI风格列表
- async getAiBeautyFamily() {
- // const spaceName = this.spaceTypes[this.spaceObj.spaceType - 1].title;
- let res = await requestConfig("getHardboundEffects", {
- "houseId": this.$route.query.houseId || this.$store.state.houseId,
- "spaceType": this.spaceObj.layoutSpaceType || this.spaceObj.spaceType,
- "spaceName": this.spaceObj.layoutSpaceName || this.spaceObj.spaceName,
- });
- this.styleList = [];
- // alert("***getAiBeautyFamily-res***"+JSON.stringify(res))
- if (res.success) {
- let list = res.list;
- this.styleList = list;
- }
- if(!this.styleList || this.styleList.length==0){
- // this.disableAble = true;//不能点击-没有风格
- let unit = this.screenWidth / 750;//单位rpm 对应 px 的值
- this.$parent.canvasHeight = this.screenHeight - (208 * unit);
- this.$parent.camera.aspect = this.screenWidth / this.$parent.canvasHeight;
- this.$parent.camera.updateProjectionMatrix();
- this.$parent.renderer.setSize(this.screenWidth, this.$parent.canvasHeight);
- }else{
- // this.disableAble = false;//可以点击
- }
- },
- //风格选择
- selectStyle(idx){
- if(this.curStyleIndex == idx){
- return false
- }
- this.curStyleIndex = idx;
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080416', //点击ID
- clkName: 'AIstyle_ret_clk', //点击前往的页面名称
- clkParams: {
- locusName: "AI 风格",
- type:this.styleList[this.curStyleIndex].styleName
- }
- };
- util.trackRequest(param);
- },
- // AI渲染
- async aiSubmit(type) {
- if(!this.styleList || this.styleList.length==0 || this.curStyleIndex==-1
- || !this.styleList[this.curStyleIndex].prompt || !this.styleList[this.curStyleIndex].negativePrompt){
- return false;
- }
- // 防止连续点击处理
- if (this.aiFlag) {
- return
- }
- this.aiFlag = true;
- this.aiImage = "";//清空AI结果图
- this.stopInterval();
- // if(!this.showAIImage){
- this.myloading = true;
- this.loadingMsg = "启动中";
- // }
-
- // this.inputBase64Url = await this.shottingAction(2);//开始截图-返回的是base64的数据
- // this.startServer();
- // let base64 = await this.$parent.shottingAction(2);//开始截图-返回的是base64
- let shottingImg = await this.$parent.shottingAction();//开始截图-返回的是图片地址
- if(!shottingImg){
- this.showToast("渲染失败,请重试");
- return false;
- }
- // ?x-oss-process=image/auto-orient,1/quality,Q_46/format,jpg //阿里OSS
- // "?imageMogr2/auto-orient/format/webp/blur/1x0/quality/75";//七牛云压缩图片
- shottingImg += "?x-oss-process=image/auto-orient,1/quality,Q_46/format,jpg";//压缩图片
- this.shottingImg = shottingImg;
- this.changeImg2Base64(this.shottingImg, false);
- console.warn("***shottingImg***",this.shottingImg)
- if(type==1){
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080413', //点击ID
- clkName: 'AIcreate_clk', //点击前往的页面名称
- clkParams: {
- locusName: "AI生成",
- userparamter:{
- shottingImg:this.shottingImg,
- style:this.styleList[this.curStyleIndex].styleName
- }
- }
- };
- util.trackRequest(param);
- }else{
- let param = {
- type: 'CLK', //埋点类型
- clkId: 'clk_2cmina_23080419', //点击ID
- clkName: 'continueAIcreate_clk', //点击前往的页面名称
- clkParams: {
- locusName: "继续生成",
- userparamter:{
- shottingImg:this.shottingImg,
- style:this.styleList[this.curStyleIndex].styleName
- }
- }
- };
- util.trackRequest(param);
- }
- },
- image2Base64(imgUrl) {//导入的图片路径
- var toBase64= new Promise(function(resolve, reject){
- window.URL = window.URL || window.webkitURL;
- var xhr = new XMLHttpRequest();
- xhr.open("get", imgUrl, true);
- // 至关重要
- xhr.responseType = "blob";//文件流
- xhr.onload = function (res) {
- if (res.currentTarget.status == 200) {
- //得到一个blob对象
- var blob = res.currentTarget.response;
- // 至关重要
- let oFileReader = new FileReader();
- oFileReader.onloadend = function (e) {
- let base64 = e.target.result;//base64
- resolve(base64)
- };
- oFileReader.readAsDataURL(blob);
- }
- }
- xhr.send();
- });
- return toBase64;
- },
- changeImg2Base64(url, isRepeat) {
- var self = this;
- if(isRepeat && self.inputBase64Url){//重复使用
- self.startServer();
- }else{
- self.startServer();
- }
- },
- //前置逻辑
- async prevHandle(parmas){
- //不是首次请求,则无需前置判断 用户上传的也不需要前置处理
- if(this.subType!=0){
- return '';
- }
- return new Promise(async (resolve, reject) => {
- let _data = JSON.parse(JSON.stringify(parmas))
- delete _data.webhook;
- let res = await requestConfig("img2img_local", _data);
- if (res.success) {
- if (res.success && res.single) {
- let resultImg = res.single;
- resolve(resultImg);
- }else{
- resolve('');
- }
- }else{
- resolve('');
- }
- })
- },
- //开始图生图流程
- async startServer() {
- if(!this.styleList || this.curStyleIndex < 0 || !this.styleList[this.curStyleIndex].imgUrl){
- return false
- }
- if(this.$parent.pvCurPageName!="room_show"){//说明用户切换页面了
- console.warn("***用户已经退出页面***")
- return false;
- }
- let imgUrl = this.styleList[this.curStyleIndex].imgUrl;
- let subType = this.subType;
- let session_hash = Date.now();
- let theme = this.themeList[this.themeIndex];//主题数据
- let prompt = this.styleList[this.curStyleIndex].prompt + theme.prot;
- let noPromot = this.styleList[this.curStyleIndex].negativePrompt;
- let unit = 768 / this.screenWidth;
- this.imageWidth = parseInt((this.screenWidth * unit).toFixed());
- this.imageHeight = parseInt((this.$parent.canvasHeight * unit).toFixed());
- this.timeOut = this.modelType==1? 20000: 50000;
- var parmas = {
- negativePrompt: noPromot,
- prompt: prompt,
- "batchSize": 1,
- brandId: $config.brandId,
- height: this.imageHeight,
- width: this.imageWidth,
- "moduleType": "AI_Biography",
- "steps":20,
- "sampler":"DDIM",
- "controlNetSessionHash":session_hash,
- "cfgScale":12,
- "denoising":0.9,
- image:this.shottingImg,
- styleImage:imgUrl,
- keyword: "replicate",
- model: this.modelType==1? "lucataco/ssd-1b" : 'catio-apps/controlnet-interior-design',
- };
- if(this.modelType==2){//精准设计
- if(subType == 0){//首次
- parmas.model = this.reqList[0].model;
- }else{//重试
- parmas.model = this.reqListReset[0].model;
- }
- }else{//创意设计
- if(subType == 0){//首次
- parmas.model = "lucataco/ssd-1b";
- }else{//重试
- parmas.model = "deployments/elabgroup/elab-ssd-1b";
- }
- }
- this.curSpaceId = this.spaceObj.spaceId;//生成时的空间id
- let result = await this.prevHandle(parmas);
- console.warn("***prevHandle***",result)
- if(result && result.length>0){
- this.randomHandle(result)
- return false;
- }else{
- if(this.modelType==2){//精准设计
- this.otherHandle(parmas);//发送其他AI请求
- }
- let res = await requestConfig("generateTaskImgToImgForAliyun", parmas);
- console.log("图生图结果:", res);
- let that = this;
- if (res.success && res.single) {
- this.aiPicId = res.single;
- if(this.modelType==2){//精准设计
- if(subType == 0){//首次
- this.reqList[0].aiPicId = this.aiPicId;
- }else{//重试
- this.reqListReset[0].aiPicId = this.aiPicId;
- }
- }
- if (this.aiPicId) {
- this.startInterval();//开始轮询AI生成图的结果
- }else{
- this.stopInterval()
- }
- }else{
- this.stopInterval()
- this.showToast("渲染失败,请重试")
- }
- }
- },
- stopInterval() {
- if (this.randomTimer) {
- clearInterval(this.randomTimer);
- this.randomTimer = null;
- }
- if (this.timer) {
- // clearInterval(this.timer);
- this.timer = null;
- }
- if (this.outTimer) {
- clearTimeout(this.outTimer)
- this.outTimer = null
- }
- this.subType = 0;
- this.reqList.forEach(it=>{
- it.aiPicId = '';
- it.resultImg = '';
- })
- this.reqListReset.forEach(it=>{
- it.aiPicId = '';
- it.resultImg = '';
- })
- // this.random = 1;
- this.myloading = false;
- this.aiFlag = false;
- },
- //随机处理
- randomHandle(resultImg){
- if(this.timer || !resultImg){
- return false;
- }
- let self = this;
- var count = 1;
- var process = 0;//进度
- var randomNum = Math.floor(Math.random() * 4 + 5);//5-8随机数
- this.randomTimer = setInterval(function() {
- process = parseInt(count * 100/(randomNum));
- if(process>=100){
- process = 99;
- }
- if (count < randomNum) {//没有到上限
- // if(!self.showAIImage){
- self.myloading = true;
- self.loadingMsg = '生成中…' + process + '%';
- // }
- }else{
- self.myloading = false;
- self.resultHandle(resultImg);
- }
- count = count + 1;
- }, 1000);
- },
-
- otherHandle(parmas){
- let reqList = [];
- if(this.subType == 0){//首次
- reqList = this.reqList;
- }else{//重试
- reqList = this.reqListReset;
- }
- reqList.forEach(async (it,index)=>{
- let _data = JSON.parse(JSON.stringify(parmas));
- if(index>0){
- _data.model = it.model;
- let res = await requestConfig("generateTaskImgToImgForAliyun", _data);
- if (res.success) {
- console.log('生成结果123:', res);
- it.aiPicId = res.single || '';
- }
- }
- })
- },
- //开始生成AI图的轮询,每隔1s轮询一次
- startInterval() {
- if(this.timer){
- return false;
- }
- let self = this;
- this.count = 1;//轮询次数
- var random = 0;
- this.currentImg = false;//当次生成图还没有结果
- this.timer = 1;
- this.getOutPicture();//不在轮询,而是等结果
- this.setOutTimer();//设置超时逻辑
- },
- //设置一个超时逻辑,到底指定时间后停止轮询,当前是90s
- setOutTimer() {
- if (this.outTimer) {
- clearTimeout(this.outTimer)
- this.outTimer = null
- }
- var self = this;
- this.outTimer = setTimeout(function() {
- if (self.timer) {
- let hasResult = false;
- if(self.modelType==2){//精准设计
- let reqList = [];
- if(self.subType == 0){//首次
- reqList = self.reqList;
- }else{//重试
- reqList = self.reqListReset;
- }
- reqList.some((item,index) => {
- if(item.resultImg){
- hasResult = true;
- self.resultHandle(item.resultImg)
- }
- });
- }
- console.warn("***hasResult***",hasResult)
- if(!hasResult){//没有结果
- self.stopInterval();//停止轮询
- // self.$message.warning("AI开了个小差,请稍后再试");
- // self.dialogVisible = true;//弹出超时提示
- MessageBox.confirm('',{
- title: '提示',
- message: '当前AI使用火爆,请继续尝试?',
- showCancelButton: true,
- confirmButtonText:'继续尝试',
- cancelButtonText:'取消等待',
- }).then(action => {
- console.warn("***MessageBox-action***",action)
- if(action == 'confirm'){
- self.confirmHandle(1);
- }
- }).catch(err=>{
- console.warn("***MessageBox-err***",err)
- if(err == 'cancel'){
- self.cancelHandle();
- }
- });
- // MessageBox.confirm('确定执行此操作?')
- }
- }
- clearTimeout(self.outTimer);
- self.outTimer = null
- }, this.timeOut);
- },
- confirmHandle(type){
- console.log('用户点击确定')
- this.subType = type || 0;
- this.startServer();
- },
- cancelHandle(){
- console.log('用户点击取消')
- this.subType = 0;
- },
- //获取生成图结果
- getOutPicture() {
- if(this.timer==null){
- console.warn("***当前轮询已经结束了1***")
- return false;
- }
- if(this.modelType==2){//精准设计
- let reqList = [];
- if(this.subType == 0){//首次
- reqList = this.reqList;
- }else{//重试
- reqList = this.reqListReset;
- }
- reqList.forEach((item,index) => {
- this.singleHandle(item)
- });
- }
- else{
- this.singleHandle({aiPicId:this.aiPicId})
- }
- },
- //发出获取结果请求获取AI生成结果
- async singleHandle(model){
- if(!model || !model.aiPicId){
- return false;
- }
- var parmas = {
- id: model.aiPicId,
- };
- let res = await requestConfig("getPredictions", parmas);
- if (res.success && res.single) {
- if(this.currentImg){//当前已经有生成图了
- console.warn("***当前已经有最高级生成图了***")
- return false;
- }
- if(this.timer==null){
- console.warn("***当前轮询已经结束了***")
- return false;
- }
- if (res.single.status == 'succeeded' && res.single.output) {
- if(this.modelType==2){//精准设计
- model.resultImg = res.single.output;
- if(model.level=='5'){//最高级了
- this.currentImg = true;
- setTimeout(()=>{
- this.resultHandle(res.single.output)
- },1500)
- }
- }else{
- this.currentImg = true;
- setTimeout(()=>{
- this.resultHandle(res.single.output)
- },1500)
- }
- console.warn("***有生成图了***",model)
- }
- if(this.modelType==2){//最高优先级返回没有图片,则处理进度条事宜
- if(model.level=='5'){
- this.processHandle(res.single);
- }
- }else{//普通设计
- this.processHandle(res.single);
- }
- } else if (!res.success) {
- this.stopInterval();//停止轮询
- this.showToast("渲染失败,请重试")
- }
- },
- //进度处理
- processHandle(single){
- console.warn("***single***",single.status,single.progress,this.count,single);
- let self = this;
- if(single.status=='starting'){//启动中的逻辑
- if(this.count>=20){
- this.stopInterval();//停止轮询
- // this.dialogAIVisible = true;
- MessageBox.confirm('',{
- title: '提示',
- message: 'AI开了小差,是否重新生成?',
- showCancelButton: true,
- confirmButtonText:'继续生成',
- cancelButtonText:'放弃生成',
- }).then(action => {
- console.warn("***MessageBox-action***",action)
- if(action == 'confirm'){
- this.confirmHandle(0);
- }
- }).catch(err=>{
- if(err == 'cancel'){
- this.cancelHandle();
- }
- });
- }else{
- this.myloading = true;
- this.loadingMsg = '启动中';
- }
- this.count = this.count + 1;
- this.getOutPicture();
- }else if(single.status=='processing'){
- let random = single.progress || 0;
- if(random >= 100){
- random = 99;
- }
- this.myloading = true;
- this.loadingMsg = '生成中…' + parseInt(random) + '%';
- this.getOutPicture();
- }else if(single.status=='succeeded'){
- this.myloading = true;
- this.loadingMsg = '生成中…100%';
- }
- },
- //返回结果处理
- resultHandle(resultImg){
- this.currentImg = true;
- // this.random = 100;
- this.aiImage = resultImg;
- this.showAIFlag = true;
- let newImage = resultImg;
- let aiStyleName = this.styleList[this.curStyleIndex].styleName;
- let _data = {
- image:newImage,
- checked:false,
- list:[newImage],
- spaceId:this.curSpaceId,//记录下当前生成的AI图是哪个空间的,因为会切换空间
- }
- this.aiImagesList.push(_data);
- this.showAIImage = true;//显示AI结果集合-因为生成了AI图片
- if(this.$parent && typeof(this.$parent.clearHandle)=="function" ){
- this.$parent.clearHandle();
- }
- this.stopInterval();
- if(this.aiImagesList.length>1){
- setTimeout(()=>{
- this.$refs.carousel.setActiveItem(this.aiImagesList.length-1);//切换到最后一张
- }, 1000);
- }
- },
- //空间数据整理
- spaceAIHandle(){
- let curHouseObj = this.$parent.curHouseObj;//当前户型数据对象
- let aiImagesList = this.aiImagesList;//AI生成图数据对象
- let wallList = this.$parent.wallList;//当前户型墙体数据列表对象
- let spaceList = JSON.parse(curHouseObj.houseJson);
-
- //遍历当前户型下的每一个空间
- spaceList.forEach((item)=>{
- let space = item;
- let list = aiImagesList.filter(it=>it.spaceId==space.spaceId);//找到AI生成图中的当前空间
- if(list && list.length>0){
- list = list.filter(it=>it.checked==true);//过滤选中的
- list = list.map(it=>{
- return it.image
- })
- space.hardboundEffect = list;
- console.warn("submitHouse: ", list,space);
- }
- let wall = wallList.find(it=>it.id==space.wallId);
- if(wall){//找到当前空间的墙体数据
- space.wallList = JSON.stringify(wall);
- }
- })
- setStorage('spaceList', spaceList);//把空间选择的数据存入本地缓存里面,方便后续使用
- console.warn("**spaceList**",spaceList)
- },
- //进入下一步
- gonext(){
- this.spaceAIHandle()
- router.push({
- name: "webgl_rxdz_env",
- query:{
- houseId:this.$route.query.houseId || this.$store.state.houseId,
- }
- });
- },
- showToast(title,time=3000){
- this.myloading = false;
- this.$store.state.loading = true;
- this.$store.state.loadingMsg = title || "";
- setTimeout(()=>{
- this.$store.state.loading = false;
- }, time);
- }
- }
- }
- </script>
- <style lang="scss" scoped>
- @import "./viewMask.scss";
- </style>
- <style lang="css" scoped>
- /* @import "@/common/css/common.css"; */
- </style>
|